tree-sitter-analyzer 0.8.1__py3-none-any.whl → 0.8.3__py3-none-any.whl

tree_sitter_analyzer/__init__.py

@@ -11,7 +11,7 @@ Architecture:
 - Data Models: Generic and language-specific code element representations
 """
 
-__version__ = "0.4.0"
+__version__ = "0.8.3"
 __author__ = "aisheng.yu"
 __email__ = "aimasteracc@gmail.com"
 

tree_sitter_analyzer/cli/commands/base_command.py

@@ -47,7 +47,9 @@ class BaseCommand(ABC):
             return False
 
         # Security validation
-        is_valid, error_msg = self.security_validator.validate_file_path(self.args.file_path)
+        is_valid, error_msg = self.security_validator.validate_file_path(
+            self.args.file_path, base_path=self.project_root
+        )
         if not is_valid:
             output_error(f"Invalid file path: {error_msg}")
             return False

tree_sitter_analyzer/cli/commands/query_command.py

@@ -24,7 +24,7 @@ class QueryCommand(BaseCommand):
                 query_to_execute = query_loader.get_query(language, sanitized_query_key)
                 if query_to_execute is None:
                     output_error(
-                        f"ERROR: Query '{sanitized_query_key}' not found for language '{language}'"
+                        f"Query '{sanitized_query_key}' not found for language '{language}'"
                     )
                     return 1
             except ValueError as e:

tree_sitter_analyzer/mcp/server.py

@@ -71,8 +71,22 @@ class TreeSitterAnalyzerMCPServer:
     def __init__(self, project_root: str = None) -> None:
         """Initialize the MCP server with analyzer components."""
         self.server: Server | None = None
+        self._initialization_complete = False
+
+        logger.info("Starting MCP server initialization...")
+
         self.analysis_engine = get_analysis_engine(project_root)
         self.security_validator = SecurityValidator(project_root)
+        # Ensure boundary manager exposes the exact provided project_root for consistency in tests/environments
+        try:
+            import os as _os
+            if self.security_validator.boundary_manager and project_root:
+                provided_root = _os.path.abspath(project_root)
+                self.security_validator.boundary_manager.project_root = provided_root
+                # Keep allowed directories in sync with the exposed project_root
+                self.security_validator.boundary_manager.allowed_directories = {provided_root}
+        except Exception:
+            pass
         # Use unified analysis engine instead of deprecated AdvancedAnalyzer
 
         # Initialize MCP tools with security validation
@@ -88,13 +102,24 @@ class TreeSitterAnalyzerMCPServer:
         self.name = MCP_INFO["name"]
         self.version = MCP_INFO["version"]
 
-        logger.info(f"Initializing {self.name} v{self.version}")
+        self._initialization_complete = True
+        logger.info(f"MCP server initialization complete: {self.name} v{self.version}")
+
+    def is_initialized(self) -> bool:
+        """Check if the server is fully initialized."""
+        return self._initialization_complete
+
+    def _ensure_initialized(self) -> None:
+        """Ensure the server is initialized before processing requests."""
+        if not self._initialization_complete:
+            raise RuntimeError("Server not fully initialized. Please wait for initialization to complete.")
 
     @handle_mcp_errors("analyze_code_scale")
     async def _analyze_code_scale(self, arguments: dict[str, Any]) -> dict[str, Any]:
         """
         Analyze code scale and complexity metrics by delegating to the universal_analyze_tool.
         """
+        self._ensure_initialized()
         # Delegate the execution to the already initialized tool
         return await self.universal_analyze_tool.execute(arguments)
 
@@ -168,6 +193,9 @@ class TreeSitterAnalyzerMCPServer:
         ) -> list[TextContent]:
             """Handle tool calls with security validation."""
             try:
+                # Ensure server is fully initialized
+                self._ensure_initialized()
+
                 # Security validation for tool name
                 sanitized_name = self.security_validator.sanitize_input(name, max_length=100)