jarvis-ai-assistant 0.7.0__py3-none-any.whl → 0.7.8__py3-none-any.whl

jarvis/jarvis_agent/language_extractors/javascript_extractor.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+"""JavaScript language symbol extractor."""
+
+from typing import Optional, Any, List
+
+from jarvis.jarvis_agent.file_context_handler import register_language_extractor
+
+
+def create_javascript_extractor() -> Optional[Any]:
+    """Create JavaScript symbol extractor using tree-sitter."""
+    try:
+        from tree_sitter import Language, Parser
+        import tree_sitter_javascript
+        from jarvis.jarvis_code_agent.code_analyzer.symbol_extractor import Symbol
+        
+        JS_LANGUAGE = tree_sitter_javascript.language()
+        JS_SYMBOL_QUERY = """
+        (function_declaration
+          name: (identifier) @function.name)
+        
+        (method_definition
+          name: (property_identifier) @method.name)
+        
+        (class_declaration
+          name: (identifier) @class.name)
+        
+        (variable_declaration
+          (variable_declarator
+            name: (identifier) @variable.name))
+        """
+        
+        class JSSymbolExtractor:
+            def __init__(self):
+                # 如果传入的是 PyCapsule，需要转换为 Language 对象
+                if not isinstance(JS_LANGUAGE, Language):
+                    self.language = Language(JS_LANGUAGE)
+                else:
+                    self.language = JS_LANGUAGE
+                self.parser = Parser()
+                # 使用 language 属性而不是 set_language 方法
+                self.parser.language = self.language
+                self.symbol_query = JS_SYMBOL_QUERY
+            
+            def extract_symbols(self, file_path: str, content: str) -> List[Any]:
+                try:
+                    tree = self.parser.parse(bytes(content, "utf8"))
+                    query = self.language.query(self.symbol_query)
+                    captures = query.captures(tree.root_node)
+                    
+                    symbols = []
+                    for node, name in captures:
+                        kind_map = {
+                            "function.name": "function",
+                            "method.name": "method",
+                            "class.name": "class",
+                            "variable.name": "variable",
+                        }
+                        symbol_kind = kind_map.get(name)
+                        if symbol_kind:
+                            symbols.append(Symbol(
+                                name=node.text.decode('utf8'),
+                                kind=symbol_kind,
+                                file_path=file_path,
+                                line_start=node.start_point[0] + 1,
+                                line_end=node.end_point[0] + 1,
+                            ))
+                    return symbols
+                except Exception:
+                    return []
+        
+        return JSSymbolExtractor()
+    except (ImportError, Exception):
+        return None
+
+
+def register_javascript_extractor() -> None:
+    """Register JavaScript extractor for .js and .jsx files."""
+    register_language_extractor(['.js', '.jsx'], create_javascript_extractor)
+

jarvis/jarvis_agent/language_extractors/python_extractor.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+"""Python language symbol extractor."""
+
+from typing import Optional, Any
+
+from jarvis.jarvis_agent.file_context_handler import register_language_extractor
+
+
+def create_python_extractor() -> Optional[Any]:
+    """Create Python symbol extractor using AST."""
+    try:
+        from jarvis.jarvis_code_agent.code_analyzer.languages.python_language import PythonSymbolExtractor
+        return PythonSymbolExtractor()
+    except (ImportError, Exception):
+        return None
+
+
+def register_python_extractor() -> None:
+    """Register Python extractor for .py and .pyw files."""
+    register_language_extractor(['.py', '.pyw'], create_python_extractor)
+

jarvis/jarvis_agent/language_extractors/rust_extractor.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+"""Rust language symbol extractor."""
+
+from typing import Optional, Any
+
+from jarvis.jarvis_agent.file_context_handler import register_language_extractor
+
+
+def create_rust_extractor() -> Optional[Any]:
+    """Create Rust symbol extractor using tree-sitter."""
+    try:
+        from jarvis.jarvis_code_agent.code_analyzer.languages.rust_language import RustSymbolExtractor
+        return RustSymbolExtractor()
+    except (ImportError, RuntimeError, Exception):
+        return None
+
+
+def register_rust_extractor() -> None:
+    """Register Rust extractor for .rs files."""
+    register_language_extractor('.rs', create_rust_extractor)
+

jarvis/jarvis_agent/language_extractors/typescript_extractor.py

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+"""TypeScript language symbol extractor."""
+
+from typing import Optional, Any, List
+
+from jarvis.jarvis_agent.file_context_handler import register_language_extractor
+
+
+def create_typescript_extractor() -> Optional[Any]:
+    """Create TypeScript symbol extractor using tree-sitter."""
+    try:
+        from tree_sitter import Language, Parser
+        import tree_sitter_typescript
+        from jarvis.jarvis_code_agent.code_analyzer.symbol_extractor import Symbol
+        
+        # tree-sitter-typescript 使用 language_typescript() 和 language_tsx()
+        TS_LANGUAGE = tree_sitter_typescript.language_typescript()
+        TS_SYMBOL_QUERY = """
+        (function_declaration
+          name: (identifier) @function.name)
+        
+        (method_definition
+          name: (property_identifier) @method.name)
+        
+        (class_declaration
+          name: (type_identifier) @class.name)
+        
+        (interface_declaration
+          name: (type_identifier) @interface.name)
+        
+        (variable_declaration
+          (variable_declarator
+            name: (identifier) @variable.name))
+        """
+        
+        class TSSymbolExtractor:
+            def __init__(self):
+                # 如果传入的是 PyCapsule，需要转换为 Language 对象
+                if not isinstance(TS_LANGUAGE, Language):
+                    self.language = Language(TS_LANGUAGE)
+                else:
+                    self.language = TS_LANGUAGE
+                self.parser = Parser()
+                # 使用 language 属性而不是 set_language 方法
+                self.parser.language = self.language
+                self.symbol_query = TS_SYMBOL_QUERY
+            
+            def extract_symbols(self, file_path: str, content: str) -> List[Any]:
+                try:
+                    tree = self.parser.parse(bytes(content, "utf8"))
+                    query = self.language.query(self.symbol_query)
+                    captures = query.captures(tree.root_node)
+                    
+                    symbols = []
+                    for node, name in captures:
+                        kind_map = {
+                            "function.name": "function",
+                            "method.name": "method",
+                            "class.name": "class",
+                            "interface.name": "interface",
+                            "variable.name": "variable",
+                        }
+                        symbol_kind = kind_map.get(name)
+                        if symbol_kind:
+                            symbols.append(Symbol(
+                                name=node.text.decode('utf8'),
+                                kind=symbol_kind,
+                                file_path=file_path,
+                                line_start=node.start_point[0] + 1,
+                                line_end=node.end_point[0] + 1,
+                            ))
+                    return symbols
+                except Exception:
+                    return []
+        
+        return TSSymbolExtractor()
+    except (ImportError, Exception):
+        return None
+
+
+def register_typescript_extractor() -> None:
+    """Register TypeScript extractor for .ts and .tsx files."""
+    register_language_extractor(['.ts', '.tsx'], create_typescript_extractor)
+