Add JSON output for ccc search

Author: faysou

src/cocoindex_code/cli.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import functools
+import json as _json
 import os
 import sys
 from collections.abc import Callable
@@ -173,6 +174,28 @@ def print_search_results(response: SearchResponse) -> None:
         _typer.echo(r.content)
 
 
+def print_search_results_json(response: SearchResponse) -> None:
+    """Print search results as machine-readable JSON."""
+    payload = {
+        "success": response.success,
+        "results": [
+            {
+                "file_path": r.file_path,
+                "language": r.language,
+                "content": r.content,
+                "start_line": r.start_line,
+                "end_line": r.end_line,
+                "score": r.score,
+            }
+            for r in response.results
+        ],
+        "total_returned": response.total_returned,
+        "offset": response.offset,
+        "message": response.message,
+    }
+    _typer.echo(_json.dumps(payload, indent=2))
+
+
 def _run_index_with_progress(project_root: str) -> None:
     """Run indexing with streaming progress display. Exits on failure."""
     from rich.console import Console as _Console
@@ -543,6 +566,7 @@ def search(
     offset: int = _typer.Option(0, "--offset", help="Number of results to skip"),
     limit: int = _typer.Option(10, "--limit", help="Maximum results to return"),
     refresh: bool = _typer.Option(False, "--refresh", help="Refresh index before searching"),
+    json_output: bool = _typer.Option(False, "--json", help="Print results as JSON"),
 ) -> None:
     """Semantic search across the codebase."""
     project_root = str(require_project_root())
@@ -568,7 +592,10 @@ def search(
         limit=limit,
         offset=offset,
     )
-    print_search_results(resp)
+    if json_output:
+        print_search_results_json(resp)
+    else:
+        print_search_results(resp)
 
 
 @app.command()

tests/test_cli_helpers.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 
+import json
 from pathlib import Path
 
 import pytest
+from typer.testing import CliRunner
 
 from cocoindex_code import cli
 from cocoindex_code.cli import (
@@ -13,6 +15,7 @@
     require_project_root,
     resolve_default_path,
 )
+from cocoindex_code.protocol import SearchResponse, SearchResult
 
 
 def test_require_project_root_success(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
@@ -84,6 +87,55 @@ def test_resolve_default_path_outside_project(
     assert result is None
 
 
+def test_search_help_includes_json_option() -> None:
+    runner = CliRunner()
+
+    result = runner.invoke(cli.app, ["search", "--help"], catch_exceptions=False)
+
+    assert result.exit_code == 0
+    assert "--json" in result.output
+
+
+def test_print_search_results_json_outputs_machine_readable_payload(
+    capsys: pytest.CaptureFixture[str],
+) -> None:
+    response = SearchResponse(
+        success=True,
+        results=[
+            SearchResult(
+                file_path="src/main.py",
+                language="python",
+                content="def main():\n    return 1",
+                start_line=10,
+                end_line=11,
+                score=0.875,
+            )
+        ],
+        total_returned=1,
+        offset=5,
+        message=None,
+    )
+
+    cli.print_search_results_json(response)
+
+    assert json.loads(capsys.readouterr().out) == {
+        "success": True,
+        "results": [
+            {
+                "file_path": "src/main.py",
+                "language": "python",
+                "content": "def main():\n    return 1",
+                "start_line": 10,
+                "end_line": 11,
+                "score": 0.875,
+            }
+        ],
+        "total_returned": 1,
+        "offset": 5,
+        "message": None,
+    }
+
+
 # ---------------------------------------------------------------------------
 # .gitignore helpers
 # ---------------------------------------------------------------------------