Make debug-coverage a proper command (databrickslabs#940)

Author: vil1

Makefile

@@ -50,19 +50,11 @@ python_coverage_report:
 	hatch -e sqlglot-latest run python src/databricks/labs/remorph/coverage/sqlglot_tsql_transpilation_coverage.py
 
 antlr_coverage_report: build_core_jar
-	java -jar $(wildcard core/target/remorph-core-*-SNAPSHOT.jar) '{"command": "debug-coverage", "flags":{"src": "$(abspath ${INPUT_DIR_PARENT}/snowflake)", "dst":"$(abspath ${OUTPUT_DIR})", "source-dialect": "Snow", "extractor": "full"}}'
-	java -jar $(wildcard core/target/remorph-core-*-SNAPSHOT.jar) '{"command": "debug-coverage", "flags":{"src": "$(abspath ${INPUT_DIR_PARENT}/tsql)", "dst":"$(abspath ${OUTPUT_DIR})", "source-dialect": "Tsql", "extractor": "full"}}'
+	java -jar $(wildcard core/target/remorph-core-*-SNAPSHOT.jar) '{"command": "debug-coverage", "flags":{"src": "$(abspath ${INPUT_DIR_PARENT})", "dst":"$(abspath ${OUTPUT_DIR})", "extractor": "full"}}'
 
 dialect_coverage_report: clean_coverage_dir antlr_coverage_report python_coverage_report
 	hatch run python src/databricks/labs/remorph/coverage/local_report.py
 
-antlr-coverage: build_core_jar
-	echo "Running coverage for snowflake"
-	java -jar $(wildcard core/target/remorph-core-*-SNAPSHOT.jar) '{"command": "coverage", "flags":{"src": "$(abspath ${INPUT_DIR_PARENT}/snowflake)", "dst":"$(abspath ${OUTPUT_DIR})", "source-dialect": "Snow", "extractor": "full"}}'
-	echo "Running coverage for tsql"
-	java -jar $(wildcard core/target/remorph-core-*-SNAPSHOT.jar) '{"command": "coverage", "flags":{"src": "$(abspath ${INPUT_DIR_PARENT}/tsql)", "dst":"$(abspath ${OUTPUT_DIR})", "source-dialect": "Tsql", "extractor": "full"}}'
-	OUTPUT_DIR=.venv/antlr-coverage hatch run python src/databricks/labs/remorph/coverage/local_report.py
-
 antlr-lint:
 	mvn compile -DskipTests exec:java -pl linter --file pom.xml -Dexec.args="-i core/src/main/antlr4 -o .venv/linter/grammar -c true"
 

labs.yml

@@ -62,3 +62,12 @@ commands:
         description: sql dialect
   - name: debug-me
     description: "[INTERNAL] Debug SDK connectivity"
+  - name: debug-coverage
+    description: "[INTERNAL] Run coverage tests"
+    flags:
+      - name: src
+        description: The parent directory under which test queries are laid out
+      - name: dst
+        description: The directory under which the report files will be written
+      - name: extractor
+        description: The strategy for extracting queries from the test files. Valid strategies are "full" (when files contain only one input query) and "comment" (when files contain an input query and the corresponding translation, separated by a comment stating the dialect of each query).

src/databricks/labs/remorph/cli.py

@@ -26,6 +26,7 @@ def raise_validation_exception(msg: str) -> Exception:
 
 proxy_command(remorph, "debug-script")
 proxy_command(remorph, "debug-me")
+proxy_command(remorph, "debug-coverage")
 
 
 @remorph.command