from bs4 import BeautifulSoup
from Src.Utilities.convert import get_TMDb_id_from_IMDb_id
from Src.Utilities.info import get_info_tmdb, is_movie, get_info_imdb
import Src.Utilities.config as config
import json
import random
import re
from urllib.parse import urlparse, parse_qs
from fake_headers import Headers  
from Src.Utilities.loadenv import load_env  
env_vars = load_env()
#Get domain
SC_DOMAIN = config.SC_DOMAIN
Public_Instance = config.Public_Instance
Alternative_Link = env_vars.get('ALTERNATIVE_LINK')

headers = Headers()
#GET VERSION OF STREAMING COMMUNITY:
async def get_version(client):
    #Extract the version from the main page of the site


    try:
        random_headers = headers.generate()
        random_headers['Referer'] = "https://streamingcommunity.buzz/"
        random_headers['Origin'] = "https://streamingcommunity.buzz"
        base_url = f'https://streamingcommunity.{SC_DOMAIN}/richiedi-un-titolo' 
        response = await client.get(base_url, headers=random_headers, allow_redirects = True, impersonate="chrome120")
        #Soup the response
        soup = BeautifulSoup(response.text, "lxml")

        # Extract version
        version = json.loads(soup.find("div", {"id": "app"}).get("data-page"))['version']
        return version
    except Exception as e:
        print("Couldn't find the version",e)
        version = "65e52dcf34d64173542cd2dc6b8bb75b"
        return version

async def search(query,date,ismovie, client,SC_FAST_SEARCH):
    random_headers = headers.generate()
    random_headers['Referer'] = "https://streamingcommunity.buzz/"
    random_headers['Origin'] = "https://streamingcommunity.buzz"
    #Do a request to get the ID of serie/move and it's slug in the URL
    response = await client.get(query, headers = random_headers, allow_redirects=True,impersonate = "chrome120")
    print(response)
    response = response.json()

    for item in response['data']:
        tid = item['id']
        slug = item['slug']
        type = item['type']
        if type == "tv":
            type = 0
        elif type == "movie":
            type = 1
        if type == ismovie: 
            #Added a Check to see if the result is what it is supposed to be
            if SC_FAST_SEARCH == "0":
                random_headers = headers.generate()
                random_headers['Referer'] = "https://streamingcommunity.buzz/"
                random_headers['Origin'] = "https://streamingcommunity.buzz"
                response = await client.get ( f'https://streamingcommunity.{SC_DOMAIN}/titles/{tid}-{slug}', headers = random_headers, allow_redirects=True,impersonate = "chrome120")
                pattern = r'<div[^>]*class="features"[^>]*>.*?<span[^>]*>(.*?)<\/span>'
                match = re.search(pattern, response.text)
                print(match.group(1).split("-")[0])
                first_air_year = match.group(1).split("-")[0]
                date = int(date)
                first_air_year = int(first_air_year)
                if first_air_year == date:
                    return tid,slug
            elif SC_FAST_SEARCH == "1":
                 return tid,slug
        else:
            print("Couldn't find anything")

        
async def get_film(tid,version,client):  
    random_headers = headers.generate()
    random_headers['Referer'] = "https://streamingcommunity.buzz/"
    random_headers['Origin'] = "https://streamingcommunity.buzz"
    random_headers['x-inertia'] = "true"
    random_headers['x-inertia-version'] = version
    #Access the iframe
    url = f'https://streamingcommunity.{SC_DOMAIN}/iframe/{tid}'
    response = await client.get(url, headers=random_headers, allow_redirects=True,impersonate = "chrome120")
    iframe = BeautifulSoup(response.text, 'lxml')
    #Get the link of iframe
    iframe = iframe.find('iframe').get("src")
    #Get the ID containted in the src of iframe
    vixid = iframe.split("/embed/")[1].split("?")[0]
    parsed_url = urlparse(iframe)
    query_params = parse_qs(parsed_url.query)
    random_headers = headers.generate()
    random_headers['Referer'] = "https://streamingcommunity.buzz/"
    random_headers['Origin'] = "https://streamingcommunity.buzz"
    random_headers['x-inertia'] = "true"
    random_headers['x-inertia-version'] = version
    #Get real token and expires by looking at the page in the iframe, vixcloud/embed
    resp = await client.get(iframe, headers = random_headers, allow_redirects=True,impersonate= "chrome120")
    soup=  BeautifulSoup(resp.text, "lxml")
    script = soup.find("body").find("script").text
    token = re.search(r"'token':\s*'(\w+)'", script).group(1)
    expires = re.search(r"'expires':\s*'(\d+)'", script).group(1)
    quality = re.search(r'"quality":(\d+)', script).group(1)
    #Example url  https://vixcloud.co/playlist/231315?b=1&token=bce060eec3dc9d1965a5d258dc78c964&expires=1728995040&rendition=1080p
    url = f'https://vixcloud.co/playlist/{vixid}?token={token}&expires={expires}'
    if 'canPlayFHD' in query_params:
       canPlayFHD = 'h=1'
       url += "&h=1"
    if 'b' in query_params:
       b = 'b=1'
       url += "&b=1"
    url720 = f'https://vixcloud.co/playlist/{vixid}'
    return url,url720,quality,

async def get_season_episode_id(tid,slug,season,episode,version,client):
    random_headers = headers.generate()
    random_headers['Referer'] = "https://streamingcommunity.buzz/"
    random_headers['Origin'] = "https://streamingcommunity.buzz"
    random_headers['x-inertia'] = "true"
    random_headers['x-inertia-version'] = version
    #Set some basic headers for the request  
      #Get episode ID 
    response = await client.get(f'https://streamingcommunity.{SC_DOMAIN}/titles/{tid}-{slug}/stagione-{season}', headers=random_headers, allow_redirects=True, impersonate = "chrome120")
    # Print the json got
    json_response = response.json().get('props', {}).get('loadedSeason', {}).get('episodes', [])
    for dict_episode in json_response:
        if dict_episode['number'] == episode:
            return dict_episode['id']

async def get_episode_link(episode_id,tid,version,client):
    #The parameters for the request
    random_headers = headers.generate()
    random_headers['Referer'] = "https://streamingcommunity.buzz/"
    random_headers['Origin'] = "https://streamingcommunity.buzz"
    params = {
                'episode_id': episode_id, 
                'next_episode': '1'
            }
    #Let's try to get the link from iframe source
        # Make a request to get iframe source
    response = await client.get(f"https://streamingcommunity.{SC_DOMAIN}/iframe/{tid}", params=params, headers = random_headers, allow_redirects=True, impersonate = "chrome120")

    # Parse response with BeautifulSoup to get iframe source
    soup = BeautifulSoup(response.text, "lxml")
    iframe = soup.find("iframe").get("src")
    vixid = iframe.split("/embed/")[1].split("?")[0]
    random_headers = headers.generate()
    random_headers['Referer'] = "https://streamingcommunity.buzz/"
    random_headers['Origin'] = "https://streamingcommunity.buzz"
    random_headers['x-inertia'] = "true"
    random_headers['x-inertia-version'] = version
    parsed_url = urlparse(iframe)
    query_params = parse_qs(parsed_url.query)
    #Get real token and expires by looking at the page in the iframe, vixcloud/embed
    resp = await client.get(iframe, headers = random_headers, allow_redirects=True, impersonate = "chrome120")
    soup=  BeautifulSoup(resp.text, "lxml")
    script = soup.find("body").find("script").text
    token = re.search(r"'token':\s*'(\w+)'", script).group(1)
    expires = re.search(r"'expires':\s*'(\d+)'", script).group(1)
    quality = re.search(r'"quality":(\d+)', script).group(1)
    #Example url  https://vixcloud.co/playlist/231315?b=1&token=bce060eec3dc9d1965a5d258dc78c964&expires=1728995040&rendition=1080p
    url = f'https://vixcloud.co/playlist/{vixid}?token={token}&expires={expires}'
    if 'canPlayFHD' in query_params:
       canPlayFHD = 'h=1'
       url += "&h=1"
    if 'b' in query_params:
       b = 'b=1'
       url += "&b=1"
    url720 = f'https://vixcloud.co/playlist/{vixid}'
    return url,url720,quality


async def streaming_community(imdb,client,SC_FAST_SEARCH):
    try:
        if Public_Instance == "1":
            Weird_Link = json.loads(Alternative_Link)
            link_post = random.choice(Weird_Link)
            response = await client.get(f"{link_post}fetch-data/{imdb}")
            url_streaming_community = response.headers.get('x-url-streaming-community')
            url_720_streaming_community = response.headers.get('x-url-720-streaming-community')
            quality_sc = response.headers.get('x-quality-sc')
            print(quality_sc,url_streaming_community)
            return url_streaming_community,url_720_streaming_community,quality_sc
        general = is_movie(imdb)
        ismovie = general[0]
        imdb_id = general[1]
        if ismovie == 0 : 
            season = int(general[2])
            episode = int(general[3])
            #Check if fast search is enabled or disabled
            if SC_FAST_SEARCH == "1":
                type = "StreamingCommunityFS"
                if "tt" in imdb:
                #Get showname
                    showname = await get_info_imdb(imdb_id,ismovie,type,client)
                    date = None
                else:
                    #I just set n season to None to avoid bugs, but it is not needed if Fast search is enabled
                    date = None
                    #else just equals them
                    tmdba = imdb_id.replace("tmdb:","")
                    showname = get_info_tmdb(tmdba,ismovie,type)
            elif SC_FAST_SEARCH == "0":
                type = "StreamingCommunity"
                tmdba = await get_TMDb_id_from_IMDb_id(imdb_id,client)
                showname,date = get_info_tmdb(tmdba,ismovie,type) 
        #HERE THE CASE IF IT IS A MOVIE
        else:
            if SC_FAST_SEARCH == "1":
                type = "StreamingCommunityFS"
                if "tt" in imdb:
                    #Get showname
                    date = None
                    showname = await get_info_imdb(imdb_id,ismovie,type,client)
                else:
                        date = None
                        tmdba = imdb_id.replace("tmdb:","")
                        showname = get_info_tmdb(tmdba,ismovie,type) 
            elif SC_FAST_SEARCH == "0":
                type = "StreamingCommunity"
                if "tt" in imdb:
                    #Get showname
                    showname,date = await get_info_imdb(imdb_id,ismovie,type,client)
                else:
                        tmdba = imdb_id.replace("tmdb:","")
                        showname,date = get_info_tmdb(tmdba,ismovie,type) 
        
        showname = showname.replace(" ", "+").replace("–", "+").replace("—","+")
        query = f'https://streamingcommunity.{SC_DOMAIN}/api/search?q={showname}'
        version = await get_version(client)
        tid,slug = await search(query,date,ismovie,client,SC_FAST_SEARCH)
        if ismovie == 1:
            #TID means temporaly ID
            url,url720,quality = await get_film(tid,version,client)
            print("MammaMia found results for StreamingCommunity")
            return url,url720,quality
        if ismovie == 0:
            #Uid = URL ID
            episode_id = await get_season_episode_id(tid,slug,season,episode,version,client)
            url,url720,quality = await get_episode_link(episode_id,tid,version,client)
            print("MammaMia found results for StreamingCommunity")
            return url,url720,quality
    except Exception as e:
        print("MammaMia: StreamingCommunity failed",e)
        return None,None,None
'''
async def test_animeworld():
    from curl_cffi.requests import AsyncSession
    async with AsyncSession() as client:
        # Replace with actual id, for example 'anime_id:episode' format
        test_id = "tt16288804:1:1"  # This is an example ID format
        results = await streaming_community(test_id, client,"0")

if __name__ == "__main__":
    import asyncio
    asyncio.run(test_animeworld())
'''