@0dai-dev/cli 4.3.5 → 4.3.7

package/lib/python/provider_registry_cli.py

@@ -0,0 +1,125 @@
+#!/usr/bin/env python3
+"""CLI wrapper for local BYOK provider registry list/switch/clear commands."""
+from __future__ import annotations
+
+import argparse
+import json
+import pathlib
+import sys
+from typing import Any
+
+import provider_registry as pr
+
+
+def _provider_row(row: dict[str, Any]) -> dict[str, Any]:
+    """Return only non-secret provider metadata for CLI output."""
+    return {
+        "name": str(row.get("name") or ""),
+        "endpoint_kind": str(row.get("endpoint_kind") or ""),
+        "base_url": str(row.get("base_url") or ""),
+        "has_credentials": bool(row.get("has_credentials")),
+        "source": str(row.get("source") or ""),
+        "active": bool(row.get("active")),
+    }
+
+
+def _profile_payload(profile: dict[str, Any]) -> dict[str, Any]:
+    """Return a switch result without raw provider secret material."""
+    return {
+        "provider_name": str(profile.get("provider_name") or ""),
+        "base_url": str(profile.get("base_url") or ""),
+        "api_key_ref": str(profile.get("api_key_ref") or ""),
+        "model": str(profile.get("model") or ""),
+        "updated_at": str(profile.get("updated_at") or ""),
+    }
+
+
+def _print_json(payload: dict[str, Any]) -> None:
+    print(json.dumps(payload, sort_keys=True))
+
+
+def _cmd_list(args: argparse.Namespace) -> int:
+    target = pathlib.Path(args.target).resolve()
+    providers = [_provider_row(row) for row in pr.list_providers_for_repo(target)]
+    if args.json:
+        _print_json({"target": str(target), "providers": providers})
+        return 0
+
+    print(f"Provider Status for {target.name}")
+    print("=" * 60)
+    for row in providers:
+        active = "->" if row["active"] else "  "
+        source = f"[{row['source']}]" if row["source"] else ""
+        creds = "yes" if row["has_credentials"] else "no"
+        print(f"  {active} {row['name']:<12} creds={creds:<3} {source:<10} {row['base_url']}")
+    print("")
+    return 0
+
+
+def _cmd_switch(args: argparse.Namespace) -> int:
+    target = pathlib.Path(args.target).resolve()
+    profile = _profile_payload(pr.switch_provider(target, args.provider))
+    if args.json:
+        _print_json({"target": str(target), "switched": True, "profile": profile})
+        return 0
+
+    print(f"Switched {target.name} to provider: {profile['provider_name']}")
+    print(f"  base_url: {profile['base_url']}")
+    print("  audit log: ~/.0dai/audit/provider-changes.jsonl")
+    return 0
+
+
+def _cmd_clear(args: argparse.Namespace) -> int:
+    target = pathlib.Path(args.target).resolve()
+    cleared = pr.clear_repo_profile(target)
+    if args.json:
+        _print_json({"target": str(target), "cleared": bool(cleared)})
+        return 0
+
+    if cleared:
+        print(f"Cleared per-repo profile for {target.name}")
+        print("Will fall back to global default or env vars")
+    else:
+        print(f"No per-repo profile to clear for {target.name}")
+    return 0
+
+
+def build_parser() -> argparse.ArgumentParser:
+    common = argparse.ArgumentParser(add_help=False)
+    common.add_argument("--target", default=".", help="project directory")
+    common.add_argument("--json", action="store_true", help="emit JSON")
+
+    parser = argparse.ArgumentParser(
+        description="Manage local provider registry profiles",
+        parents=[common],
+    )
+    sub = parser.add_subparsers(dest="command", required=True)
+
+    list_cmd = sub.add_parser("list", parents=[common], help="list configured provider availability")
+    list_cmd.set_defaults(func=_cmd_list)
+
+    switch_cmd = sub.add_parser("switch", parents=[common], help="switch this project to a provider")
+    switch_cmd.add_argument("--provider", required=True, help="provider name")
+    switch_cmd.set_defaults(func=_cmd_switch)
+
+    clear_cmd = sub.add_parser("clear", parents=[common], help="clear this project's provider override")
+    clear_cmd.set_defaults(func=_cmd_clear)
+
+    return parser
+
+
+def main(argv: list[str] | None = None) -> int:
+    parser = build_parser()
+    args = parser.parse_args(argv)
+    try:
+        return int(args.func(args) or 0)
+    except pr.ProviderRegistryError as exc:
+        if getattr(args, "json", False):
+            _print_json({"error": str(exc)})
+        else:
+            print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())