Source code for voca.cli

"""
Module that contains the command line app.

Why does this file exist, and why not put this in __main__?

  You might be tempted to import things from __main__ later, but that will cause
  problems: the code will get executed twice:

  - When you run `python -mvoca` python will execute
    ``__main__.py`` as a script. That means there won't be any
    ``voca.__main__`` in ``sys.modules``.
  - When you import __main__ it will get executed again (as a module) because
    there's no ``voca.__main__`` in ``sys.modules``.

  Also see (1) from http://click.pocoo.org/5/setuptools/#setuptools-integration
"""
import sys
import types
import os

import click
import eliot

from voca import app
from voca import listen
from voca import manager
from voca import worker
from voca import log
from voca import config


CONTEXT_SETTINGS = {"auto_envvar_prefix": "VOCA"}


def log_cli_call(f):
    def wrap(*args, **kwargs):
        with eliot.start_action(
            action_type="cli_command", pid=os.getpid(), argv=sys.argv
        ):
            return log.log_call(f)(*args, **kwargs)

    return wrap


@click.group(context_settings=CONTEXT_SETTINGS)
@click.option("--log/--no-log", "should_log", is_flag=True, default=True)
@click.pass_context
def cli(ctx, **kwargs):
    ctx.obj = types.SimpleNamespace()
    ctx.obj.kwargs = kwargs.copy()
    app.main(**kwargs)


@cli.command("mic")
def _detect_mic(**kwargs):
    del sys.argv[1]
    import voca.mic


@cli.command("listen")
@click.option("--server", "-s", default="127.0.0.1")
@click.option("--port", "-p", default=8019, type=int)
@click.option("--device", "-d", default=7, type=int)
@click.option("--hypotheses", type=bool, default=True)
@click.option("--save-adaptation-state", type=bool)
@click.option("--send-adaptation-state", type=bool)
@click.option("--audio-gate", default=0, type=int)
@log_cli_call
def _listen(**kwargs):
    content_type = "audio/x-raw, layout=(string)interleaved, rate=(int)16000, format=(string)S16LE, channels=(int)1"
    path = "client/ws/speech"

    listen.run(
        args=types.SimpleNamespace(**kwargs), path=path, content_type=content_type
    )


@cli.command("manage")
@click.option("--import-path", "-i", "module_names", multiple=True, default=None)
@click.option("--num-workers", type=int, default=5)
@click.pass_obj
@log_cli_call
def _manage(obj, **kwargs):
    log_filename = config.get_config_dir() / "log.jsonl"
    with open(log_filename, "w") as log_file:
        eliot.add_destinations(log.json_to_file(log_file))

        manager.main(**obj.kwargs, **kwargs)


@cli.command("worker")
@click.option("import_paths", "-i", multiple=True)
@click.option("--patch-caster", is_flag=True, default=False, envvar="VOCA_PATCH_CASTER")
@click.option(
    "--backup-modules/--no-backup-modules",
    "use_backup_modules",
    is_flag=True,
    default=True,
)
@click.pass_obj
@log_cli_call
def _worker(obj, patch_caster, **kwargs):

    eliot.add_destinations(log.json_to_file(sys.stdout))

    if patch_caster:
        from voca import caster_adapter

        caster_adapter.patch_all()
    worker.main(**kwargs)