hanzo-mcp 0.7.7__py3-none-any.whl → 0.8.1__py3-none-any.whl

hanzo_mcp/tools/database/sql_search.py

@@ -1,17 +1,16 @@
 """SQL search tool for text search in database."""
 
 import sqlite3
-from typing import Annotated, Optional, TypedDict, Unpack, final, override
+from typing import Unpack, Optional, Annotated, TypedDict, final, override
 
-from mcp.server.fastmcp import Context as MCPContext
 from pydantic import Field
+from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
 from hanzo_mcp.tools.database.database_manager import DatabaseManager
 
-
 SearchPattern = Annotated[
     str,
     Field(
@@ -67,7 +66,9 @@ class SqlSearchParams(TypedDict, total=False):
 class SqlSearchTool(BaseTool):
     """Tool for searching text in SQLite database."""
 
-    def __init__(self, permission_manager: PermissionManager, db_manager: DatabaseManager):
+    def __init__(
+        self, permission_manager: PermissionManager, db_manager: DatabaseManager
+    ):
         """Initialize the SQL search tool.
 
         Args:
@@ -146,8 +147,9 @@ Use sql_query for complex queries with joins, conditions, etc."""
                 project_db = self.db_manager.get_project_db(project_path)
             else:
                 import os
+
                 project_db = self.db_manager.get_project_for_path(os.getcwd())
-                
+
             if not project_db:
                 return "Error: Could not find project database"
 
@@ -163,17 +165,17 @@ Use sql_query for complex queries with joins, conditions, etc."""
         try:
             conn = project_db.get_sqlite_connection()
             cursor = conn.cursor()
-            
+
             # Get searchable columns for the table
             if column:
                 # Validate column exists
                 cursor.execute(f"PRAGMA table_info({table})")
                 columns_info = cursor.fetchall()
                 column_names = [col[1] for col in columns_info]
-                
+
                 if column not in column_names:
                     return f"Error: Column '{column}' not found in table '{table}'. Available columns: {', '.join(column_names)}"
-                
+
                 search_columns = [column]
             else:
                 # Get all text columns
@@ -184,7 +186,7 @@ Use sql_query for complex queries with joins, conditions, etc."""
             # Build WHERE clause
             where_conditions = [f"{col} LIKE ?" for col in search_columns]
             where_clause = " OR ".join(where_conditions)
-            
+
             # Build query
             if table == "files":
                 query = f"""
@@ -194,7 +196,7 @@ Use sql_query for complex queries with joins, conditions, etc."""
                     LIMIT ?
                 """
                 params_list = [pattern] * len(search_columns) + [max_results]
-                
+
             elif table == "symbols":
                 query = f"""
                     SELECT name, type, file_path, line_start, signature
@@ -204,7 +206,7 @@ Use sql_query for complex queries with joins, conditions, etc."""
                     LIMIT ?
                 """
                 params_list = [pattern] * len(search_columns) + [max_results]
-                
+
             else:  # metadata
                 query = f"""
                     SELECT key, value, updated_at
@@ -213,17 +215,17 @@ Use sql_query for complex queries with joins, conditions, etc."""
                     LIMIT ?
                 """
                 params_list = [pattern] * len(search_columns) + [max_results]
-            
+
             # Execute search
             cursor.execute(query, params_list)
             results = cursor.fetchall()
-            
+
             if not results:
                 return f"No results found for pattern '{pattern}' in {table}"
-            
+
             # Format results
             output = self._format_results(table, results, pattern, search_columns)
-            
+
             return f"Found {len(results)} result(s) in {table}:\n\n{output}"
 
         except sqlite3.Error as e:
@@ -240,21 +242,23 @@ Use sql_query for complex queries with joins, conditions, etc."""
         """Get text columns for a table."""
         cursor.execute(f"PRAGMA table_info({table})")
         columns_info = cursor.fetchall()
-        
+
         # Get TEXT columns
         text_columns = []
         for col in columns_info:
             col_name = col[1]
             col_type = col[2].upper()
-            if 'TEXT' in col_type or 'CHAR' in col_type or col_type == '':
+            if "TEXT" in col_type or "CHAR" in col_type or col_type == "":
                 text_columns.append(col_name)
-        
+
         return text_columns
 
-    def _format_results(self, table: str, results: list, pattern: str, search_columns: list[str]) -> str:
+    def _format_results(
+        self, table: str, results: list, pattern: str, search_columns: list[str]
+    ) -> str:
         """Format search results based on table type."""
         output = []
-        
+
         if table == "files":
             output.append(f"Searched columns: {', '.join(search_columns)}\n")
             for row in results:
@@ -264,12 +268,12 @@ Use sql_query for complex queries with joins, conditions, etc."""
                 output.append(f"Modified: {modified}")
                 if snippet:
                     # Highlight pattern in snippet
-                    snippet = snippet.replace('\n', ' ')
+                    snippet = snippet.replace("\n", " ")
                     if len(snippet) > 150:
                         snippet = snippet[:150] + "..."
                     output.append(f"Content: {snippet}")
                 output.append("-" * 60)
-                
+
         elif table == "symbols":
             output.append(f"Searched columns: {', '.join(search_columns)}\n")
             for row in results:
@@ -279,7 +283,7 @@ Use sql_query for complex queries with joins, conditions, etc."""
                 if signature:
                     output.append(f"Signature: {signature}")
                 output.append("-" * 60)
-                
+
         else:  # metadata
             output.append(f"Searched columns: {', '.join(search_columns)}\n")
             for row in results:
@@ -288,9 +292,9 @@ Use sql_query for complex queries with joins, conditions, etc."""
                 output.append(f"Value: {value}")
                 output.append(f"Updated: {updated}")
                 output.append("-" * 60)
-        
+
         return "\n".join(output)
 
     def register(self, mcp_server) -> None:
         """Register this tool with the MCP server."""
-        pass
+        pass

hanzo_mcp/tools/database/sql_stats.py

@@ -1,17 +1,16 @@
 """SQL statistics tool for database insights."""
 
 import sqlite3
-from typing import Annotated, Optional, TypedDict, Unpack, final, override
+from typing import Unpack, Optional, Annotated, TypedDict, final, override
 
-from mcp.server.fastmcp import Context as MCPContext
 from pydantic import Field
+from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
 from hanzo_mcp.tools.database.database_manager import DatabaseManager
 
-
 ProjectPath = Annotated[
     Optional[str],
     Field(
@@ -40,7 +39,9 @@ class SqlStatsParams(TypedDict, total=False):
 class SqlStatsTool(BaseTool):
     """Tool for getting SQLite database statistics."""
 
-    def __init__(self, permission_manager: PermissionManager, db_manager: DatabaseManager):
+    def __init__(
+        self, permission_manager: PermissionManager, db_manager: DatabaseManager
+    ):
         """Initialize the SQL stats tool.
 
         Args:
@@ -103,8 +104,9 @@ Examples:
                 project_db = self.db_manager.get_project_db(project_path)
             else:
                 import os
+
                 project_db = self.db_manager.get_project_for_path(os.getcwd())
-                
+
             if not project_db:
                 return "Error: Could not find project database"
 
@@ -113,49 +115,53 @@ Examples:
         except Exception as e:
             return f"Error accessing project database: {str(e)}"
 
-        await tool_ctx.info(f"Getting statistics for project: {project_db.project_path}")
+        await tool_ctx.info(
+            f"Getting statistics for project: {project_db.project_path}"
+        )
 
         # Collect statistics
         conn = None
         try:
             conn = project_db.get_sqlite_connection()
             cursor = conn.cursor()
-            
+
             output = []
             output.append(f"=== SQLite Database Statistics ===")
             output.append(f"Project: {project_db.project_path}")
             output.append(f"Database: {project_db.sqlite_path}")
-            
+
             # Get database size
             db_size = project_db.sqlite_path.stat().st_size
             output.append(f"Database Size: {self._format_size(db_size)}")
             output.append("")
-            
+
             # Get table statistics
-            cursor.execute("SELECT name FROM sqlite_master WHERE type='table' ORDER BY name")
+            cursor.execute(
+                "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name"
+            )
             tables = cursor.fetchall()
-            
+
             output.append("=== Tables ===")
             total_rows = 0
-            
+
             for (table_name,) in tables:
-                if table_name.startswith('sqlite_'):
+                if table_name.startswith("sqlite_"):
                     continue
-                    
+
                 # Get row count
                 cursor.execute(f"SELECT COUNT(*) FROM {table_name}")
                 row_count = cursor.fetchone()[0]
                 total_rows += row_count
-                
+
                 # Get table info
                 cursor.execute(f"PRAGMA table_info({table_name})")
                 columns = cursor.fetchall()
                 col_count = len(columns)
-                
+
                 output.append(f"\n{table_name}:")
                 output.append(f"  Rows: {row_count:,}")
                 output.append(f"  Columns: {col_count}")
-                
+
                 if detailed and row_count > 0:
                     # Show column details
                     output.append("  Columns:")
@@ -164,44 +170,50 @@ Examples:
                         col_type = col[2]
                         is_pk = col[5]
                         not_null = col[3]
-                        
+
                         flags = []
                         if is_pk:
                             flags.append("PRIMARY KEY")
                         if not_null:
                             flags.append("NOT NULL")
-                        
+
                         flag_str = f" ({', '.join(flags)})" if flags else ""
                         output.append(f"    - {col_name}: {col_type}{flag_str}")
-                    
+
                     # Show sample data for specific tables
                     if table_name == "files" and row_count > 0:
-                        cursor.execute(f"SELECT COUNT(DISTINCT SUBSTR(path, -3)) as ext_count FROM {table_name}")
+                        cursor.execute(
+                            f"SELECT COUNT(DISTINCT SUBSTR(path, -3)) as ext_count FROM {table_name}"
+                        )
                         ext_count = cursor.fetchone()[0]
                         output.append(f"  File types: ~{ext_count}")
-                        
+
                     elif table_name == "symbols" and row_count > 0:
-                        cursor.execute(f"SELECT type, COUNT(*) as count FROM {table_name} GROUP BY type ORDER BY count DESC LIMIT 5")
+                        cursor.execute(
+                            f"SELECT type, COUNT(*) as count FROM {table_name} GROUP BY type ORDER BY count DESC LIMIT 5"
+                        )
                         symbol_types = cursor.fetchall()
                         output.append("  Symbol types:")
                         for sym_type, count in symbol_types:
                             output.append(f"    - {sym_type}: {count}")
-                
+
                 # Get indexes
                 cursor.execute(f"PRAGMA index_list({table_name})")
                 indexes = cursor.fetchall()
                 if indexes:
                     output.append(f"  Indexes: {len(indexes)}")
-            
+
             output.append(f"\nTotal Rows: {total_rows:,}")
-            
+
             # Get index statistics
-            cursor.execute("SELECT name FROM sqlite_master WHERE type='index' AND sql IS NOT NULL ORDER BY name")
+            cursor.execute(
+                "SELECT name FROM sqlite_master WHERE type='index' AND sql IS NOT NULL ORDER BY name"
+            )
             indexes = cursor.fetchall()
             if indexes:
                 output.append(f"\n=== Indexes ===")
                 output.append(f"Total Indexes: {len(indexes)}")
-                
+
                 if detailed:
                     for (idx_name,) in indexes:
                         cursor.execute(f"PRAGMA index_info({idx_name})")
@@ -209,26 +221,26 @@ Examples:
                         if idx_info:
                             cols = [info[2] for info in idx_info]
                             output.append(f"  {idx_name}: ({', '.join(cols)})")
-            
+
             # Database properties
             if detailed:
                 output.append("\n=== Database Properties ===")
-                
+
                 # Page size
                 cursor.execute("PRAGMA page_size")
                 page_size = cursor.fetchone()[0]
                 output.append(f"Page Size: {page_size:,} bytes")
-                
+
                 # Page count
                 cursor.execute("PRAGMA page_count")
                 page_count = cursor.fetchone()[0]
                 output.append(f"Page Count: {page_count:,}")
-                
+
                 # Cache size
                 cursor.execute("PRAGMA cache_size")
                 cache_size = cursor.fetchone()[0]
                 output.append(f"Cache Size: {abs(cache_size):,} pages")
-            
+
             return "\n".join(output)
 
         except sqlite3.Error as e:
@@ -243,7 +255,7 @@ Examples:
 
     def _format_size(self, size: int) -> str:
         """Format file size in human-readable format."""
-        for unit in ['B', 'KB', 'MB', 'GB']:
+        for unit in ["B", "KB", "MB", "GB"]:
             if size < 1024.0:
                 return f"{size:.1f} {unit}"
             size /= 1024.0
@@ -251,4 +263,4 @@ Examples:
 
     def register(self, mcp_server) -> None:
         """Register this tool with the MCP server."""
-        pass
+        pass

hanzo_mcp/tools/editor/neovim_command.py

@@ -1,19 +1,18 @@
 """Execute Neovim commands and macros."""
 
 import os
-import subprocess
 import shutil
 import tempfile
-from typing import Annotated, Optional, TypedDict, Unpack, final, override, List
+import subprocess
+from typing import List, Unpack, Optional, Annotated, TypedDict, final, override
 
-from mcp.server.fastmcp import Context as MCPContext
 from pydantic import Field
+from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
 
-
 Command = Annotated[
     Optional[str],
     Field(
@@ -159,7 +158,9 @@ Note: Requires Neovim to be installed.
             return "Error: Must provide either 'command', 'commands', or 'macro'"
 
         if sum(bool(x) for x in [command, commands, macro]) > 1:
-            return "Error: Can only use one of 'command', 'commands', or 'macro' at a time"
+            return (
+                "Error: Can only use one of 'command', 'commands', or 'macro' at a time"
+            )
 
         # Check if Neovim is available
         nvim_cmd = shutil.which("nvim")
@@ -168,7 +169,7 @@ Note: Requires Neovim to be installed.
 
         # Prepare commands list
         nvim_commands = []
-        
+
         if command:
             nvim_commands.append(command)
         elif commands:
@@ -178,65 +179,61 @@ Note: Requires Neovim to be installed.
             # Escape special characters
             escaped_macro = macro.replace('"', '\\"')
             nvim_commands.append(f':normal "{escaped_macro}"')
-        
+
         # Add save command if requested
         if save_after:
             nvim_commands.append(":w")
-        
+
         # Always quit at the end
         nvim_commands.append(":q")
 
         # Build Neovim command line
         cmd = [nvim_cmd, "-n", "-i", "NONE"]  # No swap file, no shada file
-        
+
         # Add commands
         for vim_cmd in nvim_commands:
             cmd.extend(["-c", vim_cmd])
-        
+
         # Add file if specified
         if file_path:
             file_path = os.path.abspath(file_path)
-            
+
             # Check permissions
             if not self.permission_manager.has_permission(file_path):
                 return f"Error: No permission to access {file_path}"
-            
+
             if not os.path.exists(file_path):
                 return f"Error: File not found: {file_path}"
-            
+
             cmd.append(file_path)
         else:
             # Create empty buffer
             cmd.append("-")
-        
+
         await tool_ctx.info(f"Executing Neovim commands: {nvim_commands}")
-        
+
         try:
             # Execute Neovim
             if return_output:
                 # Capture output by redirecting messages
-                output_file = tempfile.NamedTemporaryFile(mode='w+', delete=False)
+                output_file = tempfile.NamedTemporaryFile(mode="w+", delete=False)
                 output_file.close()
-                
+
                 # Add command to redirect messages
                 cmd.insert(3, "-c")
                 cmd.insert(4, f":redir! > {output_file.name}")
-                
+
                 # Execute
-                result = subprocess.run(
-                    cmd,
-                    capture_output=True,
-                    text=True
-                )
-                
+                result = subprocess.run(cmd, capture_output=True, text=True)
+
                 # Read output
                 output_content = ""
                 try:
-                    with open(output_file.name, 'r') as f:
+                    with open(output_file.name, "r") as f:
                         output_content = f.read().strip()
                 finally:
                     os.unlink(output_file.name)
-                
+
                 if result.returncode == 0:
                     response = "Commands executed successfully"
                     if file_path:
@@ -254,7 +251,7 @@ Note: Requires Neovim to be installed.
             else:
                 # Just execute without capturing output
                 result = subprocess.run(cmd)
-                
+
                 if result.returncode == 0:
                     response = "Commands executed successfully"
                     if file_path:
@@ -262,7 +259,7 @@ Note: Requires Neovim to be installed.
                     return response
                 else:
                     return f"Neovim exited with code {result.returncode}"
-        
+
         except Exception as e:
             await tool_ctx.error(f"Failed to execute Neovim commands: {str(e)}")
             return f"Error executing Neovim commands: {str(e)}"

hanzo_mcp/tools/editor/neovim_edit.py

@@ -1,19 +1,17 @@
 """Open files in Neovim editor."""
 
 import os
-import subprocess
 import shutil
-from typing import Annotated, Optional, TypedDict, Unpack, final, override
-from pathlib import Path
+import subprocess
+from typing import Unpack, Optional, Annotated, TypedDict, final, override
 
-from mcp.server.fastmcp import Context as MCPContext
 from pydantic import Field
+from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
 
-
 FilePath = Annotated[
     str,
     Field(
@@ -170,7 +168,7 @@ Note: Requires Neovim to be installed and available in PATH.
                 if os.path.exists(path):
                     nvim_cmd = path
                     break
-            
+
             if not nvim_cmd:
                 return """Error: Neovim (nvim) not found. Install it with:
 
@@ -187,18 +185,18 @@ Or visit: https://neovim.io/"""
 
         # Convert to absolute path
         file_path = os.path.abspath(file_path)
-        
+
         # Check permissions
         if not self.permission_manager.has_permission(file_path):
             return f"Error: No permission to access {file_path}"
 
         # Build Neovim command
         cmd = [nvim_cmd]
-        
+
         # Add read-only flag
         if read_only:
             cmd.append("-R")
-        
+
         # Add split mode
         if split:
             if split == "vsplit":
@@ -209,10 +207,10 @@ Or visit: https://neovim.io/"""
                 cmd.extend(["-c", "tabnew"])
             else:
                 return f"Error: Invalid split mode '{split}'. Use 'vsplit', 'split', or 'tab'"
-        
+
         # Add file path
         cmd.append(file_path)
-        
+
         # Add line/column positioning
         if line_number:
             if column_number:
@@ -221,9 +219,9 @@ Or visit: https://neovim.io/"""
             else:
                 # Go to specific line
                 cmd.append(f"+{line_number}")
-        
+
         await tool_ctx.info(f"Opening {file_path} in Neovim")
-        
+
         try:
             # Determine how to run Neovim
             if in_terminal and not wait:
@@ -233,9 +231,9 @@ Or visit: https://neovim.io/"""
                     if shutil.which("osascript"):
                         # Build AppleScript to open in iTerm2 or Terminal
                         nvim_cmd_str = " ".join(f'"{arg}"' for arg in cmd)
-                        
+
                         # Try iTerm2 first
-                        applescript = f'''tell application "System Events"
+                        applescript = f"""tell application "System Events"
                             if exists application process "iTerm2" then
                                 tell application "iTerm"
                                     activate
@@ -252,35 +250,35 @@ Or visit: https://neovim.io/"""
                                     do script "{nvim_cmd_str}"
                                 end tell
                             end if
-                        end tell'''
-                        
+                        end tell"""
+
                         subprocess.run(["osascript", "-e", applescript])
                         return f"Opened {file_path} in Neovim (new terminal window)"
-                
+
                 elif shutil.which("gnome-terminal"):
                     # Linux with GNOME
                     subprocess.Popen(["gnome-terminal", "--"] + cmd)
                     return f"Opened {file_path} in Neovim (new terminal window)"
-                
+
                 elif shutil.which("xterm"):
                     # Fallback to xterm
                     subprocess.Popen(["xterm", "-e"] + cmd)
                     return f"Opened {file_path} in Neovim (new terminal window)"
-                
+
                 else:
                     # Can't open in terminal, fall back to subprocess
                     subprocess.Popen(cmd)
                     return f"Opened {file_path} in Neovim (background process)"
-            
+
             else:
                 # Run and wait for completion
                 result = subprocess.run(cmd)
-                
+
                 if result.returncode == 0:
                     return f"Successfully edited {file_path} in Neovim"
                 else:
                     return f"Neovim exited with code {result.returncode}"
-        
+
         except Exception as e:
             await tool_ctx.error(f"Failed to open Neovim: {str(e)}")
             return f"Error opening Neovim: {str(e)}"