tree-sitter-analyzer 1.9.17.1__py3-none-any.whl

tree_sitter_analyzer/mcp/tools/universal_analyze_tool.py

@@ -0,0 +1,653 @@
+#!/usr/bin/env python3
+"""
+Universal Analyze Tool for MCP
+
+This tool provides universal code analysis capabilities through the MCP protocol,
+supporting multiple programming languages with automatic language detection.
+"""
+
+from pathlib import Path
+from typing import Any
+
+from ...constants import (
+    ELEMENT_TYPE_CLASS,
+    ELEMENT_TYPE_FUNCTION,
+    ELEMENT_TYPE_IMPORT,
+    ELEMENT_TYPE_PACKAGE,
+    ELEMENT_TYPE_VARIABLE,
+    is_element_of_type,
+)
+from ...core.analysis_engine import AnalysisRequest, get_analysis_engine
+from ...language_detector import detect_language_from_file, is_language_supported
+from ...mcp.utils import get_performance_monitor
+from ...utils import setup_logger
+from ..utils.error_handler import handle_mcp_errors
+from .base_tool import BaseMCPTool
+
+# Set up logging
+logger = setup_logger(__name__)
+
+
+class UniversalAnalyzeTool(BaseMCPTool):
+    """
+    Universal MCP Tool for code analysis across multiple languages.
+
+    This tool provides comprehensive code analysis capabilities through the MCP protocol,
+    supporting both basic and detailed analysis with language-specific optimizations.
+    """
+
+    def __init__(self, project_root: str | None = None) -> None:
+        """Initialize the universal analyze tool."""
+        super().__init__(project_root)
+        self.analysis_engine = get_analysis_engine(project_root)
+        logger.info("UniversalAnalyzeTool initialized with security validation")
+
+    def set_project_path(self, project_path: str) -> None:
+        """
+        Update the project path for all components.
+
+        Args:
+            project_path: New project root directory
+        """
+        super().set_project_path(project_path)
+        self.analysis_engine = get_analysis_engine(project_path)
+        logger.info(f"UniversalAnalyzeTool project path updated to: {project_path}")
+
+    def get_tool_definition(self) -> dict[str, Any]:
+        """
+        Get MCP tool definition for universal code analysis
+
+        Returns:
+            Tool definition dictionary
+        """
+        return {
+            "name": "analyze_code_universal",
+            "description": "Universal code analysis for multiple programming languages with automatic language detection",
+            "inputSchema": {
+                "type": "object",
+                "properties": {
+                    "file_path": {
+                        "type": "string",
+                        "description": "Path to the code file to analyze",
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Programming language (optional, auto-detected if not specified)",
+                    },
+                    "analysis_type": {
+                        "type": "string",
+                        "enum": ["basic", "detailed", "structure", "metrics"],
+                        "description": "Type of analysis to perform",
+                        "default": "basic",
+                    },
+                    "include_ast": {
+                        "type": "boolean",
+                        "description": "Include AST information in the analysis",
+                        "default": False,
+                    },
+                    "include_queries": {
+                        "type": "boolean",
+                        "description": "Include available query information",
+                        "default": False,
+                    },
+                },
+                "required": ["file_path"],
+                "additionalProperties": False,
+            },
+        }
+
+    @handle_mcp_errors("universal_analyze")
+    async def execute(self, arguments: dict[str, Any]) -> dict[str, Any]:
+        """
+        Execute universal code analysis
+
+        Args:
+            arguments: Tool arguments containing file_path and optional parameters
+
+        Returns:
+            Dictionary containing analysis results
+
+        Raises:
+            ValueError: If required arguments are missing or invalid
+            FileNotFoundError: If the specified file doesn't exist
+        """
+        # Validate required arguments
+        if "file_path" not in arguments:
+            raise ValueError("file_path is required")
+
+        file_path = arguments["file_path"]
+        language = arguments.get("language")
+        analysis_type = arguments.get("analysis_type", "basic")
+
+        # Resolve file path to absolute path
+        resolved_file_path = self.path_resolver.resolve(file_path)
+        logger.info(f"Analyzing file: {file_path} (resolved to: {resolved_file_path})")
+
+        # Security validation using resolved path
+        is_valid, error_msg = self.security_validator.validate_file_path(
+            resolved_file_path
+        )
+        if not is_valid:
+            logger.warning(
+                f"Security validation failed for file path: {resolved_file_path} - {error_msg}"
+            )
+            raise ValueError(f"Invalid file path: {error_msg}")
+
+        # Sanitize inputs
+        if language:
+            language = self.security_validator.sanitize_input(language, max_length=50)
+        if analysis_type:
+            analysis_type = self.security_validator.sanitize_input(
+                analysis_type, max_length=50
+            )
+        include_ast = arguments.get("include_ast", False)
+        include_queries = arguments.get("include_queries", False)
+
+        # Validate file exists
+        if not Path(resolved_file_path).exists():
+            raise ValueError("Invalid file path: file does not exist")
+
+        # Detect language if not specified
+        if not language:
+            language = detect_language_from_file(resolved_file_path)
+            if language == "unknown":
+                raise ValueError(
+                    f"Could not detect language for file: {resolved_file_path}"
+                )
+
+        # Check if language is supported
+        if not is_language_supported(language):
+            raise ValueError(f"Language '{language}' is not supported by tree-sitter")
+
+        # Validate analysis_type
+        valid_analysis_types = ["basic", "detailed", "structure", "metrics"]
+        if analysis_type not in valid_analysis_types:
+            raise ValueError(
+                f"Invalid analysis_type '{analysis_type}'. Valid types: {', '.join(valid_analysis_types)}"
+            )
+
+        logger.info(
+            f"Analyzing {resolved_file_path} (language: {language}, type: {analysis_type})"
+        )
+
+        try:
+            monitor = get_performance_monitor()
+            with monitor.measure_operation("universal_analyze"):
+                # Get appropriate analyzer
+                if language == "java":
+                    # Use advanced analyzer for Java
+                    result = await self._analyze_with_advanced_analyzer(
+                        resolved_file_path, language, analysis_type, include_ast
+                    )
+                else:
+                    # Use universal analyzer for other languages
+                    result = await self._analyze_with_universal_analyzer(
+                        resolved_file_path, language, analysis_type, include_ast
+                    )
+
+                # Add query information if requested
+                if include_queries:
+                    result["available_queries"] = await self._get_available_queries(
+                        language
+                    )
+
+                logger.info(f"Successfully analyzed {resolved_file_path}")
+                return result
+
+        except Exception as e:
+            logger.error(f"Error analyzing {resolved_file_path}: {e}")
+            raise
+
+    async def _analyze_with_advanced_analyzer(
+        self, file_path: str, language: str, analysis_type: str, include_ast: bool
+    ) -> dict[str, Any]:
+        """
+        Analyze using the advanced analyzer (Java-specific)
+
+        Args:
+            file_path: Path to the file to analyze
+            language: Programming language
+            analysis_type: Type of analysis to perform
+            include_ast: Whether to include AST information
+
+        Returns:
+            Analysis results dictionary
+        """
+        # Use unified analysis engine instead of deprecated advanced_analyzer
+        request = AnalysisRequest(
+            file_path=file_path,
+            language=language,
+            include_complexity=True,
+            include_details=True,
+        )
+        analysis_result = await self.analysis_engine.analyze(request)
+
+        if analysis_result is None:
+            raise RuntimeError(f"Failed to analyze file: {file_path}")
+
+        # Build base result
+        result: dict[str, Any] = {
+            "file_path": file_path,
+            "language": language,
+            "analyzer_type": "advanced",
+            "analysis_type": analysis_type,
+        }
+
+        if analysis_type == "basic":
+            result.update(self._extract_basic_metrics(analysis_result))
+        elif analysis_type == "detailed":
+            result.update(self._extract_detailed_metrics(analysis_result))
+        elif analysis_type == "structure":
+            result.update(self._extract_structure_info(analysis_result))
+        elif analysis_type == "metrics":
+            result.update(self._extract_comprehensive_metrics(analysis_result))
+
+        if include_ast:
+            result["ast_info"] = {
+                "node_count": getattr(
+                    analysis_result, "line_count", 0
+                ),  # Approximation
+                "depth": 0,  # Advanced analyzer doesn't provide this, use 0 instead of string
+            }
+
+        return result
+
+    async def _analyze_with_universal_analyzer(
+        self, file_path: str, language: str, analysis_type: str, include_ast: bool
+    ) -> dict[str, Any]:
+        """
+        Analyze using the universal analyzer
+
+        Args:
+            file_path: Path to the file to analyze
+            language: Programming language
+            analysis_type: Type of analysis to perform
+            include_ast: Whether to include AST information
+
+        Returns:
+            Analysis results dictionary
+        """
+        request = AnalysisRequest(
+            file_path=file_path,
+            language=language,
+            include_details=(analysis_type == "detailed"),
+        )
+        analysis_result = await self.analysis_engine.analyze(request)
+
+        if not analysis_result or not analysis_result.success:
+            error_message = (
+                analysis_result.error_message if analysis_result else "Unknown error"
+            )
+            raise RuntimeError(f"Failed to analyze file: {file_path} - {error_message}")
+
+        # Convert AnalysisResult to dictionary for consistent processing
+        analysis_dict = analysis_result.to_dict()
+
+        # Build base result
+        result: dict[str, Any] = {
+            "file_path": file_path,
+            "language": language,
+            "analyzer_type": "universal",
+            "analysis_type": analysis_type,
+        }
+
+        if analysis_type == "basic":
+            result.update(self._extract_universal_basic_metrics(analysis_dict))
+        elif analysis_type == "detailed":
+            result.update(self._extract_universal_detailed_metrics(analysis_dict))
+        elif analysis_type == "structure":
+            result.update(self._extract_universal_structure_info(analysis_dict))
+        elif analysis_type == "metrics":
+            result.update(self._extract_universal_comprehensive_metrics(analysis_dict))
+
+        if include_ast:
+            result["ast_info"] = analysis_dict.get("ast_info", {})
+
+        return result
+
+    def _extract_basic_metrics(self, analysis_result: Any) -> dict[str, Any]:
+        """Extract basic metrics from advanced analyzer result"""
+        stats = analysis_result.get_statistics()
+
+        return {
+            "metrics": {
+                "lines_total": analysis_result.line_count,
+                "lines_code": stats.get("lines_of_code", 0),
+                "lines_comment": stats.get("comment_lines", 0),
+                "lines_blank": stats.get("blank_lines", 0),
+                "elements": {
+                    "classes": len(
+                        [
+                            e
+                            for e in analysis_result.elements
+                            if is_element_of_type(e, ELEMENT_TYPE_CLASS)
+                        ]
+                    ),
+                    "methods": len(
+                        [
+                            e
+                            for e in analysis_result.elements
+                            if is_element_of_type(e, ELEMENT_TYPE_FUNCTION)
+                        ]
+                    ),
+                    "fields": len(
+                        [
+                            e
+                            for e in analysis_result.elements
+                            if is_element_of_type(e, ELEMENT_TYPE_VARIABLE)
+                        ]
+                    ),
+                    "imports": len(
+                        [
+                            e
+                            for e in analysis_result.elements
+                            if is_element_of_type(e, ELEMENT_TYPE_IMPORT)
+                        ]
+                    ),
+                    "annotations": len(getattr(analysis_result, "annotations", [])),
+                    "packages": len(
+                        [
+                            e
+                            for e in analysis_result.elements
+                            if is_element_of_type(e, ELEMENT_TYPE_PACKAGE)
+                        ]
+                    ),
+                    "total": (
+                        len(
+                            [
+                                e
+                                for e in analysis_result.elements
+                                if is_element_of_type(e, ELEMENT_TYPE_CLASS)
+                            ]
+                        )
+                        + len(
+                            [
+                                e
+                                for e in analysis_result.elements
+                                if is_element_of_type(e, ELEMENT_TYPE_FUNCTION)
+                            ]
+                        )
+                        + len(
+                            [
+                                e
+                                for e in analysis_result.elements
+                                if is_element_of_type(e, ELEMENT_TYPE_VARIABLE)
+                            ]
+                        )
+                        + len(
+                            [
+                                e
+                                for e in analysis_result.elements
+                                if is_element_of_type(e, ELEMENT_TYPE_IMPORT)
+                            ]
+                        )
+                        + len(
+                            [
+                                e
+                                for e in analysis_result.elements
+                                if is_element_of_type(e, ELEMENT_TYPE_PACKAGE)
+                            ]
+                        )
+                    ),
+                },
+            }
+        }
+
+    def _extract_detailed_metrics(self, analysis_result: Any) -> dict[str, Any]:
+        """Extract detailed metrics from advanced analyzer result"""
+        basic = self._extract_basic_metrics(analysis_result)
+
+        # Add complexity metrics
+        methods = [
+            e
+            for e in analysis_result.elements
+            if is_element_of_type(e, ELEMENT_TYPE_FUNCTION)
+        ]
+        total_complexity = sum(
+            getattr(method, "complexity_score", 0) or 0 for method in methods
+        )
+
+        basic["metrics"]["complexity"] = {
+            "total": total_complexity,
+            "average": round(total_complexity / len(methods) if methods else 0, 2),
+            "max": max(
+                (getattr(method, "complexity_score", 0) or 0 for method in methods),
+                default=0,
+            ),
+        }
+
+        return basic
+
+    def _extract_structure_info(self, analysis_result: Any) -> dict[str, Any]:
+        """Extract structure information from advanced analyzer result"""
+        return {
+            "structure": {
+                "package": (
+                    analysis_result.package.name if analysis_result.package else None
+                ),
+                "classes": [
+                    (
+                        cls.to_summary_item()
+                        if hasattr(cls, "to_summary_item")
+                        else {"name": getattr(cls, "name", "unknown")}
+                    )
+                    for cls in [
+                        e
+                        for e in analysis_result.elements
+                        if is_element_of_type(e, ELEMENT_TYPE_CLASS)
+                    ]
+                ],
+                "methods": [
+                    (
+                        method.to_summary_item()
+                        if hasattr(method, "to_summary_item")
+                        else {"name": getattr(method, "name", "unknown")}
+                    )
+                    for method in [
+                        e
+                        for e in analysis_result.elements
+                        if is_element_of_type(e, ELEMENT_TYPE_FUNCTION)
+                    ]
+                ],
+                "fields": [
+                    (
+                        field.to_summary_item()
+                        if hasattr(field, "to_summary_item")
+                        else {"name": getattr(field, "name", "unknown")}
+                    )
+                    for field in [
+                        e
+                        for e in analysis_result.elements
+                        if is_element_of_type(e, ELEMENT_TYPE_VARIABLE)
+                    ]
+                ],
+                "imports": [
+                    (
+                        imp.to_summary_item()
+                        if hasattr(imp, "to_summary_item")
+                        else {"name": getattr(imp, "name", "unknown")}
+                    )
+                    for imp in [
+                        e
+                        for e in analysis_result.elements
+                        if is_element_of_type(e, ELEMENT_TYPE_IMPORT)
+                    ]
+                ],
+                "annotations": [
+                    (
+                        ann.to_summary_item()
+                        if hasattr(ann, "to_summary_item")
+                        else {"name": getattr(ann, "name", "unknown")}
+                    )
+                    for ann in getattr(analysis_result, "annotations", [])
+                ],
+            }
+        }
+
+    def _extract_comprehensive_metrics(self, analysis_result: Any) -> dict[str, Any]:
+        """Extract comprehensive metrics from advanced analyzer result"""
+        detailed = self._extract_detailed_metrics(analysis_result)
+        structure = self._extract_structure_info(analysis_result)
+
+        # Combine both
+        result = detailed.copy()
+        result.update(structure)
+
+        return result
+
+    def _extract_universal_basic_metrics(
+        self, analysis_result: dict[str, Any]
+    ) -> dict[str, Any]:
+        """Extract basic metrics from universal analyzer result"""
+        elements = analysis_result.get("elements", [])
+        return {
+            "metrics": {
+                "lines_total": analysis_result.get("line_count", 0),
+                "lines_code": analysis_result.get("line_count", 0),  # Approximation
+                "lines_comment": 0,  # Not available in universal analyzer
+                "lines_blank": 0,  # Not available in universal analyzer
+                "elements": {
+                    "classes": len(
+                        [
+                            e
+                            for e in elements
+                            if hasattr(e, "element_type") and e.element_type == "class"
+                        ]
+                    ),
+                    "methods": len(
+                        [
+                            e
+                            for e in elements
+                            if hasattr(e, "element_type")
+                            and e.element_type == "function"
+                        ]
+                    ),
+                    "fields": len(
+                        [
+                            e
+                            for e in elements
+                            if hasattr(e, "element_type")
+                            and e.element_type == "variable"
+                        ]
+                    ),
+                    "imports": len(
+                        [
+                            e
+                            for e in elements
+                            if hasattr(e, "element_type") and e.element_type == "import"
+                        ]
+                    ),
+                    "annotations": 0,  # Not available in universal analyzer
+                },
+            }
+        }
+
+    def _extract_universal_detailed_metrics(
+        self, analysis_result: dict[str, Any]
+    ) -> dict[str, Any]:
+        """Extract detailed metrics from universal analyzer result"""
+        basic = self._extract_universal_basic_metrics(analysis_result)
+
+        # Add query results if available
+        if "query_results" in analysis_result:
+            basic["query_results"] = analysis_result["query_results"]
+
+        return basic
+
+    def _extract_universal_structure_info(
+        self, analysis_result: dict[str, Any]
+    ) -> dict[str, Any]:
+        """Extract structure information from universal analyzer result"""
+        return {
+            "structure": analysis_result.get("structure", {}),
+            "queries_executed": analysis_result.get("queries_executed", []),
+        }
+
+    def _extract_universal_comprehensive_metrics(
+        self, analysis_result: dict[str, Any]
+    ) -> dict[str, Any]:
+        """Extract comprehensive metrics from universal analyzer result"""
+        detailed = self._extract_universal_detailed_metrics(analysis_result)
+        structure = self._extract_universal_structure_info(analysis_result)
+
+        # Combine both
+        result = detailed.copy()
+        result.update(structure)
+
+        return result
+
+    async def _get_available_queries(self, language: str) -> dict[str, Any]:
+        """
+        Get available queries for the specified language
+
+        Args:
+            language: Programming language
+
+        Returns:
+            Dictionary containing available queries information
+        """
+        try:
+            if language == "java":
+                # For Java, we don't have predefined queries in the advanced analyzer
+                return {
+                    "language": language,
+                    "queries": [],
+                    "note": "Advanced analyzer uses built-in analysis logic",
+                }
+            else:
+                # For other languages, get from universal analyzer
+                queries = self.analysis_engine.get_supported_languages()
+                return {"language": language, "queries": queries, "count": len(queries)}
+        except Exception as e:
+            logger.warning(f"Failed to get queries for {language}: {e}")
+            return {"language": language, "queries": [], "error": str(e)}
+
+    def validate_arguments(self, arguments: dict[str, Any]) -> bool:
+        """
+        Validate tool arguments against the schema.
+
+        Args:
+            arguments: Arguments to validate
+
+        Returns:
+            True if arguments are valid
+
+        Raises:
+            ValueError: If arguments are invalid
+        """
+        # Check required fields
+        if "file_path" not in arguments:
+            raise ValueError("Required field 'file_path' is missing")
+
+        # Validate file_path
+        file_path = arguments["file_path"]
+        if not isinstance(file_path, str):
+            raise ValueError("file_path must be a string")
+        if not file_path.strip():
+            raise ValueError("file_path cannot be empty")
+
+        # Validate optional fields
+        if "language" in arguments:
+            language = arguments["language"]
+            if not isinstance(language, str):
+                raise ValueError("language must be a string")
+
+        if "analysis_type" in arguments:
+            analysis_type = arguments["analysis_type"]
+            if not isinstance(analysis_type, str):
+                raise ValueError("analysis_type must be a string")
+            valid_types = ["basic", "detailed", "structure", "metrics"]
+            if analysis_type not in valid_types:
+                raise ValueError(f"analysis_type must be one of {valid_types}")
+
+        if "include_ast" in arguments:
+            include_ast = arguments["include_ast"]
+            if not isinstance(include_ast, bool):
+                raise ValueError("include_ast must be a boolean")
+
+        if "include_queries" in arguments:
+            include_queries = arguments["include_queries"]
+            if not isinstance(include_queries, bool):
+                raise ValueError("include_queries must be a boolean")
+
+        return True