#!/usr/bin/env python

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

# Copyright 2007-2008 VEB IT

# See COPYING for distribution information.

# $Id$

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

from constants.VERSION import VERSION

__author__ = 'Pascal Volk <p.volk@veb-it.de>'

__version__ = VERSION

__revision__ = 'rev '+'$Rev$'.split()[1]

__date__ = '$Date$'.split()[1]

from Exceptions import VMMAccountException

from Domain import Domain

from Transport import Transport

from MailLocation import MailLocation

import constants.ERROR as ERR

class Account:

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

    def __init__(self, dbh, address, password=None):

        self._dbh = dbh

        self._base = None

        self._addr = address

        self._localpart = None

        self._name = None

        self._uid = 0

        self._gid = 0

        self._mid = 0

        self._tid = 0

        self._passwd = password

        self._setAddr(address)

        self._exists()

        if self._isAlias():

            raise VMMAccountException(

            ('There is already an alias with address »%s«' % address,

                ERR.ALIAS_EXISTS))

    def _exists(self):

        dbc = self._dbh.cursor()

        dbc.execute("SELECT uid, mid, tid FROM users \

WHERE gid=%s AND local_part=%s",

                self._gid, self._localpart)

        result = dbc.fetchone()

        dbc.close()

        if result is not None:

            self._uid, self._mid, self._tid = result

            return True

        else:

            return False

    def _isAlias(self):

        dbc = self._dbh.cursor()

        dbc.execute("SELECT gid FROM alias WHERE gid=%s AND address=%s",

                self._gid, self._localpart)

        gid = dbc.fetchone()

        dbc.close()

        if gid is not None:

            return True

        else:

            return False

    def _setAddr(self, address):

        self._localpart, d = address.split('@')

        dom = Domain(self._dbh, d)

        self._gid = dom.getID()

        if self._gid == 0:

            raise VMMAccountException(("Domain »%s« doesn't exist." % d,

                ERR.NO_SUCH_DOMAIN))

        self._base = dom.getDir()

        self._tid = dom.getTransportID()

    def _setID(self):

        dbc = self._dbh.cursor()

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

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

        dbc.close()

    def _prepare(self, maillocation):

        self._setID()

        self._mid = MailLocation(self._dbh, maillocation=maillocation).getID()

    def _switchState(self, state, service):

        if not isinstance(state, bool):

            return False

        if not service in ['smtp', 'pop3', 'imap', 'managesieve', 'all', None]:

            raise VMMAccountException(("Unknown service »%s«" % service,

                ERR.UNKNOWN_SERVICE))

        if self._uid < 1:

            raise VMMAccountException(("Account doesn't exists",

                ERR.NO_SUCH_ACCOUNT))

        dbc = self._dbh.cursor()

        if service in ['smtp', 'pop3', 'imap', 'managesieve']:

            dbc.execute(

                    "UPDATE users SET %s=%s WHERE local_part='%s' AND gid=%s"

                    % (service, state, self._localpart, self._gid))

        elif state:

            dbc.execute("UPDATE users SET smtp = TRUE, pop3 = TRUE,\

 imap = TRUE, managesieve = TRUE WHERE local_part = %s AND gid = %s",

                self._localpart, self._gid)

        else:

            dbc.execute("UPDATE users SET smtp = FALSE, pop3 = FALSE,\

 imap = FALSE, managesieve = FALSE WHERE local_part = %s AND gid = %s",

                self._localpart, self._gid)

        if dbc.rowcount > 0:

            self._dbh.commit()

        dbc.close()

    def getUID(self):

        return self._uid

    def getGID(self):

        return self._gid

    def getDir(self, directory):

        if directory == 'domain':

            return '%s' % self._base

        elif directory == 'home':

            return '%s/%i' % (self._base, self._uid)

    def enable(self, service=None):

        self._switchState(True, service)

    def disable(self, service=None):

        self._switchState(False, service)

    def save(self, maillocation, smtp, pop3, imap, managesieve):

        if self._uid < 1:

            self._prepare(maillocation)

            dbc = self._dbh.cursor()

            dbc.execute("""INSERT INTO users (local_part, passwd, uid, gid,\

 mid, tid, smtp, pop3, imap, managesieve)\

 VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)""",

                self._localpart, self._passwd, self._uid, self._gid, self._mid,

                self._tid, smtp, pop3, imap, managesieve)

            self._dbh.commit()

            dbc.close()

        else:

            raise VMMAccountException(('Account already exists.',

                ERR.ACCOUNT_EXISTS))

    def modify(self, what, value):

        if self._uid == 0:

            raise VMMAccountException(("Account doesn't exists",

                ERR.NO_SUCH_ACCOUNT))

        if what not in ['name', 'password', 'transport']:

            return False

        dbc = self._dbh.cursor()

        if what == 'password':

            dbc.execute("UPDATE users SET passwd=%s WHERE local_part=%s AND\

 gid=%s", value, self._localpart, self._gid)

        elif what == 'transport':

            self._tid = Transport(self._dbh, transport=value).getID()

            dbc.execute("UPDATE users SET tid=%s WHERE local_part=%s AND\

 gid=%s", self._tid, self._localpart, self._gid)

        else:

            dbc.execute("UPDATE users SET name=%s WHERE local_part=%s AND\

 gid=%s", value, self._localpart, self._gid)

        if dbc.rowcount > 0:

            self._dbh.commit()

        dbc.close()

    def getInfo(self):

        dbc = self._dbh.cursor()

        dbc.execute("SELECT name, uid, gid, mid, tid, smtp, pop3, imap, \

 managesieve FROM users WHERE local_part=%s AND gid=%s",

            self._localpart, self._gid)

        info = dbc.fetchone()

        dbc.close()

        if info is None:

            raise VMMAccountException(("Account doesn't exists",

                ERR.NO_SUCH_ACCOUNT))

        else:

            keys = ['name', 'uid', 'gid', 'maildir', 'transport', 'smtp',

                    'pop3', 'imap', 'managesieve']

            info = dict(zip(keys, info))

            for service in ['smtp', 'pop3', 'imap', 'managesieve']:

                if bool(info[service]):

                    info[service] = 'enabled'

                else:

                    info[service] = 'disabled'

            info['address'] = self._addr

            info['maildir'] = '%s/%s/%s' % (self._base, info['uid'],

                    MailLocation(self._dbh,

                        mid=info['maildir']).getMailLocation())

            info['transport'] = Transport(self._dbh,

                    tid=info['transport']).getTransport()

            return info

    def delete(self):

        if self._uid > 0:

            dbc = self._dbh.cursor()

            dbc.execute("DELETE FROM users WHERE gid=%s AND local_part=%s",

                    self._gid, self._localpart)

            if dbc.rowcount > 0:

                self._dbh.commit()

            dbc.close()

        else:

            raise VMMAccountException(("Account doesn't exists",

                ERR.NO_SUCH_ACCOUNT))

def getAccountByID(uid, dbh):

    try:

        uid = long(uid)

    except ValueError:

        raise VMMAccountException(('uid must be an int/long.',

            ERR.INVALID_AGUMENT))

    if uid < 1:

        raise VMMAccountException(('uid must be greater than 0.',

            ERR.INVALID_AGUMENT))

    dbc = dbh.cursor()

    dbc.execute("SELECT local_part||'@'||domains.domainname AS address, uid,\

 gid FROM users LEFT JOIN domains USING(gid) WHERE uid=%s", uid)

    info = dbc.fetchone()

    dbc.close()

    if info is None:

        raise VMMAccountException(("Account doesn't exists",

            ERR.NO_SUCH_ACCOUNT))

    keys = ['address', 'uid', 'gid']

    info = dict(zip(keys, info))

    return info