#!/usr/bin/python2
import logging
from argparse import ArgumentParser
from alias_server.config import config
import daemon
import daemon.pidfile
from alias_server.component import ObjectComponent
import os.path
import sys

if sys.version_info < (3, 0):
    reload(sys)
    sys.setdefaultencoding('utf8')
else:
    raw_input = input

if __name__ == '__main__':
    commandline = ArgumentParser(description = 'Connect the alias \
                                component to a given server')
    commandline.add_argument('-p', '--port',
                             help = 'Port to connect to',
                             type = int)
    commandline.add_argument('-s', '--secret',
                             help = 'password')
    commandline.add_argument('-n', '--name',
                             help = 'Name the component will have')
    commandline.add_argument('-r', '--root',
                             help = 'Root directory of the user files')
    commandline.add_argument('-c', '--config',
                             help = 'Name of the config file to use')
    commandline.add_argument('-d', '--debug',
                             help = 'Set log level to DEBUG',
                             action = 'store_const',
                             const = logging.DEBUG,
                             default = logging.INFO)
    commandline.add_argument('-o', '--host',
                             help = 'Host to connect to')
    commandline.add_argument('-b', '--background',
                             help = 'run the server in the background',
                             action = 'store_true')
    commandline.add_argument('--logfile',
                             help = 'location of the log file (default /var/log/${name}.log')
    commandline.add_argument('--pidfile',
                             help = 'location of the pid file (default /var/run/${name}.pid')
    args = commandline.parse_args()

    if args.config is None:
        config.name = args.name
        config.port = args.port
        config.secret = args.secret
        config.root = args.root
        config.host = args.host
        config.background = args.background
        config.logfile = args.logfile
        config.pidfile = args.pidfile
    else:
        filename = args.config
        config.read(filename)
    if config.logfile is None:
        config.logfile = os.path.join('/var/log/', config.name + '.log')
    if config.pidfile is None:
        config.pidfile = os.path.join('/var/run/', config.name + '.pid')
   
    #set up the root logger
    logging.getLogger('').setLevel(args.debug)
    formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
    if config.background:
        #save logs in a file
        fh = logging.FileHandler(config.logfile)
        fh.setFormatter(formatter)
        logging.getLogger('').addHandler(fh)
    else:
        #save logs to the console
        ch = logging.StreamHandler()
        ch.setFormatter(formatter)
        logging.getLogger('').addHandler(ch)

    if config.background:
        context = daemon.DaemonContext(detach_process = True,
                                       pidfile = daemon.pidfile.TimeoutPIDLockFile(config.pidfile,10),
                                       files_preserve=[fh.stream.fileno()],
                                       working_directory=os.path.abspath(config.root),
                                       stdout=sys.stdout, stderr=sys.stderr)
    else:
        context = daemon.DaemonContext(detach_process = False,
                                       stdout=sys.stdout, stderr=sys.stderr,
                                       working_directory=os.path.abspath(config.root))
    
    with context:
        component = ObjectComponent(config.name, config.secret,
                                    config.host, config.port,
                                    config.root)
        if component.connect():  
            logging.info('Component {} connected'.format(component.boundjid))
            component.process(block=False)
        else:
            logging.error("Component {} couldn't connect".format(component.boundjid))