# -*- coding: UTF-8 -*-
#/*
# *      Copyright (C) 2013 Libor Zoubek
# *
# *
# *  This Program is free software; you can redistribute it and/or modify
# *  it under the terms of the GNU General Public License as published by
# *  the Free Software Foundation; either version 2, or (at your option)
# *  any later version.
# *
# *  This Program is distributed in the hope that it will be useful,
# *  but WITHOUT ANY WARRANTY; without even the implied warranty of
# *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# *  GNU General Public License for more details.
# *
# *  You should have received a copy of the GNU General Public License
# *  along with this program; see the file COPYING.  If not, write to
# *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
# *  http://www.gnu.org/copyleft/gpl.html
# *
# */
import re,random,util,sys,os,traceback,hashlib
import urllib
#Python 2
try: 
    import cookielib
    import urllib2
    import urlparse
    import elementtree.ElementTree as ET
    #import sys
    # reload(sys)  # Reload does the trick!
    # sys.setdefaultencoding('UTF8')
#Python 3
except:
    import http.cookiejar
    cookielib = http.cookiejar
    urllib2 = urllib.request
    urlparse = urllib.parse
    # urllib = urllib.parse
    import xml.etree.ElementTree as ET

from provider import ContentProvider
from provider import ResolveException
from crypto.md5crypt import md5crypt

def loadurl(url, req, headers):
    req = urllib.parse.urlencode(req).encode('utf-8')
    req = urllib.request.Request(url, req, headers=headers)
    with urllib.request.urlopen(req) as f:
        try:
            if f.getcode()==200:
                response = f.read()
                return response
        except:
            return False


class WebshareContentProvider(ContentProvider):

    def __init__(self,username=None,password=None,filter=None,tmp_dir='.'):
        ContentProvider.__init__(self,'webshare.cz','https://webshare.cz/',username,password,filter,tmp_dir)
        opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookielib.LWPCookieJar()))
        urllib2.install_opener(opener)
        self.token = ''

    def capabilities(self):
        return ['login', 'search', 'resolve']

    def search(self, keyword):
        return self.list('what=%s'%urllib.parse.quote(keyword))

    def _create_request(self, url, base):
        args = dict(urlparse.parse_qsl(url))
        headers = {'X-Requested-With':'XMLHttpRequest','Accept':'text/xml; charset=UTF-8','Referer':self.base_url}
        req = base.copy()
        for key in req:
            if key in args.keys():
                req[key] = args[key]
        return headers, req


    def login(self):
        if not self.username and not self.password:
            return True # fall back to free account
        elif self.username and self.password and len(self.username)>0 and len(self.password)>0:
            self.info('Login user=%s, pass=*****' % (self.username))
            # get salt
            headers,req = self._create_request('',{'username_or_email':self.username})
            data = util.post(self._url('api/salt/'),req,headers=headers)
            xml = ET.fromstring(data)
            if not xml.find('status').text == 'OK':
                self.error('Server returned error status, response: %s' % data)
                return False
            salt = xml.find('salt').text
            # create hashes
            password = hashlib.sha1(md5crypt(self.password, salt).encode('utf-8')).hexdigest()
            digest = hashlib.md5((self.username + ':Webshare:' + self.password).encode('utf-8')).hexdigest()
            # login
            headers,req = self._create_request('',{'username_or_email':self.username,'password':password,'digest':digest,'keep_logged_in':1})
            data = util.post(self._url('api/login/'),req,headers=headers)
            xml = ET.fromstring(data)
            if not xml.find('status').text == 'OK':
                self.error('Server returned error status, response: %s' % data)
                return False
            self.token = xml.find('token').text
            self.info('Login successfull')
            return True
        return False


    def list(self, url):
        self.info(' ==== list ==========')
        result = []
        headers, req = self._create_request( url, {'what':'','offset':0,'limit':25,'category':'','sort':'','wst':self.token} )
        self.info(' ==== headers ==========' )
        self.info(headers)
        self.info(type(headers))
        self.info(' ==== req ==========')
        self.info(req)
        self.info(type(req))

        data = loadurl(self._url('api/search/'), req, headers=headers)
        # with urllib.request.urlopen(req, data=data) as f:
        #     resp = f.read()

        self.info(data)
        xml = ET.fromstring(data)
        if not xml.find('status').text == 'OK':
            self.error('Server returned error status, response: %s' % data)
            return []
        total = int(xml.find('total').text)
        for file in xml.findall('file'):
            item = self.video_item()
            item['title'] = file.find('name').text
            item['url'] = 'ident=%s' % file.find('ident').text
            size = int(file.find('size').text)
            item['size'] = '%d MB' % (int(size)/1024/1024)
            img = file.find('img').text
            if img:
                item['img'] = self._url(img)
            self._filter(result,item)
        listed = int(req['limit']) + int(req['offset'])
        if total > listed:
            req['offset'] = listed
            item = self.dir_item()
            item['type'] = 'next'
            item['url'] = urllib.parse.urlencode(req)
            result.append(item)
        return result


    def resolve(self,item,captcha_cb=None,select_cb=None):
        item = item.copy()        
        util.init_urllib()
        headers, req = self._create_request(item['url'],{'ident':'','wst':self.token})
        #data = util.post(self._url('api/file_link/'),req,headers=headers)
        data = loadurl(self._url('api/file_link/'), req, headers=headers)
        self.info(data)
        xml = ET.fromstring(data)
        if not xml.find('status').text == 'OK':
            self.error('Server returned error status, response: %s' % data)
            raise ResolveException(xml.find('message').text)
        item['url'] = xml.find('link').text
        return item