commands.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""inReach Cursor-on-Target Gateway Commands."""

import asyncio
import argparse
import configparser
import logging
import sys
import urllib

import pytak

import inrcot

# Python 3.6 support:
if sys.version_info[:2] >= (3, 7):
    from asyncio import get_running_loop
else:
    from asyncio import _get_running_loop as get_running_loop  # NOQA pylint: disable=no-name-in-module


__author__ = "Greg Albrecht W2GMD <oss@undef.net>"
__copyright__ = "Copyright 2021 Greg Albrecht."
__license__ = "Apache License, Version 2.0"


async def main(config):
    """Main program function, executes workers, et al."""
    tx_queue: asyncio.Queue = asyncio.Queue()
    rx_queue: asyncio.Queue = asyncio.Queue()

    cot_url: urllib.parse.ParseResult = urllib.parse.urlparse(
        config["inrcot"].get("COT_URL"))

    # Create our CoT Event Queue Worker
    reader, writer = await pytak.protocol_factory(cot_url)
    write_worker = pytak.EventTransmitter(tx_queue, writer)
    read_worker = pytak.EventReceiver(rx_queue, reader)

    message_worker = inrcot.InrWorker(tx_queue, config)

    await tx_queue.put(pytak.hello_event("inrcot"))

    done, _ = await asyncio.wait(
        {message_worker.run(), read_worker.run(), write_worker.run()},
        return_when=asyncio.FIRST_COMPLETED)

    for task in done:
        print(f"Task completed: {task}")


def cli():
    """Command Line interface for inReach Cursor-on-Target Gateway."""

    # Get cli arguments:
    parser = argparse.ArgumentParser()

    parser.add_argument("-c", "--CONFIG_FILE", dest="CONFIG_FILE",
                        default="config.ini", type=str)
    namespace = parser.parse_args()
    cli_args = {k: v for k, v in vars(namespace).items() if v is not None}

    # Read config file:
    config = configparser.ConfigParser()

    config_file = cli_args.get("CONFIG_FILE")
    logging.info("Reading configuration from %s", config_file)
    config.read(config_file)

    if sys.version_info[:2] >= (3, 7):
        asyncio.run(main(config), debug=config["inrcot"].getboolean("DEBUG"))
    else:
        loop = get_running_loop()
        try:
            loop.run_until_complete(main(config))
        finally:
            loop.close()


if __name__ == '__main__':
    cli()