import base64
import os
from urllib import addinfourl, pathname2url
from urlparse import urlparse,urlunparse
import urllib2
from cgi import parse_qs
from httpsession import HTTPSession
from BitQueue.log import get_logger

urlopener = None

def urldecode(data):
    dict = parse_qs(data)
    ret = {}
    for key in dict.keys():
        ret[key] = dict[key][0]
    return ret

class AURLOpener:
    def __init__(self,user=None,passwd=None):
        self.user = user
        self.passwd = passwd
        self.log = get_logger()

    def set_default(self):
        global urlopener
        urlopener = self

    def urlopen(self,url,data=None,referer=None):
        self.log.debug('urlopen: %s %s\n' % (url,str(data)))
        if self.user and self.passwd:
            req = urllib2.Request(url)
            raw = '%s:%s' % (self.user,self.passwd)
            auth = 'Basic %s' % base64.encodestring(raw).strip()
            req.add_header('Authorization',auth)
            if not referer:
                referer = url
            if referer:
                req.add_header('Referer',referer)
            opener = urllib2.build_opener()
            web = opener.open(req,data)
        else:
            web = urllib2.urlopen(url,data)
        return web

class CookieAURLOpener(AURLOpener):
    def __init__(self,user=None,passwd=None):
        AURLOpener.__init__(self,user,passwd)
        self.session = HTTPSession(debug_level=0)
        self.session.add_header('User-Agent',
                                'Mozilla/4.0 (compatible; Windows; Linux)')
        if self.user and self.passwd:
            self.session.set_basic_auth(self.user,self.passwd)
        self.log = get_logger()

    def urlopen(self,url,data=None,method='get',referer=None):
        self.log.debug('urlopen: %s %s\n' % (url,str(data)))
        scheme = urlparse(url)[0]
        if not scheme in ['http','https']:
            return urllib2.urlopen(url,data)
        if method == 'post':
            req = self.session.post(url)
        else:
            req = self.session.get(url)

        if not referer:
            referer = url
        if referer:
            req.add_header('Referer',referer)

        if data:
            params = urldecode(data)
            req.add_params(params)
        rc,msg,hdr = req.getreply()
        fd = req.getfile()
        return addinfourl(fd,hdr,req.url)

def urlopen(uri,data=None,user=None,passwd=None,method='get',referer=None):
    global urlopener
    if not urlopener:
        urlopener = CookieAURLOpener(user,passwd)
    if not data:
        ut = list(urlparse(uri))
        data = ut[4]
        ut[4] = ''
        ut[5] = ''
        uri = urlunparse(tuple(ut))
    if uri.find('://') == -1:
       uri = 'file://' + pathname2url(os.path.realpath(uri))
    return urlopener.urlopen(uri,data,method=method,referer=referer)