spud-py/spud/cli.py

"""
Command-line application entry-point.

Logic for the command-line spud tooling.
"""

import json
import pathlib

import click
import uvicorn

from spud.base import Configuration

DEFAULT_CONFIGURATION_PATH = "~/.config/spud/config.json"


@click.group()
@click.option(
    "--config",
    type=str,
    default=None,
    help="Configuration path",
)
@click.pass_context
def cli(context, config):
    """CLI root"""
    context.ensure_object(dict)
    config = config if config is not None else DEFAULT_CONFIGURATION_PATH
    context.obj["config_path"] = pathlib.Path(config).expanduser()


@click.command()
@click.pass_context
def init(context):
    """Generates a default configuration and writes it to disk."""
    config_path = context.obj["config_path"]

    config_path.parent.mkdir(parents=True, exist_ok=True)

    if config_path.exists():
        raise RuntimeError(
            f"File already exists ({str(config_path)}), cannot initialize."
        )

    default_configuration = Configuration()

    with open(config_path, "w", encoding="utf8") as config_file:
        config_file.write(default_configuration.model_dump_json(indent=2))


@click.command()
@click.pass_context
def print_config(context):
    """Prints the current configuration."""
    config_path = context.obj["config_path"]

    if not config_path.exists():
        raise RuntimeError(f"Configuration file not found at {str(config_path)}.")

    with open(config_path, "r", encoding="utf8") as config_file:
        config_json = json.loads(config_file.read())

    config = Configuration(**config_json)

    click.echo(config.model_dump_json(indent=2))


@click.command()
@click.option(
    "--reload",
    type=bool,
    default=False,
    is_flag=True,
    help="Hot reloads on code updates.",
)
def daemon(reload):
    """Starts the daemon process."""
    uvicorn.run("spud.daemon:app", reload=reload)


cli.add_command(init)
cli.add_command(print_config)
cli.add_command(daemon)

if __name__ == "__main__":
    cli(None, None)