code2llm 0.5.115__tar.gz → 0.5.117__tar.gz

{code2llm-0.5.115 → code2llm-0.5.117}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code2llm
-Version: 0.5.115
+Version: 0.5.117
 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
@@ -67,7 +67,7 @@ Dynamic: requires-python
 
 ## AI Cost Tracking
 
-![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.115-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
+![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.117-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
 ![AI Cost](https://img.shields.io/badge/AI%20Cost-$7.50-orange) ![Human Time](https://img.shields.io/badge/Human%20Time-57.3h-blue) ![Model](https://img.shields.io/badge/Model-openrouter%2Fqwen%2Fqwen3--coder--next-lightgrey)
 
 - 🤖 **LLM usage:** $7.5000 (166 commits)

{code2llm-0.5.115 → code2llm-0.5.117}/README.md

@@ -3,7 +3,7 @@
 
 ## AI Cost Tracking
 
-![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.115-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
+![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.117-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
 ![AI Cost](https://img.shields.io/badge/AI%20Cost-$7.50-orange) ![Human Time](https://img.shields.io/badge/Human%20Time-57.3h-blue) ![Model](https://img.shields.io/badge/Model-openrouter%2Fqwen%2Fqwen3--coder--next-lightgrey)
 
 - 🤖 **LLM usage:** $7.5000 (166 commits)

{code2llm-0.5.115 → code2llm-0.5.117}/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.115"
+__version__ = "0.5.117"
 __author__ = "STTS Project"
 
 # Core analysis components (lightweight, always needed)

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm/analysis/call_graph.py

@@ -5,6 +5,7 @@ from typing import Optional, Set, List, Dict
 
 from code2llm.core.config import Config
 from code2llm.core.models import AnalysisResult, FlowEdge
+from code2llm.analysis.utils import ast_unparse
 
 
 class CallGraphExtractor(ast.NodeVisitor):
@@ -202,10 +203,4 @@ class CallGraphExtractor(ast.NodeVisitor):
         return None
         
     def _expr_to_str(self, node: ast.AST) -> str:
-        """Convert AST expression to string."""
-        if node is None:
-            return ""
-        try:
-            return ast.unparse(node) if hasattr(ast, 'unparse') else str(node)
-        except:
-            return str(node)
+        return ast_unparse(node, default_none="")

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm/analysis/cfg.py

@@ -6,6 +6,7 @@ from typing import Optional
 
 from code2llm.core.config import Config
 from code2llm.core.models import AnalysisResult, FlowNode, FlowEdge, FunctionInfo
+from code2llm.analysis.utils import ast_unparse
 
 
 class CFGExtractor(ast.NodeVisitor):
@@ -275,13 +276,7 @@ class CFGExtractor(ast.NodeVisitor):
             return str(node)[:50]
             
     def _expr_to_str(self, node: ast.AST) -> str:
-        """Convert AST expression to string."""
-        if node is None:
-            return "None"
-        try:
-            return ast.unparse(node) if hasattr(ast, 'unparse') else str(node)
-        except:
-            return str(node)
+        return ast_unparse(node)
             
     def _format_except(self, handler: ast.ExceptHandler) -> str:
         """Format except handler."""

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm/analysis/dfg.py

@@ -6,6 +6,7 @@ from typing import Set, Dict, List
 
 from code2llm.core.config import Config
 from code2llm.core.models import AnalysisResult, FlowEdge, DataFlow, Mutation
+from code2llm.analysis.utils import ast_unparse
 
 
 class DFGExtractor(ast.NodeVisitor):
@@ -207,13 +208,7 @@ class DFGExtractor(ast.NodeVisitor):
         return names
         
     def _expr_to_str(self, node: ast.AST) -> str:
-        """Convert AST expression to string."""
-        if node is None:
-            return "None"
-        try:
-            return ast.unparse(node) if hasattr(ast, 'unparse') else str(node)
-        except:
-            return str(node)
+        return ast_unparse(node)
             
     def _build_data_flow_edges(self):
         """Build DFG edges from data flow records."""

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm/cli_exports/formats.py

@@ -215,30 +215,27 @@ def _export_mermaid_pngs(args, output_dir: Path) -> None:
                 print(f"  - PNG: Skipped (install with: make install-mermaid)")
 
 
-def _export_calls(args, result, output_dir: Path):
-    """Export standalone calls.yaml (structured call graph YAML).
-
-    Generates calls.yaml with structured call graph data:
-    - nodes: functions with metadata (CC, calls_in/out)
-    - edges: caller -> callee relationships
-    - modules: grouping by module
-    - stats: summary statistics
-    """
+def _export_calls_format(args, result, output_dir: Path, toon: bool = False) -> None:
+    """Shared helper: export call graph in YAML or toon format."""
     yaml_exporter = YAMLExporter()
-    yaml_exporter.export_calls(result, str(output_dir / 'calls.yaml'))
-    if args.verbose:
-        print(f"  - CALLS (call graph YAML): {output_dir / 'calls.yaml'}")
+    if toon:
+        yaml_exporter.export_calls_toon(result, str(output_dir / 'calls.toon.yaml'))
+        if args.verbose:
+            print(f"  - CALLS (toon format): {output_dir / 'calls.toon.yaml'}")
+    else:
+        yaml_exporter.export_calls(result, str(output_dir / 'calls.yaml'))
+        if args.verbose:
+            print(f"  - CALLS (call graph YAML): {output_dir / 'calls.yaml'}")
 
 
-def _export_calls_toon(args, result, output_dir: Path):
-    """Export calls.toon.yaml (call graph in human-readable toon format).
+def _export_calls(args, result, output_dir: Path):
+    """Export standalone calls.yaml (structured call graph YAML)."""
+    _export_calls_format(args, result, output_dir, toon=False)
 
-    Generates calls.toon.yaml with hubs, modules, and edges sections.
-    """
-    yaml_exporter = YAMLExporter()
-    yaml_exporter.export_calls_toon(result, str(output_dir / 'calls.toon.yaml'))
-    if args.verbose:
-        print(f"  - CALLS (toon format): {output_dir / 'calls.toon.yaml'}")
+
+def _export_calls_toon(args, result, output_dir: Path):
+    """Export calls.toon.yaml (call graph in human-readable toon format)."""
+    _export_calls_format(args, result, output_dir, toon=True)
 
 
 def _export_mermaid(args, result, output_dir: Path):

code2llm-0.5.117/code2llm/core/lang/__init__.py

@@ -0,0 +1,173 @@
+"""Language-specific analyzers for non-Python source files.
+
+Provides:
+  - LanguageParser ABC — abstract base class for all language parsers
+  - LANGUAGE_REGISTRY — dict mapping extensions to parser functions
+  - register_language — decorator for auto-registration
+  - get_parser(extension) — lookup parser by file extension
+"""
+
+from abc import ABC, abstractmethod
+from typing import Dict, Any, Callable, Set, Optional
+from pathlib import Path
+
+from code2llm.core.models import ModuleInfo, FunctionInfo, ClassInfo
+
+
+# Type alias for parser results
+ParserResult = Dict[str, Any]
+
+
+class LanguageParser(ABC):
+    """Abstract base class for language-specific parsers.
+
+    All language parsers must inherit from this class and implement:
+      - analyze() method to parse file content
+      - supported_extensions class attribute
+
+    Example:
+        @register_language('.go', '.golang')
+        class GoParser(LanguageParser):
+            supported_extensions = ('.go', '.golang')
+
+            def analyze(self, content, file_path, module_name, stats):
+                # Parse Go code
+                return {'module': ..., 'functions': ..., 'classes': ...}
+    """
+
+    supported_extensions: tuple = ()
+    language_name: str = ""
+
+    @abstractmethod
+    def analyze(
+        self,
+        content: str,
+        file_path: str,
+        module_name: str,
+        stats: Dict[str, Any]
+    ) -> ParserResult:
+        """Analyze file content and return parsed structure.
+
+        Args:
+            content: File content as string
+            file_path: Absolute path to file
+            module_name: Logical module name
+            stats: Statistics dict to update
+
+        Returns:
+            Dict with 'module', 'functions', 'classes', 'nodes', 'edges'
+        """
+        pass
+
+    def can_parse(self, file_path: str) -> bool:
+        """Check if this parser can handle the given file."""
+        ext = Path(file_path).suffix.lower()
+        return ext in self.supported_extensions
+
+
+# Legacy parser function type
+LegacyParser = Callable[[str, str, str, str, Dict[str, Any]], ParserResult]
+
+
+# Registry: extension -> parser instance or function
+LANGUAGE_REGISTRY: Dict[str, Any] = {}
+
+
+def register_language(*extensions: str, name: str = ""):
+    """Decorator to register a language parser.
+
+    Args:
+        *extensions: File extensions this parser handles (e.g., '.go', '.rs')
+        name: Human-readable language name
+
+    Example:
+        @register_language('.go', '.golang', name='Go')
+        class GoParser(LanguageParser):
+            ...
+
+        @register_language('.rb', name='Ruby')
+        def analyze_ruby(content, file_path, module_name, ext, stats):
+            ...
+    """
+    def decorator(cls_or_func):
+        if isinstance(cls_or_func, type) and issubclass(cls_or_func, LanguageParser):
+            # It's a class
+            parser = cls_or_func()
+            parser.language_name = name or cls_or_func.__name__.replace('Parser', '')
+            for ext in extensions:
+                LANGUAGE_REGISTRY[ext.lower()] = parser
+        else:
+            # It's a function
+            for ext in extensions:
+                LANGUAGE_REGISTRY[ext.lower()] = cls_or_func
+        return cls_or_func
+    return decorator
+
+
+def get_parser(extension: str) -> Optional[Any]:
+    """Get parser for a file extension.
+
+    Args:
+        extension: File extension (e.g., '.go', '.rs')
+
+    Returns:
+        LanguageParser instance or legacy function, or None if not found
+    """
+    return LANGUAGE_REGISTRY.get(extension.lower())
+
+
+def list_parsers() -> Dict[str, Any]:
+    """List all registered parsers."""
+    return dict(LANGUAGE_REGISTRY)
+
+
+# Legacy function imports (for backward compatibility)
+from .typescript import analyze_typescript_js
+from .go_lang import analyze_go
+from .rust import analyze_rust
+from .java import analyze_java
+from .cpp import analyze_cpp
+from .csharp import analyze_csharp
+from .php import analyze_php
+from .ruby import analyze_ruby
+from .generic import analyze_generic
+
+
+# Register legacy parsers
+register_language('.ts', '.tsx', '.js', '.jsx', '.mjs', '.cjs')(analyze_typescript_js)
+register_language('.go')(analyze_go)
+register_language('.rs')(analyze_rust)
+register_language('.java')(analyze_java)
+register_language('.cpp', '.cc', '.cxx', '.hpp', '.h', '.c')(analyze_cpp)
+register_language('.cs')(analyze_csharp)
+register_language('.php')(analyze_php)
+
+# Import and register RubyParser class (demonstrating new ABC pattern)
+from .ruby import RubyParser
+LANGUAGE_REGISTRY['.rb'] = RubyParser()
+LANGUAGE_REGISTRY['.rbw'] = RubyParser()
+
+# Generic parser as fallback for unknown extensions
+LANGUAGE_REGISTRY['*'] = analyze_generic
+
+
+__all__ = [
+    'LanguageParser',
+    'ParserResult',
+    'register_language',
+    'LANGUAGE_REGISTRY',
+    'get_parser',
+    'list_parsers',
+    # New class-based parsers
+    'RubyParser',
+    # Legacy exports
+    'analyze_typescript_js',
+    'analyze_go',
+    'analyze_rust',
+    'analyze_java',
+    'analyze_cpp',
+    'analyze_csharp',
+    'analyze_php',
+    'analyze_ruby',
+    'analyze_generic',
+]

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm/core/lang/ruby.py

@@ -138,6 +138,27 @@ def analyze_ruby(content: str, file_path: str, module_name: str,
         }
     
     extract_calls_regex(content, module_name, result)
-    
+
     stats['files_processed'] += 1
     return result
+
+
+# New LanguageParser ABC implementation (demonstrating the new pattern)
+# Note: Imports at the bottom to avoid circular imports
+# The __init__.py handles the actual registration
+class RubyParser:
+    """Ruby language parser - registered via @register_language in __init__.py."""
+
+    supported_extensions = ('.rb', '.rbw')
+    language_name = 'Ruby'
+
+    def analyze(
+        self,
+        content: str,
+        file_path: str,
+        module_name: str,
+        stats: Dict
+    ) -> Dict:
+        """Analyze Ruby file content."""
+        # Delegate to the existing legacy function for now
+        return analyze_ruby(content, file_path, module_name, '.rb', stats)

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm/exporters/base.py

@@ -94,6 +94,22 @@ class BaseExporter(ABC):
 Exporter = BaseExporter
 
 
+class ViewGeneratorMixin:
+    """Mixin providing the shared ``generate`` implementation for view generators.
+
+    Eliminates the identical ``generate`` method duplicated across
+    ``ArticleViewGenerator``, ``ContextViewGenerator``, and ``ToonViewGenerator``.
+
+    Subclasses must implement ``_render(data) -> List[str]``.
+    """
+
+    def generate(self, data: Dict[str, Any], output_path: str) -> None:
+        lines = self._render(data)
+        Path(output_path).parent.mkdir(parents=True, exist_ok=True)
+        with open(output_path, "w", encoding="utf-8") as f:
+            f.write("\n".join(lines) + "\n")
+
+
 # Export registry: format_name -> exporter_class
 EXPORT_REGISTRY: Dict[str, Type[BaseExporter]] = {}
 

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm/nlp/__init__.py

@@ -4,7 +4,7 @@ Provides query normalization, intent matching, and entity resolution
 with multilingual support and fuzzy matching.
 """
 
-__version__ = "0.5.115"
+__version__ = "0.5.117"
 
 from .pipeline import NLPPipeline
 from .normalization import QueryNormalizer

{code2llm-0.5.115 → code2llm-0.5.117}/code2llm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code2llm
-Version: 0.5.115
+Version: 0.5.117
 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
@@ -67,7 +67,7 @@ Dynamic: requires-python
 
 ## AI Cost Tracking
 
-![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.115-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
+![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.117-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
 ![AI Cost](https://img.shields.io/badge/AI%20Cost-$7.50-orange) ![Human Time](https://img.shields.io/badge/Human%20Time-57.3h-blue) ![Model](https://img.shields.io/badge/Model-openrouter%2Fqwen%2Fqwen3--coder--next-lightgrey)
 
 - 🤖 **LLM usage:** $7.5000 (166 commits)

{code2llm-0.5.115 → code2llm-0.5.117}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "code2llm"
-version = "0.5.115"
+version = "0.5.117"
 description = "High-performance Python code flow analysis with optimized TOON format - CFG, DFG, call graphs, and intelligent code queries"
 readme = "README.md"
 requires-python = ">=3.8"

{code2llm-0.5.115 → code2llm-0.5.117}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 import os
 
 # Read version
-version = "0.5.114"
+version = "0.5.116"
 
 # Read long description
 def read_readme():

code2llm-0.5.115/code2llm/core/lang/__init__.py

@@ -1,11 +0,0 @@
-"""Language-specific analyzers for non-Python source files."""
-
-from .typescript import analyze_typescript_js
-from .go_lang import analyze_go
-from .rust import analyze_rust
-from .java import analyze_java
-from .cpp import analyze_cpp
-from .csharp import analyze_csharp
-from .php import analyze_php
-from .ruby import analyze_ruby
-from .generic import analyze_generic