#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# Copyright 2007-2008 VEB IT
# See COPYING for distribution information.
# $Id$

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

from VirtualMailManager.constants.VERSION import VERSION

__author__ = 'Pascal Volk <p.volk@veb-it.de>'
__version__ = VERSION
__revision__ = 'rev '+'$Rev$'.split()[1]
__date__ = '$Date$'.split()[1]

import os
import sys
from getpass import getpass

from VirtualMailManager.VirtualMailManager import VirtualMailManager
from VirtualMailManager.Config import VMMConfig
import VirtualMailManager.Exceptions as VMME
import VirtualMailManager.constants.EXIT as EXIT

__prog__ = os.path.basename(sys.argv[0])

def usage(excode=0, errMsg=None):
    sys.stderr.write("""\
Usage: %s SUBCOMMAND OBJECT ARGS*
  short long
  subcommand            object           args (* = optional)

  da    domainadd       domain.tld       transport*
  di    domaininfo      domain.tld       detailed*
  dt    domaintransport domain.tld       transport force*
  dd    domaindelete    domain.tld       delalias*|deluser*|delall*
  ua    useradd         user@domain.tld  password*
  ui    userinfo        user@domain.tld  du*
  un    username        user@domain.tld  'Users Name'
  up    userpassword    user@domain.tld  password*
  ut    usertransport   user@domain.tld  transport
  u0    userdisable     user@domain.tld  smtp*|pop3*|imap*|managesieve*|all*
  u1    userenable      user@domain.tld  smtp*|pop3*|imap*|managesieve*|all*
  ud    userdelete      user@domain.tld
  aa    aliasadd        alias@domain.tld user@domain.tld
  ai    aliasinfo       alias@domain.tld
  ad    aliasdelete     alias@domain.tld user@domain.tld*
  gu    getuser         userid
  cf    configure                        section*
  h     help
  v     version

""" % __prog__)
    if not errMsg is None:
        sys.stderr.write('Error: %s\n' % errMsg)
    sys.exit(excode)

def getVMM():
    try:
        vmm = VirtualMailManager()
        return vmm
    except (VMME.VMMException, VMME.VMMNotRootException, VMME.VMMPermException,
            VMME.VMMConfigException), e:
        sys.stderr.write("\aERROR: %s\n" % e[0][0])
        sys.exit(e[0][1])

def configure():
    try:
        if len(sys.argv) < 3:
            vmm.configure()
        else:
            vmm.configure(sys.argv[2])
    except (EOFError, KeyboardInterrupt):
        sys.stderr.write('\nOuch!\n')
        sys.exit(EXIT.USER_INTERRUPT)
    except VMME.VMMConfigException, e:
        sys.stderr.write(str(e))
        sys.exit(EXIT.CONF_ERROR)
    sys.exit(0)

def _readpass():
    clear0 = ''
    clear1 = '1'
    while clear0 != clear1:
        while len(clear0) < 1:
            clear0 = getpass(prompt='Enter new password: ')
            if len(clear0) < 1:
                sys.stderr.write('Sorry, empty passwords are not permitted\n')
        clear1 = getpass(prompt='Retype new password: ')
        if clear0 != clear1:
            clear0 = ''
            sys.stderr.write('Sorry, passwords do not match\n')
    return clear0

def _getOrder():
    global argc
    order = ()
    if sys.argv[1] in ['di', 'domaininfo']:
        order = (('domainname', 0), ('gid', 1), ('transport', 0),
                ('domaindir', 0), ('accounts', 0), ('aliases', 0))
    elif sys.argv[1] in ['ui', 'userinfo']:
        if argc == 3:
            order = (('address', 0), ('name', 0), ('uid', 1), ('gid', 1),
                    ('transport', 0), ('maildir', 0), ('smtp', 1), ('pop3', 1),
                    ('imap', 1), ('managesieve', 1))
        else:
            order = (('address', 0), ('name', 0), ('uid', 1), ('gid', 1),
                    ('transport', 0), ('maildir', 0), ('disk usage', 0),
                    ('smtp', 1), ('pop3', 1), ('imap', 1), ('managesieve', 1))
    elif sys.argv[1] in ['gu', 'getuser']:
        order = (('uid', 1), ('gid', 1), ('address', 0))
    return order


def _printInfo(info, title):
    msg = title+' information'
    print '%s\n%s' % (msg, '-'*len(msg))
    for k,u in _getOrder():
        if u:
            print '\t%s: %s' % (k.upper().ljust(15, '.'), info[k])
        else:
            print '\t%s: %s' % (k.title().ljust(15, '.'), info[k])
    print

def _printUsers(users, title):
    msg = 'Available '+title
    print '%s\n%s' % (msg, '-'*len(msg))
    if len(users) > 0:
        for user in users:
            print '\t%s' % user
    else:
        print '\tNone'
    print

def _printAliases(alias, targets):
    msg = 'Alias information'
    print '%s\n%s' % (msg, '-'*len(msg))
    print '\tMail for %s goes to:' % alias
    if len(targets) > 0:
        for target in targets:
            print '\t     -> %s' % target
    else:
        print '\tNone'
    print

def domain_add():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing domain name.')
    elif argc < 4:
        vmm.domain_add(sys.argv[2].lower())
    else:
        vmm.domain_add(sys.argv[2].lower(), sys.argv[3])

def domain_delete():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing domain name.')
    elif argc < 4:
        vmm.domain_delete(sys.argv[2].lower())
    else:
        vmm.domain_delete(sys.argv[2].lower(), sys.argv[3])

def domain_info():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing domain name.')
    elif argc < 4:
        _printInfo(vmm.domain_info(sys.argv[2].lower()), 'Domain')
    else:
        infos = vmm.domain_info(sys.argv[2].lower(), sys.argv[3])
        _printInfo(infos[0], 'Domain')
        _printUsers(infos[1], 'accounts')
        _printUsers(infos[2], 'aliases')

def domain_transport():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing domain name and new transport.')
    if argc < 4:
        usage(EXIT.MISSING_ARGS, 'Missing new transport.')
    elif argc < 5:
        vmm.domain_transport(sys.argv[2].lower(), sys.argv[3])
    else:
        vmm.domain_transport(sys.argv[2].lower(), sys.argv[3], sys.argv[4])

def user_add():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address.')
    elif argc < 4:
        password = _readpass()
    else:
        password = sys.argv[3]
    vmm.user_add(sys.argv[2].lower(), password)

def user_delete():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address.')
    else:
        vmm.user_delete(sys.argv[2].lower())

def user_info():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address.')
    elif argc < 4:
        _printInfo(vmm.user_info(sys.argv[2].lower()), 'Account')
    else:
        _printInfo(vmm.user_info(sys.argv[2].lower(), True), 'Account')

def user_name():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address and users name.')
    if argc < 4:
        usage(EXIT.MISSING_ARGS, 'Missing users name.')
    else:
        vmm.user_name(sys.argv[2].lower(), sys.argv[3])

def user_transport():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address and transport.')
    if argc <4:
        usage(EXIT.MISSING_ARGS, 'Missing transport.')
    else:
        vmm.user_transport(sys.argv[2].lower(), sys.argv[3])

def user_enable():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address.')
    elif argc < 4:
        vmm.user_enable(sys.argv[2].lower())
    else:
        vmm.user_enable(sys.argv[2].lower(), sys.argv[3].lower())

def user_disable():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address.')
    elif argc < 4:
        vmm.user_disable(sys.argv[2].lower())
    else:
        vmm.user_disable(sys.argv[2].lower(), sys.argv[3].lower())

def user_password():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing e-mail address.')
    elif argc < 4:
        password = _readpass()
    else:
        password = sys.argv[3]
    vmm.user_password(sys.argv[2].lower(), password)

def alias_add():
    global argc
    if argc < 4:
        usage(EXIT.MISSING_ARGS, 'Missing alias address and destination.')
    else:
        vmm.alias_add(sys.argv[2].lower(), sys.argv[3])

def alias_info():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing alias address')
    else:
        _printAliases(sys.argv[2], vmm.alias_info(sys.argv[2].lower()))

def alias_delete():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing alias address')
    elif argc < 4:
        vmm.alias_delete(sys.argv[2].lower())
    else:
        vmm.alias_delete(sys.argv[2].lower(), sys.argv[3].lower())

def user_byID():
    global argc
    if argc < 3:
        usage(EXIT.MISSING_ARGS, 'Missing userid')
    else:
        _printInfo(vmm.user_byID(sys.argv[2]), 'Account')

def showWarnings():
    if vmm.hasWarnings():
        print '\nWarnings:'
        for w in vmm.getWarnings():
            print " * ",w

#def main():
if __name__ == '__main__':
    argc = len(sys.argv)
    if argc < 2:
        usage(EXIT.MISSING_ARGS) # -> exit
    vmm = getVMM()
    try:
        if sys.argv[1] in ['cf',  'configure'] or not vmm.setupIsDone():
            configure()
    except VMME.VMMConfigException, e:
        sys.stderr.write(str(e))
        sys.exit(EXIT.CONF_ERROR)
    except VMME.VMMException, e:
        sys.stderr.write("\aERROR: %s\n" % e[0][0])
        sys.exit(e[0][1])
    try:
        if sys.argv[1] in ['da', 'domainadd']:
            domain_add()
        elif sys.argv[1] in ['di', 'domaininfo']:
            domain_info()
        elif sys.argv[1] in ['dt', 'domaintransport']:
            domain_transport()
        elif sys.argv[1] in ['dd', 'domaindelete']:
            domain_delete()
        elif sys.argv[1] in ['ua', 'useradd']:
            user_add()
        elif sys.argv[1] in ['ui', 'userinfo']:
            user_info()
        elif sys.argv[1] in ['un', 'username']:
            user_name()
        elif sys.argv[1] in ['up', 'userpassword']:
            user_password()
        elif sys.argv[1] in ['ut', 'usertransport']:
            user_transport()
        elif sys.argv[1] in ['u0', 'userdisable']:
            user_disable()
        elif sys.argv[1] in ['u1', 'userenable']:
            user_enable()
        elif sys.argv[1] in ['ud', 'userdelete']:
            user_delete()
        elif sys.argv[1] in ['aa', 'aliasadd']:
            alias_add()
        elif sys.argv[1] in ['ai', 'aliasinfo']:
            alias_info()
        elif sys.argv[1] in ['ad', 'aliasdelete']:
            alias_delete()
        elif sys.argv[1] in ['gu', 'getuser']:
            user_byID()
        elif sys.argv[1] in ['h', 'help']:
            usage()
        elif sys.argv[1] in ['v', 'version']:
            print "%s, version %s (%s from %s)\n" % (__prog__, __version__,
                    __revision__, __date__)
        else:
            usage(EXIT.UNKNOWN_COMMAND,
                    'Unknown subcommand: »%s«' % sys.argv[1])
        showWarnings()
    except (EOFError, KeyboardInterrupt):
        sys.stderr.write('\nOuch!\n')
        sys.exit(EXIT.USER_INTERRUPT)
    except VMME.VMMException, e:
        sys.stderr.write("\aERROR: %s\n" % e[0][0])
        sys.exit(e[0][1])