import os
import sys
import logging
import traceback
import logging.config
from config import WormConfiguration
from model import MONKEY_ARG, DROPPER_ARG
from dropper import MonkeyDrops
from monkey import ChaosMonkey

__author__ = 'itamar'

LOG = None

LOG_CONFIG = {'version': 1,
              'disable_existing_loggers': False,
              'formatters': {'standard': {'format': '%(asctime)s [%(process)d:%(levelname)s] %(module)s.%(funcName)s.%(lineno)d: %(message)s'},
                            },
              'handlers': {'console': {'class': 'logging.StreamHandler',
                                       'level': 'DEBUG',
                                       'formatter': 'standard'},
                            'file': {'class': 'logging.FileHandler',
                                     'level': 'DEBUG',
                                     'formatter': 'standard',
                                     'filename': None}
                            },
              'root': {'level': 'DEBUG',
                       'handlers': ['console']},
               }

def main():
    global LOG

    if 2 > len(sys.argv):
        return True

    monkey_mode = sys.argv[1]
    monkey_args = sys.argv[2:]

    try:
        if MONKEY_ARG == monkey_mode:
            log_path = WormConfiguration.monkey_log_path
            monkey_cls = ChaosMonkey
        elif DROPPER_ARG == monkey_mode:
            log_path = WormConfiguration.dropper_log_path
            monkey_cls = MonkeyDrops
        else:
            return True
    except ValueError:
        return True

    if WormConfiguration.use_file_logging:
        LOG_CONFIG['handlers']['file']['filename'] = log_path
        LOG_CONFIG['root']['handlers'].append('file')

    logging.config.dictConfig(LOG_CONFIG)
    LOG = logging.getLogger()

    def log_uncaught_exceptions(ex_cls, ex, tb):
        LOG.critical(''.join(traceback.format_tb(tb)))
        LOG.critical('{0}: {1}'.format(ex_cls, ex))

    sys.excepthook = log_uncaught_exceptions

    LOG.info(">>>>>>>>>> Initializing monkey (%s): PID %s <<<<<<<<<<",
             monkey_cls.__name__, os.getpid())

    monkey = monkey_cls(monkey_args)
    monkey.initialize()

    try:
        monkey.start()
        return True
    finally:
        monkey.cleanup()

if "__main__" == __name__:
    if not main():
        sys.exit(1)