code-puppy 0.0.135__py3-none-any.whl → 0.0.137__py3-none-any.whl

code_puppy/tui/screens/mcp_install_wizard.py

@@ -4,24 +4,12 @@ MCP Install Wizard Screen - TUI interface for installing MCP servers.
 
 import json
 import os
-from typing import Dict, List, Optional
 
 from textual import on
 from textual.app import ComposeResult
-from textual.containers import Container, Horizontal, Vertical
+from textual.containers import Container, Horizontal
 from textual.screen import ModalScreen
-from textual.widgets import (
-    Button, 
-    Input, 
-    Label, 
-    ListItem, 
-    ListView, 
-    Static, 
-    Select,
-    TextArea
-)
-
-from code_puppy.messaging import emit_info
+from textual.widgets import Button, Input, ListItem, ListView, Static, TextArea
 
 
 class MCPInstallWizardScreen(ModalScreen):
@@ -140,7 +128,7 @@ class MCPInstallWizardScreen(ModalScreen):
         width: 2fr;
         border: solid $primary;
     }
-    
+
     #custom-json-container {
         width: 100%;
         height: 1fr;
@@ -148,7 +136,7 @@ class MCPInstallWizardScreen(ModalScreen):
         display: none;
         padding: 1;
     }
-    
+
     #custom-json-header {
         width: 100%;
         height: 2;
@@ -156,13 +144,13 @@ class MCPInstallWizardScreen(ModalScreen):
         color: $warning;
         margin-bottom: 1;
     }
-    
+
     #custom-name-input {
         width: 100%;
         margin-bottom: 1;
         border: solid $primary;
     }
-    
+
     #custom-json-input {
         width: 100%;
         height: 1fr;
@@ -170,7 +158,7 @@ class MCPInstallWizardScreen(ModalScreen):
         margin-bottom: 1;
         background: $surface-darken-1;
     }
-    
+
     #custom-json-button {
         width: auto;
         height: 3;
@@ -183,24 +171,30 @@ class MCPInstallWizardScreen(ModalScreen):
         """Create the wizard layout."""
         with Container(id="wizard-container"):
             yield Static("🔌 MCP Server Install Wizard", id="wizard-header")
-            
+
             # Step 1: Search and select server
             with Container(id="search-container"):
-                yield Input(placeholder="Search MCP servers (e.g. 'github', 'postgres')...", id="search-input")
+                yield Input(
+                    placeholder="Search MCP servers (e.g. 'github', 'postgres')...",
+                    id="search-input",
+                )
                 yield ListView(id="results-list")
-            
+
             # Step 2: Configure server (hidden initially)
             with Container(id="config-container"):
                 yield Static("Server Configuration", id="config-header")
                 yield Container(id="server-info")
                 yield Container(id="env-vars-container")
-            
+
             # Step 3: Custom JSON configuration (hidden initially)
             with Container(id="custom-json-container"):
                 yield Static("📝 Custom JSON Configuration", id="custom-json-header")
-                yield Input(placeholder="Server name (e.g. 'my-sqlite-db')", id="custom-name-input")
+                yield Input(
+                    placeholder="Server name (e.g. 'my-sqlite-db')",
+                    id="custom-name-input",
+                )
                 yield TextArea(id="custom-json-input")
-            
+
             # Navigation buttons
             with Horizontal(id="button-container"):
                 yield Button("Cancel", id="cancel-button", variant="default")
@@ -213,7 +207,7 @@ class MCPInstallWizardScreen(ModalScreen):
         """Initialize the wizard."""
         self._show_search_step()
         self._load_popular_servers()
-        
+
         # Focus the search input
         search_input = self.query_one("#search-input", Input)
         search_input.focus()
@@ -225,7 +219,7 @@ class MCPInstallWizardScreen(ModalScreen):
         self.query_one("#search-container").display = True
         self.query_one("#config-container").display = False
         self.query_one("#custom-json-container").display = False
-        
+
         self.query_one("#back-button").display = False
         self.query_one("#custom-json-button").display = True
         self.query_one("#next-button").display = True
@@ -238,14 +232,14 @@ class MCPInstallWizardScreen(ModalScreen):
         self.query_one("#search-container").display = False
         self.query_one("#config-container").display = True
         self.query_one("#custom-json-container").display = False
-        
+
         self.query_one("#back-button").display = True
         self.query_one("#custom-json-button").display = False
         self.query_one("#next-button").display = False
         self.query_one("#install-button").display = True
-        
+
         self._setup_server_config()
-    
+
     def _show_custom_json_step(self) -> None:
         """Show the custom JSON configuration step."""
         self.step = "custom_json"
@@ -253,16 +247,16 @@ class MCPInstallWizardScreen(ModalScreen):
         self.query_one("#search-container").display = False
         self.query_one("#config-container").display = False
         self.query_one("#custom-json-container").display = True
-        
+
         self.query_one("#back-button").display = True
         self.query_one("#custom-json-button").display = False
         self.query_one("#next-button").display = False
         self.query_one("#install-button").display = True
-        
+
         # Pre-populate with SQLite example
         name_input = self.query_one("#custom-name-input", Input)
         name_input.value = "my-sqlite-db"
-        
+
         json_input = self.query_one("#custom-json-input", TextArea)
         json_input.text = """{
   "type": "stdio",
@@ -270,7 +264,7 @@ class MCPInstallWizardScreen(ModalScreen):
   "args": ["-y", "@modelcontextprotocol/server-sqlite", "./database.db"],
   "timeout": 30
 }"""
-        
+
         # Focus the name input
         name_input.focus()
 
@@ -278,65 +272,79 @@ class MCPInstallWizardScreen(ModalScreen):
         """Load all available servers into the list."""
         self.search_counter += 1
         counter = self.search_counter
-        
+
         try:
             from code_puppy.mcp.server_registry_catalog import catalog
+
             # Load ALL servers instead of just popular ones
             servers = catalog.servers
-            
+
             results_list = self.query_one("#results-list", ListView)
             # Force clear by removing all children
             results_list.remove_children()
-            
+
             if servers:
                 # Sort servers to show popular and verified first
-                sorted_servers = sorted(servers, key=lambda s: (not s.popular, not s.verified, s.display_name))
-                
+                sorted_servers = sorted(
+                    servers,
+                    key=lambda s: (not s.popular, not s.verified, s.display_name),
+                )
+
                 for i, server in enumerate(sorted_servers):
                     indicators = []
                     if server.verified:
                         indicators.append("✓")
                     if server.popular:
                         indicators.append("⭐")
-                    
+
                     display_name = f"{server.display_name} {''.join(indicators)}"
-                    description = server.description[:60] + "..." if len(server.description) > 60 else server.description
-                    
+                    description = (
+                        server.description[:60] + "..."
+                        if len(server.description) > 60
+                        else server.description
+                    )
+
                     item_text = f"{display_name}\n[dim]{description}[/dim]"
                     # Use counter to ensure globally unique IDs
                     item = ListItem(Static(item_text), id=f"item-{counter}-{i}")
                     item.server_data = server
                     results_list.append(item)
             else:
-                no_servers_item = ListItem(Static("No servers found"), id=f"no-results-{counter}")
+                no_servers_item = ListItem(
+                    Static("No servers found"), id=f"no-results-{counter}"
+                )
                 results_list.append(no_servers_item)
-                
+
         except ImportError:
             results_list = self.query_one("#results-list", ListView)
             results_list.remove_children()
-            error_item = ListItem(Static("[red]Server registry not available[/red]"), id=f"error-{counter}")
+            error_item = ListItem(
+                Static("[red]Server registry not available[/red]"),
+                id=f"error-{counter}",
+            )
             results_list.append(error_item)
 
     @on(Input.Changed, "#search-input")
     def on_search_changed(self, event: Input.Changed) -> None:
         """Handle search input changes."""
         query = event.value.strip()
-        
+
         if not query:
             self._load_popular_servers()  # This now loads all servers
             return
-        
+
         self.search_counter += 1
         counter = self.search_counter
-        
+
         try:
             from code_puppy.mcp.server_registry_catalog import catalog
+
             servers = catalog.search(query)
-            
+
             results_list = self.query_one("#results-list", ListView)
             # Force clear by removing all children
             results_list.remove_children()
-            
+
             if servers:
                 for i, server in enumerate(servers[:15]):  # Limit results
                     indicators = []
@@ -344,29 +352,39 @@ class MCPInstallWizardScreen(ModalScreen):
                         indicators.append("✓")
                     if server.popular:
                         indicators.append("⭐")
-                    
+
                     display_name = f"{server.display_name} {''.join(indicators)}"
-                    description = server.description[:60] + "..." if len(server.description) > 60 else server.description
-                    
+                    description = (
+                        server.description[:60] + "..."
+                        if len(server.description) > 60
+                        else server.description
+                    )
+
                     item_text = f"{display_name}\n[dim]{description}[/dim]"
                     # Use counter to ensure globally unique IDs
                     item = ListItem(Static(item_text), id=f"item-{counter}-{i}")
                     item.server_data = server
                     results_list.append(item)
             else:
-                no_results_item = ListItem(Static(f"No servers found for '{query}'"), id=f"no-results-{counter}")
+                no_results_item = ListItem(
+                    Static(f"No servers found for '{query}'"),
+                    id=f"no-results-{counter}",
+                )
                 results_list.append(no_results_item)
-                
+
         except ImportError:
             results_list = self.query_one("#results-list", ListView)
             results_list.remove_children()
-            error_item = ListItem(Static("[red]Server registry not available[/red]"), id=f"error-{counter}")
+            error_item = ListItem(
+                Static("[red]Server registry not available[/red]"),
+                id=f"error-{counter}",
+            )
             results_list.append(error_item)
 
     @on(ListView.Selected, "#results-list")
     def on_server_selected(self, event: ListView.Selected) -> None:
         """Handle server selection."""
-        if hasattr(event.item, 'server_data'):
+        if hasattr(event.item, "server_data"):
             self.selected_server = event.item.server_data
 
     @on(Button.Pressed, "#next-button")
@@ -391,7 +409,7 @@ class MCPInstallWizardScreen(ModalScreen):
     def on_custom_json_clicked(self) -> None:
         """Handle custom JSON button click."""
         self._show_custom_json_step()
-    
+
     @on(Button.Pressed, "#install-button")
     def on_install_clicked(self) -> None:
         """Handle install button click."""
@@ -409,17 +427,17 @@ class MCPInstallWizardScreen(ModalScreen):
         """Setup the server configuration step."""
         if not self.selected_server:
             return
-        
+
         # Show server info
         server_info = self.query_one("#server-info", Container)
         server_info.remove_children()
-        
+
         info_text = f"""[bold]{self.selected_server.display_name}[/bold]
 {self.selected_server.description}
 
 [yellow]Category:[/yellow] {self.selected_server.category}
-[yellow]Type:[/yellow] {getattr(self.selected_server, 'type', 'stdio')}"""
-        
+[yellow]Type:[/yellow] {getattr(self.selected_server, "type", "stdio")}"""
+
         # Show requirements summary
         requirements = self.selected_server.get_requirements()
         req_items = []
@@ -429,17 +447,17 @@ class MCPInstallWizardScreen(ModalScreen):
             req_items.append(f"Env vars: {len(requirements.environment_vars)}")
         if requirements.command_line_args:
             req_items.append(f"Config args: {len(requirements.command_line_args)}")
-        
+
         if req_items:
             info_text += f"\n[yellow]Requirements:[/yellow] {' | '.join(req_items)}"
-        
+
         server_info.mount(Static(info_text))
-        
+
         # Setup configuration requirements
         config_container = self.query_one("#env-vars-container", Container)
         config_container.remove_children()
         config_container.mount(Static("[bold]Server Configuration:[/bold]"))
-        
+
         # Add server name input
         config_container.mount(Static("\n[bold blue]Server Name:[/bold blue]"))
         name_row = Horizontal(classes="env-var-row")
@@ -449,40 +467,42 @@ class MCPInstallWizardScreen(ModalScreen):
             placeholder=f"Default: {self.selected_server.name}",
             value=self.selected_server.name,
             classes="env-var-input",
-            id="server-name-input"
+            id="server-name-input",
         )
         name_row.mount(name_input)
-        
+
         try:
             # Check system requirements first
             self._setup_system_requirements(config_container)
-            
+
             # Setup environment variables
             self._setup_environment_variables(config_container)
-            
+
             # Setup command line arguments
             self._setup_command_line_args(config_container)
-            
+
             # Show package dependencies info
             self._setup_package_dependencies(config_container)
-                
+
         except Exception as e:
-            config_container.mount(Static(f"[red]Error loading configuration: {e}[/red]"))
+            config_container.mount(
+                Static(f"[red]Error loading configuration: {e}[/red]")
+            )
 
     def _setup_system_requirements(self, parent: Container) -> None:
         """Setup system requirements validation."""
         required_tools = self.selected_server.get_required_tools()
-        
+
         if not required_tools:
             return
-        
+
         parent.mount(Static("\n[bold cyan]System Tools:[/bold cyan]"))
-        
+
         # Import here to avoid circular imports
         from code_puppy.mcp.system_tools import detector
-        
+
         tool_status = detector.detect_tools(required_tools)
-        
+
         for tool_name, tool_info in tool_status.items():
             if tool_info.available:
                 status_text = f"✅ {tool_name}"
@@ -492,7 +512,7 @@ class MCPInstallWizardScreen(ModalScreen):
             else:
                 status_text = f"❌ {tool_name} - {tool_info.error or 'Not found'}"
                 parent.mount(Static(f"[red]{status_text}[/red]"))
-                
+
                 # Show installation suggestions
                 suggestions = detector.get_installation_suggestions(tool_name)
                 if suggestions:
@@ -501,97 +521,106 @@ class MCPInstallWizardScreen(ModalScreen):
     def _setup_environment_variables(self, parent: Container) -> None:
         """Setup environment variables inputs."""
         env_vars = self.selected_server.get_environment_vars()
-        
+
         if not env_vars:
             return
-        
+
         parent.mount(Static("\n[bold yellow]Environment Variables:[/bold yellow]"))
-        
+
         for var in env_vars:
             # Check if already set
             import os
+
             current_value = os.environ.get(var, "")
-            
+
             row_container = Horizontal(classes="env-var-row")
             parent.mount(row_container)
-            
+
             status_indicator = "✅" if current_value else "📝"
-            row_container.mount(Static(f"{status_indicator} {var}:", classes="env-var-label"))
-            
+            row_container.mount(
+                Static(f"{status_indicator} {var}:", classes="env-var-label")
+            )
+
             env_input = Input(
-                placeholder=f"Enter {var} value..." if not current_value else "Already set",
+                placeholder=f"Enter {var} value..."
+                if not current_value
+                else "Already set",
                 value=current_value,
-                classes="env-var-input", 
-                id=f"env-{var}"
+                classes="env-var-input",
+                id=f"env-{var}",
             )
             row_container.mount(env_input)
 
     def _setup_command_line_args(self, parent: Container) -> None:
         """Setup command line arguments inputs."""
         cmd_args = self.selected_server.get_command_line_args()
-        
+
         if not cmd_args:
             return
-        
+
         parent.mount(Static("\n[bold green]Command Line Arguments:[/bold green]"))
-        
+
         for arg_config in cmd_args:
             name = arg_config.get("name", "")
             prompt = arg_config.get("prompt", name)
             default = arg_config.get("default", "")
             required = arg_config.get("required", True)
-            
+
             row_container = Horizontal(classes="env-var-row")
             parent.mount(row_container)
-            
+
             indicator = "⚡" if required else "🔧"
             label_text = f"{indicator} {prompt}:"
             if not required:
                 label_text += " (optional)"
-            
+
             row_container.mount(Static(label_text, classes="env-var-label"))
-            
+
             arg_input = Input(
                 placeholder=f"Default: {default}" if default else f"Enter {name}...",
                 value=default,
                 classes="env-var-input",
-                id=f"arg-{name}"
+                id=f"arg-{name}",
             )
             row_container.mount(arg_input)
 
     def _setup_package_dependencies(self, parent: Container) -> None:
         """Setup package dependencies information."""
         packages = self.selected_server.get_package_dependencies()
-        
+
         if not packages:
             return
-        
+
         parent.mount(Static("\n[bold magenta]Package Dependencies:[/bold magenta]"))
-        
+
         # Import here to avoid circular imports
         from code_puppy.mcp.system_tools import detector
-        
+
         package_status = detector.check_package_dependencies(packages)
-        
+
         for package, available in package_status.items():
             if available:
                 parent.mount(Static(f"✅ {package} (installed)"))
             else:
-                parent.mount(Static(f"[yellow]📦 {package} (will be installed automatically)[/yellow]"))
+                parent.mount(
+                    Static(
+                        f"[yellow]📦 {package} (will be installed automatically)[/yellow]"
+                    )
+                )
 
     def _install_server(self) -> None:
         """Install the selected server with configuration."""
         if not self.selected_server:
             return
-        
+
         try:
             # Collect configuration inputs
             env_vars = {}
             cmd_args = {}
             server_name = self.selected_server.name  # Default fallback
-            
+
             all_inputs = self.query(Input)
-            
+
             for input_widget in all_inputs:
                 if input_widget.id == "server-name-input":
                     custom_name = input_widget.value.strip()
@@ -607,168 +636,168 @@ class MCPInstallWizardScreen(ModalScreen):
                     value = input_widget.value.strip()
                     if value:
                         cmd_args[arg_name] = value
-            
+
             # Set environment variables in the current environment
             for var, value in env_vars.items():
                 os.environ[var] = value
-            
+
             # Get server config with command line argument overrides
             config_dict = self.selected_server.to_server_config(server_name, **cmd_args)
-            
+
             # Update the config with actual environment variable values
-            if 'env' in config_dict:
-                for env_key, env_value in config_dict['env'].items():
+            if "env" in config_dict:
+                for env_key, env_value in config_dict["env"].items():
                     # If it's a placeholder like $GITHUB_TOKEN, replace with actual value
-                    if env_value.startswith('$'):
+                    if env_value.startswith("$"):
                         var_name = env_value[1:]  # Remove the $
                         if var_name in env_vars:
-                            config_dict['env'][env_key] = env_vars[var_name]
-            
+                            config_dict["env"][env_key] = env_vars[var_name]
+
             # Create and register the server
             from code_puppy.mcp import ServerConfig
             from code_puppy.mcp.manager import get_mcp_manager
-            
+
             server_config = ServerConfig(
                 id=server_name,
                 name=server_name,
-                type=config_dict.pop('type'),
+                type=config_dict.pop("type"),
                 enabled=True,
-                config=config_dict
+                config=config_dict,
             )
-            
+
             manager = get_mcp_manager()
             server_id = manager.register_server(server_config)
-            
+
             if server_id:
                 # Save to mcp_servers.json
                 from code_puppy.config import MCP_SERVERS_FILE
-                
+
                 if os.path.exists(MCP_SERVERS_FILE):
-                    with open(MCP_SERVERS_FILE, 'r') as f:
+                    with open(MCP_SERVERS_FILE, "r") as f:
                         data = json.load(f)
                         servers = data.get("mcp_servers", {})
                 else:
                     servers = {}
                     data = {"mcp_servers": servers}
-                
+
                 servers[server_name] = config_dict
-                servers[server_name]['type'] = server_config.type
-                
+                servers[server_name]["type"] = server_config.type
+
                 os.makedirs(os.path.dirname(MCP_SERVERS_FILE), exist_ok=True)
-                with open(MCP_SERVERS_FILE, 'w') as f:
+                with open(MCP_SERVERS_FILE, "w") as f:
                     json.dump(data, f, indent=2)
-                
+
                 # Reload MCP servers
                 from code_puppy.agent import reload_mcp_servers
+
                 reload_mcp_servers()
-                
-                self.dismiss({
-                    "success": True,
-                    "message": f"Successfully installed '{server_name}' from {self.selected_server.display_name}",
-                    "server_name": server_name
-                })
+
+                self.dismiss(
+                    {
+                        "success": True,
+                        "message": f"Successfully installed '{server_name}' from {self.selected_server.display_name}",
+                        "server_name": server_name,
+                    }
+                )
             else:
-                self.dismiss({
-                    "success": False,
-                    "message": "Failed to register server"
-                })
-                
+                self.dismiss({"success": False, "message": "Failed to register server"})
+
         except Exception as e:
-            self.dismiss({
-                "success": False,
-                "message": f"Installation failed: {str(e)}"
-            })
+            self.dismiss(
+                {"success": False, "message": f"Installation failed: {str(e)}"}
+            )
 
     def _install_custom_json(self) -> None:
         """Install server from custom JSON configuration."""
         try:
             name_input = self.query_one("#custom-name-input", Input)
             json_input = self.query_one("#custom-json-input", TextArea)
-            
+
             server_name = name_input.value.strip()
             json_text = json_input.text.strip()
-            
+
             if not server_name:
                 # Show error - need a name
                 return
-            
+
             if not json_text:
                 # Show error - need JSON config
                 return
-            
+
             # Parse JSON
             try:
                 config_dict = json.loads(json_text)
-            except json.JSONDecodeError as e:
+            except json.JSONDecodeError:
                 # Show error - invalid JSON
                 return
-            
+
             # Validate required fields
-            if 'type' not in config_dict:
+            if "type" not in config_dict:
                 # Show error - missing type
                 return
-            
+
             # Extract type and create server config
-            server_type = config_dict.pop('type')
-            
+            server_type = config_dict.pop("type")
+
             # Create and register the server
             from code_puppy.mcp import ServerConfig
             from code_puppy.mcp.manager import get_mcp_manager
-            
+
             server_config = ServerConfig(
                 id=server_name,
                 name=server_name,
                 type=server_type,
                 enabled=True,
-                config=config_dict
+                config=config_dict,
             )
-            
+
             manager = get_mcp_manager()
             server_id = manager.register_server(server_config)
-            
+
             if server_id:
                 # Save to mcp_servers.json
                 from code_puppy.config import MCP_SERVERS_FILE
-                
+
                 if os.path.exists(MCP_SERVERS_FILE):
-                    with open(MCP_SERVERS_FILE, 'r') as f:
+                    with open(MCP_SERVERS_FILE, "r") as f:
                         data = json.load(f)
                         servers = data.get("mcp_servers", {})
                 else:
                     servers = {}
                     data = {"mcp_servers": servers}
-                
+
                 # Add the full config including type
                 full_config = config_dict.copy()
-                full_config['type'] = server_type
+                full_config["type"] = server_type
                 servers[server_name] = full_config
-                
+
                 os.makedirs(os.path.dirname(MCP_SERVERS_FILE), exist_ok=True)
-                with open(MCP_SERVERS_FILE, 'w') as f:
+                with open(MCP_SERVERS_FILE, "w") as f:
                     json.dump(data, f, indent=2)
-                
+
                 # Reload MCP servers
                 from code_puppy.agent import reload_mcp_servers
+
                 reload_mcp_servers()
-                
-                self.dismiss({
-                    "success": True,
-                    "message": f"Successfully installed custom server '{server_name}'",
-                    "server_name": server_name
-                })
+
+                self.dismiss(
+                    {
+                        "success": True,
+                        "message": f"Successfully installed custom server '{server_name}'",
+                        "server_name": server_name,
+                    }
+                )
             else:
-                self.dismiss({
-                    "success": False,
-                    "message": "Failed to register custom server"
-                })
-                
+                self.dismiss(
+                    {"success": False, "message": "Failed to register custom server"}
+                )
+
         except Exception as e:
-            self.dismiss({
-                "success": False,
-                "message": f"Installation failed: {str(e)}"
-            })
-    
+            self.dismiss(
+                {"success": False, "message": f"Installation failed: {str(e)}"}
+            )
+
     def on_key(self, event) -> None:
         """Handle key events."""
         if event.key == "escape":
-            self.on_cancel_clicked()
+            self.on_cancel_clicked()