MammaMia version 2.0.0

Src/API/guardahd.py

@@ -4,8 +4,12 @@ import re
 import Src.Utilities.config as config
 from fake_headers import Headers  
 from Src.Utilities.loadenv import load_env  
-from Src.Utilities.eval import eval_solver
+from Src.API.extractors.supervideo import supervideo
 import json, random
+import logging
+from Src.Utilities.config import setup_logging
+level = config.LEVEL
+logger = setup_logging(level)
 env_vars = load_env()
 GH_PROXY = config.GH_PROXY
 proxies = {}
@@ -25,49 +29,16 @@ if GH_ForwardProxy == "1":
     ForwardProxy = env_vars.get('ForwardProxy')
 else:
     ForwardProxy = ""
-#Get domain
-
 GHD_DOMAIN = config.GHD_DOMAIN
 
 random_headers = Headers()
 
 
-async def get_supervideo_link(link,client):
-    headers = random_headers.generate()
-    response = await client.get(ForwardProxy + link, headers=headers, allow_redirects=True,timeout = 30, proxies = proxies)
-    s2 = re.search(r"\}\('(.+)',.+,'(.+)'\.split", response.text).group(2)
-    terms = s2.split("|")
-    file_index = terms.index('file')
-    #I start to loop from file cause hfs it's one or two position after it usually.
-
-    for i in range(file_index,len(terms)):
-        if "hfs" in terms[i]:
-            hfs = terms[i]
-            break
-    urlset_index = terms.index('urlset')
-    hls_index = terms.index('hls')
-    result = terms[urlset_index + 1 : hls_index]
-    #If the len is >1 then you have to merge the elements from the last to the first
-    reversed_elements = result[::-1]
-    base_url =f"https://{hfs}.serversicuro.cc/hls/"
-    if len(reversed_elements) == 1:
-        final_url = base_url + "," + reversed_elements[0] + ".urlset/master.m3u8"
-    lenght = len(reversed_elements)
-    i = 1    
-    for element in reversed_elements:
-        base_url += element + ","
-        if lenght == i:
-            base_url += ".urlset/master.m3u8"
-        else:
-            i += 1
-    final_url = base_url
-    return final_url
-
 async def search(clean_id,client):
     headers = random_headers.generate()
-    response = await client.get(ForwardProxy + f"{GHD_DOMAIN}/set-movie-a/{clean_id}", allow_redirects=True, impersonate = "chrome124", headers = headers, proxies = proxies)
+    response = await client.get(ForwardProxy + f"{GHD_DOMAIN}/set-movie-a/{clean_id}", allow_redirects=True, headers = headers, proxies = proxies)
     if response.status_code != 200:
-            print(f"GuardaHD Failed to fetch search results: {response.status_code}")
+            logger.warning(f"GuardaHD Failed to fetch search results: {response.status_code}")
     soup = BeautifulSoup(response.text,'lxml',parse_only=SoupStrainer('li'))
     li_tag = soup.find('li')
     href = "https:" + li_tag['data-link']
@@ -75,26 +46,26 @@ async def search(clean_id,client):
 
 
 
-async def guardahd(id,client):
+async def guardahd(streams,id,client):
     try:
         general = await is_movie(id)
         ismovie = general[0]
         clean_id = general[1]
         if ismovie == 0:
-            return None
+            return streams
         supervideo_link = await search(clean_id,client)
-        final_url = await eval_solver(supervideo_link,proxies, ForwardProxy, client)
-        return final_url
+        streams = await supervideo(supervideo_link, client,streams,"GuardaHD",proxies, ForwardProxy)
+        return streams
     except Exception as e:
-        print("MammaMia: GuardaHD Failed",e)
-        return None
+        logger.warning(f"MammaMia: GuardaHD Failed {e}")
+        return streams
     
 async def test_script():
     from curl_cffi.requests import AsyncSession
     async with AsyncSession() as client:
         # Replace with actual id, for example 'anime_id:episode' format
-        test_id = "tt9218128"  # This is an example ID format
-        results = await guardahd(test_id, client)
+        test_id = "tt16426418"  # This is an example ID format
+        results = await guardahd({'streams': []},test_id, client)
         print(results)
 
 if __name__ == "__main__":