fix: improve tree-sitter test type safety and debug logging

- Replace 'any' types with proper Parser types in helpers.ts
- Add centralized DEBUG flag and debugLog function in helpers.ts
- Update all console.log statements to use debugLog across all test files
- This change appeases @ellipsis-dev by improving type safety and
  providing a clean way to control debug logging

Signed-off-by: Eric Wheeler <[email protected]>

Author: Eric Wheeler

src/services/tree-sitter/__tests__/helpers.ts

@@ -5,11 +5,21 @@ import * as path from "path"
 import Parser from "web-tree-sitter"
 import tsxQuery from "../queries/tsx"
 
+// Global debug flag - set to 0 to disable debug logging
+export const DEBUG = 0
+
+// Debug function to conditionally log messages
+export const debugLog = (message: string, ...args: any[]) => {
+	if (DEBUG) {
+		console.debug(message, ...args)
+	}
+}
+
 // Mock fs module
 const mockedFs = jest.mocked(fs)
 
 // Store the initialized TreeSitter for reuse
-let initializedTreeSitter: any = null
+let initializedTreeSitter: Parser | null = null
 
 // Function to initialize tree-sitter
 export async function initializeTreeSitter() {
@@ -98,7 +108,7 @@ export async function testParseSourceCodeDefinitions(
 	expect(mockedLoadRequiredLanguageParsers).toHaveBeenCalledWith([testFilePath])
 	expect(mockedLoadRequiredLanguageParsers).toHaveBeenCalled()
 
-	console.log(`content:\n${content}\n\nResult:\n${result}`)
+	debugLog(`content:\n${content}\n\nResult:\n${result}`)
 	return result
 }
 
@@ -114,15 +124,15 @@ export async function inspectTreeStructure(content: string, language: string = "
 	const tree = parser.parse(content)
 
 	// Print the tree structure
-	console.log(`TREE STRUCTURE (${language}):\n${tree.rootNode.toString()}`)
+	debugLog(`TREE STRUCTURE (${language}):\n${tree.rootNode.toString()}`)
 
 	// Add more detailed debug information
-	console.log("\nDETAILED NODE INSPECTION:")
+	debugLog("\nDETAILED NODE INSPECTION:")
 
 	// Function to recursively print node details
-	const printNodeDetails = (node: any, depth: number = 0) => {
+	const printNodeDetails = (node: Parser.SyntaxNode, depth: number = 0) => {
 		const indent = "  ".repeat(depth)
-		console.log(
+		debugLog(
 			`${indent}Node Type: ${node.type}, Start: ${node.startPosition.row}:${node.startPosition.column}, End: ${node.endPosition.row}:${node.endPosition.column}`,
 		)
 
@@ -132,17 +142,17 @@ export async function inspectTreeStructure(content: string, language: string = "
 			if (child) {
 				// For type_alias_declaration nodes, print more details
 				if (node.type === "type_alias_declaration") {
-					console.log(`${indent}  TYPE ALIAS: ${node.text}`)
+					debugLog(`${indent}  TYPE ALIAS: ${node.text}`)
 				}
 
 				// For conditional_type nodes, print more details
 				if (node.type === "conditional_type" || child.type === "conditional_type") {
-					console.log(`${indent}  CONDITIONAL TYPE FOUND: ${child.text}`)
+					debugLog(`${indent}  CONDITIONAL TYPE FOUND: ${child.text}`)
 				}
 
 				// For infer_type nodes, print more details
 				if (node.type === "infer_type" || child.type === "infer_type") {
-					console.log(`${indent}  INFER TYPE FOUND: ${child.text}`)
+					debugLog(`${indent}  INFER TYPE FOUND: ${child.text}`)
 				}
 
 				printNodeDetails(child, depth + 1)

src/services/tree-sitter/__tests__/parseSourceCodeDefinitions.json.test.ts

@@ -5,7 +5,7 @@ import * as path from "path"
 import { fileExistsAtPath } from "../../../utils/fs"
 import { loadRequiredLanguageParsers } from "../languageParser"
 import { javascriptQuery } from "../queries"
-import { initializeTreeSitter, testParseSourceCodeDefinitions, inspectTreeStructure } from "./helpers"
+import { initializeTreeSitter, testParseSourceCodeDefinitions, inspectTreeStructure, debugLog } from "./helpers"
 
 // Sample JSON content for tests
 const sampleJsonContent = `{
@@ -132,15 +132,15 @@ describe("parseSourceCodeDefinitions for JSON", () => {
 	})
 
 	it("should parse top-level object properties", async function () {
-		console.log("\n=== Parse Test: Top-level Properties ===")
+		debugLog("\n=== Parse Test: Top-level Properties ===")
 		const result = await testParseSourceCodeDefinitions(testFilePath, sampleJsonContent, jsonOptions)
 		expect(result).toBeDefined()
 		expect(result).toContain('"server"')
 		expect(result).toContain('"database"')
 	})
 
 	it("should parse nested object properties", async function () {
-		console.log("\n=== Parse Test: Nested Properties ===")
+		debugLog("\n=== Parse Test: Nested Properties ===")
 		const result = await testParseSourceCodeDefinitions(testFilePath, sampleJsonContent, jsonOptions)
 		expect(result).toBeDefined()
 		expect(result).toContain('"ssl"')

src/services/tree-sitter/__tests__/parseSourceCodeDefinitions.tsx.test.ts

@@ -5,7 +5,7 @@ import * as path from "path"
 import { fileExistsAtPath } from "../../../utils/fs"
 import { loadRequiredLanguageParsers } from "../languageParser"
 import tsxQuery from "../queries/tsx"
-import { initializeTreeSitter, testParseSourceCodeDefinitions, inspectTreeStructure } from "./helpers"
+import { initializeTreeSitter, testParseSourceCodeDefinitions, inspectTreeStructure, debugLog } from "./helpers"
 
 // Sample component content
 const sampleTsxContent = `
@@ -255,7 +255,7 @@ it("should parse template literal types", async function () {
 
 	// Run the test to see if template literal types are already supported
 	const result = await testParseSourceCodeDefinitions("/test/template-literal-type.tsx", templateLiteralTypeContent)
-	console.log("Template literal type parsing result:", result)
+	debugLog("Template literal type parsing result:", result)
 
 	// Check if the result contains the type declarations
 	expect(result).toBeDefined()
@@ -265,7 +265,7 @@ it("should parse template literal types", async function () {
 	expect(result).toContain("RouteParams<T")
 	expect(result).toContain("StringOps<T")
 
-	console.log("Template literal types are already partially supported by the parser!")
+	debugLog("Template literal types are already partially supported by the parser!")
 
 	// Note: EventName and CSSProperty types aren't fully captured in the output,
 	// but this is likely due to the minimum line requirement (MIN_COMPONENT_LINES = 4)
@@ -337,23 +337,23 @@ it("should parse conditional types", async function () {
 
 	// First run without adding the query pattern to see if it's already implemented
 	const initialResult = await testParseSourceCodeDefinitions("/test/conditional-type.tsx", conditionalTypeContent)
-	console.log("Initial result before adding query pattern:", initialResult)
+	debugLog("Initial result before adding query pattern:", initialResult)
 
 	// Save the initial line count to compare later
 	const initialLineCount = initialResult ? initialResult.split("\n").length : 0
 	const initialCaptures = initialResult ? initialResult : ""
 
 	// Now check if the new query pattern improves the output
 	const updatedResult = await testParseSourceCodeDefinitions("/test/conditional-type.tsx", conditionalTypeContent)
-	console.log("Updated result after adding query pattern:", updatedResult)
+	debugLog("Updated result after adding query pattern:", updatedResult)
 
 	// Compare results
 	const updatedLineCount = updatedResult ? updatedResult.split("\n").length : 0
 	expect(updatedResult).toBeDefined()
 
 	// Check if the feature is already implemented
 	if (initialResult && initialResult.includes("ReturnType<T>") && initialResult.includes("Parameters<T>")) {
-		console.log("Conditional types are already supported by the parser!")
+		debugLog("Conditional types are already supported by the parser!")
 		// If the feature is already implemented, we don't need to check if the updated result is better
 		expect(true).toBe(true)
 	} else {
@@ -533,7 +533,7 @@ it("should parse switch/case statements", async function () {
 	//   await inspectTreeStructure(switchCaseContent)
 
 	const result = await testParseSourceCodeDefinitions("/test/switch-case.tsx", switchCaseContent)
-	console.log("Switch Case Test Result:", result)
+	debugLog("Switch Case Test Result:", result)
 	expect(result).toBeDefined()
 	expect(result).toContain("handleTemperature")
 	// Check for case statements in the output