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

tree_sitter_analyzer/mcp/tools/read_partial_tool.py

@@ -0,0 +1,302 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Read Code Partial MCP Tool
+
+This tool provides partial file reading functionality through the MCP protocol,
+allowing selective content extraction with line and column range support.
+"""
+
+import json
+import logging
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+from ...file_handler import read_file_partial
+from ...utils import log_performance, setup_logger
+
+# Set up logging
+logger = setup_logger(__name__)
+
+
+class ReadPartialTool:
+    """
+    MCP Tool for reading partial content from code files.
+
+    This tool integrates with existing file_handler functionality to provide
+    selective file content reading through the MCP protocol.
+    """
+
+    def __init__(self) -> None:
+        """Initialize the read partial tool."""
+        logger.info("ReadPartialTool initialized")
+
+    def get_tool_schema(self) -> Dict[str, Any]:
+        """
+        Get the MCP tool schema for read_code_partial.
+
+        Returns:
+            Dictionary containing the tool schema
+        """
+        return {
+            "type": "object",
+            "properties": {
+                "file_path": {
+                    "type": "string",
+                    "description": "Path to the code file to read",
+                },
+                "start_line": {
+                    "type": "integer",
+                    "description": "Starting line number (1-based)",
+                    "minimum": 1,
+                },
+                "end_line": {
+                    "type": "integer",
+                    "description": "Ending line number (1-based, optional - reads to end if not specified)",
+                    "minimum": 1,
+                },
+                "start_column": {
+                    "type": "integer",
+                    "description": "Starting column number (0-based, optional)",
+                    "minimum": 0,
+                },
+                "end_column": {
+                    "type": "integer",
+                    "description": "Ending column number (0-based, optional)",
+                    "minimum": 0,
+                },
+                "format": {
+                    "type": "string",
+                    "description": "Output format for the content",
+                    "enum": ["text", "json"],
+                    "default": "text",
+                },
+            },
+            "required": ["file_path", "start_line"],
+            "additionalProperties": False,
+        }
+
+    async def execute(self, arguments: Dict[str, Any]) -> Dict[str, Any]:
+        """
+        Execute the read_code_partial tool.
+
+        Args:
+            arguments: Tool arguments containing file_path, line/column ranges, and format
+
+        Returns:
+            Dictionary containing the partial file content and metadata (CLI --partial-read compatible format)
+
+        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")
+
+        if "start_line" not in arguments:
+            raise ValueError("start_line is required")
+
+        file_path = arguments["file_path"]
+        start_line = arguments["start_line"]
+        end_line = arguments.get("end_line")
+        start_column = arguments.get("start_column")
+        end_column = arguments.get("end_column")
+        output_format = arguments.get("format", "text")
+
+        # Validate file exists
+        if not Path(file_path).exists():
+            raise FileNotFoundError(f"File not found: {file_path}")
+
+        # Validate line numbers
+        if start_line < 1:
+            raise ValueError("start_line must be >= 1")
+
+        if end_line is not None and end_line < start_line:
+            raise ValueError("end_line must be >= start_line")
+
+        # Validate column numbers
+        if start_column is not None and start_column < 0:
+            raise ValueError("start_column must be >= 0")
+
+        if end_column is not None and end_column < 0:
+            raise ValueError("end_column must be >= 0")
+
+        logger.info(
+            f"Reading partial content from {file_path}: lines {start_line}-{end_line or 'end'}"
+        )
+
+        try:
+            # Use existing file_handler functionality
+            # Use performance monitoring with proper context manager
+            from ...mcp.utils import get_performance_monitor
+
+            with get_performance_monitor().measure_operation("read_code_partial"):
+                content = self._read_file_partial(
+                    file_path, start_line, end_line, start_column, end_column
+                )
+
+                if content is None:
+                    raise RuntimeError(
+                        f"Failed to read partial content from file: {file_path}"
+                    )
+
+                # Build result structure compatible with CLI --partial-read format
+                result_data = {
+                    "file_path": file_path,
+                    "range": {
+                        "start_line": start_line,
+                        "end_line": end_line,
+                        "start_column": start_column,
+                        "end_column": end_column,
+                    },
+                    "content": content,
+                    "content_length": len(content),
+                }
+
+                # Format as JSON string like CLI does
+                json_output = json.dumps(result_data, indent=2, ensure_ascii=False)
+
+                # Build range info for header
+                range_info = f"行 {start_line}"
+                if end_line:
+                    range_info += f"-{end_line}"
+
+                # Build CLI-compatible output with header and JSON (without log message)
+                cli_output = (
+                    f"--- 部分読み込み結果 ---\n"
+                    f"ファイル: {file_path}\n"
+                    f"範囲: {range_info}\n"
+                    f"読み込み文字数: {len(content)}\n"
+                    f"{json_output}"
+                )
+
+                logger.info(
+                    f"Successfully read {len(content)} characters from {file_path}"
+                )
+
+                return {"partial_content_result": cli_output}
+
+        except Exception as e:
+            logger.error(f"Error reading partial content from {file_path}: {e}")
+            raise
+
+    def _read_file_partial(
+        self,
+        file_path: str,
+        start_line: int,
+        end_line: Optional[int] = None,
+        start_column: Optional[int] = None,
+        end_column: Optional[int] = None,
+    ) -> Optional[str]:
+        """
+        Internal method to read partial file content.
+
+        This method wraps the existing read_file_partial function from file_handler.
+
+        Args:
+            file_path: Path to the file to read
+            start_line: Starting line number (1-based)
+            end_line: Ending line number (1-based, optional)
+            start_column: Starting column number (0-based, optional)
+            end_column: Ending column number (0-based, optional)
+
+        Returns:
+            Partial file content as string, or None if error
+        """
+        return read_file_partial(
+            file_path, start_line, end_line, start_column, end_column
+        )
+
+    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
+        """
+        schema = self.get_tool_schema()
+        required_fields = schema.get("required", [])
+
+        # Check required fields
+        for field in required_fields:
+            if field not in arguments:
+                raise ValueError(f"Required field '{field}' is missing")
+
+        # Validate file_path
+        if "file_path" in arguments:
+            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 start_line
+        if "start_line" in arguments:
+            start_line = arguments["start_line"]
+            if not isinstance(start_line, int):
+                raise ValueError("start_line must be an integer")
+            if start_line < 1:
+                raise ValueError("start_line must be >= 1")
+
+        # Validate end_line
+        if "end_line" in arguments:
+            end_line = arguments["end_line"]
+            if not isinstance(end_line, int):
+                raise ValueError("end_line must be an integer")
+            if end_line < 1:
+                raise ValueError("end_line must be >= 1")
+            if "start_line" in arguments and end_line < arguments["start_line"]:
+                raise ValueError("end_line must be >= start_line")
+
+        # Validate column numbers
+        for col_field in ["start_column", "end_column"]:
+            if col_field in arguments:
+                col_value = arguments[col_field]
+                if not isinstance(col_value, int):
+                    raise ValueError(f"{col_field} must be an integer")
+                if col_value < 0:
+                    raise ValueError(f"{col_field} must be >= 0")
+
+        # Validate format
+        if "format" in arguments:
+            format_value = arguments["format"]
+            if not isinstance(format_value, str):
+                raise ValueError("format must be a string")
+            if format_value not in ["text", "json"]:
+                raise ValueError("format must be 'text' or 'json'")
+
+        return True
+
+    def get_tool_definition(self) -> Any:
+        """
+        Get the MCP tool definition for read_code_partial.
+
+        Returns:
+            Tool definition object compatible with MCP server
+        """
+        try:
+            from mcp.types import Tool
+
+            return Tool(
+                name="read_code_partial",
+                description="Read partial content from code files with line and column range support (equivalent to CLI --partial-read option)",
+                inputSchema=self.get_tool_schema(),
+            )
+        except ImportError:
+            # Fallback for when MCP is not available
+            return {
+                "name": "read_code_partial",
+                "description": "Read partial content from code files with line and column range support (equivalent to CLI --partial-read option)",
+                "inputSchema": self.get_tool_schema(),
+            }
+
+
+# Tool instance for easy access
+read_partial_tool = ReadPartialTool()

tree_sitter_analyzer/mcp/tools/table_format_tool.py

@@ -0,0 +1,359 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Table Format MCP Tool
+
+This tool provides table-formatted output for code analysis results through the MCP protocol,
+equivalent to the CLI --table=full option functionality.
+"""
+
+import logging
+import os
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+from ...core.analysis_engine import get_analysis_engine, AnalysisRequest
+from ...language_detector import detect_language_from_file
+from ...table_formatter import TableFormatter
+from ...utils import setup_logger
+from ..utils import get_performance_monitor
+
+# Set up logging
+logger = setup_logger(__name__)
+
+
+class TableFormatTool:
+    """
+    MCP Tool for formatting code analysis results as tables.
+
+    This tool integrates with existing table_formatter and analyzer components
+    to provide table-formatted output through the MCP protocol, equivalent to
+    the CLI --table=full option.
+    """
+
+    def __init__(self) -> None:
+        """Initialize the table format tool."""
+        self.logger = logger
+        self.analysis_engine = get_analysis_engine()
+        logger.info("TableFormatTool initialized")
+
+    def get_tool_schema(self) -> Dict[str, Any]:
+        """
+        Get the MCP tool schema for format_table.
+
+        Returns:
+            Dictionary containing the tool schema
+        """
+        return {
+            "type": "object",
+            "properties": {
+                "file_path": {
+                    "type": "string",
+                    "description": "Path to the code file to analyze and format",
+                },
+                "format_type": {
+                    "type": "string",
+                    "description": "Table format type",
+                    "enum": ["full", "compact", "csv"],
+                    "default": "full",
+                },
+                "language": {
+                    "type": "string",
+                    "description": "Programming language (optional, auto-detected if not specified)",
+                },
+            },
+            "required": ["file_path"],
+            "additionalProperties": False,
+        }
+
+    def validate_arguments(self, arguments: Dict[str, Any]) -> bool:
+        """
+        Validate tool arguments.
+
+        Args:
+            arguments: Dictionary of 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 format_type if provided
+        if "format_type" in arguments:
+            format_type = arguments["format_type"]
+            if not isinstance(format_type, str):
+                raise ValueError("format_type must be a string")
+            if format_type not in ["full", "compact", "csv"]:
+                raise ValueError("format_type must be one of: full, compact, csv")
+
+        # Validate language if provided
+        if "language" in arguments:
+            language = arguments["language"]
+            if not isinstance(language, str):
+                raise ValueError("language must be a string")
+
+        return True
+
+    def _convert_parameters(self, parameters):
+        """Convert parameters to expected format"""
+        result = []
+        for param in parameters:
+            if isinstance(param, dict):
+                result.append({
+                    "name": param.get('name', 'param'),
+                    "type": param.get('type', 'Object')
+                })
+            else:
+                result.append({
+                    "name": getattr(param, 'name', 'param'),
+                    "type": getattr(param, 'param_type', 'Object')
+                })
+        return result
+
+    def _get_method_modifiers(self, method) -> list:
+        """Extract method modifiers as a list"""
+        modifiers = []
+        if getattr(method, 'is_static', False):
+            modifiers.append('static')
+        if getattr(method, 'is_final', False):
+            modifiers.append('final')
+        if getattr(method, 'is_abstract', False):
+            modifiers.append('abstract')
+        return modifiers
+
+    def _get_method_parameters(self, method):
+        """Get method parameters in the correct format for TableFormatter"""
+        parameters = getattr(method, 'parameters', [])
+        
+        # If parameters is already a list of strings (like "int value"), convert to dict format
+        if parameters and isinstance(parameters[0], str):
+            result = []
+            for param_str in parameters:
+                parts = param_str.strip().split()
+                if len(parts) >= 2:
+                    param_type = ' '.join(parts[:-1])  # Everything except last part is type
+                    param_name = parts[-1]  # Last part is name
+                    result.append({
+                        "name": param_name,
+                        "type": param_type
+                    })
+                elif len(parts) == 1:
+                    # Only type, no name
+                    result.append({
+                        "name": "param",
+                        "type": parts[0]
+                    })
+            return result
+        
+        # Fallback to original conversion method
+        return self._convert_parameters(parameters)
+
+    def _get_field_modifiers(self, field) -> list:
+        """Extract field modifiers as a list"""
+        modifiers = []
+        
+        # Add visibility to modifiers for CLI compatibility
+        visibility = getattr(field, 'visibility', 'private')
+        if visibility and visibility != 'package':
+            modifiers.append(visibility)
+        
+        if getattr(field, 'is_static', False):
+            modifiers.append('static')
+        if getattr(field, 'is_final', False):
+            modifiers.append('final')
+        return modifiers
+
+    def _convert_analysis_result_to_dict(self, result) -> Dict[str, Any]:
+        """Convert AnalysisResult to dictionary format expected by TableFormatter"""
+        # Extract elements by type
+        classes = [e for e in result.elements if e.__class__.__name__ == 'Class']
+        methods = [e for e in result.elements if e.__class__.__name__ == 'Function']
+        fields = [e for e in result.elements if e.__class__.__name__ == 'Variable']
+        imports = [e for e in result.elements if e.__class__.__name__ == 'Import']
+        packages = [e for e in result.elements if e.__class__.__name__ == 'Package']
+        
+        # Convert package to expected format
+        package_info = None
+        if packages:
+            package_info = {"name": packages[0].name}
+        
+        return {
+            "file_path": result.file_path,
+            "language": result.language,
+            "package": package_info,
+            "classes": [
+                {
+                    "name": getattr(cls, 'name', 'unknown'),
+                    "line_range": {
+                        "start": getattr(cls, 'start_line', 0),
+                        "end": getattr(cls, 'end_line', 0)
+                    },
+                    "type": getattr(cls, 'class_type', 'class'),
+                    "visibility": "public",  # Force all classes to public for CLI compatibility
+                    "extends": getattr(cls, 'extends_class', None),
+                    "implements": getattr(cls, 'implements_interfaces', []),
+                    "annotations": []
+                } for cls in classes
+            ],
+            "methods": [
+                {
+                    "name": getattr(method, 'name', 'unknown'),
+                    "line_range": {
+                        "start": getattr(method, 'start_line', 0),
+                        "end": getattr(method, 'end_line', 0)
+                    },
+                    "return_type": getattr(method, 'return_type', 'void'),
+                    "parameters": self._get_method_parameters(method),
+                    "visibility": getattr(method, 'visibility', 'public'),
+                    "is_static": getattr(method, 'is_static', False),
+                    "is_constructor": getattr(method, 'is_constructor', False),
+                    "complexity_score": getattr(method, 'complexity_score', 0),
+                    "modifiers": self._get_method_modifiers(method),
+                    "annotations": []
+                } for method in methods
+            ],
+            "fields": [
+                {
+                    "name": getattr(field, 'name', 'unknown'),
+                    "type": getattr(field, 'field_type', 'Object'),
+                    "line_range": {
+                        "start": getattr(field, 'start_line', 0),
+                        "end": getattr(field, 'end_line', 0)
+                    },
+                    "visibility": getattr(field, 'visibility', 'private'),
+                    "modifiers": self._get_field_modifiers(field),
+                    "annotations": []
+                } for field in fields
+            ],
+            "imports": [
+                {
+                    "name": getattr(imp, 'name', 'unknown'),
+                    "statement": getattr(imp, 'name', ''),  # Use name for CLI compatibility
+                    "is_static": getattr(imp, 'is_static', False),
+                    "is_wildcard": getattr(imp, 'is_wildcard', False)
+                } for imp in imports
+            ],
+            "statistics": {
+                "class_count": len(classes),
+                "method_count": len(methods),
+                "field_count": len(fields),
+                "import_count": len(imports),
+                "total_lines": result.line_count
+            }
+        }
+
+    async def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
+        """Execute table formatting tool."""
+        try:
+            # Validate arguments first
+            if "file_path" not in args:
+                raise ValueError("file_path is required")
+
+            file_path = args["file_path"]
+            format_type = args.get("format_type", "full")
+            language = args.get("language")
+
+            # Validate file exists
+            if not Path(file_path).exists():
+                raise FileNotFoundError(f"File not found: {file_path}")
+
+            # Detect language if not provided
+            if not language:
+                language = detect_language_from_file(file_path)
+
+            # Use performance monitoring
+            monitor = get_performance_monitor()
+            with monitor.measure_operation("table_format_analysis"):
+                # Analyze structure using the unified analysis engine
+                request = AnalysisRequest(
+                    file_path=file_path,
+                    language=language,
+                    include_complexity=True,
+                    include_details=True
+                )
+                structure_result = await self.analysis_engine.analyze(request)
+
+                if structure_result is None:
+                    raise RuntimeError(
+                        f"Failed to analyze structure for file: {file_path}"
+                    )
+
+                # Create table formatter
+                formatter = TableFormatter(format_type)
+
+                # Convert AnalysisResult to dict format for TableFormatter
+                structure_dict = self._convert_analysis_result_to_dict(structure_result)
+                
+                # Format table
+                table_output = formatter.format_structure(structure_dict)
+
+                # Ensure output format matches CLI exactly
+                # Fix line ending differences: normalize to Unix-style LF (\n)
+                table_output = table_output.replace("\r\n", "\n").replace("\r", "\n")
+
+                # CLI uses sys.stdout.buffer.write() which doesn't add trailing newline
+                # Ensure MCP output matches this behavior exactly
+                # Remove any trailing whitespace and newlines to match CLI output
+                table_output = table_output.rstrip()
+
+                # Extract metadata from structure dict
+                metadata = {}
+                if "statistics" in structure_dict:
+                    stats = structure_dict["statistics"]
+                    metadata = {
+                        "classes_count": stats.get("class_count", 0),
+                        "methods_count": stats.get("method_count", 0),
+                        "fields_count": stats.get("field_count", 0),
+                        "total_lines": stats.get("total_lines", 0),
+                    }
+
+                return {
+                    "table_output": table_output,
+                    "format_type": format_type,
+                    "file_path": file_path,
+                    "language": language,
+                    "metadata": metadata,
+                }
+
+        except Exception as e:
+            self.logger.error(f"Error in table format tool: {e}")
+            raise
+
+    def get_tool_definition(self) -> Any:
+        """
+        Get the MCP tool definition for format_table.
+
+        Returns:
+            Tool definition object compatible with MCP server
+        """
+        try:
+            from mcp.types import Tool
+
+            return Tool(
+                name="format_table",
+                description="Format code analysis results as tables (equivalent to CLI --table=full option)",
+                inputSchema=self.get_tool_schema(),
+            )
+        except ImportError:
+            # Fallback for when MCP is not available
+            return {
+                "name": "format_table",
+                "description": "Format code analysis results as tables (equivalent to CLI --table=full option)",
+                "inputSchema": self.get_tool_schema(),
+            }
+
+
+# Tool instance for easy access
+table_format_tool = TableFormatTool()