hanzo-mcp 0.3.4__py3-none-any.whl → 0.5.0__py3-none-any.whl

hanzo_mcp/tools/common/permissions.py

@@ -2,6 +2,8 @@
 
 import json
 import os
+import sys
+import tempfile
 from collections.abc import Awaitable, Callable
 from pathlib import Path
 from typing import Any, TypeVar, final
@@ -18,9 +20,14 @@ class PermissionManager:
     def __init__(self) -> None:
         """Initialize the permission manager."""
         # Allowed paths
-        self.allowed_paths: set[Path] = set(
-            [Path("/tmp").resolve(), Path("/var").resolve()]
-        )
+        self.allowed_paths: set[Path] = set()
+
+        # Allowed paths based on platform
+        if sys.platform == "win32":  # Windows
+            self.allowed_paths.add(Path(tempfile.gettempdir()).resolve())
+        else:  # Unix/Linux/Mac
+            self.allowed_paths.add(Path("/tmp").resolve())
+            self.allowed_paths.add(Path("/var").resolve())
 
         # Excluded paths
         self.excluded_paths: set[Path] = set()
@@ -33,17 +40,14 @@ class PermissionManager:
         """Add default exclusions for sensitive files and directories."""
         # Sensitive directories
         sensitive_dirs: list[str] = [
-            # ".git" is now allowed by default
             ".ssh",
             ".gnupg",
-            ".config",
             "node_modules",
             "__pycache__",
             ".venv",
             "venv",
             "env",
             ".idea",
-            ".vscode",
             ".DS_Store",
         ]
         self.excluded_patterns.extend(sensitive_dirs)
@@ -69,9 +73,7 @@ class PermissionManager:
         Args:
             path: The path to allow
         """
-        # Expand user path (e.g., ~/ or $HOME)
-        expanded_path = os.path.expanduser(path)
-        resolved_path: Path = Path(expanded_path).resolve()
+        resolved_path: Path = Path(path).resolve()
         self.allowed_paths.add(resolved_path)
 
     def remove_allowed_path(self, path: str) -> None:
@@ -110,9 +112,7 @@ class PermissionManager:
         Returns:
             True if the path is allowed, False otherwise
         """
-        # Expand user path (e.g., ~/ or $HOME)
-        expanded_path = os.path.expanduser(path)
-        resolved_path: Path = Path(expanded_path).resolve()
+        resolved_path: Path = Path(path).resolve()
 
         # Check exclusions first
         if self._is_path_excluded(resolved_path):

hanzo_mcp/tools/common/thinking_tool.py

@@ -0,0 +1,153 @@
+"""Thinking tool implementation.
+
+This module provides the ThinkingTool for Claude to engage in structured thinking.
+"""
+
+from typing import Annotated, TypedDict, Unpack, final, override
+
+from fastmcp import Context as MCPContext
+from fastmcp import FastMCP
+from fastmcp.server.dependencies import get_context
+from pydantic import Field
+
+from hanzo_mcp.tools.common.base import BaseTool
+from hanzo_mcp.tools.common.context import create_tool_context
+
+
+Thought = Annotated[
+    str,
+    Field(
+        description="The detailed thought process to record",
+        min_length=1,
+    ),
+]
+
+
+class ThinkingToolParams(TypedDict):
+    """Parameters for the ThinkingTool.
+
+    Attributes:
+        thought: The detailed thought process to record
+    """
+
+    thought: Thought
+
+
+@final
+class ThinkingTool(BaseTool):
+    """Tool for Claude to engage in structured thinking."""
+
+    @property
+    @override
+    def name(self) -> str:
+        """Get the tool name.
+
+        Returns:
+            Tool name
+        """
+        return "think"
+
+    @property
+    @override
+    def description(self) -> str:
+        """Get the tool description.
+
+        Returns:
+            Tool description
+        """
+        return """Use the tool to think about something. It will not obtain new information or make any changes to the repository, but just log the thought. Use it when complex reasoning or brainstorming is needed. 
+Ensure thinking content is concise and accurate, without needing to include code details
+
+Common use cases:
+1. When exploring a repository and discovering the source of a bug, call this tool to brainstorm several unique ways of fixing the bug, and assess which change(s) are likely to be simplest and most effective
+2. After receiving test results, use this tool to brainstorm ways to fix failing tests
+3. When planning a complex refactoring, use this tool to outline different approaches and their tradeoffs
+4. When designing a new feature, use this tool to think through architecture decisions and implementation details
+5. When debugging a complex issue, use this tool to organize your thoughts and hypotheses
+6. When considering changes to the plan or shifts in thinking that the user has not previously mentioned, consider whether it is necessary to confirm with the user. 
+
+<think_example>
+Feature Implementation Planning
+- New code search feature requirements:
+* Search for code patterns across multiple files
+* Identify function usages and references
+* Analyze import relationships
+* Generate summary of matching patterns
+- Implementation considerations:
+* Need to leverage existing search mechanisms
+* Should use regex for pattern matching
+* Results need consistent format with other search methods
+* Must handle large codebases efficiently
+- Design approach:
+1. Create new CodeSearcher class that follows existing search patterns
+2. Implement core pattern matching algorithm
+3. Add result formatting methods
+4. Integrate with file traversal system
+5. Add caching for performance optimization
+- Testing strategy:
+* Unit tests for search accuracy
+* Integration tests with existing components
+* Performance tests with large codebases
+</think_example>"""
+
+    def __init__(self) -> None:
+        """Initialize the thinking tool."""
+        pass
+
+    @override
+    async def call(
+        self,
+        ctx: MCPContext,
+        **params: Unpack[ThinkingToolParams],
+    ) -> str:
+        """Execute the tool with the given parameters.
+
+        Args:
+            ctx: MCP context
+            **params: Tool parameters
+
+        Returns:
+            Tool result
+        """
+        tool_ctx = create_tool_context(ctx)
+        tool_ctx.set_tool_info(self.name)
+
+        # Extract parameters
+        thought = params.get("thought")
+
+        # Validate required thought parameter
+        if not thought:
+            await tool_ctx.error(
+                "Parameter 'thought' is required but was None or empty"
+            )
+            return "Error: Parameter 'thought' is required but was None or empty"
+
+        if thought.strip() == "":
+            await tool_ctx.error("Parameter 'thought' cannot be empty")
+            return "Error: Parameter 'thought' cannot be empty"
+
+        # Log the thought but don't take action
+        await tool_ctx.info("Thinking process recorded")
+
+        # Return confirmation
+        return "I've recorded your thinking process. You can continue with your next action based on this analysis."
+
+    @override
+    def register(self, mcp_server: FastMCP) -> None:
+        """Register this thinking tool with the MCP server.
+
+        Creates a wrapper function with explicitly defined parameters that match
+        the tool's parameter schema and registers it with the MCP server.
+
+        Args:
+            mcp_server: The FastMCP server instance
+        """
+        tool_self = self  # Create a reference to self for use in the closure
+
+        @mcp_server.tool(name=self.name, description=self.description)
+        async def think(
+            ctx: MCPContext,
+            thought: Thought,
+        ) -> str:
+            ctx = get_context()
+            return await tool_self.call(ctx, thought=thought)

hanzo_mcp/tools/common/validation.py

@@ -3,7 +3,7 @@
 This module provides utilities for validating parameters in tool functions.
 """
 
-from typing import Any, TypeVar, final
+from typing import TypeVar, final
 
 T = TypeVar("T")
 
@@ -32,48 +32,6 @@ class ValidationResult:
         return not self.is_valid
 
 
-def validate_parameter(
-    parameter: Any, parameter_name: str, allow_empty: bool = False
-) -> ValidationResult:
-    """Validate a single parameter.
-
-    Args:
-        parameter: The parameter value to validate
-        parameter_name: The name of the parameter (for error messages)
-        allow_empty: Whether to allow empty strings, lists, etc.
-
-    Returns:
-        A ValidationResult indicating whether the parameter is valid
-    """
-    # Check for None
-    if parameter is None:
-        return ValidationResult(
-            is_valid=False,
-            error_message=f"Parameter '{parameter_name}' is required but was None",
-        )
-
-    # Check for empty strings
-    if isinstance(parameter, str) and not allow_empty and parameter.strip() == "":
-        return ValidationResult(
-            is_valid=False,
-            error_message=f"Parameter '{parameter_name}' is required but was empty string",
-        )
-
-    # Check for empty collections
-    if (
-        isinstance(parameter, (list, tuple, dict, set))
-        and not allow_empty
-        and len(parameter) == 0
-    ):
-        return ValidationResult(
-            is_valid=False,
-            error_message=f"Parameter '{parameter_name}' is required but was empty {type(parameter).__name__}",
-        )
-
-    # Parameter is valid
-    return ValidationResult(is_valid=True)
-
-
 def validate_path_parameter(
     path: str | None, parameter_name: str = "path"
 ) -> ValidationResult:
@@ -102,23 +60,3 @@ def validate_path_parameter(
 
     # Path is valid
     return ValidationResult(is_valid=True)
-
-
-def validate_parameters(**kwargs: Any) -> ValidationResult:
-    """Validate multiple parameters.
-
-    Accepts keyword arguments where the key is the parameter name and the value is the parameter value.
-
-    Args:
-        **kwargs: Parameters to validate as name=value pairs
-
-    Returns:
-        A ValidationResult for the first invalid parameter, or a valid result if all are valid
-    """
-    for name, value in kwargs.items():
-        result = validate_parameter(value, name)
-        if result.is_error:
-            return result
-
-    # All parameters are valid
-    return ValidationResult(is_valid=True)

hanzo_mcp/tools/filesystem/__init__.py

@@ -4,91 +4,138 @@ This package provides tools for interacting with the filesystem, including readi
 and editing files, directory navigation, and content searching.
 """
 
-from mcp.server.fastmcp import FastMCP
+from fastmcp import FastMCP
 
 from hanzo_mcp.tools.common.base import BaseTool, ToolRegistry
-from hanzo_mcp.tools.common.context import DocumentContext
+
 from hanzo_mcp.tools.common.permissions import PermissionManager
 from hanzo_mcp.tools.filesystem.content_replace import ContentReplaceTool
 from hanzo_mcp.tools.filesystem.directory_tree import DirectoryTreeTool
-from hanzo_mcp.tools.filesystem.edit_file import EditFileTool
-from hanzo_mcp.tools.filesystem.get_file_info import GetFileInfoTool
-from hanzo_mcp.tools.filesystem.read_files import ReadFilesTool
-from hanzo_mcp.tools.filesystem.search_content import SearchContentTool
-from hanzo_mcp.tools.filesystem.write_file import WriteFileTool
+from hanzo_mcp.tools.filesystem.edit import Edit
+from hanzo_mcp.tools.filesystem.grep import Grep
+from hanzo_mcp.tools.filesystem.grep_ast_tool import GrepAstTool
+from hanzo_mcp.tools.filesystem.multi_edit import MultiEdit
+from hanzo_mcp.tools.filesystem.read import ReadTool
+from hanzo_mcp.tools.filesystem.write import Write
 
 # Export all tool classes
 __all__ = [
-    "ReadFilesTool",
-    "WriteFileTool",
-    "EditFileTool",
+    "ReadTool",
+    "Write",
+    "Edit",
+    "MultiEdit",
     "DirectoryTreeTool",
-    "GetFileInfoTool",
-    "SearchContentTool",
+    "Grep",
     "ContentReplaceTool",
+    "GrepAstTool",
     "get_filesystem_tools",
     "register_filesystem_tools",
 ]
 
+
 def get_read_only_filesystem_tools(
-            document_context: DocumentContext, permission_manager: PermissionManager
+    permission_manager: PermissionManager,
 ) -> list[BaseTool]:
     """Create instances of read-only filesystem tools.
-    
+
     Args:
-        document_context: Document context for tracking file contents
         permission_manager: Permission manager for access control
 
     Returns:
         List of read-only filesystem tool instances
     """
     return [
-        ReadFilesTool(document_context, permission_manager),
-        DirectoryTreeTool(document_context, permission_manager),
-        GetFileInfoTool(document_context, permission_manager),
-        SearchContentTool(document_context, permission_manager),
+        ReadTool(permission_manager),
+        DirectoryTreeTool(permission_manager),
+        Grep(permission_manager),
+        GrepAstTool(permission_manager),
     ]
 
 
-def get_filesystem_tools(
-    document_context: DocumentContext, permission_manager: PermissionManager
-) -> list[BaseTool]:
+def get_filesystem_tools(permission_manager: PermissionManager) -> list[BaseTool]:
     """Create instances of all filesystem tools.
-    
+
     Args:
-        document_context: Document context for tracking file contents
         permission_manager: Permission manager for access control
-        
+
     Returns:
         List of filesystem tool instances
     """
     return [
-        ReadFilesTool(document_context, permission_manager),
-        WriteFileTool(document_context, permission_manager),
-        EditFileTool(document_context, permission_manager),
-        DirectoryTreeTool(document_context, permission_manager),
-        GetFileInfoTool(document_context, permission_manager),
-        SearchContentTool(document_context, permission_manager),
-        ContentReplaceTool(document_context, permission_manager),
+        ReadTool(permission_manager),
+        Write(permission_manager),
+        Edit(permission_manager),
+        MultiEdit(permission_manager),
+        DirectoryTreeTool(permission_manager),
+        Grep(permission_manager),
+        ContentReplaceTool(permission_manager),
+        GrepAstTool(permission_manager),
     ]
 
 
 def register_filesystem_tools(
     mcp_server: FastMCP,
-    document_context: DocumentContext,
     permission_manager: PermissionManager,
     disable_write_tools: bool = False,
-) -> None:
-    """Register all filesystem tools with the MCP server.
-    
+    disable_search_tools: bool = False,
+    enabled_tools: dict[str, bool] | None = None,
+) -> list[BaseTool]:
+    """Register filesystem tools with the MCP server.
+
     Args:
         mcp_server: The FastMCP server instance
-        document_context: Document context for tracking file contents
         permission_manager: Permission manager for access control
-        disable_write_tools: Whether to disable write/edit tools (default: False)
+        disable_write_tools: Whether to disable write tools (default: False)
+        disable_search_tools: Whether to disable search tools (default: False)
+        enabled_tools: Dictionary of individual tool enable states (default: None)
+
+    Returns:
+        List of registered tools
     """
-    if disable_write_tools:
-        tools = get_read_only_filesystem_tools(document_context, permission_manager)
+    # Define tool mapping
+    tool_classes = {
+        "read": ReadTool,
+        "write": Write,
+        "edit": Edit,
+        "multi_edit": MultiEdit,
+        "directory_tree": DirectoryTreeTool,
+        "grep": Grep,
+        "grep_ast": GrepAstTool,
+        "content_replace": ContentReplaceTool,
+    }
+    
+    tools = []
+    
+    if enabled_tools:
+        # Use individual tool configuration
+        for tool_name, enabled in enabled_tools.items():
+            if enabled and tool_name in tool_classes:
+                tool_class = tool_classes[tool_name]
+                tools.append(tool_class(permission_manager))
     else:
-        tools = get_filesystem_tools(document_context, permission_manager)
+        # Use category-level configuration (backward compatibility)
+        if disable_write_tools and disable_search_tools:
+            # Only read and directory tools
+            tools = [
+                ReadTool(permission_manager),
+                DirectoryTreeTool(permission_manager),
+            ]
+        elif disable_write_tools:
+            # Read-only tools including search
+            tools = get_read_only_filesystem_tools(permission_manager)
+        elif disable_search_tools:
+            # Write tools but no search
+            tools = [
+                ReadTool(permission_manager),
+                Write(permission_manager),
+                Edit(permission_manager),
+                MultiEdit(permission_manager),
+                DirectoryTreeTool(permission_manager),
+                ContentReplaceTool(permission_manager),
+            ]
+        else:
+            # All tools
+            tools = get_filesystem_tools(permission_manager)
+    
     ToolRegistry.register_tools(mcp_server, tools)
+    return tools

hanzo_mcp/tools/filesystem/base.py

@@ -8,27 +8,29 @@ from abc import ABC
 from pathlib import Path
 from typing import Any
 
-from mcp.server.fastmcp import Context as MCPContext
+from fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import FileSystemTool
 from hanzo_mcp.tools.common.context import ToolContext, create_tool_context
 
 
-class FilesystemBaseTool(FileSystemTool,ABC):
+class FilesystemBaseTool(FileSystemTool, ABC):
     """Enhanced base class for all filesystem tools.
-    
+
     Provides additional utilities specific to filesystem operations beyond
     the base functionality in FileSystemTool.
     """
-    
-    async def check_path_allowed(self, path: str, tool_ctx: Any, error_prefix: str = "Error") -> tuple[bool, str]:
+
+    async def check_path_allowed(
+        self, path: str, tool_ctx: Any, error_prefix: str = "Error"
+    ) -> tuple[bool, str]:
         """Check if a path is allowed and log an error if not.
-        
+
         Args:
             path: Path to check
             tool_ctx: Tool context for logging
             error_prefix: Prefix for error messages
-            
+
         Returns:
             tuple of (is_allowed, error_message)
         """
@@ -37,15 +39,17 @@ class FilesystemBaseTool(FileSystemTool,ABC):
             await tool_ctx.error(message)
             return False, f"{error_prefix}: {message}"
         return True, ""
-        
-    async def check_path_exists(self, path: str, tool_ctx: Any, error_prefix: str = "Error") -> tuple[bool, str]:
+
+    async def check_path_exists(
+        self, path: str, tool_ctx: Any, error_prefix: str = "Error"
+    ) -> tuple[bool, str]:
         """Check if a path exists and log an error if not.
-        
+
         Args:
             path: Path to check
             tool_ctx: Tool context for logging
             error_prefix: Prefix for error messages
-            
+
         Returns:
             tuple of (exists, error_message)
         """
@@ -55,15 +59,17 @@ class FilesystemBaseTool(FileSystemTool,ABC):
             await tool_ctx.error(message)
             return False, f"{error_prefix}: {message}"
         return True, ""
-        
-    async def check_is_file(self, path: str, tool_ctx: Any, error_prefix: str = "Error") -> tuple[bool, str]:
+
+    async def check_is_file(
+        self, path: str, tool_ctx: Any, error_prefix: str = "Error"
+    ) -> tuple[bool, str]:
         """Check if a path is a file and log an error if not.
-        
+
         Args:
             path: Path to check
             tool_ctx: Tool context for logging
             error_prefix: Prefix for error messages
-            
+
         Returns:
             tuple of (is_file, error_message)
         """
@@ -73,15 +79,17 @@ class FilesystemBaseTool(FileSystemTool,ABC):
             await tool_ctx.error(message)
             return False, f"{error_prefix}: {message}"
         return True, ""
-        
-    async def check_is_directory(self, path: str, tool_ctx: Any, error_prefix: str = "Error") -> tuple[bool, str]:
+
+    async def check_is_directory(
+        self, path: str, tool_ctx: Any, error_prefix: str = "Error"
+    ) -> tuple[bool, str]:
         """Check if a path is a directory and log an error if not.
-        
+
         Args:
             path: Path to check
             tool_ctx: Tool context for logging
             error_prefix: Prefix for error messages
-            
+
         Returns:
             tuple of (is_directory, error_message)
         """
@@ -91,22 +99,22 @@ class FilesystemBaseTool(FileSystemTool,ABC):
             await tool_ctx.error(message)
             return False, f"{error_prefix}: {message}"
         return True, ""
-        
+
     def create_tool_context(self, ctx: MCPContext) -> ToolContext:
         """Create a tool context with the tool name.
-        
+
         Args:
             ctx: MCP context
-            
+
         Returns:
             Tool context
         """
         tool_ctx = create_tool_context(ctx)
         return tool_ctx
-        
+
     def set_tool_context_info(self, tool_ctx: ToolContext) -> None:
         """Set the tool info on the context.
-        
+
         Args:
             tool_ctx: Tool context
         """