from Common import *

from MiscUtils.Funcs import positive_id

# If this is Python 2.2 or greater, import the standard Cookie module
# as CookieEngine. Otherwise, import WebUtils.Cookie as CookieEngine.
# This is because there is a nasty bug in the Cookie.py module
# included in Python 2.1 and earlier.

pyVer = getattr(sys, 'version_info', None)
if pyVer and pyVer[:2] >= (2, 2):
    # Get Python's Cookie module. We have to do some work since
    # it has the same name as we do. So we'll strip out anything
    # from the path that might cause us to import from the WebKit
    # directory, then import Cookie using that restricted path --
    # that ought to ensure that we're using Python's module.
    import imp
    path = []
    thisDir = os.path.abspath(os.path.dirname(__file__))
    for dir in sys.path:
        if not dir or dir == '.':
            continue
        if os.path.abspath(dir) == thisDir:
            continue
        path.append(dir)
    file, pathname, description = imp.find_module('Cookie', path)
    try:
        CookieEngine = imp.load_module('Cookie', file, pathname, description)
    finally:
        if file:
            file.close()
else:
    # For Python versions < 2.2, we are including a copy of the
    # standard Cookie.py module from Python 2.2, but modified to
    # work with Python 1.5.2 and up.
    from WebUtils import Cookie
    CookieEngine = Cookie
    del Cookie


class Cookie(Object):
    """Delicious cookies.

    Cookie is used to create cookies that have additional attributes
    beyond their value.

    Note that web browsers don't typically send any information with
    the cookie other than its value. Therefore `HTTPRequest.cookie`
    simply returns a value such as an integer or a string.

    When the server sends cookies back to the browser, it can send
    a cookie that simply has a value, or the cookie can be
    accompanied by various attributes (domain, path, max-age, ...)
    as described in `RFC 2109`_. Therefore, in HTTPResponse,
    `setCookie` can take either an instance of the Cookie class,
    as defined in this module, or a value.

    Note that Cookies values get pickled (see the `pickle` module),
    so you can set and get cookies that are integers, lists,
    dictionaries, etc.

    .. _`RFC 2109`: ftp://ftp.isi.edu/in-notes/rfc2109.txt

    """

    # Future
    #
    # * This class should provide error checking in the setFoo()
    #   methods. Or maybe our internal Cookie implementation
    #   already does that?
    # * This implementation is probably not as efficient as it
    #   should be, [a] it works and [b] the interface is stable.
    #   We can optimize later.


    ## Init ##

    def __init__(self, name, value):
        """Create a cookie.

        Properties other than `name` and `value` are set with methods.

        """

        self._cookies = CookieEngine.SimpleCookie()
        self._name = name
        self._value = value
        self._cookies[name] = value
        self._cookie = self._cookies[name]

    def __repr__(self):
        return ('%s(id=0x%x, name=%r, domain=%r, '
                'path=%r, value=%r, expires=%r, maxAge=%r)'
            % (self.__class__.__name__, positive_id(self),
                self.name(), self.domain(),
                self.path(), self.value(), self.expires(), self.maxAge()))


    ## Accessors ##

    def comment(self):
        return self._cookie['comment']

    def domain(self):
        return self._cookie['domain']

    def maxAge(self):
        return self._cookie['max-age']

    def expires(self):
        return self._cookie['expires']

    def name(self):
        return self._name

    def path(self):
        return self._cookie['path']

    def isSecure(self):
        return self._cookie['secure']

    def value(self):
        return self._value

    def version(self):
        return self._cookie['version']


    ## Setters ##

    def setComment(self, comment):
        self._cookie['comment'] = comment

    def setDomain(self, domain):
        self._cookie['domain'] = domain

    def setExpires(self, expires):
        self._cookie['expires'] = expires

    def setMaxAge(self, maxAge):
        self._cookie['max-age'] = maxAge

    def setPath(self, path):
        self._cookie['path'] = path

    def setSecure(self, secure=True):
        self._cookie['secure'] = secure

    def setValue(self, value):
        self._value = value
        self._cookies[self._name] = value

    def setVersion(self, version):
        self._cookie['version'] = version


    ## Misc ##

    def delete(self):
        """Delete a cookie.

        When sent, this should delete the cookie from the user's
        browser, by making it empty, expiring it in the past,
        and setting its max-age to 0. One of these will delete
        the cookie for any browser (which one actually works
        depends on the browser).

        """
        self._value = ''
        self._cookie['expires'] = "Mon, 01-Jan-1900 00:00:00 GMT"
        self._cookie['max-age'] = 0

    def headerValue(self):
        """Return header value.

        Returns a string with the value that should be used
        in the HTTP headers.

        """
        items = self._cookies.items()
        assert len(items) == 1
        return items[0][1].OutputString()