#!/usr/bin/env python

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

# Copyright 2007 - 2009, VEB IT

# See COPYING for distribution information.

"""This is the vmm main script."""

import gettext

from time import strftime, strptime

from VirtualMailManager import *

from VirtualMailManager.VirtualMailManager import VirtualMailManager

import VirtualMailManager.Exceptions as VMME

import VirtualMailManager.constants.EXIT as EXIT

def usage(excode=0, errMsg=None):

    u_head = _(u"""\

Usage: %s SUBCOMMAND OBJECT ARGS*

  short long

  subcommand               object             args (* = optional)\n""")\

          % __prog__

    u_body = u"""\

  da    domainadd          domain.tld         transport*

  di    domaininfo         domain.tld         details*

  dt    domaintransport    domain.tld         transport force*

  dd    domaindelete       domain.tld         delalias*|deluser*|delall*

  ada   aliasdomainadd     aliasdomain.tld    domain.tld

  adi   aliasdomaininfo    aliasdomain.tld

  ads   aliasdomainswitch  aliasdomain.tld    domain.tld

  add   aliasdomaindelete  aliasdomain.tld

  ua    useradd            user@domain.tld    password*

  ui    userinfo           user@domain.tld    details*

  un    username           user@domain.tld    'Users Name'

  up    userpassword       user@domain.tld    password*

  ut    usertransport      user@domain.tld    transport

  u0    userdisable        user@domain.tld    service*

  u1    userenable         user@domain.tld    service*

  ud    userdelete         user@domain.tld    delalias*

  aa    aliasadd           alias@domain.tld   user@domain.tld

  ai    aliasinfo          alias@domain.tld

  ad    aliasdelete        alias@domain.tld   user@domain.tld*

  ra    relocatedadd       exaddr@domain.tld  user@domain.tld

  ri    relocatedinfo      exaddr@domain.tld

  rf    relocateddelete    exaddr@domain.tld

  gu    getuser            userid

  ld    listdomains                           pattern*

  cf    configure                             section*

  h     help

  v     version

"""

    if excode > 0:

        if errMsg is None:

            w_err(excode, u_head, u_body)

        else:

            w_err(excode, u_head, u_body, '%s: %s\n' % (_(u'Error'), errMsg))

    else:

        w_std(u_head, u_body)

        os.sys.exit(excode)

def get_vmm():

    try:

        vmm = VirtualMailManager()

        return vmm

    except (VMME.VMMException, VMME.VMMNotRootException, VMME.VMMPermException,

            VMME.VMMConfigException), e:

        w_err(e.code(), "%s: %s\n" % (_(u'Error'), e.msg()))

def _getOrder():

    order = ()

    if vmm.cfgGetInt('misc', 'dovecotvers') > 11:

        sieve_name = u'sieve'

    else:

        sieve_name = u'managesieve'

    if argv[1] in (u'di', u'domaininfo'):

        order = ((u'domainname', 0), (u'gid', 1), (u'transport', 0),

                (u'domaindir', 0), (u'aliasdomains', 0), (u'accounts', 0),

                (u'aliases', 0), (u'relocated', 0))

    elif argv[1] in (u'ui', u'userinfo'):

        if argc == 4 and argv[3] != u'aliases'\

        or vmm.cfgGetBoolean('maildir', 'diskusage'):

            order = ((u'address', 0), (u'name', 0), (u'uid', 1), (u'gid', 1),

                    (u'transport', 0), (u'maildir', 0), (u'disk usage', 0),

                    (u'smtp', 1), (u'pop3', 1), (u'imap', 1), (sieve_name, 1))

        else:

            order = ((u'address', 0), (u'name', 0), (u'uid', 1), (u'gid', 1),

                    (u'transport', 0), (u'maildir', 0), (u'smtp', 1),

                    (u'pop3', 1), (u'imap', 1), (sieve_name, 1))

    elif argv[1] in (u'gu', u'getuser'):

        order = ((u'uid', 1), (u'gid', 1), (u'address', 0))

    return order

def _printInfo(info, title):

    msg = u'%s %s' % (title, _(u'information'))

    w_std (u'%s\n%s' % (msg, u'-'*len(msg)))

    for k,u in _getOrder():

        if u:

            w_std(u'\t%s: %s' % (k.upper().ljust(15, u'.'), info[k]))

        else:

            w_std(u'\t%s: %s' % (k.title().ljust(15, u'.'), info[k]))

    print

def _printList(alist, title):

    msg = u'%s %s' % (_(u'Available'), title)

    w_std(u'%s\n%s' % (msg, u'-'*len(msg)))

    if len(alist) > 0:

        if title != _(u'alias domains'):

            for val in alist:

                w_std(u'\t%s' % val)

        else:

            for dom in alist:

                if not dom.startswith('xn--'):

                    w_std(u'\t%s' % dom)

                else:

                    w_std(u'\t%s (%s)' % (dom, vmm.ace2idna(dom)))

    else:

        w_std(_(u'\tNone'))

    print

def _printAliases(alias, targets):

    msg = _(u'Alias information')

    w_std(u'%s\n%s' % (msg, u'-'*len(msg)))

    w_std(_(u'\tMail for %s will be redirected to:') % alias)

    if len(targets) > 0:

        for target in targets:

            w_std(u'\t     * %s' % target)

    else:

        w_std(_(u'\tNone'))

    print

def _printRelocated(addr_dest):

    msg = _(u'Relocated information')

    w_std(u'%s\n%s' % (msg, u'-'*len(msg)))

    w_std(_(u'\tUser “%(addr)s” has moved to “%(dest)s”') % addr_dest)

    print

def _formatDom(domain, main=True):

    if domain.startswith('xn--'):

        domain = u'%s (%s)' % (domain, vmm.ace2idna(domain))

    if main:

        return u'\t[+] %s' %  domain

    else:

        return u'\t[-]     %s' % domain

def _printDomList(dids, domains):

    if argc < 3:

        msg = _('Available domains')

    else:

        msg = _('Matching domains')

    w_std('%s\n%s' % (msg, '-'*len(msg)))

    if not len(domains):

        w_std(_('\tNone'))

    else:

        for id in dids:

            if domains[id][0] is not None:

                w_std(_formatDom(domains[id][0]))

            if len(domains[id]) > 1:

                for alias in domains[id][1:]:

                    w_std(_formatDom(alias, main=False))

    print

def _printAliasDomInfo(info):

    msg = _('Alias domain information')

    for k in ['alias', 'domain']:

        if info[k].startswith('xn--'):

            info[k] = "%s (%s)" % (info[k], vmm.ace2idna(info[k]))

    w_std('%s\n%s' % (msg, '-'*len(msg)))

    w_std(

        _('\tThe alias domain %(alias)s belongs to:\n\t    * %(domain)s')%info)

    print

def configure():

    if need_setup or len(argv) < 3:

        vmm.configure()

    else:

        vmm.configure(argv[2])

def domain_add():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing domain name.'))

    elif argc < 4:

        vmm.domainAdd(argv[2].lower())

    else:

        vmm.domainAdd(argv[2].lower(), argv[3])

def domain_delete():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing domain name.'))

    elif argc < 4:

        vmm.domainDelete(argv[2].lower())

    else:

        vmm.domainDelete(argv[2].lower(), argv[3])

def domain_info():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing domain name.'))

    try:

        if argc < 4:

            _printInfo(vmm.domainInfo(argv[2].lower()), _(u'Domain'))

        else:

            details = argv[3].lower()

            infos = vmm.domainInfo(argv[2].lower(), details)

            _printInfo(infos[0], _(u'Domain'))

            if details == u'accounts':

                _printList(infos[1], _(u'accounts'))

            elif details == u'aliasdomains':

                _printList(infos[1], _(u'alias domains'))

            elif details == u'aliases':

                _printList(infos[1], _(u'aliases'))

            elif details == u'relocated':

                _printList(infos[1], _(u'relocated users'))

            else:

                _printList(infos[1], _(u'alias domains'))

                _printList(infos[2], _(u'accounts'))

                _printList(infos[3], _(u'aliases'))

                _printList(infos[4], _(u'relocated users'))

    except VMME.VMMDomainException, e:

        if e.code() is ERR.DOMAIN_ALIAS_EXISTS:

            w_std(plan_a_b % {'subcommand': u'aliasdomaininfo',

                'address': argv[2].lower()})

            alias_domain_info()

        else:

            raise e

def domain_transport():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing domain name and new transport.'))

    if argc < 4:

        usage(EXIT.MISSING_ARGS, _(u'Missing new transport.'))

    elif argc < 5:

        vmm.domainTransport(argv[2].lower(), argv[3])

    else:

        vmm.domainTransport(argv[2].lower(), argv[3], argv[4])

def alias_domain_add():

    if argc < 3:

        usage(EXIT.MISSING_ARGS,

                _(u'Missing alias domain name and target domain name.'))

    elif argc < 4:

        usage(EXIT.MISSING_ARGS, _(u'Missing target domain name.'))

    else:

        vmm.aliasDomainAdd(argv[2].lower(), argv[3].lower())

def alias_domain_info():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing alias domain name.'))

    try:

        _printAliasDomInfo(vmm.aliasDomainInfo(argv[2].lower()))

    except VMME.VMMAliasDomainException, e:

        if e.code() is ERR.ALIASDOMAIN_ISDOMAIN:

            w_std(plan_a_b % {'subcommand': u'domaininfo',

                'address': argv[2].lower()})

            argv[1] = u'di' # necessary manipulation to get the order

            domain_info()

        else:

            raise e

def alias_domain_switch():

    if argc < 3:

        usage(EXIT.MISSING_ARGS,

                _(u'Missing alias domain name and target domain name.'))

    elif argc < 4:

        usage(EXIT.MISSING_ARGS, _(u'Missing target domain name.'))

    else:

        vmm.aliasDomainSwitch(argv[2].lower(), argv[3].lower())

def alias_domain_delete():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing alias domain name.'))

    else:

        vmm.aliasDomainDelete(argv[2].lower())

def user_add():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing e-mail address.'))

    elif argc < 4:

        password = None

    else:

        password = argv[3]

    vmm.userAdd(argv[2].lower(), password)

def user_delete():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing e-mail address.'))

    elif argc < 4:

        vmm.userDelete(argv[2].lower())

    else:

        vmm.userDelete(argv[2].lower(), argv[3].lower())

def user_info():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing e-mail address.'))

    try:

        if argc < 4:

            _printInfo(vmm.userInfo(argv[2].lower()), u'Account')

        else:

            arg3 = argv[3].lower()

            info = vmm.userInfo(argv[2].lower(), arg3)

            if arg3 in ['aliases', 'full']:

                _printInfo(info[0], u'Account')

                _printList(info[1], _(u'alias addresses'))

            else:

                _printInfo(info, u'Account')

    except VMME.VMMAccountException, e:

        if e.code() is ERR.ALIAS_EXISTS:

            w_std(plan_a_b % {'subcommand': u'aliasinfo',

                'address': argv[2].lower()})

            alias_info()

        elif e.code() is ERR.RELOCATED_EXISTS:

            w_std(plan_a_b % {'subcommand': u'relocatedinfo',

                'address': argv[2].lower()})

            relocated_info()

        else:

            raise e

def user_name():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing e-mail address and users name.'))

    if argc < 4:

        usage(EXIT.MISSING_ARGS, _('Missing users name.'))

    else:

        vmm.userName(argv[2].lower(), argv[3])

def user_transport():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing e-mail address and transport.'))

    if argc <4:

        usage(EXIT.MISSING_ARGS, _(u'Missing transport.'))

    else:

        vmm.userTransport(argv[2].lower(), argv[3])

def user_enable():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing e-mail address.'))

    elif argc < 4:

        vmm.userEnable(argv[2].lower())

    else:

        vmm.userEnable(argv[2].lower(), argv[3].lower())

def user_disable():

    if argc < 3:

        usage(EXIT.MISSING_ARGS, _(u'Missing e-mail address.'))

    elif argc < 4:

        vmm.userDisable(argv[2].lower())

    else: