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

hanzo_mcp/tools/common/think_tool.py

@@ -1,123 +0,0 @@
-"""Thinking tool implementation.
-
-This module provides the ThinkingTool for Claude to engage in structured thinking.
-"""
-
-from typing import Any, final, override
-
-from mcp.server.fastmcp import Context as MCPContext
-from mcp.server.fastmcp import FastMCP
-
-from hanzo_mcp.tools.common.base import BaseTool
-from hanzo_mcp.tools.common.context import create_tool_context
-
-
-@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. For example, if you explore the repo and discover 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. Alternatively, if you receive some test results, call this tool to brainstorm ways to fix the failing tests."""
-        
-    @property
-    @override
-    def parameters(self) -> dict[str, Any]:
-        """Get the parameter specifications for the tool.
-        
-        Returns:
-            Parameter specifications
-        """
-        return {
-            "properties": {
-                "thought": {
-                    "title": "Thought",
-                    "type": "string"
-                }
-            },
-            "required": ["thought"],
-            "title": "thinkArguments",
-            "type": "object"
-        }
-        
-    @property
-    @override
-    def required(self) -> list[str]:
-        """Get the list of required parameter names.
-        
-        Returns:
-            List of required parameter names
-        """
-        return ["thought"]
-        
-    def __init__(self) -> None:
-        """Initialize the thinking tool."""
-        pass
-        
-    @override
-    async def call(self, ctx: MCPContext, **params: Any) -> 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.mcp_description)
-        async def think(thought: str, ctx: MCPContext) -> str:
-            return await tool_self.call(ctx, thought=thought)

hanzo_mcp/tools/common/version_tool.py

@@ -1,120 +0,0 @@
-"""Version tool for displaying project version information."""
-
-from typing import Any, Dict, TypedDict, final, override
-
-from mcp.server.fastmcp import Context as MCPContext
-from mcp.server.fastmcp import FastMCP
-
-from hanzo_mcp.tools.common.base import BaseTool
-from hanzo_mcp.tools.common.context import create_tool_context
-
-
-class VersionToolResponse(TypedDict):
-    """Response from the version tool."""
-
-    version: str
-    package_name: str
-
-
-@final
-class VersionTool(BaseTool):
-    """Tool for displaying version information about the Hanzo MCP package."""
-
-    @property
-    @override
-    def name(self) -> str:
-        """Get the tool name.
-        
-        Returns:
-            Tool name
-        """
-        return "version"
-    
-    @property
-    @override
-    def description(self) -> str:
-        """Get the tool description.
-        
-        Returns:
-            Tool description
-        """
-        return "Display the current version of hanzo-mcp"
-    
-    @property
-    @override
-    def parameters(self) -> dict[str, Any]:
-        """Get the parameter specifications for the tool.
-        
-        Returns:
-            Parameter specifications
-        """
-        return {
-            "properties": {},
-            "required": [],
-            "title": "versionArguments",
-            "type": "object"
-        }
-    
-    @property
-    @override
-    def required(self) -> list[str]:
-        """Get the list of required parameter names.
-        
-        Returns:
-            List of required parameter names
-        """
-        return []
-        
-    def __init__(self, mcp_server: FastMCP) -> None:
-        """Initialize the version tool and register it with the server.
-
-        Args:
-            mcp_server: The MCP server to register with
-        """
-        self.register(mcp_server)
-
-    @override
-    async def call(self, ctx: MCPContext, **params: Any) -> str:
-        """Execute the tool with the given parameters.
-        
-        Args:
-            ctx: MCP context
-            **params: Tool parameters
-            
-        Returns:
-            Tool result with version information
-        """
-        tool_ctx = create_tool_context(ctx)
-        tool_ctx.set_tool_info(self.name)
-        
-        version_info = self.get_version()
-        await tool_ctx.info(f"Hanzo MCP version: {version_info['version']}")
-        
-        return f"Hanzo MCP version: {version_info['version']}"
-
-    def get_version(self) -> VersionToolResponse:
-        """Get the current version of the hanzo-mcp package.
-
-        Returns:
-            A dictionary containing the package name and version
-        """
-        # Directly use the __version__ from the hanzo_mcp package
-        from hanzo_mcp import __version__
-        
-        return {"version": __version__, "package_name": "hanzo-mcp"}
-        
-    @override
-    def register(self, mcp_server: FastMCP) -> None:
-        """Register this version tool with the MCP server.
-        
-        Creates a wrapper function that calls this tool's call method 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.mcp_description)
-        async def version(ctx: MCPContext) -> str:
-            return await tool_self.call(ctx)

hanzo_mcp/tools/filesystem/edit_file.py

@@ -1,287 +0,0 @@
-"""Edit file tool implementation.
-
-This module provides the EditFileTool for making line-based edits to text files.
-"""
-
-from difflib import unified_diff
-from pathlib import Path
-from typing import Any, final, override
-
-from mcp.server.fastmcp import Context as MCPContext
-from mcp.server.fastmcp import FastMCP
-
-from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
-
-
-@final
-class EditFileTool(FilesystemBaseTool):
-    """Tool for making line-based edits to files."""
-    
-    @property
-    @override
-    def name(self) -> str:
-        """Get the tool name.
-        
-        Returns:
-            Tool name
-        """
-        return "edit_file"
-        
-    @property
-    @override
-    def description(self) -> str:
-        """Get the tool description.
-        
-        Returns:
-            Tool description
-        """
-        return """Make line-based edits to a text file.
-
-Each edit replaces exact line sequences with new content.
-Returns a git-style diff showing the changes made.
-Only works within allowed directories."""
-        
-    @property
-    @override
-    def parameters(self) -> dict[str, Any]:
-        """Get the parameter specifications for the tool.
-        
-        Returns:
-            Parameter specifications
-        """
-        return {
-            "properties": {
-                "path": {
-                    "type": "string",
-                    "description": "The absolute path to the file to modify (must be absolute, not relative)",
-                },
-                "edits": {
-                    "items": {
-                        "properties": {
-                            "oldText": {
-                                "type": "string",
-                                "description": "The text to replace (must be unique within the file, and must match the file contents exactly, including all whitespace and indentation)",
-                                },
-                            "newText": {
-                                "type": "string",
-                                "description": "The edited text to replace the old_string",
-                                },
-                            },
-                        "additionalProperties": {
-                            "type": "string"
-                        },
-                        "type": "object"
-                    },
-                    "description":"List of edit operations [{\"oldText\": \"...\", \"newText\": \"...\"}]",
-                    "type": "array"
-                },
-                "dry_run": {
-                    "default": False,
-                    "type": "boolean",
-                    "description": "If true, do not write changes to the file"
-                }
-            },
-            "required": ["path", "edits"],
-            "title": "edit_fileArguments",
-            "type": "object"
-        }
-        
-    @property
-    @override
-    def required(self) -> list[str]:
-        """Get the list of required parameter names.
-        
-        Returns:
-            List of required parameter names
-        """
-        return ["path", "edits"]
-        
-    @override
-    async def call(self, ctx: MCPContext, **params: Any) -> str:
-        """Execute the tool with the given parameters.
-        
-        Args:
-            ctx: MCP context
-            **params: Tool parameters
-            
-        Returns:
-            Tool result
-        """
-        tool_ctx = self.create_tool_context(ctx)
-        self.set_tool_context_info(tool_ctx)
-        
-        # Extract parameters
-        path = params.get("path")
-        edits = params.get("edits")
-        dry_run = params.get("dry_run", False)  # Default to False if not provided
-        
-        if not path:
-            await tool_ctx.error("Parameter 'path' is required but was None")
-            return "Error: Parameter 'path' is required but was None"
-
-        if path.strip() == "":
-            await tool_ctx.error("Parameter 'path' cannot be empty")
-            return "Error: Parameter 'path' cannot be empty"
-
-        # Validate parameters
-        path_validation = self.validate_path(path)
-        if path_validation.is_error:
-            await tool_ctx.error(path_validation.error_message)
-            return f"Error: {path_validation.error_message}"
-
-        if not edits:
-            await tool_ctx.error("Parameter 'edits' is required but was None")
-            return "Error: Parameter 'edits' is required but was None"
-
-        if not edits:  # Check for empty list
-            await tool_ctx.warning("No edits specified")
-            return "Error: No edits specified"
-
-        # Validate each edit to ensure oldText is not empty
-        for i, edit in enumerate(edits):
-            old_text = edit.get("oldText", "")
-            if not old_text or old_text.strip() == "":
-                await tool_ctx.error(
-                    f"Parameter 'oldText' in edit at index {i} is empty"
-                )
-                return f"Error: Parameter 'oldText' in edit at index {i} cannot be empty - must provide text to match"
-
-        # dry_run parameter can be None safely as it has a default value in the function signature
-
-        await tool_ctx.info(f"Editing file: {path}")
-
-        # Check if file is allowed to be edited
-        allowed, error_msg = await self.check_path_allowed(path, tool_ctx)
-        if not allowed:
-            return error_msg
-
-        # Additional check already verified by is_path_allowed above
-        await tool_ctx.info(f"Editing file: {path}")
-
-        try:
-            file_path = Path(path)
-
-            # Check file exists
-            exists, error_msg = await self.check_path_exists(path, tool_ctx)
-            if not exists:
-                return error_msg
-                
-            # Check is a file
-            is_file, error_msg = await self.check_is_file(path, tool_ctx)
-            if not is_file:
-                return error_msg
-
-            # Read the file
-            try:
-                with open(file_path, "r", encoding="utf-8") as f:
-                    original_content = f.read()
-
-                # Apply edits
-                modified_content = original_content
-                edits_applied = 0
-
-                for edit in edits:
-                    old_text = edit.get("oldText", "")
-                    new_text = edit.get("newText", "")
-
-                    if old_text in modified_content:
-                        modified_content = modified_content.replace(
-                            old_text, new_text
-                        )
-                        edits_applied += 1
-                    else:
-                        # Try line-by-line matching for whitespace flexibility
-                        old_lines = old_text.splitlines()
-                        content_lines = modified_content.splitlines()
-
-                        for i in range(len(content_lines) - len(old_lines) + 1):
-                            current_chunk = content_lines[i : i + len(old_lines)]
-
-                            # Compare with whitespace normalization
-                            matches = all(
-                                old_line.strip() == content_line.strip()
-                                for old_line, content_line in zip(
-                                    old_lines, current_chunk
-                                )
-                            )
-
-                            if matches:
-                                # Replace the matching lines
-                                new_lines = new_text.splitlines()
-                                content_lines[i : i + len(old_lines)] = new_lines
-                                modified_content = "\n".join(content_lines)
-                                edits_applied += 1
-                                break
-
-                if edits_applied < len(edits):
-                    await tool_ctx.warning(
-                        f"Some edits could not be applied: {edits_applied}/{len(edits)}"
-                    )
-
-                # Generate diff
-                original_lines = original_content.splitlines(keepends=True)
-                modified_lines = modified_content.splitlines(keepends=True)
-
-                diff_lines = list(
-                    unified_diff(
-                        original_lines,
-                        modified_lines,
-                        fromfile=f"{path} (original)",
-                        tofile=f"{path} (modified)",
-                        n=3,
-                    )
-                )
-
-                diff_text = "".join(diff_lines)
-
-                # Determine the number of backticks needed
-                num_backticks = 3
-                while f"```{num_backticks}" in diff_text:
-                    num_backticks += 1
-
-                # Format diff with appropriate number of backticks
-                formatted_diff = (
-                    f"```{num_backticks}diff\n{diff_text}```{num_backticks}\n"
-                )
-
-                # Write the file if not a dry run
-                if not dry_run and diff_text:  # Only write if there are changes
-                    with open(file_path, "w", encoding="utf-8") as f:
-                        f.write(modified_content)
-
-                    # Update document context
-                    self.document_context.update_document(path, modified_content)
-
-                    await tool_ctx.info(
-                        f"Successfully edited file: {path} ({edits_applied} edits applied)"
-                    )
-                    return f"Successfully edited file: {path} ({edits_applied} edits applied)\n\n{formatted_diff}"
-                elif not diff_text:
-                    return f"No changes made to file: {path}"
-                else:
-                    await tool_ctx.info(
-                        f"Dry run: {edits_applied} edits would be applied"
-                    )
-                    return f"Dry run: {edits_applied} edits would be applied\n\n{formatted_diff}"
-            except UnicodeDecodeError:
-                await tool_ctx.error(f"Cannot edit binary file: {path}")
-                return f"Error: Cannot edit binary file: {path}"
-        except Exception as e:
-            await tool_ctx.error(f"Error editing file: {str(e)}")
-            return f"Error editing file: {str(e)}"
-            
-    @override
-    def register(self, mcp_server: FastMCP) -> None:
-        """Register this edit file 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.mcp_description)
-        async def edit_file(ctx: MCPContext, path: str, edits: list[dict[str, str]], dry_run: bool = False) -> str:
-            return await tool_self.call(ctx, path=path, edits=edits, dry_run=dry_run)

hanzo_mcp/tools/filesystem/get_file_info.py

@@ -1,170 +0,0 @@
-"""Get file info tool implementation.
-
-This module provides the GetFileInfoTool for retrieving metadata about files and directories.
-"""
-
-import time
-from pathlib import Path
-from typing import Any, final, override
-
-from mcp.server.fastmcp import Context as MCPContext
-from mcp.server.fastmcp import FastMCP
-
-from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
-
-
-@final
-class GetFileInfoTool(FilesystemBaseTool):
-    """Tool for retrieving metadata about files and directories."""
-    
-    @property
-    @override
-    def name(self) -> str:
-        """Get the tool name.
-        
-        Returns:
-            Tool name
-        """
-        return "get_file_info"
-        
-    @property
-    @override
-    def description(self) -> str:
-        """Get the tool description.
-        
-        Returns:
-            Tool description
-        """
-        return """Retrieve detailed metadata about a file or directory.
-
-Returns comprehensive information including size, creation time,
-last modified time, permissions, and type. This tool is perfect for
-understanding file characteristics without reading the actual content.
-Only works within allowed directories."""
-        
-    @property
-    @override
-    def parameters(self) -> dict[str, Any]:
-        """Get the parameter specifications for the tool.
-        
-        Returns:
-            Parameter specifications
-        """
-        return {
-            "properties": {
-                "path": {
-                    "type": "string",
-                    "description": "path to the file or directory to inspect"
-                }
-            },
-            "required": ["path"],
-            "type": "object"
-        }
-        
-    @property
-    @override
-    def required(self) -> list[str]:
-        """Get the list of required parameter names.
-        
-        Returns:
-            List of required parameter names
-        """
-        return ["path"]
-        
-    @override
-    async def call(self, ctx: MCPContext, **params: Any) -> str:
-        """Execute the tool with the given parameters.
-        
-        Args:
-            ctx: MCP context
-            **params: Tool parameters
-            
-        Returns:
-            Tool result
-        """
-        tool_ctx = self.create_tool_context(ctx)
-        
-        # Extract parameters
-        path = params.get("path")
-
-        if not path:
-            await tool_ctx.error("Parameter 'path' is required but was None")
-            return "Error: Parameter 'path' is required but was None"
-
-        if path.strip() == "":
-            await tool_ctx.error("Parameter 'path' cannot be empty")
-            return "Error: Parameter 'path' cannot be empty"
-        
-        # Validate path parameter
-        path_validation = self.validate_path(path)
-        if path_validation.is_error:
-            await tool_ctx.error(path_validation.error_message)
-            return f"Error: {path_validation.error_message}"
-
-        await tool_ctx.info(f"Getting file info: {path}")
-
-        # Check if path is allowed
-        allowed, error_msg = await self.check_path_allowed(path, tool_ctx)
-        if not allowed:
-            return error_msg
-
-        try:
-            file_path = Path(path)
-
-            # Check if path exists
-            exists, error_msg = await self.check_path_exists(path, tool_ctx)
-            if not exists:
-                return error_msg
-
-            # Get file stats
-            stats = file_path.stat()
-
-            # Format timestamps
-            created_time = time.strftime(
-                "%Y-%m-%d %H:%M:%S", time.localtime(stats.st_ctime)
-            )
-            modified_time = time.strftime(
-                "%Y-%m-%d %H:%M:%S", time.localtime(stats.st_mtime)
-            )
-            accessed_time = time.strftime(
-                "%Y-%m-%d %H:%M:%S", time.localtime(stats.st_atime)
-            )
-
-            # Format permissions in octal
-            permissions = oct(stats.st_mode)[-3:]
-
-            # Build info dictionary
-            file_info: dict[str, Any] = {
-                "name": file_path.name,
-                "type": "directory" if file_path.is_dir() else "file",
-                "size": stats.st_size,
-                "created": created_time,
-                "modified": modified_time,
-                "accessed": accessed_time,
-                "permissions": permissions,
-            }
-
-            # Format the output
-            result = [f"{key}: {value}" for key, value in file_info.items()]
-
-            await tool_ctx.info(f"Retrieved info for {path}")
-            return "\n".join(result)
-        except Exception as e:
-            await tool_ctx.error(f"Error getting file info: {str(e)}")
-            return f"Error getting file info: {str(e)}"
-            
-    @override
-    def register(self, mcp_server: FastMCP) -> None:
-        """Register this get file info 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.mcp_description)
-        async def get_file_info(path: str, ctx: MCPContext) -> str:
-            return await tool_self.call(ctx, path=path)