Merge pull request #280 from DevanshuNEU/fix/ts-dependency-analyzer

fix: TypeScript dependency analysis -- proper parser, .js resolution, include_paths (OPE-120)

Author: DevanshuNEU

backend/requirements.txt

@@ -18,6 +18,7 @@ hiredis>=2.3.0
 tree-sitter>=0.23.0
 tree-sitter-python>=0.23.0
 tree-sitter-javascript>=0.23.0
+tree-sitter-typescript>=0.23.0
 
 # AI/ML
 openai>=1.54.0

backend/services/dependency_analyzer.py

@@ -9,6 +9,7 @@
 # Tree-sitter
 import tree_sitter_python as tspython
 import tree_sitter_javascript as tsjavascript
+import tree_sitter_typescript as tstypescript
 from tree_sitter import Language, Parser
 
 from services.observability import logger, metrics
@@ -22,7 +23,8 @@ def __init__(self):
         self.parsers = {
             'python': Parser(Language(tspython.language())),
             'javascript': Parser(Language(tsjavascript.language())),
-            'typescript': Parser(Language(tsjavascript.language())),
+            'typescript': Parser(Language(tstypescript.language_typescript())),
+            'tsx': Parser(Language(tstypescript.language_tsx())),
         }
         logger.info("DependencyAnalyzer initialized")
 
@@ -34,7 +36,7 @@ def _detect_language(self, file_path: str) -> str:
             '.js': 'javascript',
             '.jsx': 'javascript',
             '.ts': 'typescript',
-            '.tsx': 'typescript',
+            '.tsx': 'tsx',
         }
         return lang_map.get(ext, 'unknown')
 
@@ -122,8 +124,8 @@ def analyze_file_dependencies(self, file_path: str) -> Dict:
             logger.error("Error analyzing file", file_path=file_path, error=str(e))
             return {"file": str(file_path), "imports": [], "language": language, "error": str(e)}
 
-    def build_dependency_graph(self, repo_path: str) -> Dict:
-        """Build complete dependency graph for repository"""
+    def build_dependency_graph(self, repo_path: str, include_paths: List[str] = None) -> Dict:
+        """Build dependency graph. If include_paths set, only analyze those dirs."""
         repo_path = Path(repo_path)
 
         # Discover code files
@@ -136,8 +138,16 @@ def build_dependency_graph(self, repo_path: str) -> Dict:
                 continue
             if any(skip in file_path.parts for skip in skip_dirs):
                 continue
-            if file_path.suffix in extensions:
-                code_files.append(file_path)
+            if file_path.suffix not in extensions:
+                continue
+            if include_paths:
+                rel_parts = file_path.relative_to(repo_path).parts
+                if not any(
+                    rel_parts[:len(Path(p).parts)] == Path(p).parts
+                    for p in include_paths
+                ):
+                    continue
+            code_files.append(file_path)
 
         logger.info("Building dependency graph", file_count=len(code_files))
 
@@ -236,6 +246,10 @@ def _resolve_import_to_file(
         source_path = Path(source_file)
         source_dir = source_path.parent
 
+        # TS imports use .js extension but actual file is .ts on disk
+        if import_path.endswith('.js') or import_path.endswith('.jsx'):
+            import_path = re.sub(r'\.(jsx?)$', '', import_path)
+        
         # Relative imports
         if import_path.startswith('.'):
             clean_import = import_path.lstrip('./')
@@ -250,7 +264,7 @@ def _resolve_import_to_file(
             else:
                 potential_base = source_dir / clean_import
 
-            extensions = ['', '.ts', '.tsx', '.js', '.jsx', '.py']
+            extensions = ['', '.ts', '.tsx', '.d.ts', '.js', '.jsx', '.py']
 
             for ext in extensions:
                 # Build the potential path
@@ -269,7 +283,7 @@ def _resolve_import_to_file(
         if not import_path.startswith('.'):
             module_path = import_path.replace('.', '/')
 
-            for ext in ['.py', '.js', '.ts']:
+            for ext in ['', '.ts', '.tsx', '.d.ts', '.js', '.jsx', '.py']:
                 test_path = module_path + ext
                 if test_path in internal_files:
                     return test_path