import re,xbmc,urllib,urlparse
from ..scraper import Scraper
import requests
from ..common import clean_title,clean_search, filter_host, get_rd_domains
User_Agent = 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/55.0.2883.87 Safari/537.36'

# kept movies off

class twoddl(Scraper):
    domains = ['http://twoddl.co']
    name = "TwoDDL"
    sources = []

    def __init__(self):
        self.base_link = 'http://twoddl.co'
        self.sources = []

    # def scrape_movie(self, title, year, imdb, debrid=False):
        # try:            
            # start_url = "%s/search/%s+%s/" % (self.base_link, title.replace(' ','+').lower(),year)
            
            # headers = {'User_Agent':User_Agent}
            # OPEN = open_url(start_url,headers=headers,timeout=5).content
            
            # content = re.compile('<h2><a href="([^"]+)"',re.DOTALL).findall(OPEN)
            # for url in content:
                # self.get_source(url)                        
            # return self.sources
        # except Exception, argument:
            # return self.sources           
            

    def scrape_episode(self,title, show_year, year, season, episode, imdb, tvdb, debrid = False):
        try:
            if not debrid:
                return []
            season_url = "0%s"%season if len(season)<2 else season
            episode_url = "0%s"%episode if len(episode)<2 else episode
            sea_epi ='s%se%s'%(season_url,episode_url)
            
            start_url = "%s/?s=%s+%s" % (self.base_link, title.replace(' ','+').lower(),sea_epi)
            headers = {'User_Agent':User_Agent}
            OPEN = requests.get(start_url,headers=headers,timeout=5).content
            content = re.compile('<h2><a href="([^"]+)"',re.DOTALL).findall(OPEN)
            for url in content:
                if clean_title(title).lower() in clean_title(url).lower():
                    self.get_source(url)                        
            return self.sources
        except Exception, argument:
            return self.sources  

            
    def get_source(self,url):
        try:        
            headers = {'User_Agent':User_Agent}
            links = requests.get(url,headers=headers,timeout=3).content   
            LINK = re.compile('href="([^"]+)" rel="nofollow"',re.DOTALL).findall(links)
                        
            for url in LINK:
                if '.rar' not in url:
                    if '.srt' not in url:
                        if '1080' in url:
                            res = '1080p'
                        elif '720' in url:
                            res = '720p'
                        elif 'HDTV' in url:
                            res = 'HD'
                        else:
                            pass

                        host = url.split('//')[1].replace('www.','')
                        host = host.split('/')[0].lower()

                        # if not filter_host(host):
                            # continue
                                # if debrid == "true":
                        rd_domains = get_rd_domains()
                        if host in rd_domains:
                            self.sources.append({'source': host,'quality': res,'scraper': self.name,'url': url,'direct': False, 'debridonly': True})

        except:pass

