add bazelisk command

Author: ohmayr

.generator/cli.py

@@ -17,6 +17,7 @@
 import logging
 import os
 import sys
+import subprocess
 
 logger = logging.getLogger()
 
@@ -66,7 +67,31 @@ def handle_generate():
             f"failed to read {LIBRARIAN_DIR}/{GENERATE_REQUEST_FILE}"
         ) from e
 
-    logger.info(json.dumps(request_data, indent=2))
+    library_id = request_data.get("id")
+    if not library_id:
+        raise ValueError("Request file is missing required 'id' field.")
+
+    for api in request_data.get("apis", []):
+        api_path = api.get("path")
+        if api_path:
+            try:
+                query = f'filter("-py$", kind("rule", //{api_path}/...:*))'
+                command = ["bazelisk", "query", query]
+                result = subprocess.run(
+                    command, capture_output=True, text=True, check=True
+                )
+
+                bazel_rule = result.stdout.strip()
+                if not bazel_rule:
+                    raise ValueError(
+                        f"Bazel query `{query}` returned an empty bazel rule."
+                    )
+
+                logger.info(f"Found Bazel rule: {bazel_rule}")
+            except Exception as e:
+                raise ValueError(f"Bazelisk query `{query}` failed") from e
+
+        logger.info(json.dumps(request_data, indent=2))
 
     # TODO(https://github.com/googleapis/librarian/issues/448): Implement generate command and update docstring.
     logger.info("'generate' command executed.")

.generator/test_cli.py

@@ -17,7 +17,7 @@
 import json
 import logging
 
-from unittest.mock import mock_open
+from unittest.mock import mock_open, MagicMock
 
 from cli import (
     _read_json_file,
@@ -60,16 +60,30 @@ def test_handle_configure_success(caplog, mock_generate_request_file):
     assert "'configure' command executed." in caplog.text
 
 
-def test_handle_generate_success(caplog, mock_generate_request_file):
+def test_handle_generate_success(caplog, mock_generate_request_file, mocker):
     """
     Tests the successful execution path of handle_generate.
     """
     caplog.set_level(logging.INFO)
 
+    mock_query_result = MagicMock(
+        stdout="//google/cloud/language/v1:google-cloud-language-v1-py\n", returncode=0
+    )
+
+    mock_subprocess = mocker.patch(
+        "cli.subprocess.run", side_effect=[mock_query_result]
+    )
+
     handle_generate()
 
+    # captured = capsys.readouterr()
     assert "google-cloud-language" in caplog.text
+    assert (
+        "Found Bazel rule: //google/cloud/language/v1:google-cloud-language-v1-py"
+        in caplog.text
+    )
     assert "'generate' command executed." in caplog.text
+    assert mock_subprocess.call_count == 1
 
 
 def test_handle_generate_fail(caplog):