code2llm 0.5.45__tar.gz → 0.5.46__tar.gz

{code2llm-0.5.45 → code2llm-0.5.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code2llm
-Version: 0.5.45
+Version: 0.5.46
 Summary: High-performance Python code flow analysis with optimized TOON format - CFG, DFG, call graphs, and intelligent code queries
 Home-page: https://github.com/wronai/stts
 Author: STTS Project

{code2llm-0.5.45 → code2llm-0.5.46}/code2llm/__init__.py

@@ -8,7 +8,7 @@ Includes NLP Processing Pipeline for query normalization, intent matching,
 and entity resolution with multilingual support.
 """
 
-__version__ = "0.5.45"
+__version__ = "0.5.46"
 __author__ = "STTS Project"
 
 # Core analysis components (lightweight, always needed)

{code2llm-0.5.45 → code2llm-0.5.46}/code2llm/core/analyzer.py

@@ -5,7 +5,7 @@ from concurrent.futures import ProcessPoolExecutor, as_completed
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple
 
-from .config import Config, FAST_CONFIG
+from .config import Config, FAST_CONFIG, ALL_EXTENSIONS, LANGUAGE_EXTENSIONS
 from .models import AnalysisResult, FlowEdge, FlowNode, Pattern
 from ..analysis.call_graph import CallGraphExtractor
 
@@ -84,23 +84,28 @@ class ProjectAnalyzer:
         return merged
     
     def _collect_files(self, project_path: Path) -> List[Tuple[str, str]]:
-        """Collect all Python files with their module names."""
+        """Collect all source files with their module names for all supported languages."""
         files = []
         
-        for py_file in project_path.rglob("*.py"):
-            file_str = str(py_file)
-            if not self.file_filter.should_process(file_str):
-                continue
-            
-            # Calculate module name
-            rel_path = py_file.relative_to(project_path)
-            parts = list(rel_path.parts)[:-1]  # Remove .py
-            if py_file.name == '__init__.py':
-                module_name = '.'.join(parts) if parts else project_path.name
-            else:
-                module_name = '.'.join(parts + [py_file.stem])
-            
-            files.append((file_str, module_name))
+        # Collect files for all supported extensions
+        for ext in ALL_EXTENSIONS:
+            for src_file in project_path.rglob(f"*{ext}"):
+                file_str = str(src_file)
+                if not self.file_filter.should_process(file_str):
+                    continue
+                
+                # Calculate module name
+                rel_path = src_file.relative_to(project_path)
+                parts = list(rel_path.parts)[:-1]  # Remove filename
+                
+                # Handle init files for various languages
+                is_init = src_file.name in ('__init__.py', 'index.js', 'index.ts', 'mod.rs', 'lib.rs')
+                if is_init:
+                    module_name = '.'.join(parts) if parts else project_path.name
+                else:
+                    module_name = '.'.join(parts + [src_file.stem])
+                
+                files.append((file_str, module_name))
         
         return files
     

{code2llm-0.5.45 → code2llm-0.5.46}/code2llm/core/config.py

@@ -150,6 +150,37 @@ ANALYSIS_MODES = {
 }
 
 
+# Supported language extensions
+LANGUAGE_EXTENSIONS = {
+    'python': ['.py'],
+    'typescript': ['.ts', '.tsx'],
+    'javascript': ['.js', '.jsx', '.mjs', '.cjs'],
+    'go': ['.go'],
+    'rust': ['.rs'],
+    'java': ['.java'],
+    'cpp': ['.cpp', '.cc', '.cxx', '.hpp', '.hh', '.h'],
+    'c': ['.c', '.h'],
+    'csharp': ['.cs'],
+    'ruby': ['.rb'],
+    'php': ['.php'],
+    'swift': ['.swift'],
+    'kotlin': ['.kt', '.kts'],
+    'scala': ['.scala'],
+    'r': ['.r', '.R'],
+    'matlab': ['.m'],
+    'shell': ['.sh', '.bash', '.zsh'],
+    'dart': ['.dart'],
+    'elixir': ['.ex', '.exs'],
+    'clojure': ['.clj', '.cljs'],
+    'haskell': ['.hs'],
+    'lua': ['.lua'],
+    'perl': ['.pl', '.pm'],
+    'raku': ['.raku', '.rakumod'],
+}
+
+# All supported extensions flat list
+ALL_EXTENSIONS = [ext for exts in LANGUAGE_EXTENSIONS.values() for ext in exts]
+
 # Node types
 NODE_TYPES = {
     'FUNC': 'Function definition',