wild-pc/cli/src/castle_cli/commands/gateway.py

"""castle gateway - manage the Caddy reverse proxy gateway."""

from __future__ import annotations

import argparse
import subprocess

from castle_core.config import SPECS_DIR
from castle_core.generators.caddyfile import generate_caddyfile_from_registry
from castle_core.registry import REGISTRY_PATH, load_registry

from castle_cli.config import CastleConfig, ensure_dirs, load_config

GATEWAY_COMPONENT = "castle-gateway"
GATEWAY_UNIT = "castle-castle-gateway.service"


def _write_generated_files() -> None:
    """Write generated Caddyfile from registry."""
    ensure_dirs()

    if not REGISTRY_PATH.exists():
        print("Error: no registry found. Run 'castle apply' first.")
        return

    registry = load_registry()
    caddyfile_path = SPECS_DIR / "Caddyfile"
    caddyfile_path.write_text(generate_caddyfile_from_registry(registry))
    print(f"  Generated {caddyfile_path}")


def run_gateway(args: argparse.Namespace) -> int:
    """Manage the Caddy gateway."""
    if not args.gateway_command:
        print("Usage: castle gateway {start|stop|reload|status}")
        return 1

    config = load_config()

    if args.gateway_command == "start":
        if getattr(args, "dry_run", False):
            return _gateway_dry_run()
        return _gateway_start(config)
    elif args.gateway_command == "stop":
        return _gateway_stop()
    elif args.gateway_command == "reload":
        if getattr(args, "dry_run", False):
            return _gateway_dry_run()
        return _gateway_reload()
    elif args.gateway_command == "status":
        return _gateway_status()

    return 1


def _gateway_dry_run() -> int:
    """Print generated Caddyfile without applying."""
    if not REGISTRY_PATH.exists():
        print("Error: no registry found. Run 'castle apply' first.")
        return 1

    registry = load_registry()
    print("# Caddyfile")
    print(generate_caddyfile_from_registry(registry))
    return 0


def _gateway_start(config: CastleConfig) -> int:
    """Generate config and enable the gateway service."""
    from castle_cli.commands.service import _service_enable

    if GATEWAY_COMPONENT not in config.services:
        print(f"Error: '{GATEWAY_COMPONENT}' not found in services section")
        return 1

    print("Generating gateway configuration...")
    _write_generated_files()

    print(f"\nStarting gateway on port {config.gateway.port}...")
    return _service_enable(config, GATEWAY_COMPONENT)


def _gateway_stop() -> int:
    """Stop the gateway service."""
    from castle_cli.commands.service import _service_disable

    return _service_disable(GATEWAY_COMPONENT)


def _gateway_reload() -> int:
    """Regenerate config and reload Caddy."""
    print("Regenerating gateway configuration...")
    _write_generated_files()

    result = subprocess.run(
        ["systemctl", "--user", "reload", GATEWAY_UNIT],
        capture_output=True,
        text=True,
    )

    if result.returncode == 0:
        print("Gateway reloaded.")
    else:
        print("Reload signal sent. Verifying...")
        result = subprocess.run(
            ["systemctl", "--user", "is-active", GATEWAY_UNIT],
            capture_output=True,
            text=True,
        )
        if result.stdout.strip() == "active":
            print("Gateway running.")
        else:
            print("Warning: gateway may not be running. Try: castle gateway start")

    return 0


def _gateway_status() -> int:
    """Show gateway status + the full route table (static, proxy, remote)."""
    result = subprocess.run(
        ["systemctl", "--user", "is-active", GATEWAY_UNIT],
        capture_output=True,
        text=True,
    )
    status = result.stdout.strip()
    print(f"Gateway: {'running' if status == 'active' else status}")

    if not REGISTRY_PATH.exists():
        print("  (no registry — run 'castle apply')")
        return 0

    from castle_core.generators.caddyfile import compute_routes

    routes = compute_routes(load_registry())
    if not routes:
        print("  No routes configured.")
        return 0

    # Each route: address → target, tagged by kind. static = files served in
    # place; proxy/remote = reverse-proxied to a process. (Caddyfile order is
    # precedence-sensitive; this table is alphabetical.)
    print(f"\n  {'ADDRESS':24} {'KIND':7} TARGET")
    for r in sorted(routes, key=lambda r: r.address):
        target = r.target.replace("localhost:", ":") if r.kind != "static" else r.target
        print(f"  {r.address:24} {r.kind:7} {target}")
    return 0