Fix code formatting and linting issues

- Format all code with ruff formatter
- Fix test imports to use updated MCPMProxyFactory constructor
- Remove deprecated RouterConfig usage in tests
- Fix import ordering and line length issues
- Add missing newlines and remove trailing whitespace

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: niechen

src/mcpm/cli.py

@@ -9,7 +9,6 @@
 
 from mcpm import __version__
 from mcpm.clients.client_config import ClientConfigManager
-from mcpm.utils.logging_config import setup_logging
 from mcpm.commands import (
     add,
     client,
@@ -27,6 +26,7 @@
     usage,
 )
 from mcpm.commands.share import share
+from mcpm.utils.logging_config import setup_logging
 
 console = Console()
 client_config_manager = ClientConfigManager()

src/mcpm/clients/client_registry.py

@@ -49,7 +49,9 @@ class ClientRegistry:
     }
 
     @classmethod
-    def get_client_manager(cls, client_name: str, config_path_override: Optional[str] = None) -> Optional[BaseClientManager]:
+    def get_client_manager(
+        cls, client_name: str, config_path_override: Optional[str] = None
+    ) -> Optional[BaseClientManager]:
         """
         Get the client manager for a given client name
 

src/mcpm/commands/client.py

@@ -217,7 +217,9 @@ def list_clients(verbose):
 @client.command(name="edit", context_settings=dict(help_option_names=["-h", "--help"]))
 @click.argument("client_name")
 @click.option("-e", "--external", is_flag=True, help="Open config file in external editor instead of interactive mode")
-@click.option("-f", "--file", "config_path_override", type=click.Path(), help="Specify a custom path to the client's config file.")
+@click.option(
+    "-f", "--file", "config_path_override", type=click.Path(), help="Specify a custom path to the client's config file."
+)
 def edit_client(client_name, external, config_path_override):
     """Enable/disable MCPM-managed servers in the specified client configuration.
 

src/mcpm/commands/profile/run.py

@@ -9,7 +9,11 @@
 from mcpm.fastmcp_integration.proxy import create_mcpm_proxy
 from mcpm.profile.profile_config import ProfileConfigManager
 from mcpm.utils.config import DEFAULT_PORT
-from mcpm.utils.logging_config import setup_dependency_logging, ensure_dependency_logging_suppressed, get_uvicorn_log_level
+from mcpm.utils.logging_config import (
+    ensure_dependency_logging_suppressed,
+    get_uvicorn_log_level,
+    setup_dependency_logging,
+)
 
 profile_config_manager = ProfileConfigManager()
 logger = logging.getLogger(__name__)
@@ -50,10 +54,10 @@ async def run_profile_fastmcp(profile_servers, profile_name, http_mode=False, po
         )
 
         logger.debug(f"FastMCP proxy initialized with: {[s.name for s in profile_servers]}")
-        
+
         # Set up dependency logging for FastMCP/MCP libraries
         setup_dependency_logging()
-        
+
         # Re-suppress library logging after FastMCP initialization
         ensure_dependency_logging_suppressed()
 
@@ -72,18 +76,20 @@ async def run_profile_fastmcp(profile_servers, profile_name, http_mode=False, po
             http_url = f"http://127.0.0.1:{actual_port}/mcp/"
             console.print(f"[bold green]Profile '{profile_name}' is now running at:[/]")
             console.print(f"[cyan]{http_url}[/]")
-            
+
             # Show servers in the profile
-            console.print(f"[bold green]Servers:[/]")
+            console.print("[bold green]Servers:[/]")
             for server_config in profile_servers:
                 console.print(f"  • [cyan]{server_config.name}[/]")
-            
+
             console.print("[dim]Press Ctrl+C to stop the profile[/]")
-            
+
             logger.debug(f"Starting FastMCP proxy for profile '{profile_name}' on port {actual_port}")
 
             # Run the aggregated proxy over HTTP with uvicorn logging control
-            await proxy.run_http_async(host="127.0.0.1", port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()})
+            await proxy.run_http_async(
+                host="127.0.0.1", port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()}
+            )
         else:
             # Run the aggregated proxy over stdio (default)
             logger.info(f"Starting profile '{profile_name}' over stdio")
@@ -116,7 +122,7 @@ def run(profile_name, http, port):
         mcpm profile run web-dev                    # Run over stdio (default)
         mcpm profile run --http web-dev             # Run over HTTP on port 6276
         mcpm profile run --http --port 9000 ai      # Run over HTTP on port 9000
-    
+
     Debug logging: Set MCPM_DEBUG=1 for verbose output
     """
     # Validate profile name

src/mcpm/commands/profile/share.py

@@ -10,8 +10,12 @@
 from mcpm.core.tunnel import Tunnel
 from mcpm.fastmcp_integration.proxy import create_mcpm_proxy
 from mcpm.profile.profile_config import ProfileConfigManager
-from mcpm.utils.config import DEFAULT_SHARE_ADDRESS, DEFAULT_PORT
-from mcpm.utils.logging_config import setup_dependency_logging, ensure_dependency_logging_suppressed, get_uvicorn_log_level
+from mcpm.utils.config import DEFAULT_PORT, DEFAULT_SHARE_ADDRESS
+from mcpm.utils.logging_config import (
+    ensure_dependency_logging_suppressed,
+    get_uvicorn_log_level,
+    setup_dependency_logging,
+)
 
 console = Console()
 profile_config_manager = ProfileConfigManager()
@@ -42,6 +46,7 @@ async def share_profile_fastmcp(profile_servers, profile_name, port, address, ht
     api_key = None
     if not no_auth:
         from mcpm.utils.config import ConfigManager
+
         config_manager = ConfigManager()
         auth_config = config_manager.get_auth_config()
         api_key = auth_config.get("api_key")
@@ -63,10 +68,10 @@ async def share_profile_fastmcp(profile_servers, profile_name, port, address, ht
         )
 
         logger.debug(f"FastMCP proxy created with {server_count} server(s)")
-        
+
         # Set up dependency logging for FastMCP/MCP libraries
         setup_dependency_logging()
-        
+
         # Re-suppress library logging after FastMCP initialization
         ensure_dependency_logging_suppressed()
 
@@ -79,7 +84,11 @@ async def share_profile_fastmcp(profile_servers, profile_name, port, address, ht
         logger.debug(f"Starting HTTP server on port {actual_port}")
 
         # Start the FastMCP proxy as a streamable HTTP server in a background task
-        server_task = asyncio.create_task(proxy.run_http_async(host="127.0.0.1", port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()}))
+        server_task = asyncio.create_task(
+            proxy.run_http_async(
+                host="127.0.0.1", port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()}
+            )
+        )
 
         # Wait a moment for server to start
         await asyncio.sleep(2)
@@ -113,17 +122,17 @@ async def share_profile_fastmcp(profile_servers, profile_name, port, address, ht
             http_url = f"{public_url}/mcp/"
             console.print(f"[bold green]Profile '{profile_name}' is now shared at:[/]")
             console.print(f"[cyan]{http_url}[/]")
-            
+
             if not no_auth and api_key:
                 console.print(f"[bold green]API Key:[/] [cyan]{api_key}[/]")
             else:
                 console.print("[bold red]Warning:[/] Anyone with the URL can access your servers.")
-            
-            # Show available servers 
-            console.print(f"[bold green]Shared servers:[/]")
+
+            # Show available servers
+            console.print("[bold green]Shared servers:[/]")
             for server_config in profile_servers:
                 console.print(f"  • [cyan]{server_config.name}[/]")
-            
+
             console.print("[dim]Press Ctrl+C to stop sharing[/]")
 
             # Keep running until interrupted

src/mcpm/commands/run.py

@@ -12,7 +12,11 @@
 from mcpm.fastmcp_integration.proxy import create_mcpm_proxy
 from mcpm.global_config import GlobalConfigManager
 from mcpm.utils.config import DEFAULT_PORT
-from mcpm.utils.logging_config import setup_dependency_logging, ensure_dependency_logging_suppressed, get_uvicorn_log_level
+from mcpm.utils.logging_config import (
+    ensure_dependency_logging_suppressed,
+    get_uvicorn_log_level,
+    setup_dependency_logging,
+)
 
 global_config_manager = GlobalConfigManager()
 logger = logging.getLogger(__name__)
@@ -98,10 +102,10 @@ async def run_server_with_fastmcp(server_config, server_name, http_mode=False, p
             name=f"mcpm-run-{server_name}",
             stdio_mode=not http_mode,  # stdio_mode=False for HTTP
         )
-        
+
         # Set up dependency logging for FastMCP/MCP libraries
         setup_dependency_logging()
-        
+
         # Re-suppress library logging after FastMCP initialization
         ensure_dependency_logging_suppressed()
 
@@ -116,11 +120,13 @@ async def run_server_with_fastmcp(server_config, server_name, http_mode=False, p
             console.print(f"[bold green]Server '{server_name}' is now running at:[/]")
             console.print(f"[cyan]{http_url}[/]")
             console.print("[dim]Press Ctrl+C to stop the server[/]")
-            
+
             logger.debug(f"Starting FastMCP proxy for server '{server_name}' on port {actual_port}")
 
             # Run FastMCP proxy in HTTP mode with uvicorn logging control
-            await proxy.run_http_async(host="127.0.0.1", port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()})
+            await proxy.run_http_async(
+                host="127.0.0.1", port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()}
+            )
         else:
             # Run FastMCP proxy in stdio mode (default)
             logger.info(f"Starting server '{server_name}' over stdio")

src/mcpm/commands/share.py

@@ -14,8 +14,12 @@
 from mcpm.core.tunnel import Tunnel
 from mcpm.fastmcp_integration.proxy import create_mcpm_proxy
 from mcpm.global_config import GlobalConfigManager
-from mcpm.utils.config import DEFAULT_SHARE_ADDRESS, DEFAULT_PORT
-from mcpm.utils.logging_config import setup_dependency_logging, ensure_dependency_logging_suppressed, get_uvicorn_log_level
+from mcpm.utils.config import DEFAULT_PORT, DEFAULT_SHARE_ADDRESS
+from mcpm.utils.logging_config import (
+    ensure_dependency_logging_suppressed,
+    get_uvicorn_log_level,
+    setup_dependency_logging,
+)
 
 console = Console()
 global_config_manager = GlobalConfigManager()
@@ -49,7 +53,9 @@ async def find_available_port(preferred_port, max_attempts=10):
     return preferred_port
 
 
-async def start_fastmcp_proxy(server_config, server_name, port: Optional[int] = None, auth_enabled: bool = True, api_key: Optional[str] = None) -> int:
+async def start_fastmcp_proxy(
+    server_config, server_name, port: Optional[int] = None, auth_enabled: bool = True, api_key: Optional[str] = None
+) -> int:
     """
     Start FastMCP proxy in HTTP mode for sharing a single server.
 
@@ -89,10 +95,10 @@ async def start_fastmcp_proxy(server_config, server_name, port: Optional[int] =
         )
 
         logger.debug(f"FastMCP proxy ready on port {actual_port}")
-        
+
         # Set up dependency logging for FastMCP/MCP libraries
         setup_dependency_logging()
-        
+
         # Re-suppress library logging after FastMCP initialization
         ensure_dependency_logging_suppressed()
 
@@ -182,6 +188,7 @@ async def _share_async(server_config, server_name, port, remote_host, remote_por
 
     if not no_auth:
         from mcpm.utils.config import ConfigManager
+
         config_manager = ConfigManager()
         auth_config = config_manager.get_auth_config()
         api_key = auth_config.get("api_key")
@@ -193,10 +200,14 @@ async def _share_async(server_config, server_name, port, remote_host, remote_por
     try:
         # Start FastMCP proxy
         logger.debug(f"Starting FastMCP proxy to share server '{server_name}'")
-        actual_port, proxy = await start_fastmcp_proxy(server_config, server_name, port, auth_enabled=not no_auth, api_key=api_key)
+        actual_port, proxy = await start_fastmcp_proxy(
+            server_config, server_name, port, auth_enabled=not no_auth, api_key=api_key
+        )
 
         # Start the FastMCP proxy as an HTTP server in a background task
-        server_task = asyncio.create_task(proxy.run_http_async(port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()}))
+        server_task = asyncio.create_task(
+            proxy.run_http_async(port=actual_port, uvicorn_config={"log_level": get_uvicorn_log_level()})
+        )
 
         # Wait a moment for server to start
         await asyncio.sleep(2)
@@ -223,12 +234,12 @@ async def _share_async(server_config, server_name, port, remote_host, remote_por
         http_url = f"{share_url}/mcp/"
         console.print(f"[bold green]Server '{server_name}' is now shared at:[/]")
         console.print(f"[cyan]{http_url}[/]")
-        
+
         if not no_auth and api_key:
             console.print(f"[bold green]API Key:[/] [cyan]{api_key}[/]")
         else:
             console.print("[bold red]Warning:[/] Anyone with the URL can access your server.")
-        
+
         console.print("[dim]Press Ctrl+C to stop sharing[/]")
 
         # Keep running until interrupted

src/mcpm/core/tunnel.py

@@ -188,4 +188,4 @@ def _raise_tunnel_error():
 
         if self.http:
             url = url.replace("https://", "http://")
-        return url
+        return url

src/mcpm/fastmcp_integration/middleware.py

@@ -115,31 +115,32 @@ async def on_request(self, context, call_next):
             # Multiple approaches to get headers
             headers = None
             auth_header = None
-            
+
             # Method 1: Try FastMCP's built-in helper
             try:
                 from fastmcp.server.dependencies import get_http_headers
+
                 headers = get_http_headers()
                 auth_header = headers.get("authorization") or headers.get("Authorization")
             except (RuntimeError, ImportError):
                 pass
-            
+
             # Method 2: Try accessing from context
-            if not auth_header and hasattr(context, 'request'):
+            if not auth_header and hasattr(context, "request"):
                 request = context.request
-                if hasattr(request, 'headers'):
+                if hasattr(request, "headers"):
                     auth_header = request.headers.get("Authorization") or request.headers.get("authorization")
-            
+
             # Method 3: Try direct context headers
-            if not auth_header and hasattr(context, 'headers'):
+            if not auth_header and hasattr(context, "headers"):
                 headers = context.headers
                 auth_header = headers.get("Authorization") or headers.get("authorization")
-                
+
             # Method 4: Check for auth in context metadata
-            if not auth_header and hasattr(context, 'metadata'):
+            if not auth_header and hasattr(context, "metadata"):
                 metadata = context.metadata
                 auth_header = metadata.get("authorization") or metadata.get("Authorization")
-            
+
             if not auth_header:
                 # For debugging: print available context attributes
                 # print(f"DEBUG: Context type: {type(context)}, attrs: {dir(context)}")