"""HAI-Relay CLI entry point."""

from __future__ import annotations

import argparse
import asyncio
import shlex
from pathlib import Path

from .authority import DenyAllAuthority, SubprocessAuthority
from .custody import ensure_state_dir
from .daemon import RelayDaemon
from .dispatcher import AcpDispatchPort
from .protocol import RelayProtocol
from .registry import AgentRegistry
from .service import RelayService
from .store import RelayStore


def main() -> None:
    parser = argparse.ArgumentParser(prog="hai-relay")
    sub = parser.add_subparsers(dest="command", required=True)
    serve = sub.add_parser("serve")
    serve.add_argument("--state-dir", type=Path, required=True)
    serve.add_argument("--socket", type=Path, required=True)
    serve.add_argument("--authority-command")
    args = parser.parse_args()

    if args.command == "serve":
        state_dir = ensure_state_dir(args.state_dir)
        authority = (
            SubprocessAuthority(tuple(shlex.split(args.authority_command)))
            if args.authority_command
            else DenyAllAuthority()
        )
        store = RelayStore(state_dir / "relay.sqlite3", state_dir=state_dir)
        service = RelayService(store, AgentRegistry(), authority, AcpDispatchPort())
        protocol = RelayProtocol(service)
        try:
            asyncio.run(RelayDaemon(args.socket, protocol).serve())
        except KeyboardInterrupt:
            pass
        finally:
            store.close()


if __name__ == "__main__":
    main()