# urllibcompat.py - adapters to ease using urllib2 on Py2 and urllib on Py3

#

# Copyright 2017 Google, Inc.

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

from . import pycompat

_sysstr = pycompat.sysstr

class _pycompatstub(object):

    def __init__(self):

        self._aliases = {}

    def _registeraliases(self, origin, items):

        """Add items that will be populated at the first access"""

        items = map(_sysstr, items)

        self._aliases.update(

    def _registeralias(self, origin, attr, name):

        """Alias ``origin``.``attr`` as ``name``"""

        self._aliases[_sysstr(name)] = (origin, _sysstr(attr))

    def __getattr__(self, name):

        try:

            origin, item = self._aliases[name]

        except KeyError:

            raise AttributeError(name)

        self.__dict__[name] = obj = getattr(origin, item)

        return obj

httpserver = _pycompatstub()

urlreq = _pycompatstub()

urlerr = _pycompatstub()

if pycompat.ispy3:

    import urllib.parse

    urlreq._registeralias(urllib.parse, "parse_qs", "parseqs")

    urlreq._registeralias(urllib.parse, "parse_qsl", "parseqsl")

    urlreq._registeralias(urllib.parse, "unquote_to_bytes", "unquote")

    import urllib.request

    import urllib.response

    import urllib.error

    import http.server

    # urllib.parse.quote() accepts both str and bytes, decodes bytes

    # (if necessary), and returns str. This is wonky. We provide a custom

    # implementation that only accepts bytes and emits bytes.

    def quote(s, safe=r'/'):

        # bytestr has an __iter__ that emits characters. quote_from_bytes()

        # does an iteration and expects ints. We coerce to bytes to appease it.

        if isinstance(s, pycompat.bytestr):

            s = bytes(s)

        s = urllib.parse.quote_from_bytes(s, safe=safe)

        return s.encode('ascii', 'strict')

    # urllib.parse.urlencode() returns str. We use this function to make

    # sure we return bytes.

    def urlencode(query, doseq=False):

    urlreq.quote = quote

    urlreq.urlencode = urlencode

    def getfullurl(req):

        return req.full_url

    def gethost(req):

        return req.host

    def getselector(req):

        return req.selector

    def getdata(req):

        return req.data

    def hasdata(req):

        return req.data is not None

else:

    import BaseHTTPServer

    import CGIHTTPServer

    import SimpleHTTPServer

    import urllib2

    import urllib

    import urlparse

    urlreq._registeralias(urlparse, "parse_qs", "parseqs")

    urlreq._registeralias(urlparse, "parse_qsl", "parseqsl")

    def gethost(req):

        return req.get_host()

    def getselector(req):

        return req.get_selector()

    def getfullurl(req):

        return req.get_full_url()

    def getdata(req):

        return req.get_data()

    def hasdata(req):

        return req.has_data()