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

# Copyright (c) 2007 - 2011, Pascal Volk

# See COPYING for distribution information.

"""

    VirtualMailManager.account

    ~~~~~~~~~~~~~~~~~~~~~~~~~~

    Virtual Mail Manager's Account class to manage e-mail accounts.

"""

from VirtualMailManager.common import version_str

from VirtualMailManager.constants import \

     ACCOUNT_EXISTS, ACCOUNT_MISSING_PASSWORD, ALIAS_PRESENT, \

     INVALID_ARGUMENT, INVALID_MAIL_LOCATION, NO_SUCH_ACCOUNT, \

     NO_SUCH_DOMAIN, UNKNOWN_SERVICE, VMM_ERROR

from VirtualMailManager.domain import Domain

from VirtualMailManager.emailaddress import EmailAddress

from VirtualMailManager.errors import VMMError, AccountError as AErr

from VirtualMailManager.maillocation import MailLocation

from VirtualMailManager.password import pwhash

from VirtualMailManager.quotalimit import QuotaLimit

from VirtualMailManager.transport import Transport

__all__ = ('SERVICES', 'Account', 'get_account_by_uid')

SERVICES = ('imap', 'pop3', 'smtp', 'sieve')

_ = lambda msg: msg

cfg_dget = lambda option: None

class Account(object):

    """Class to manage e-mail accounts."""

    __slots__ = ('_addr', '_dbh', '_domain', '_mail', '_new', '_passwd',

                 '_qlimit', '_transport', '_uid')

    def __init__(self, dbh, address):

        """Creates a new Account instance.

        When an account with the given *address* could be found in the

        database all relevant data will be loaded.

        Arguments:

        `dbh` : pyPgSQL.PgSQL.Connection

          A database connection for the database access.

        `address` : VirtualMailManager.EmailAddress.EmailAddress

          The e-mail address of the (new) Account.

        """

        if not isinstance(address, EmailAddress):

            raise TypeError("Argument 'address' is not an EmailAddress")

        self._addr = address

        self._dbh = dbh

        self._domain = Domain(self._dbh, self._addr.domainname)

        if not self._domain.gid:

            # TP: Hm, what “quotation marks” should be used?

            # If you are unsure have a look at:

            # http://en.wikipedia.org/wiki/Quotation_mark,_non-English_usage

            raise AErr(_(u"The domain '%s' does not exist.") %

                       self._addr.domainname, NO_SUCH_DOMAIN)

        self._uid = 0

        self._mail = None

        self._qlimit = self._domain.quotalimit

        self._transport = self._domain.transport

        self._passwd = None

        self._new = True

        self._load()

    def __nonzero__(self):

        """Returns `True` if the Account is known, `False` if it's new."""

        return not self._new

    def _load(self):

        """Load 'uid', 'mid', 'qid' and 'tid' from the database and set

        _new to `False` - if the user could be found. """

        dbc = self._dbh.cursor()

        dbc.execute('SELECT uid, mid, qid, tid FROM users WHERE gid = %s AND '

                    'local_part=%s', (self._domain.gid, self._addr.localpart))

        result = dbc.fetchone()

        dbc.close()

        if result:

            self._uid, _mid, _qid, _tid = result

            if _qid != self._qlimit.qid:

                self._qlimit = QuotaLimit(self._dbh, qid=_qid)

            if _tid != self._transport.tid:

                self._transport = Transport(self._dbh, tid=_tid)

            self._mail = MailLocation(self._dbh, mid=_mid)

            self._new = False

    def _set_uid(self):

        """Set the unique ID for the new Account."""

        assert self._uid == 0

        dbc = self._dbh.cursor()

        dbc.execute("SELECT nextval('users_uid')")

        self._uid = dbc.fetchone()[0]

        dbc.close()

    def _prepare(self, maillocation):

        """Check and set different attributes - before we store the

        information in the database.

        """

        if maillocation.dovecot_version > cfg_dget('misc.dovecot_version'):

            raise AErr(_(u"The mailbox format '%(mbfmt)s' requires Dovecot "

                         u">= v%(version)s.") % {

                       'mbfmt': maillocation.mbformat,

                       'version': version_str(maillocation.dovecot_version)},

                       INVALID_MAIL_LOCATION)

        if not maillocation.postfix and \

          self._transport.transport.lower() in ('virtual:', 'virtual'):

            raise AErr(_(u"Invalid transport '%(transport)s' for mailbox "

                         u"format '%(mbfmt)s'.") %

                       {'transport': self._transport,

                        'mbfmt': maillocation.mbformat}, INVALID_MAIL_LOCATION)

        self._mail = maillocation

        self._set_uid()

    def _update_services(self, activate, *services):

        """Activate or deactivate the Account's services.

        Arguments:

        `activate`: bool

          When `True` the Account's user will be able to login to the

          services, otherwise the login will fail.

        `*services`

          No or one or more of the services: imap, pop3, smtp and sieve

        """

        self._chk_state()

        if services:

            services = set(services)

            for service in services:

                if service not in SERVICES:

                    raise AErr(_(u"Unknown service: '%s'") % service,

                               UNKNOWN_SERVICE)

        else:

            services = SERVICES

        state = ('FALSE', 'TRUE')[activate]

        sql = 'UPDATE users SET %s WHERE uid = %u' % (

                    (' = %(s)s, '.join(services) + ' = %(s)s') % {'s': state},

                    self._uid)

        if 'sieve' in services and \

           cfg_dget('misc.dovecot_version') < 0x10200b02:

            sql = sql.replace('sieve', 'managesieve')

        dbc = self._dbh.cursor()

        dbc.execute(sql)

        if dbc.rowcount > 0:

            self._dbh.commit()

        dbc.close()

    def _update_tables(self, column, value):

        """Update various columns in the users table.

        Arguments:

        `column` : basestring

          Name of the table column. Currently: qid and tid

        `value` : long

          The referenced key

        """

        if column not in ('qid', 'tid'):

            raise ValueError('Unknown column: %r' % column)

        dbc = self._dbh.cursor()

        dbc.execute('UPDATE users SET %s = %%s WHERE uid = %%s' % column,

                    (value, self._uid))

        if dbc.rowcount > 0:

            self._dbh.commit()

        dbc.close()

    def _count_aliases(self):

        """Count all alias addresses where the destination address is the

        address of the Account."""

        dbc = self._dbh.cursor()

        dbc.execute('SELECT COUNT(destination) FROM alias WHERE destination '

                    '= %s', (str(self._addr),))

        a_count = dbc.fetchone()[0]

        dbc.close()

        return a_count

    def _chk_state(self):

        """Raise an AccountError if the Account is new - not yet saved in the

        database."""

        if self._new:

            raise AErr(_(u"The account '%s' does not exist.") % self._addr,

                       NO_SUCH_ACCOUNT)

    @property

    def address(self):

        """The Account's EmailAddress instance."""

        return self._addr

    @property

    def domain(self):

        """The Domain to which the Account belongs to."""

        if self._domain:

            return self._domain

        return None

    @property

    def gid(self):

        """The Account's group ID."""

        if self._domain:

            return self._domain.gid

        return None

    @property

    def home(self):

        """The Account's home directory."""

        if not self._new:

            return '%s/%s' % (self._domain.directory, self._uid)

        return None

    @property

    def mail_location(self):

        """The Account's MailLocation."""

        return self._mail

    @property

    def uid(self):

        """The Account's unique ID."""

        return self._uid

    def set_password(self, password):

        """Set a password for the new Account.

        If you want to update the password of an existing Account use

        Account.modify().

        Argument:

        `password` : basestring

          The password for the new Account.

        """

        if not self._new:

            raise AErr(_(u"The account '%s' already exists.") % self._addr,

                       ACCOUNT_EXISTS)

        if not isinstance(password, basestring) or not password:

            raise AErr(_(u"Could not accept password: '%s'") % password,

                       ACCOUNT_MISSING_PASSWORD)

        self._passwd = password

    def enable(self, *services):

        """Enable all or the given service/s for the Account.

        Possible *services* are: 'imap', 'pop3', 'sieve' and 'smtp'.

        When all services should be enabled, give no service name.

        Arguments:

        `*services` : basestring

          No or one or more of the services 'imap', 'pop3', 'smtp', and

          'sieve'.

        """

        self._update_services(True, *services)

    def disable(self, *services):

        """Disable all or the given service/s for the Account.

        For more information see: Account.enable()."""

        self._update_services(False, *services)

    def save(self):

        """Save the new Account in the database."""

        if not self._new:

            raise AErr(_(u"The account '%s' already exists.") % self._addr,

                       ACCOUNT_EXISTS)

        if not self._passwd:

            raise AErr(_(u"No password set for account: '%s'") % self._addr,

                       ACCOUNT_MISSING_PASSWORD)

        if cfg_dget('misc.dovecot_version') >= 0x10200b02:

            sieve_col = 'sieve'

        else:

            sieve_col = 'managesieve'

        self._prepare(MailLocation(self._dbh, mbfmt=cfg_dget('mailbox.format'),

                                   directory=cfg_dget('mailbox.root')))

        dbc = self._dbh.cursor()

        dbc.execute('INSERT INTO users (local_part, passwd, uid, gid, mid, '

                    'qid, tid, smtp, pop3, imap, %s) VALUES' % (sieve_col,) + \

                    '(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)',

                    (self._addr.localpart,

                     pwhash(self._passwd, user=self._addr), self._uid,

                     self._domain.gid, self._mail.mid, self._qlimit.qid,

                     self._transport.tid, cfg_dget('account.smtp'),

                     cfg_dget('account.pop3'), cfg_dget('account.imap'),

                     cfg_dget('account.sieve')))

        self._dbh.commit()

        dbc.close()

        self._new = False

    def modify(self, field, value):

        """Update the Account's *field* to the new *value*.

        Possible values for *field* are: 'name', 'password'.

        Arguments:

        `field` : basestring

          The attribute name: 'name' or 'password'

        `value` : basestring

          The new value of the attribute.

        """

        if field not in ('name', 'password'):

            raise AErr(_(u"Unknown field: '%s'") % field, INVALID_ARGUMENT)

        self._chk_state()

        dbc = self._dbh.cursor()

        if field == 'password':

            dbc.execute('UPDATE users SET passwd = %s WHERE uid = %s',

                        (pwhash(value, user=self._addr), self._uid))

        else:

            dbc.execute('UPDATE users SET name = %s WHERE uid = %s',

                        (value, self._uid))

        if dbc.rowcount > 0:

            self._dbh.commit()

        dbc.close()

    def update_quotalimit(self, quotalimit):

        """Update the user's quota limit.

        Arguments:

        `quotalimit` : VirtualMailManager.quotalimit.QuotaLimit

          the new quota limit of the domain.

        """

        if cfg_dget('misc.dovecot_version') < 0x10102f00:

            raise VMMError(_(u'PostgreSQL-based dictionary quota requires '

                             u'Dovecot >= v1.1.2.'), VMM_ERROR)

        self._chk_state()

        assert isinstance(quotalimit, QuotaLimit)

        if quotalimit == self._qlimit:

            return

        self._update_tables('qid', quotalimit.qid)

        self._qlimit = quotalimit

    def update_transport(self, transport):

        """Sets a new transport for the Account.

        Arguments:

        `transport` : VirtualMailManager.transport.Transport

          the new transport

        """

        self._chk_state()

        assert isinstance(transport, Transport)

        if transport == self._transport:

            return

        if transport.transport.lower() in ('virtual', 'virtual:') and \

           not self._mail.postfix:

            raise AErr(_(u"Invalid transport '%(transport)s' for mailbox "

                         u"format '%(mbfmt)s'.") %

                       {'transport': transport, 'mbfmt': self._mail.mbformat},

                       INVALID_MAIL_LOCATION)

        self._update_tables('tid', transport.tid)

        self._transport = transport

    def get_info(self):

        """Returns a dict with some information about the Account.

        The keys of the dict are: 'address', 'gid', 'home', 'imap'

        'mail_location', 'name', 'pop3', 'sieve', 'smtp', transport', 'uid',

        'uq_bytes', 'uq_messages', 'ql_bytes', and 'ql_messages'.

        """

        self._chk_state()

        if cfg_dget('misc.dovecot_version') >= 0x10200b02:

            sieve_col = 'sieve'

        else:

            sieve_col = 'managesieve'

        dbc = self._dbh.cursor()

        dbc.execute('SELECT name, smtp, pop3, imap, %s, CASE WHEN bytes IS '

                    'NULL THEN 0 ELSE bytes END, CASE WHEN messages IS NULL '

                    'THEN 0 ELSE messages END FROM users LEFT JOIN userquota '

                    'USING (uid) WHERE users.uid = %u' % (sieve_col,

                        self._uid))

        info = dbc.fetchone()

        dbc.close()

        if info:

            keys = ('name', 'smtp', 'pop3', 'imap', sieve_col, 'uq_bytes',

                    'uq_messages')

            info = dict(zip(keys, info))

            for service in keys[1:5]:

                if info[service]:

                    # TP: A service (pop3/imap) is enabled/usable for a user