code-search-cli 0.1.0__py3-none-any.whl

code_retrieval/__init__.py

@@ -0,0 +1,6 @@
+"""Code Search: task-aware retrieval with explainable evidence bundles."""
+
+from .engine import RetrievalEngine
+from .models import Evidence, RetrievalResult
+
+__all__ = ["Evidence", "RetrievalEngine", "RetrievalResult"]

code_retrieval/__main__.py

@@ -0,0 +1,3 @@
+from .cli import main
+
+raise SystemExit(main())

code_retrieval/cli.py

@@ -0,0 +1,144 @@
+from __future__ import annotations
+
+import argparse
+import json
+import sqlite3
+import subprocess
+import sys
+
+from .engine import RetrievalEngine
+from .index_store import PersistentIndex
+from .installer import install_codex_skill, uninstall_codex_skill
+from .models import Evidence
+from .repository import RepositoryReader
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(prog="code-search", description="Task-aware local code retrieval")
+    subparsers = parser.add_subparsers(dest="command", required=True)
+
+    retrieve = subparsers.add_parser("retrieve", help="build an evidence bundle for a task")
+    _repository_args(retrieve)
+    retrieve.add_argument("--task", required=True, help="engineering task or failure description")
+    retrieve.add_argument("--max-evidence", type=int, default=16)
+    retrieve.add_argument("--format", choices=("markdown", "json"), default="markdown")
+
+    expand = subparsers.add_parser("expand", help="expand a known source anchor")
+    _repository_args(expand)
+    expand.add_argument("--path", required=True, help="repository-relative source path")
+    anchor = expand.add_mutually_exclusive_group()
+    anchor.add_argument("--line", type=int)
+    anchor.add_argument("--symbol")
+    expand.add_argument("--relation", choices=("enclosing", "siblings", "references"), default="enclosing")
+    expand.add_argument("--format", choices=("markdown", "json"), default="markdown")
+
+    index = subparsers.add_parser("index", help="build or incrementally refresh the persistent index")
+    _repository_args(index)
+    index.add_argument("--format", choices=("text", "json"), default="text")
+
+    status = subparsers.add_parser("status", help="show persistent index freshness")
+    _repository_args(status)
+    status.add_argument("--format", choices=("text", "json"), default="text")
+
+    clean = subparsers.add_parser("clean", help="remove the persistent index for a repository")
+    _repository_args(clean)
+
+    install = subparsers.add_parser("install", help="install the bundled Codex skill")
+    install.add_argument("--force", action="store_true", help="replace a locally modified installed skill")
+
+    uninstall = subparsers.add_parser("uninstall", help="remove the bundled Codex skill")
+    uninstall.add_argument("--force", action="store_true", help="remove a locally modified installed skill")
+    return parser
+
+
+def main(argv: list[str] | None = None) -> int:
+    args = build_parser().parse_args(argv)
+    try:
+        if args.command in {"install", "uninstall"}:
+            operation = install_codex_skill if args.command == "install" else uninstall_codex_skill
+            path, action = operation(force=args.force)
+            print(f"Codex skill {action}: {path}")
+            if args.command == "install":
+                print("Restart Codex to load the skill.")
+            return 0
+
+        if args.command in {"index", "status", "clean"}:
+            store = PersistentIndex(RepositoryReader(args.repo, args.ref))
+            if args.command == "clean":
+                removed = store.clean()
+                print(f"Index {'removed' if removed else 'not found'}: {store.path}")
+                return 0
+            report = store.refresh() if args.command == "index" else store.status()
+            data = report.to_dict()
+            if args.format == "json":
+                print(json.dumps(data, ensure_ascii=False, indent=2))
+            else:
+                print(_index_text(args.command, data))
+            return 0
+
+        engine = RetrievalEngine(args.repo, args.ref)
+        if args.command == "retrieve":
+            result = engine.retrieve(args.task, args.max_evidence)
+            output = json.dumps(result.to_dict(), ensure_ascii=False, indent=2) if args.format == "json" else result.to_markdown()
+        else:
+            evidence = engine.expand(
+                args.path,
+                line=args.line,
+                symbol=args.symbol,
+                relation=args.relation,
+            )
+            output = _evidence_json(evidence) if args.format == "json" else _evidence_markdown(evidence)
+        print(output)
+        return 0
+    except (OSError, ValueError, sqlite3.Error, subprocess.SubprocessError) as error:
+        print(f"error: {error}", file=sys.stderr)
+        return 2
+
+
+def _repository_args(parser: argparse.ArgumentParser) -> None:
+    parser.add_argument("--repo", required=True, help="local repository path")
+    parser.add_argument("--ref", help="optional Git commit, tag, or branch; does not modify the working tree")
+
+
+def _evidence_json(evidence: list[Evidence]) -> str:
+    from dataclasses import asdict
+
+    return json.dumps([asdict(item) for item in evidence], ensure_ascii=False, indent=2)
+
+
+def _evidence_markdown(evidence: list[Evidence]) -> str:
+    lines = ["# Expanded evidence", ""]
+    for item in evidence:
+        lines.extend(
+            [
+                f"## `{item.path}:{item.start_line}-{item.end_line}`",
+                "",
+                "; ".join(item.reasons),
+                "",
+                "```",
+                item.content.rstrip(),
+                "```",
+                "",
+            ]
+        )
+    return "\n".join(lines)
+
+
+def _index_text(command: str, data: dict[str, object]) -> str:
+    if command == "status":
+        state = "fresh" if data["fresh"] else ("stale" if data["exists"] else "not built")
+        return (
+            f"Index {state}: {data['index_path']}\n"
+            f"Source files: {data['source_files']}; pending +{data['added']} "
+            f"~{data['updated']} -{data['removed']}"
+        )
+    return (
+        f"Index refreshed: {data['index_path']}\n"
+        f"Source files: {data['source_files']}; +{data['added']} "
+        f"~{data['updated']} -{data['removed']}; unchanged {data['unchanged']}; "
+        f"{data['elapsed_ms']} ms"
+    )
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())