All files / src/services/parsing treeSitterTest.js

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
/**
 * TreeSitter Test
 *
 * This file contains a simple test to demonstrate how to use the TreeSitterManager.
 * It can be run with: node src/services/parsing/treeSitterTest.js
 */
 
import { TreeSitterManager } from "./treeSitterManager.js";
import logger from "../../utils/logger.js";
 
/**
 * Run tests for the TreeSitterManager
 */
async function testTreeSitter() {
  try {
    // Initialize the TreeSitterManager
    const manager = new TreeSitterManager();
 
    // Load grammars for JavaScript, TypeScript, and Python
    await manager.initializeGrammars(["javascript", "typescript", "python"]);
 
    // Check what languages were loaded
    const loadedLanguages = manager.getLoadedLanguages();
    logger.info(`Loaded languages: ${loadedLanguages.join(", ")}`);
 
    // Test parsing some JavaScript code
    if (manager.hasLanguage("javascript")) {
      const jsParser = manager.getParserForLanguage("javascript");
      const jsCode = 'function hello() { return "world"; }';
      const jsTree = jsParser.parse(jsCode);
      logger.info("JavaScript AST root node type:", jsTree.rootNode.type);
      logger.info("JavaScript AST child count:", jsTree.rootNode.childCount);
    }
 
    // Test parsing some Python code
    if (manager.hasLanguage("python")) {
      const pyParser = manager.getParserForLanguage("python");
      const pyCode = 'def hello():\n    return "world"';
      const pyTree = pyParser.parse(pyCode);
      logger.info("Python AST root node type:", pyTree.rootNode.type);
      logger.info("Python AST child count:", pyTree.rootNode.childCount);
    }
 
    // Test parsing some TypeScript code
    if (manager.hasLanguage("typescript")) {
      const tsParser = manager.getParserForLanguage("typescript");
      const tsCode = 'function hello(): string { return "world"; }';
      const tsTree = tsParser.parse(tsCode);
      logger.info("TypeScript AST root node type:", tsTree.rootNode.type);
      logger.info("TypeScript AST child count:", tsTree.rootNode.childCount);
    }
 
    logger.info("TreeSitterManager tests completed successfully");
  } catch (error) {
    logger.error(`TreeSitterManager test failed: ${error.message}`, {
      error: error.stack,
    });
  }
}
 
// Run the tests
testTreeSitter().catch((err) => {
  logger.error("Unhandled error in TreeSitterManager test", {
    error: err.stack,
  });
  process.exit(1);
});