__main__.py

from collections import OrderedDict
from tempfile import NamedTemporaryFile

import argparse
import yaml
import json
import os

from crossbar import run
from twisted.logger import Logger
from twisted.python.logfile import DailyLogFile

from mdstudio.logging.brand import ascii_brand
from mdstudio.logging.impl.printing_observer import PrintingLogObserver


if __name__ == '__main__':

    temp_config = None
    log_file = None

    try:
        with open('crossbar_config.yml', 'r') as cc:
            config = yaml.safe_load(cc)

        ring0_config = [{
            "type": "class",
            "classname": "{role}.application.{component}Component".format(role=role, component=component),
            "realm": "mdstudio",
            "role": role
        } for role, component in OrderedDict([
            ('auth', 'Auth'),
            ('cache', 'Cache'),
            ('db', 'DB'),
            ('schema', 'Schema'),
            ('logger', 'Logger')
        ]).items()]

        ticket_config = {
            "type": "static",
            "principals": OrderedDict((role, {'role': role, 'ticket': role}) for role in ['auth', 'db', 'schema', 'logger'])
        }

        parser = argparse.ArgumentParser(description='MDstudio application startup script')
        parser.add_argument('--dev', action='store_true', help='Flag to run in dev mode. Allows core components to start separately.')
        parser.add_argument('-s', '--skip-startup', nargs='+', metavar='COMPONENT', help='List of core components that are not started (--dev required)')
        args = parser.parse_args()

        if args.dev:
            if args.skip_startup:
                ring0_config = [v for v in ring0_config if v['role'] not in args.skip_startup]
            else:
                ring0_config = None

            config['workers'][0]['transports'][0]['paths']['ws']['auth'].update({'ticket': ticket_config})

        if ring0_config:
            config['workers'][0]['components'] = ring0_config

        temp_config = NamedTemporaryFile(delete=False, suffix='.json')
        temp_config.write(json.dumps(config).encode('utf-8'))
        temp_config.close()

        if not os.path.isdir('logs'):
            os.mkdir('logs')
        log_file = DailyLogFile('daily.log', 'logs')
        log = Logger(observer=PrintingLogObserver(log_file))

        print(ascii_brand)
        run(['start', '--cbdir', '.', '--config', temp_config.name, '--loglevel', 'info',])

    finally:
        if temp_config:
            os.remove(temp_config.name)

        if log_file:
            log_file.flush()
            log_file.close()