lumilake-cli 0.1.0__py3-none-any.whl

lumilake_cli/cli.py

@@ -0,0 +1,54 @@
+"""Lumilake CLI entrypoint.
+
+Installs a typer-aware handler on the shared ``Lumilake`` logger root so
+records from every workspace package render with the CLI colour palette.
+"""
+
+import logging
+
+import typer
+from lumilake.log import ColorFormatter, get_default_logger
+
+from .commands import register
+from .core.typer import get_typer
+
+root_app = get_typer(help="Lumilake command line interface.")
+
+
+class _TyperHandler(logging.Handler):
+    """Routes records through ``typer.echo``: stdout for INFO/DEBUG,
+    stderr for WARNING/ERROR/CRITICAL. Honours typer's color detection.
+    Coloring is left to ``ColorFormatter``.
+    """
+
+    def emit(self, record: logging.LogRecord) -> None:
+        try:
+            message = self.format(record)
+        except Exception:
+            self.handleError(record)
+            return
+        typer.echo(message, err=record.levelno >= logging.WARNING)
+
+
+def _install_handler() -> None:
+    root = get_default_logger()
+    if any(isinstance(h, _TyperHandler) for h in root.handlers):
+        return
+    # Drop get_default_logger's StreamHandler so records aren't doubled.
+    for existing in list(root.handlers):
+        root.removeHandler(existing)
+    handler = _TyperHandler()
+    handler.setFormatter(ColorFormatter())
+    root.addHandler(handler)
+    root.propagate = False
+
+
+def main() -> None:
+    """Attach command groups and dispatch to Typer."""
+    _install_handler()
+    register(root_app)
+    root_app()
+
+
+if __name__ == "__main__":
+    main()

lumilake_cli/commands/__init__.py

@@ -0,0 +1,69 @@
+"""Command modules for the Lumilake CLI."""
+
+from importlib.util import find_spec
+
+import typer
+
+from .base import app as base_app
+from .job import app as job_app
+from .trace import app as trace_app
+from .worker import app as worker_app
+
+# The ``[deploy]`` extra pulls ``lumilake-deploy`` and
+# ``flowmesh-cli-stack``; gate on both up front so real ImportError
+# from inside .deploy still bubbles up loudly.
+_DEPLOY_AVAILABLE = (
+    find_spec("lumilake_deploy") is not None
+    and find_spec("flowmesh_cli_stack") is not None
+)
+
+if _DEPLOY_AVAILABLE:
+    from .deploy import app as deploy_app
+else:
+    _DEPLOY_INSTALL_HINT = (
+        "lumilake deploy requires the `deploy` extra. Install with "
+        "`pip install 'lumilake[cli]'` or "
+        "`pip install 'lumilake-cli[deploy]'`."
+    )
+    deploy_app = typer.Typer(
+        help=(
+            "Deploy and manage the Lumilake stack. Install with "
+            "`pip install 'lumilake[cli]'` or "
+            "`pip install 'lumilake-cli[deploy]'` to enable."
+        )
+    )
+
+    def _deploy_hint() -> None:
+        typer.echo(_DEPLOY_INSTALL_HINT, err=True)
+        raise typer.Exit(code=2)
+
+    @deploy_app.callback(invoke_without_command=True)
+    def _deploy_no_subcommand(ctx: typer.Context) -> None:
+        if ctx.resilient_parsing or ctx.invoked_subcommand is not None:
+            return
+        _deploy_hint()
+
+    for _name in (
+        "init",
+        "doctor",
+        "build",
+        "pull",
+        "up",
+        "down",
+        "clean",
+        "reset",
+        "status",
+        "restart",
+        "logs",
+        "update-flowmesh",
+    ):
+        deploy_app.command(_name)(_deploy_hint)
+
+
+def register(app: typer.Typer) -> None:
+    """Register command groups on the root app."""
+    app.add_typer(base_app)
+    app.add_typer(deploy_app, name="deploy")
+    app.add_typer(job_app, name="job")
+    app.add_typer(worker_app, name="worker")
+    app.add_typer(trace_app, name="trace")

lumilake_cli/commands/base.py

@@ -0,0 +1,41 @@
+import json
+from pathlib import Path
+
+import typer
+
+from ..core import logging
+from ..core.config import DEFAULT_CONFIG_PATH
+from ..core.http import HttpError, client_from_config, resolve_base_url
+from ..core.typer import get_typer
+
+app = get_typer()
+
+
+@app.command()
+def config(
+    config_path: Path = typer.Option(
+        DEFAULT_CONFIG_PATH, "--config", help="Path to configuration file"
+    ),
+) -> None:
+    """Show the resolved Lumilake CLI configuration."""
+    base_url, source = resolve_base_url(config_path)
+    payload = {"base_url": base_url, "source": source, "config_path": str(config_path)}
+    logging.log(json.dumps(payload, indent=2))
+
+
+@app.command()
+def info() -> None:
+    """Query server health status."""
+    client = client_from_config()
+    try:
+        response = client.get("/healthz")
+    except HttpError as exc:
+        logging.error(str(exc))
+        raise typer.Exit(code=1)
+    logging.log(json.dumps(response.json(), indent=2))
+
+
+@app.command()
+def health() -> None:
+    """Check if the Lumilake server is reachable and healthy."""
+    info()