# -*- coding: UTF-8 -*-

# Copyright (c) 2008 - 2010, Pascal Volk

# See COPYING for distribution information.

"""Virtual Mail Manager's EmailAddress class to handle e-mail addresses."""

import re

import VirtualMailManager.constants.ERROR as ERR

from VirtualMailManager import chk_domainname

from VirtualMailManager.Exceptions import VMMEmailAddressException as VMMEAE

RE_LOCALPART = """[^\w!#$%&'\*\+-\.\/=?^_`{\|}~]"""

class EmailAddress(object):

    __slots__ = ('_localpart', '_domainname')

    def __init__(self, address):

        self._localpart = None

        self._domainname = None

        self.__chkAddress(address)

    def __eq__(self, other):

        if isinstance(other, self.__class__):

            return self._localpart == other._localpart\

                    and self._domainname == other._domainname

        return NotImplemented

    def __ne__(self, other):

        if isinstance(other, self.__class__):

            return self._localpart != other._localpart\

                    or self._domainname != other._domainname

        return NotImplemented

    def __repr__(self):

        return "EmailAddress('%s@%s')" % (self._localpart, self._domainname)

    def __str__(self):

        return "%s@%s" % (self._localpart, self._domainname)

    def __chkAddress(self, address):

        try:

            localpart, domain = address.split('@')

        except ValueError:

            raise VMMEAE(_(u"Missing '@' sign in e-mail address “%s”.") %

                         address, ERR.INVALID_ADDRESS)

        except AttributeError:

            raise VMMEAE(_(u"“%s” doesn't look like an e-mail address.") %

                         address, ERR.INVALID_ADDRESS)

        if len(domain) > 0:

            domain = chk_domainname(domain)

        else:

            raise VMMEAE(_(u"Missing domain name after “%s@”.") % localpart,

                         ERR.DOMAIN_NO_NAME)

        localpart = self.__chkLocalpart(localpart)

        self._localpart, self._domainname = localpart, domain

    def __chkLocalpart(self, localpart):

        """Validates the local-part of an e-mail address.

        Arguments:

        localpart -- local-part of the e-mail address that should be validated (str)

        """

        if len(localpart) < 1:

            raise VMMEAE(_(u'No local-part specified.'),

                ERR.LOCALPART_INVALID)

        if len(localpart) > 64:

            raise VMMEAE(_(u'The local-part “%s” is too long') %

                localpart, ERR.LOCALPART_TOO_LONG)

        ic = set(re.findall(RE_LOCALPART, localpart))

        if len(ic):

            ichrs = ''

            for c in ic:

                ichrs += u"“%s” " % c

            raise VMMEAE(_(u"The local-part “%(lpart)s” contains invalid\

 characters: %(ichrs)s") % {'lpart': localpart, 'ichrs': ichrs},

                ERR.LOCALPART_INVALID)

        return localpart