hat.event.server.main

 1"""Event server main"""
 2
 3from pathlib import Path
 4import argparse
 5import asyncio
 6import contextlib
 7import logging.config
 8import sys
 9
10import appdirs
11
12from hat import aio
13from hat import json
14
15from hat.event import common
16from hat.event.server.runner import MainRunner
17
18
19mlog: logging.Logger = logging.getLogger('hat.event.server.main')
20"""Module logger"""
21
22user_conf_dir: Path = Path(appdirs.user_config_dir('hat'))
23"""User configuration directory path"""
24
25
26def create_argument_parser() -> argparse.ArgumentParser:
27    """Create argument parser"""
28    parser = argparse.ArgumentParser()
29    parser.add_argument(
30        '--conf', metavar='PATH', type=Path, default=None,
31        help="configuration defined by hat-event://server.yaml "
32             "(default $XDG_CONFIG_HOME/hat/event.{yaml|yml|toml|json})")
33    return parser
34
35
36def main():
37    """Event Server"""
38    parser = create_argument_parser()
39    args = parser.parse_args()
40    conf = json.read_conf(args.conf, user_conf_dir / 'event')
41    sync_main(conf)
42
43
44def sync_main(conf: json.Data):
45    """Sync main entry point"""
46    aio.init_asyncio()
47
48    common.json_schema_repo.validate('hat-event://server.yaml', conf)
49
50    info = common.import_backend_info(conf['backend']['module'])
51    if info.json_schema_repo and info.json_schema_id:
52        info.json_schema_repo.validate(info.json_schema_id, conf['backend'])
53
54    for module_conf in conf['modules']:
55        info = common.import_module_info(module_conf['module'])
56        if info.json_schema_repo and info.json_schema_id:
57            info.json_schema_repo.validate(info.json_schema_id, module_conf)
58
59    log_conf = conf.get('log')
60    if log_conf:
61        logging.config.dictConfig(log_conf)
62
63    with contextlib.suppress(asyncio.CancelledError):
64        aio.run_asyncio(async_main(conf))
65
66
67async def async_main(conf: json.Data):
68    """Async main entry point"""
69    main_runner = MainRunner(conf)
70
71    async def cleanup():
72        await main_runner.async_close()
73        await asyncio.sleep(0.1)
74
75    try:
76        await main_runner.wait_closing()
77
78    finally:
79        await aio.uncancellable(cleanup())
80
81
82if __name__ == '__main__':
83    sys.argv[0] = 'hat-event-server'
84    sys.exit(main())