mailcommander.py

#!/usr/bin/env python2
# coding: utf-8
#
# Copyright (C) 2011 by Edgar Merino (http://devio.us/~emerino)
#
# Licensed under the Artistic License 2.0 (The License).
# You may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
#    http://www.perlfoundation.org/artistic_license_2_0
#
# THE PACKAGE IS PROVIDED BY THE COPYRIGHT HOLDER AND CONTRIBUTORS "AS
# IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES. THE IMPLIED
# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR
# NON-INFRINGEMENT ARE DISCLAIMED TO THE EXTENT PERMITTED BY YOUR LOCAL
# LAW. UNLESS REQUIRED BY LAW, NO COPYRIGHT HOLDER OR CONTRIBUTOR WILL
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, OR CONSEQUENTIAL
# DAMAGES ARISING IN ANY WAY OUT OF THE USE OF THE PACKAGE, EVEN IF
# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

__version__ = "0.2"
__author__ = "emerino <emerino at gmail dot com>"


import sys
import os
import logging
import getopt

from devpower.mail import UnicodeParser
from devpower.util import Properties

from auth import AuthManager, AuthKey, PUBLIC_KEY
from scanner import MailScanner, ScannerException
from commander import Commander, AuthException
from notifications import EmailNotifier
from util import Parsers, find_commands, AuthNotification, SuccessNotification

def main():
    try:
        # Load system configuration
        conf = parse_arguments()

        # Send logging output to a file if specified
        configure_log(conf["log.file"])

        # Look for available commands
        commands = find_commands("cmds/", conf["home.dir"])

        # Core components
        auth = AuthManager(conf["home.dir"] + "/users.auth")
        commander = Commander(auth)
        scanner = MailScanner(Parsers(commands))
        notifier = EmailNotifier(conf["smtp.host"], conf["smtp.port"])

        # Read the email from stdin
        email = UnicodeParser().parse(sys.stdin)

        cmd_id, data, sprops, authkey = scanner.scan(email)
        auth_users = [ key.user for key in auth.keys[cmd_id] if key.user ]

        command = commands[cmd_id]
        output = commander.execute(command, data, authkey)

        if output:
            notifier.send(SuccessNotification(conf["notification.sender"], \
                                              auth_users, command, sprops))
    except AuthException, err:
        notifier.send(AuthNotification(cmd_id, conf["commander.address"], \
                                       auth_users, email, data))
    except ScannerException, err:
        logging.error(str(err))
        sys.stderr.write(str(err))
        sys.exit(1) # Exit with err code 1 so the email is bounced
    except Exception, err:
        #import traceback
        #traceback.print_exc()
        logging.error(str(err))
        usage()

def usage():
    print """
Usage: mailcommander.py <options>

Available options:
    -c, --config        path to configuration file 
                        default: /etc/MailCommander/commander.conf
    -h, --help          show this message

The email source should be passed through stdin.  """

def parse_arguments():
    short_opts = "c:h"
    long_opts = ["config=", "help"]

    config = {}
    config_file = "/etc/MailCommander/commander.conf"

    if not os.path.exists(config_file):
        config_file = None

    default_config = { # defaults
        "home.dir": "/etc/MailCommander",
        "lang": "es",
        "commander.address": None,
        "notification.sender": None,
        "smtp.host": "localhost",
        "smtp.port": 25,
        "log.file": None
    }

    opts, args = getopt.getopt(sys.argv[1:], short_opts, long_opts)

    for opt, arg in opts:
        if opt in ("--help", "-h"):
            usage()
            sys.exit(0) # =(
        elif opt in ("--config", "-c"):
           config_file = arg 

    if config_file:
        config = Properties(config_file, separator=".")
        
    for opt, value in default_config.items():
        if opt not in config:
            config[opt] = value

    if not os.path.exists(config["home.dir"]):
        raise Exception("home dir '%s' does not exist!" % (config["home.dir"]))
    elif config["home.dir"][-1] == "/":
        config["home.dir"] = config["home.dir"][:-1]

    return config

def configure_log(filename=None):
    if filename:
        logging.basicConfig(filename=filename, \
                            format="[%(asctime)s]: %(message)s", \
                            datefmt="%m/%d/%Y %I:%M:%S %p")
    else:
        logging.basicConfig(format="[%(asctime)s]: %(message)s", \
                            datefmt="%m/%d/%Y %I:%M:%S %p")


if __name__ == "__main__":
    main()