chore: import upstream snapshot with attribution

2026-07-13 12:42:18 +08:00
commit 05f60106aa
288 changed files with 76871 additions and 0 deletions
@@ -0,0 +1,125 @@
+"""Tools 21, 22: list_repos_func, cross_repo_search_func."""
+
+from __future__ import annotations
+
+import logging
+from pathlib import Path
+from typing import Any
+
+from ..graph import GraphStore
+from ..incremental import get_db_path
+from ..search import hybrid_search
+
+logger = logging.getLogger(__name__)
+
+
+# ---------------------------------------------------------------------------
+# Tool 21: list_repos  [REGISTRY]
+# ---------------------------------------------------------------------------
+
+
+def list_repos_func() -> dict[str, Any]:
+    """List all registered repositories.
+
+    [REGISTRY] Returns the list of repositories registered in the global
+    multi-repo registry at ``~/.code-review-graph/registry.json``.
+
+    Returns:
+        List of registered repos with paths and aliases.
+    """
+    from ..registry import Registry
+
+    try:
+        registry = Registry()
+        repos = registry.list_repos()
+        return {
+            "status": "ok",
+            "summary": f"{len(repos)} registered repository(ies)",
+            "repos": repos,
+        }
+    except Exception as exc:
+        return {"status": "error", "error": str(exc)}
+
+
+# ---------------------------------------------------------------------------
+# Tool 22: cross_repo_search  [REGISTRY]
+# ---------------------------------------------------------------------------
+
+
+def cross_repo_search_func(
+    query: str,
+    kind: str | None = None,
+    limit: int = 20,
+) -> dict[str, Any]:
+    """Search across all registered repositories.
+
+    [REGISTRY] Runs hybrid_search on each registered repo's graph database
+    and merges the results.
+
+    Args:
+        query: Search query string.
+        kind: Optional node kind filter (e.g. "Function", "Class").
+        limit: Maximum results per repo (default: 20).
+
+    Returns:
+        Combined search results from all registered repos.
+    """
+    from ..registry import Registry
+
+    try:
+        registry = Registry()
+        repos = registry.list_repos()
+        if not repos:
+            return {
+                "status": "ok",
+                "summary": (
+                    "No repositories registered. "
+                    "Use 'register' to add repos."
+                ),
+                "results": [],
+            }
+
+        all_results: list[dict[str, Any]] = []
+        searched_repos: list[str] = []
+
+        for repo_entry in repos:
+            repo_path = Path(repo_entry["path"])
+            db_path = get_db_path(repo_path)
+            if not db_path.exists():
+                continue
+
+            try:
+                store = GraphStore(str(db_path))
+                try:
+                    results = hybrid_search(
+                        store, query, kind=kind, limit=limit
+                    )
+                    alias = repo_entry.get("alias", repo_path.name)
+                    for r in results:
+                        r["repo"] = alias
+                        r["repo_path"] = str(repo_path)
+                    all_results.extend(results)
+                    searched_repos.append(alias)
+                finally:
+                    store.close()
+            except Exception as exc:
+                logger.warning(
+                    "Search failed for %s: %s", repo_path, exc
+                )
+
+        # Sort all results by score descending
+        all_results.sort(
+            key=lambda r: r.get("score", 0), reverse=True
+        )
+
+        return {
+            "status": "ok",
+            "summary": (
+                f"Found {len(all_results)} result(s) across "
+                f"{len(searched_repos)} repo(s) for '{query}'"
+            ),
+            "results": all_results[:limit],
+            "repos_searched": searched_repos,
+        }
+    except Exception as exc:
+        return {"status": "error", "error": str(exc)}