ai-hist

#!/usr/bin/env python3
"""Rust-first ai-hist dispatcher."""

import os
import subprocess
import sys
from pathlib import Path


ROOT = Path(__file__).resolve().parent
PYTHON_CLI = ROOT / "ai-hist-python"
RUST_MANIFEST = ROOT / "Cargo.toml"

RUST_COMMANDS = {
    "search",
    "recent",
    "session",
    "resume",
    "sync-opencode",
    "show",
    "context",
    "stats",
    "pack",
    "watch",
    "export",
    "import",
    "tag",
    "untag",
    "tags",
    "sync",
    "login",
    "admin-mint",
    "push",
    "coverage",
    "pair",
    "learn",
}

PYTHON_COMMANDS = set()

PYTHON_ONLY_FLAGS = {}


def main():
    mode = os.environ.get("AI_HIST_CLI", "auto").strip().lower()
    if mode not in {"auto", "rust", "python"}:
        print(
            "ai-hist: AI_HIST_CLI must be one of auto, rust, or python",
            file=sys.stderr,
        )
        return 2

    argv = sys.argv[1:]
    if mode == "python":
        return run_python(argv, warn=False)
    if mode == "rust":
        return run_rust(argv)

    if is_top_level_help(argv):
        print_wrapper_help()
        return 0

    route, reason = route_auto(argv)
    if route == "python":
        return run_python(argv, warn=True, reason=reason)
    return run_rust(argv)


def route_auto(argv):
    if not argv:
        return "help", None
    first = argv[0]
    if first in {"-V", "--version"}:
        return "rust", None
    if first in PYTHON_COMMANDS:
        return "python", f"`{first}` is still implemented by the legacy Python CLI"
    if first in PYTHON_ONLY_FLAGS:
        used = [arg for arg in argv[1:] if arg in PYTHON_ONLY_FLAGS[first]]
        if used:
            return (
                "python",
                f"`{first}` option(s) {', '.join(used)} are still implemented by the legacy Python CLI",
            )
    if first in RUST_COMMANDS:
        return "rust", None
    return "rust", None


def is_top_level_help(argv):
    return not argv or argv in (["-h"], ["--help"], ["help"])


def print_wrapper_help():
    print(
        """ai-hist: Rust-first AI history CLI

Usage:
  ai-hist [COMMAND] [OPTIONS]

Rust-default commands:
  search         Full-text search prompts
  recent         Show most recent entries
  session        Show all prompts in a session
  resume         Print a resume command
  sync-opencode  Sync OpenCode history explicitly
  show           Show full details for an entry by ID
  context        Show entries around an ID
  stats          Show summary statistics
  pack           Evidence bundle for LLM handoff
  watch          Continuous sync loop
  export         Export JSONL/gzip/SQLite with filters
  import         Import JSONL/gzip/SQLite with dry-run support
  setup          Install local integrations (for example git hooks)
  link           Link sessions to commits and other artifacts
  tag            Tag a session
  untag          Remove a tag from a session
  tags           List tags and tagged sessions
  login          Authenticate through Agent Relay Cloud
  admin-mint     Dev-only local relayhistory token mint
  push           Push new local history to relayhistory-cloud
  coverage       Show which machines are pushing, and how recently
  pair           Ask relayhistory-cloud for in-session warnings
  learn          Distill history into Pair signal

Rust binary override:
  AI_HIST_RUST_BIN=/path/to/ai-hist
  ai-hist-rust"""
    )


def run_python(argv, warn=False, reason=None):
    if warn:
        detail = f": {reason}" if reason else ""
        print(
            f"ai-hist: using deprecated Python fallback{detail}",
            file=sys.stderr,
        )
    return subprocess.run([sys.executable, str(PYTHON_CLI), *argv]).returncode


def run_rust(argv):
    rust_bin = os.environ.get("AI_HIST_RUST_BIN")
    if rust_bin:
        cmd = [rust_bin, *argv]
    else:
        cmd = [
            "cargo",
            "run",
            "-q",
            "-p",
            "ai-hist-cli",
            "--manifest-path",
            str(RUST_MANIFEST),
            "--",
            *argv,
        ]
    return subprocess.run(cmd).returncode


if __name__ == "__main__":
    raise SystemExit(main())