feat(parser): Add C and C++ support to ASTParser

src/file_cache.py

@@ -38,6 +38,8 @@ from pathlib import Path
 from typing import Optional, Any, List, Tuple, Dict
 import tree_sitter
 import tree_sitter_python
+import tree_sitter_cpp
+import tree_sitter_c
 import re
 
 _ast_cache: Dict[str, Tuple[float, tree_sitter.Tree]] = {}
@@ -49,11 +51,16 @@ class ASTParser:
     """
 
  def __init__(self, language: str) -> None:
-  if language != "python":
+  if language not in ("python", "cpp", "c"):
    raise ValueError(f"Language '{language}' not supported yet.")
   self.language_name = language
   # Load the tree-sitter language grammar
-  self.language = tree_sitter.Language(tree_sitter_python.language())
+  if language == "python":
+   self.language = tree_sitter.Language(tree_sitter_python.language())
+  elif language == "cpp":
+   self.language = tree_sitter.Language(tree_sitter_cpp.language())
+  elif language == "c":
+   self.language = tree_sitter.Language(tree_sitter_c.language())
   self.parser = tree_sitter.Parser(self.language)
 
  def parse(self, code: str) -> tree_sitter.Tree:

tests/test_ast_parser.py

@@ -42,6 +42,16 @@ def test_ast_parser_invalid_language() -> None:
  # If it's intended to raise or handle gracefully, test it here.
  pass
 
+def test_ast_parser_cpp_init() -> None:
+ """Verify that ASTParser can be initialized with 'cpp'."""
+ parser = ASTParser(language="cpp")
+ assert parser.language_name == "cpp"
+
+def test_ast_parser_c_init() -> None:
+ """Verify that ASTParser can be initialized with 'c'."""
+ parser = ASTParser(language="c")
+ assert parser.language_name == "c"
+
 def test_ast_parser_get_curated_view() -> None:
  """Verify that get_curated_view preserves function bodies with @core_logic or # [HOT]."""
  parser = ASTParser(language="python")