#-*- coding: utf-8 -*-
#Vstream https://github.com/Kodi-vStream/venom-xbmc-addons
#
from resources.lib.handler.requestHandler import cRequestHandler
from resources.lib.parser import cParser
from resources.hosters.hoster import iHoster
from resources.lib.comaddon import dialog, VSlog, xbmc
from resources.lib.handler.premiumHandler import cPremiumHandler

from resources.lib.config import GestionCookie

import urllib, urllib2
import re

UA = 'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:39.0) Gecko/20100101 Firefox/39.0'

class cHoster(iHoster):

    def __init__(self):
        self.__sDisplayName = '1Fichier'
        self.__sFileName = self.__sDisplayName
        self.__sHD = ''
        self.oPremiumHandler = False

    def getDisplayName(self):
        return  self.__sDisplayName

    def setDisplayName(self, sDisplayName):
        self.__sDisplayName = sDisplayName + ' [COLOR violet]' + self.__sDisplayName + '[/COLOR]'

    def setFileName(self, sFileName):
        self.__sFileName = sFileName

    def getFileName(self):
        return self.__sFileName

    def getPluginIdentifier(self):
        return 'onefichier'

    def setHD(self, sHD):
        self.__sHD = ''

    def getHD(self):
        return self.__sHD

    def isDownloadable(self):
        return True

    def isJDownloaderable(self):
        return True

    def __getIdFromUrl(self, sUrl):
        #http://kzu0y3.1fichier.com/
        #https://1fichier.com/?s6gdceia9y
        id = sUrl.replace('https://', '')
        id = id.replace('http://', '')
        id = id.replace('1fichier.com/?', '')
        id = id.replace('.1fichier.com', '')
        id = id.replace('/', '')

        return id

    def setUrl(self, sUrl):
        self.__sUrl = str(sUrl)

    def checkUrl(self, sUrl):
        return True

    def __getUrl(self, media_id):
        return

    def getMediaLink(self):

        self.oPremiumHandler = cPremiumHandler(self.getPluginIdentifier())
        #print self.oPremiumHandler.isPremiumModeAvailable()

        import sys
        if ('site=cDownload&function' not in sys.argv[2]) and not (self.oPremiumHandler.isPremiumModeAvailable()):
            oDialog = dialog().VSok('Pas de streaming sans premium\nPour voir le film passer par l\'option "DL et Visualiser" du menu contextuel.')
            return False, False

        if (self.oPremiumHandler.isPremiumModeAvailable()):
            return self.__getMediaLinkByPremiumUser()

        return self.__getMediaLinkForGuest()


    def __getMediaLinkByPremiumUser(self):
        api_call = False

        #if not self.oPremiumHandler.Authentificate():
        if not self.Authentication_test():
            return False, False

        url = 'https://1fichier.com/?' + self.__getIdFromUrl(self.__sUrl)
        #La partie ci-dessous permet d'utiliser l'option "Forcer l'affichage du menu pour les téléchargements" permettant notamment de choisir depuis l'interface web de télécharger ou d'ajouter un fichier.
        #Pour cela, on va ajouter le paramètre e=1 (cf. https://1fichier.com/hlp.html#dev ) à la requête permettant d'obtenir le lien direct
        url2 = url + '&e=1'

        sHtmlContent = self.oPremiumHandler.GetHtml(url2)

        if "L'accès à ce fichier est protégé par un mot de passe" in sHtmlContent:
            VSlog("Fichier protege par MDP")

            cookie = self.oPremiumHandler.AddCookies().replace('Cookie=', '', 1)
            data = {'pass':'annuaire-telechargement.com'}
            #Seul le Cookie est nécessaire, néanmoins autant rendre les headers cohérents
            headers = {
                'User-Agent': UA,
                'Host': '1fichier.com',
                'Referer': url ,
                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
                'Accept-Language': 'fr,fr-FR;q=0.8,en-US;q=0.5,en;q=0.3',
                'Cookie': cookie,
                'Content-Length': str(len(urllib.urlencode(data))),
                'Content-Type': 'application/x-www-form-urlencoded',
                'Connection' :  'keep-alive'
            }
            request = urllib2.Request(url, urllib.urlencode(data), headers)
            response = urllib2.urlopen(request)

            url = response.geturl()
            response.close()

            #tempo de 5s
            xbmc.sleep(5*1000)

        else:
            VSlog("Fichier non protege par MDP")
            url = 'https://1fichier.com/?' + self.__getIdFromUrl(self.__sUrl)

        VSlog( url )

        #Pour tester premiere partie
        #return False,False

        api_call = url + '|' + self.oPremiumHandler.AddCookies() + '&User-Agent=' + UA

        #VSlog( api_call )

        if (api_call):
            return True, api_call

        return False, False

    def __getMediaLinkForGuest(self):
        import random
        api_call = False
        url = 'https://1fichier.com/?' + self.__getIdFromUrl(self.__sUrl)

        headers = {'User-Agent': UA ,
                   'Host': '1fichier.com',
                   'Referer': self.__sUrl ,
                   'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
                   'Accept-Language': 'fr,fr-FR;q=0.8,en-US;q=0.5,en;q=0.3'
                   #'Content-Type': 'application/x-www-form-urlencoded'
                   }

        adcode = random.uniform(000.000000000, 999.999999999)

        Mode = ''
        #Mode = {'dl_no_ssl' : 'on' , 'dlinline' : 'on'}
        Mode = {'dl_no_ssl' : 'on' ,'adzone' : adcode}
        postdata = urllib.urlencode( Mode )

        req = urllib2.Request(url, postdata, headers)

        try:
            #import ssl
            #context = ssl._create_unverified_context()
            #response = urllib2.urlopen(req,context=context)
            response = urllib2.urlopen(req)
        except URLError, e:
            print e.read()
            print e.reason

        sHtmlContent = response.read()
        response.close()

        #fh = open('c:\\test.txt', "w")
        #fh.write(sHtmlContent)
        #fh.close()

        api_call = self.GetMedialinkDL(sHtmlContent)

        if (api_call):
            return True, api_call

        return False, False

    def GetMedialinkDL(self, sHtmlContent):

        oParser = cParser()
        api_call = False

        sPattern = 'Vous devez attendre encore [0-9]+ minutes'
        aResult = oParser.parse(sHtmlContent, sPattern)
        if (aResult[0] == True):
            dialog().VSinfo('Erreur - Limitation %s' % aResult[1][0])
            return False

        sPattern = '<a href="([^<>"]+?)"  style="float:none;margin:auto;font-weight:bold;padding: 10px;margin: 10px;font-size:\+1\.6em;border:2px solid red" class="ok btn-general btn-orange">'
        aResult = oParser.parse(sHtmlContent, sPattern)

        #print aResult

        if (aResult[0] == True):
            #xbmc.sleep(1*1000)
            #VSlog(  aResult[1][0] )
            api_call = aResult[1][0] + '|User-Agent=' + UA# + '&Referer=' + self.__sUrl
            return api_call

        return False

    def Authentication_test(self):
        if (GestionCookie().Readcookie('onefichier') != ''):
            return True
        else:
            Connection = self.oPremiumHandler.Authentificate()
            if (Connection == False):
                return False
            else:
                return True