from tmdbv3api import TMDb, Movie, TV
import requests
import logging 
from bs4 import BeautifulSoup,SoupStrainer
from datetime import datetime
import dateparser
from convert import get_TMDb_id_from_IMDb_id
from loadenv import load_env
from info import get_info, is_movie
#Get domain
_, _,SC_DOMAIN,_ ,_= load_env()

headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.10; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',
    'Accept-Language': 'en-US,en;q=0.5'
}

def search(query):
    #Do a request to get the ID of serie/move and it's slug in the URL
    response = requests.get(query).json()
    for item in response['data']:
        tid = item['id']
        slug = item['slug']
        return tid,slug

        
def get_film(tid):
    #Access the iframe
    url = f'https://streamingcommunity.{SC_DOMAIN}/iframe/{tid}'
    response = requests.get(url, headers=headers)
    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]
    #Build the url with 1080p, to get the highest resolution
    url = f'https://vixcloud.co/playlist/{vixid}'
    return url

def get_season_episode_id(tid,slug,season,episode):
    #Set some basic headers for the request
    headers = {
            'user-agent': "Mozilla/5.0 (Windows NT 10.10; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3",
            'x-inertia': 'true', 
            #Version of streaming community
            'x-inertia-version': 'c1bdf1189cb68fca28151c8e909bd053'
        }   
      #Get episode ID 
    response = requests.get(f'https://streamingcommunity.{SC_DOMAIN}/titles/{tid}-{slug}/stagione-{season}', headers=headers)
    # 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']

def get_episode_link(episode_id,tid):
    #The parameters for the request
    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 = requests.get(f"https://streamingcommunity.{SC_DOMAIN}/iframe/{tid}", params=params)

    # 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]
    url = f"https://vixcloud.co/playlist/{vixid}"
    return url


def streaming_community(imdb):
    general = is_movie(imdb)
    ismovie = general[0]
    imdb_id = general[1]
    if ismovie == 0 : 
        season = int(general[2])
        episode = int(general[3])
    if "tt" in imdb:
        #If it is a imbd then conver it to tmbd
        tmdba  = get_TMDb_id_from_IMDb_id(imdb_id)
        showname = requests.get(f'https://v3.sg.media-imdb.com/suggestion/a/{imdb_id}.json')
        showname = showname.json()
        showname = showname['d'][0]['l']
    else:
        #else just equals them
        tmdba = imdb_id.replace("tmdb:","")
        type = "StreamingCommunity"
        showname = get_info(tmdba,ismovie,type)
    showname = showname.replace(" ", "+").replace("–", "+").replace("—","+")
    query = f'https://streamingcommunity.{SC_DOMAIN}/api/search?q={showname}'
    tid,slug = search(query)
    if ismovie == 1:
        #TID means temporaly ID
        url = get_film(tid)
        print(url)
        return url
    if ismovie == 0:
        #Uid = URL ID
        episode_id = get_season_episode_id(tid,slug,season,episode)
        url = get_episode_link(episode_id,tid)
        print(url)
        return url