"""click CLI for the example app.

Two subcommands:

teesql-example migrate — one-shot schema bootstrap. Run once per

cluster (or after a burn-down) before

the first `serve`.

teesql-example serve — start the FastAPI app under uvicorn.

Holds primary + secondary connections

for the lifetime of the process.

Mirrors hivemind-core's `hivemind` click CLI shape.

"""

from __future__ import annotations

import logging

import sys

import click

import uvicorn

from .config import load_settings

from .connection import connect

from .migrate import apply_migrations

log = logging.getLogger(__name__)

def _setup_logging(level: str) -> None:

logging.basicConfig(

level=level,

format="%(asctime)s %(levelname)s %(name)s: %(message)s",

datefmt="%Y-%m-%dT%H:%M:%S%z",

)

@click.group(help="Example client for connecting to a teesql cluster.")

@click.pass_context

def cli(ctx: click.Context) -> None:

settings = load_settings()

_setup_logging(settings.log_level)

ctx.obj = settings

@cli.command(help="Idempotent schema bootstrap. Run once per cluster.")

@click.pass_obj

def migrate(settings) -> None: # type: ignore[no-untyped-def]

# Migration is a one-shot — short-lived `connect()` is fine.

with connect(settings, role=settings.primary_role) as conn:

apply_migrations(conn)

click.echo("migrations complete")

@cli.command(help="Start the FastAPI app (POST /events + GET /ws/events).")

@click.option(

"--host",

default=None,

help="Bind host (overrides TEESQL_EXAMPLE_HOST)",

)

@click.option(

"--port",

default=None,

type=int,

help="Bind port (overrides TEESQL_EXAMPLE_PORT)",

)

@click.pass_obj

def serve(settings, host: str | None, port: int | None) -> None: # type: ignore[no-untyped-def]

bind_host = host or settings.host

bind_port = port or settings.port

log.info("serving on %s:%d", bind_host, bind_port)

uvicorn.run(

"teesql_example.server:create_app",

host=bind_host,

port=bind_port,

log_level=settings.log_level.lower(),

factory=True,

)

def main() -> None:

"""Entry point exposed as `teesql-example` in pyproject.toml.

Wraps the click group in a small try/except so unexpected exceptions

surface a clear error rather than a Python traceback (which would

include the cluster_secret if it appears in a frame's locals).

"""

try:

cli()

except SystemExit:

raise

except KeyboardInterrupt:

click.echo("interrupted", err=True)

sys.exit(130)

except Exception as e: # noqa: BLE001

click.echo(f"error: {e}", err=True)

sys.exit(1)

if __name__ == "__main__":

main()