tree-sitter-analyzer 0.9.6__tar.gz → 0.9.7__tar.gz

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/CHANGELOG.md

@@ -1,4 +1,22 @@
 # Changelog
+## [0.9.7] - 2025-01-17
+
+### 🛠️ Error Handling Improvements
+
+#### 🔧 MCP Tool Enhancements
+- **Enhanced Error Decorator**: Improved `@handle_mcp_errors` decorator with tool name identification
+- **Better Error Context**: Added tool name "query_code" to error handling for improved debugging
+- **Security Validation**: Enhanced file path security validation in query tool
+
+#### 🧪 Code Quality
+- **Pre-commit Hooks**: All code quality checks passed including black, ruff, bandit, and isort
+- **Mixed Line Endings**: Fixed mixed line ending issues in query_tool.py
+- **Type Safety**: Maintained existing type annotations and code structure
+
+#### 📚 Documentation
+- **Updated Examples**: Enhanced error handling documentation
+- **Security Guidelines**: Improved security validation documentation
+
 ## [0.9.6] - 2025-01-17
 
 ### 🎉 New Feature: Advanced Query Filtering System

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tree-sitter-analyzer
-Version: 0.9.6
+Version: 0.9.7
 Summary: Extensible multi-language code analyzer framework using Tree-sitter with dynamic plugin architecture
 Project-URL: Homepage, https://github.com/aimasteracc/tree-sitter-analyzer
 Project-URL: Documentation, https://github.com/aimasteracc/tree-sitter-analyzer#readme
@@ -141,6 +141,7 @@ Description-Content-Type: text/markdown
 [![Coverage](https://img.shields.io/badge/coverage-74.36%25-green.svg)](#quality-assurance)
 [![Quality](https://img.shields.io/badge/quality-enterprise%20grade-blue.svg)](#quality-assurance)
 [![PyPI](https://img.shields.io/pypi/v/tree-sitter-analyzer.svg)](https://pypi.org/project/tree-sitter-analyzer/)
+[![Version](https://img.shields.io/badge/version-0.9.7-blue.svg)](https://github.com/aimasteracc/tree-sitter-analyzer/releases)
 [![GitHub Stars](https://img.shields.io/github/stars/aimasteracc/tree-sitter-analyzer.svg?style=social)](https://github.com/aimasteracc/tree-sitter-analyzer)
 
 ## 🚀 Break Through LLM Token Limits, Let AI Understand Code Files of Any Size
@@ -359,7 +360,12 @@ Parameters: {"file_path": "examples/BigService.java", "start_line": 100, "end_li
 }
 ```
 
-#### 🔍 **Step 4: Smart Query Filtering (NEW!)**
+#### 🔍 **Step 4: Smart Query Filtering (v0.9.6+)**
+
+**Enhanced Error Handling (v0.9.7):**
+- Improved `@handle_mcp_errors` decorator with tool name identification
+- Better error context for debugging and troubleshooting
+- Enhanced security validation for file paths
 
 **Find specific methods:**
 ```
@@ -418,7 +424,7 @@ uv run python -m tree_sitter_analyzer examples/BigService.java --partial-read --
 # Silent mode (display results only)
 uv run python -m tree_sitter_analyzer examples/BigService.java --table=full --quiet
 
-# 🔍 Query filtering examples (NEW!)
+# 🔍 Query filtering examples (v0.9.6+)
 # Find specific methods
 uv run python -m tree_sitter_analyzer examples/BigService.java --query-key methods --filter "name=main"
 

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/README.md

@@ -6,6 +6,7 @@
 [![Coverage](https://img.shields.io/badge/coverage-74.36%25-green.svg)](#quality-assurance)
 [![Quality](https://img.shields.io/badge/quality-enterprise%20grade-blue.svg)](#quality-assurance)
 [![PyPI](https://img.shields.io/pypi/v/tree-sitter-analyzer.svg)](https://pypi.org/project/tree-sitter-analyzer/)
+[![Version](https://img.shields.io/badge/version-0.9.7-blue.svg)](https://github.com/aimasteracc/tree-sitter-analyzer/releases)
 [![GitHub Stars](https://img.shields.io/github/stars/aimasteracc/tree-sitter-analyzer.svg?style=social)](https://github.com/aimasteracc/tree-sitter-analyzer)
 
 ## 🚀 Break Through LLM Token Limits, Let AI Understand Code Files of Any Size
@@ -224,7 +225,12 @@ Parameters: {"file_path": "examples/BigService.java", "start_line": 100, "end_li
 }
 ```
 
-#### 🔍 **Step 4: Smart Query Filtering (NEW!)**
+#### 🔍 **Step 4: Smart Query Filtering (v0.9.6+)**
+
+**Enhanced Error Handling (v0.9.7):**
+- Improved `@handle_mcp_errors` decorator with tool name identification
+- Better error context for debugging and troubleshooting
+- Enhanced security validation for file paths
 
 **Find specific methods:**
 ```
@@ -283,7 +289,7 @@ uv run python -m tree_sitter_analyzer examples/BigService.java --partial-read --
 # Silent mode (display results only)
 uv run python -m tree_sitter_analyzer examples/BigService.java --table=full --quiet
 
-# 🔍 Query filtering examples (NEW!)
+# 🔍 Query filtering examples (v0.9.6+)
 # Find specific methods
 uv run python -m tree_sitter_analyzer examples/BigService.java --query-key methods --filter "name=main"
 

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/README_ja.md

@@ -6,6 +6,7 @@
 [![カバレッジ](https://img.shields.io/badge/coverage-74.36%25-green.svg)](#品質保証)
 [![品質](https://img.shields.io/badge/quality-enterprise%20grade-blue.svg)](#品質保証)
 [![PyPI](https://img.shields.io/pypi/v/tree-sitter-analyzer.svg)](https://pypi.org/project/tree-sitter-analyzer/)
+[![バージョン](https://img.shields.io/badge/version-0.9.7-blue.svg)](https://github.com/aimasteracc/tree-sitter-analyzer/releases)
 [![GitHub Stars](https://img.shields.io/github/stars/aimasteracc/tree-sitter-analyzer.svg?style=social)](https://github.com/aimasteracc/tree-sitter-analyzer)
 
 ## 🚀 LLMトークン制限を突破し、AIにあらゆるサイズのコードファイルを理解させる
@@ -224,7 +225,12 @@ MCPツールextract_code_sectionを使用して指定されたコードセクシ
 }
 ```
 
-#### 🔍 **ステップ4: スマートクエリフィルタリング（NEW！）**
+#### 🔍 **ステップ4: スマートクエリフィルタリング（v0.9.6+）**
+
+**エラーハンドリングの強化（v0.9.7）：**
+- ツール名識別を追加した`@handle_mcp_errors`デコレータの改善
+- デバッグとトラブルシューティングのためのより良いエラーコンテキスト
+- ファイルパスのセキュリティ検証の強化
 
 **特定のメソッドを検索：**
 ```
@@ -283,7 +289,7 @@ uv run python -m tree_sitter_analyzer examples/BigService.java --partial-read --
 # サイレントモード（結果のみ表示）
 uv run python -m tree_sitter_analyzer examples/BigService.java --table=full --quiet
 
-# 🔍 クエリフィルタリング例（NEW！）
+# 🔍 クエリフィルタリング例（v0.9.6+）
 # 特定のメソッドを検索
 uv run python -m tree_sitter_analyzer examples/BigService.java --query-key methods --filter "name=main"
 

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/README_zh.md

@@ -6,6 +6,7 @@
 [![覆盖率](https://img.shields.io/badge/coverage-74.36%25-green.svg)](#质量保证)
 [![质量](https://img.shields.io/badge/quality-enterprise%20grade-blue.svg)](#质量保证)
 [![PyPI](https://img.shields.io/pypi/v/tree-sitter-analyzer.svg)](https://pypi.org/project/tree-sitter-analyzer/)
+[![版本](https://img.shields.io/badge/version-0.9.7-blue.svg)](https://github.com/aimasteracc/tree-sitter-analyzer/releases)
 [![GitHub Stars](https://img.shields.io/github/stars/aimasteracc/tree-sitter-analyzer.svg?style=social)](https://github.com/aimasteracc/tree-sitter-analyzer)
 
 ## 🚀 突破LLM token限制，让AI理解任意大小的代码文件
@@ -224,7 +225,12 @@ uv run python -m tree_sitter_analyzer examples/BigService.java --partial-read --
 }
 ```
 
-#### 🔍 **步骤4: 智能查询过滤（NEW！）**
+#### 🔍 **步骤4: 智能查询过滤（v0.9.6+）**
+
+**增强的错误处理（v0.9.7）：**
+- 改进了`@handle_mcp_errors`装饰器，添加工具名称标识
+- 更好的错误上下文，便于调试和故障排除
+- 增强的文件路径安全验证
 
 **查找特定方法：**
 ```
@@ -283,7 +289,7 @@ uv run python -m tree_sitter_analyzer examples/BigService.java --partial-read --
 # 静默模式（仅显示结果）
 uv run python -m tree_sitter_analyzer examples/BigService.java --table=full --quiet
 
-# 🔍 查询过滤示例（NEW！）
+# 🔍 查询过滤示例（v0.9.6+）
 # 查找特定方法
 uv run python -m tree_sitter_analyzer examples/BigService.java --query-key methods --filter "name=main"
 

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "tree-sitter-analyzer"
-version = "0.9.6"
+version = "0.9.7"
 description = "Extensible multi-language code analyzer framework using Tree-sitter with dynamic plugin architecture"
 readme = "README.md"
 requires-python = ">=3.10"

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/tree_sitter_analyzer/__init__.py

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

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/tree_sitter_analyzer/core/query_service.py

@@ -1,162 +1,162 @@
-#!/usr/bin/env python3
-"""
-Query Service
-
-Unified query service for both CLI and MCP interfaces to avoid code duplication.
-Provides core tree-sitter query functionality including predefined and custom queries.
-"""
-
-import logging
-from typing import Any
-
-from ..encoding_utils import read_file_safe
-from ..query_loader import query_loader
-from .parser import Parser
-from .query_filter import QueryFilter
-
-logger = logging.getLogger(__name__)
-
-
-class QueryService:
-    """Unified query service providing tree-sitter query functionality"""
-
-    def __init__(self, project_root: str | None = None) -> None:
-        """Initialize the query service"""
-        self.project_root = project_root
-        self.parser = Parser()
-        self.filter = QueryFilter()
-
-    async def execute_query(
-        self,
-        file_path: str,
-        language: str,
-        query_key: str | None = None,
-        query_string: str | None = None,
-        filter_expression: str | None = None,
-    ) -> list[dict[str, Any]] | None:
-        """
-        Execute a query
-
-        Args:
-            file_path: Path to the file to analyze
-            language: Programming language
-            query_key: Predefined query key (e.g., 'methods', 'class')
-            query_string: Custom query string (e.g., '(method_declaration) @method')
-            filter_expression: Filter expression (e.g., 'name=main', 'name=~get*,public=true')
-
-        Returns:
-            List of query results, each containing capture_name, node_type, start_line, end_line, content
-
-        Raises:
-            ValueError: If neither query_key nor query_string is provided
-            FileNotFoundError: If file doesn't exist
-            Exception: If query execution fails
-        """
-        if not query_key and not query_string:
-            raise ValueError("Must provide either query_key or query_string")
-
-        if query_key and query_string:
-            raise ValueError("Cannot provide both query_key and query_string")
-
-        try:
-            # Read file content
-            content, encoding = read_file_safe(file_path)
-
-            # Parse file
-            parse_result = self.parser.parse_code(content, language, file_path)
-            if not parse_result or not parse_result.tree:
-                raise Exception("Failed to parse file")
-
-            tree = parse_result.tree
-            language_obj = tree.language if hasattr(tree, "language") else None
-            if not language_obj:
-                raise Exception(f"Language object not available for {language}")
-
-            # Get query string
-            if query_key:
-                query_string = query_loader.get_query(language, query_key)
-                if not query_string:
-                    raise ValueError(
-                        f"Query '{query_key}' not found for language '{language}'"
-                    )
-
-            # Execute tree-sitter query
-            ts_query = language_obj.query(query_string)
-            captures = ts_query.captures(tree.root_node)
-
-            # Process capture results
-            results = []
-            if isinstance(captures, dict):
-                # New tree-sitter API returns dictionary
-                for capture_name, nodes in captures.items():
-                    for node in nodes:
-                        results.append(self._create_result_dict(node, capture_name))
-            else:
-                # Old tree-sitter API returns list of tuples
-                for capture in captures:
-                    if isinstance(capture, tuple) and len(capture) == 2:
-                        node, name = capture
-                        results.append(self._create_result_dict(node, name))
-
-            # Apply filters
-            if filter_expression and results:
-                results = self.filter.filter_results(results, filter_expression)
-
-            return results
-
-        except Exception as e:
-            logger.error(f"Query execution failed: {e}")
-            raise
-
-    def _create_result_dict(self, node: Any, capture_name: str) -> dict[str, Any]:
-        """
-        Create result dictionary from tree-sitter node
-
-        Args:
-            node: tree-sitter node
-            capture_name: capture name
-
-        Returns:
-            Result dictionary
-        """
-        return {
-            "capture_name": capture_name,
-            "node_type": node.type if hasattr(node, "type") else "unknown",
-            "start_line": (
-                node.start_point[0] + 1 if hasattr(node, "start_point") else 0
-            ),
-            "end_line": node.end_point[0] + 1 if hasattr(node, "end_point") else 0,
-            "content": (
-                node.text.decode("utf-8", errors="replace")
-                if hasattr(node, "text") and node.text
-                else ""
-            ),
-        }
-
-    def get_available_queries(self, language: str) -> list[str]:
-        """
-        Get available query keys for specified language
-
-        Args:
-            language: Programming language
-
-        Returns:
-            List of available query keys
-        """
-        return query_loader.list_queries(language)
-
-    def get_query_description(self, language: str, query_key: str) -> str | None:
-        """
-        Get description for query key
-
-        Args:
-            language: Programming language
-            query_key: Query key
-
-        Returns:
-            Query description, or None if not found
-        """
-        try:
-            return query_loader.get_query_description(language, query_key)
-        except Exception:
-            return None
+#!/usr/bin/env python3
+"""
+Query Service
+
+Unified query service for both CLI and MCP interfaces to avoid code duplication.
+Provides core tree-sitter query functionality including predefined and custom queries.
+"""
+
+import logging
+from typing import Any
+
+from ..encoding_utils import read_file_safe
+from ..query_loader import query_loader
+from .parser import Parser
+from .query_filter import QueryFilter
+
+logger = logging.getLogger(__name__)
+
+
+class QueryService:
+    """Unified query service providing tree-sitter query functionality"""
+
+    def __init__(self, project_root: str | None = None) -> None:
+        """Initialize the query service"""
+        self.project_root = project_root
+        self.parser = Parser()
+        self.filter = QueryFilter()
+
+    async def execute_query(
+        self,
+        file_path: str,
+        language: str,
+        query_key: str | None = None,
+        query_string: str | None = None,
+        filter_expression: str | None = None,
+    ) -> list[dict[str, Any]] | None:
+        """
+        Execute a query
+
+        Args:
+            file_path: Path to the file to analyze
+            language: Programming language
+            query_key: Predefined query key (e.g., 'methods', 'class')
+            query_string: Custom query string (e.g., '(method_declaration) @method')
+            filter_expression: Filter expression (e.g., 'name=main', 'name=~get*,public=true')
+
+        Returns:
+            List of query results, each containing capture_name, node_type, start_line, end_line, content
+
+        Raises:
+            ValueError: If neither query_key nor query_string is provided
+            FileNotFoundError: If file doesn't exist
+            Exception: If query execution fails
+        """
+        if not query_key and not query_string:
+            raise ValueError("Must provide either query_key or query_string")
+
+        if query_key and query_string:
+            raise ValueError("Cannot provide both query_key and query_string")
+
+        try:
+            # Read file content
+            content, encoding = read_file_safe(file_path)
+
+            # Parse file
+            parse_result = self.parser.parse_code(content, language, file_path)
+            if not parse_result or not parse_result.tree:
+                raise Exception("Failed to parse file")
+
+            tree = parse_result.tree
+            language_obj = tree.language if hasattr(tree, "language") else None
+            if not language_obj:
+                raise Exception(f"Language object not available for {language}")
+
+            # Get query string
+            if query_key:
+                query_string = query_loader.get_query(language, query_key)
+                if not query_string:
+                    raise ValueError(
+                        f"Query '{query_key}' not found for language '{language}'"
+                    )
+
+            # Execute tree-sitter query
+            ts_query = language_obj.query(query_string)
+            captures = ts_query.captures(tree.root_node)
+
+            # Process capture results
+            results = []
+            if isinstance(captures, dict):
+                # New tree-sitter API returns dictionary
+                for capture_name, nodes in captures.items():
+                    for node in nodes:
+                        results.append(self._create_result_dict(node, capture_name))
+            else:
+                # Old tree-sitter API returns list of tuples
+                for capture in captures:
+                    if isinstance(capture, tuple) and len(capture) == 2:
+                        node, name = capture
+                        results.append(self._create_result_dict(node, name))
+
+            # Apply filters
+            if filter_expression and results:
+                results = self.filter.filter_results(results, filter_expression)
+
+            return results
+
+        except Exception as e:
+            logger.error(f"Query execution failed: {e}")
+            raise
+
+    def _create_result_dict(self, node: Any, capture_name: str) -> dict[str, Any]:
+        """
+        Create result dictionary from tree-sitter node
+
+        Args:
+            node: tree-sitter node
+            capture_name: capture name
+
+        Returns:
+            Result dictionary
+        """
+        return {
+            "capture_name": capture_name,
+            "node_type": node.type if hasattr(node, "type") else "unknown",
+            "start_line": (
+                node.start_point[0] + 1 if hasattr(node, "start_point") else 0
+            ),
+            "end_line": node.end_point[0] + 1 if hasattr(node, "end_point") else 0,
+            "content": (
+                node.text.decode("utf-8", errors="replace")
+                if hasattr(node, "text") and node.text
+                else ""
+            ),
+        }
+
+    def get_available_queries(self, language: str) -> list[str]:
+        """
+        Get available query keys for specified language
+
+        Args:
+            language: Programming language
+
+        Returns:
+            List of available query keys
+        """
+        return query_loader.list_queries(language)
+
+    def get_query_description(self, language: str, query_key: str) -> str | None:
+        """
+        Get description for query key
+
+        Args:
+            language: Programming language
+            query_key: Query key
+
+        Returns:
+            Query description, or None if not found
+        """
+        try:
+            return query_loader.get_query_description(language, query_key)
+        except Exception:
+            return None

{tree_sitter_analyzer-0.9.6 → tree_sitter_analyzer-0.9.7}/tree_sitter_analyzer/mcp/tools/query_tool.py

@@ -74,7 +74,7 @@ class QueryTool:
             },
         }
 
-    @handle_mcp_errors
+    @handle_mcp_errors("query_code")
     async def execute(self, arguments: dict[str, Any]) -> dict[str, Any]:
         """
         Execute query tool
@@ -91,7 +91,11 @@ class QueryTool:
             raise ValueError("file_path is required")
 
         # Security validation
-        validated_path = self.security_validator.validate_file_path(file_path)
+        is_valid, error_msg = self.security_validator.validate_file_path(file_path)
+        if not is_valid:
+            raise ValueError(f"Invalid or unsafe file path: {error_msg or file_path}")
+        # Use the original path as validated path after checks
+        validated_path = file_path
 
         # Get query parameters
         query_key = arguments.get("query_key")