hat.gateway.main

 1"""Gateway 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.gateway import common
16from hat.gateway.runner import MainRunner
17
18
19mlog: logging.Logger = logging.getLogger('hat.gateway.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-gateway://main.yaml "
32             "(default $XDG_CONFIG_HOME/hat/gateway.{yaml|yml|toml|json})")
33    return parser
34
35
36def main():
37    """Gateway"""
38    parser = create_argument_parser()
39    args = parser.parse_args()
40    conf = json.read_conf(args.conf, user_conf_dir / 'gateway')
41    sync_main(conf)
42
43
44def sync_main(conf: json.Data):
45    """Sync main entry point"""
46    aio.init_asyncio()
47
48    validator = json.DefaultSchemaValidator(common.json_schema_repo)
49    validator.validate('hat-gateway://main.yaml', conf)
50
51    for device_conf in conf['devices']:
52        info = common.import_device_info(device_conf['module'])
53        if info.json_schema_repo and info.json_schema_id:
54            validator = json.DefaultSchemaValidator(info.json_schema_repo)
55            validator.validate(info.json_schema_id, device_conf)
56
57    log_conf = conf.get('log')
58    if log_conf:
59        logging.config.dictConfig(log_conf)
60
61    with contextlib.suppress(asyncio.CancelledError):
62        aio.run_asyncio(async_main(conf))
63
64
65async def async_main(conf: json.Data):
66    """Async main entry point"""
67    main_runner = MainRunner(conf)
68
69    async def cleanup():
70        await main_runner.async_close()
71        await asyncio.sleep(0.1)
72
73    try:
74        await main_runner.wait_closing()
75
76    finally:
77        await aio.uncancellable(cleanup())
78
79
80if __name__ == '__main__':
81    sys.argv[0] = 'hat-gateway'
82    sys.exit(main())