code2logic 1.0.28__tar.gz → 1.0.30__tar.gz

{code2logic-1.0.28 → code2logic-1.0.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: code2logic
-Version: 1.0.28
+Version: 1.0.30
 Summary: Convert source code to logical representation for LLM analysis
 License: Apache-2.0
 Keywords: code-analysis,llm,ast,static-analysis,tree-sitter,code-understanding,documentation,dependency-graph,nlp

{code2logic-1.0.28 → code2logic-1.0.30}/code2logic/__init__.py

@@ -18,7 +18,7 @@ Example:
     >>> print(output)
 """
 
-__version__ = "1.0.28"
+__version__ = "1.0.30"
 __author__ = "Softreck"
 __email__ = "info@softreck.dev"
 __license__ = "MIT"

{code2logic-1.0.28 → code2logic-1.0.30}/code2logic/analyzer.py

@@ -8,7 +8,7 @@ import sys
 from collections import defaultdict
 from datetime import datetime
 from pathlib import Path
-from typing import Dict, List
+from typing import Dict, List, Optional
 
 from .dependency import NETWORKX_AVAILABLE, DependencyAnalyzer
 from .intent import NLTK_AVAILABLE, SPACY_AVAILABLE
@@ -45,9 +45,13 @@ class ProjectAnalyzer:
     LANGUAGE_EXTENSIONS: Dict[str, str] = {
         '.py': 'python',
         '.js': 'javascript',
+        '.mjs': 'javascript',
+        '.cjs': 'javascript',
         '.jsx': 'javascript',
         '.ts': 'typescript',
         '.tsx': 'typescript',
+        '.mts': 'typescript',
+        '.cts': 'typescript',
         '.sql': 'sql',
         '.java': 'java',
         '.go': 'go',
@@ -82,6 +86,17 @@ class ProjectAnalyzer:
         'Cargo.lock', 'pnpm-lock.yaml',
     }
 
+    @staticmethod
+    def _language_from_shebang(first_line: str) -> Optional[str]:
+        s = (first_line or '').strip().lower()
+        if not s.startswith('#!'):
+            return None
+        if 'python' in s:
+            return 'python'
+        if 'node' in s:
+            return 'javascript'
+        return None
+
     def __init__(
         self,
         root_path: str,
@@ -177,12 +192,18 @@ class ProjectAnalyzer:
             if fp.name in self.IGNORE_FILES:
                 continue
 
-            # Check extension
             ext = fp.suffix.lower()
-            if ext not in self.LANGUAGE_EXTENSIONS:
+            language = self.LANGUAGE_EXTENSIONS.get(ext)
+            if language is None and ext == '':
+                try:
+                    with fp.open('r', encoding='utf-8', errors='ignore') as f:
+                        language = self._language_from_shebang(f.readline())
+                except Exception:
+                    language = None
+
+            if language is None:
                 continue
 
-            language = self.LANGUAGE_EXTENSIONS[ext]
             self.languages[language] += 1
 
             # Read file

{code2logic-1.0.28 → code2logic-1.0.30}/code2logic/function_logic.py

@@ -35,7 +35,7 @@ class FunctionLogicGenerator:
             lines.append("functions:")
 
             for kind, qname, func in items:
-                lines.extend(self._format_function(kind, qname, func, detail, indent=2))
+                lines.extend(self._format_function(kind, qname, func, detail, indent=2, module_language=module.language))
 
             lines.append("")
 
@@ -96,7 +96,7 @@ class FunctionLogicGenerator:
             lines.append(f"    functions[{len(items)}]{{{header}}}:")
 
             for kind, qname, func in items:
-                sig = self._build_sig(func, include_async_prefix=False)
+                sig = self._build_sig(func, include_async_prefix=False, language=m.language)
                 loc = self._build_loc(func)
                 is_async = 'true' if getattr(func, 'is_async', False) else 'false'
                 row = [
@@ -134,7 +134,7 @@ class FunctionLogicGenerator:
             modules_data.append({
                 'path': m.path,
                 'language': m.language,
-                'functions': [self._item_to_dict(kind, qname, func, detail) for kind, qname, func in items]
+                'functions': [self._item_to_dict(kind, qname, func, detail, module_language=m.language) for kind, qname, func in items]
             })
 
         data = {
@@ -156,11 +156,19 @@ class FunctionLogicGenerator:
 
         return items
 
-    def _build_sig(self, func: FunctionInfo, include_async_prefix: bool = True) -> str:
+    def _build_sig(self, func: FunctionInfo, include_async_prefix: bool = True, language: str = '') -> str:
         clean_params = remove_self_from_params((func.params or [])[:10])
         params_str = ', '.join(clean_params)
-        ret = func.return_type or 'None'
-        sig = f"({params_str}) -> {ret}" if params_str else f"() -> {ret}"
+        ret = getattr(func, 'return_type', None)
+        if isinstance(ret, str):
+            ret = ret.strip()
+
+        if language in ('javascript', 'typescript') and ret == 'None':
+            ret = None
+
+        sig = f"({params_str})" if params_str else "()"
+        if ret:
+            sig = f"{sig} -> {ret}"
         if include_async_prefix and getattr(func, 'is_async', False):
             sig = f"async {sig}"
         return sig
@@ -178,11 +186,11 @@ class FunctionLogicGenerator:
         does = does.replace('\n', ' ').replace('"', "'").strip()
         return does or '-'
 
-    def _item_to_dict(self, kind: str, qualified_name: str, func: FunctionInfo, detail: str) -> dict:
+    def _item_to_dict(self, kind: str, qualified_name: str, func: FunctionInfo, detail: str, module_language: str = '') -> dict:
         data = {
             'name': qualified_name,
             'kind': kind,
-            'sig': self._build_sig(func, include_async_prefix=True),
+            'sig': self._build_sig(func, include_async_prefix=True, language=module_language),
         }
 
         start_line = getattr(func, 'start_line', 0) or 0
@@ -224,6 +232,7 @@ class FunctionLogicGenerator:
         func: FunctionInfo,
         detail: str,
         indent: int,
+        module_language: str = '',
     ) -> List[str]:
         prefix = ' ' * indent
         sub = ' ' * (indent + 2)
@@ -232,7 +241,7 @@ class FunctionLogicGenerator:
 
         lines.append(f"{sub}kind: {kind}")
 
-        sig = self._build_sig(func, include_async_prefix=True)
+        sig = self._build_sig(func, include_async_prefix=True, language=module_language)
         lines.append(f"{sub}sig: {sig}")
 
         loc = self._build_loc(func)

{code2logic-1.0.28 → code2logic-1.0.30}/code2logic/toon_format.py

@@ -323,9 +323,13 @@ class TOONGenerator:
 
         param_str = ';'.join(params)
 
-        # Include return type
-        ret = f.return_type if f.return_type else 'None'
-        return f"({param_str})->{ret}"
+        ret = getattr(f, 'return_type', None)
+        if isinstance(ret, str):
+            ret = ret.strip()
+
+        if ret:
+            return f"({param_str})->{ret}"
+        return f"({param_str})"
 
     def _quote(self, value: Any) -> str:
         """Quote a value if necessary for TOON format."""

{code2logic-1.0.28 → code2logic-1.0.30}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "code2logic"
-version = "1.0.28"
+version = "1.0.30"
 description = "Convert source code to logical representation for LLM analysis"
 readme = "README.md"
 license = "Apache-2.0"
@@ -183,7 +183,7 @@ module = [
 ignore_missing_imports = true
 
 [tool.bumpver]
-current_version = "1.0.28"
+current_version = "1.0.30"
 version_pattern = "MAJOR.MINOR.PATCH"
 commit = false
 tag = false