feat:sse support for 'mcpm run'

Author: GabrielDrapor

src/mcpm/commands/run.py

@@ -32,7 +32,9 @@ def find_installed_server(server_name):
     return None, None
 
 
-async def run_server_with_fastmcp(server_config, server_name, http_mode=False, port=None, host="127.0.0.1"):
+async def run_server_with_fastmcp(
+    server_config, server_name, http_mode=False, sse_mode=False, port=None, host="127.0.0.1"
+):
     """Run server using FastMCP proxy (stdio or HTTP)."""
     try:
         # Use default port if none specified
@@ -41,11 +43,17 @@ async def run_server_with_fastmcp(server_config, server_name, http_mode=False, p
         # Note: Usage tracking is handled by proxy middleware
 
         # Create FastMCP proxy for single server
-        action = "run_http" if http_mode else "run"
+        if sse_mode:
+            action = "run_sse"
+        elif http_mode:
+            action = "run_http"
+        else:
+            action = "run"
+
         proxy = await create_mcpm_proxy(
             servers=[server_config],
             name=f"mcpm-run-{server_name}",
-            stdio_mode=not http_mode,  # stdio_mode=False for HTTP
+            stdio_mode=not (http_mode or sse_mode),  # stdio_mode=False for HTTP/SSE
             action=action,
         )
 
@@ -55,25 +63,35 @@ async def run_server_with_fastmcp(server_config, server_name, http_mode=False, p
         # Re-suppress library logging after FastMCP initialization
         ensure_dependency_logging_suppressed()
 
-        if http_mode:
+        if http_mode or sse_mode:
             # Try to find an available port if the requested one is taken
             actual_port = await find_available_port(port)
             if actual_port != port:
                 logger.debug(f"Port {port} is busy, using port {actual_port} instead")
 
             # Display server information in a nice panel
-            http_url = f"http://{host}:{actual_port}/mcp/"
-            panel_content = f"[bold]Server:[/] {server_name}\n[bold]URL:[/] [cyan]{http_url}[/cyan]\n\n[dim]Press Ctrl+C to stop the server[/]"
-            panel = Panel(
-                panel_content, title="🌐 Local Server Running", title_align="left", border_style="green", padding=(1, 2)
-            )
+            if sse_mode:
+                server_url = f"http://{host}:{actual_port}/sse/"
+                title = "📡 SSE Server Running"
+            else:
+                server_url = f"http://{host}:{actual_port}/mcp/"
+                title = "🌐 Local Server Running"
+
+            panel_content = f"[bold]Server:[/] {server_name}\n[bold]URL:[/] [cyan]{server_url}[/cyan]\n\n[dim]Press Ctrl+C to stop the server[/]"
+            panel = Panel(panel_content, title=title, title_align="left", border_style="green", padding=(1, 2))
             console.print(panel)
 
-            logger.debug(f"Starting FastMCP proxy for server '{server_name}' on {host}:{actual_port}")
+            mode = "SSE" if sse_mode else "HTTP"
+            logger.debug(f"Starting FastMCP proxy for server '{server_name}' in {mode} mode on {host}:{actual_port}")
 
-            # Run FastMCP proxy in HTTP mode with uvicorn logging control
+            # Run FastMCP proxy in HTTP/SSE mode with uvicorn logging control
+            transport = "sse" if sse_mode else "http"
             await proxy.run_http_async(
-                host=host, port=actual_port, show_banner=False, uvicorn_config={"log_level": get_uvicorn_log_level()}
+                host=host,
+                port=actual_port,
+                show_banner=False,
+                transport=transport,
+                uvicorn_config={"log_level": get_uvicorn_log_level()},
             )
         else:
             # Run FastMCP proxy in stdio mode (default)
@@ -84,7 +102,7 @@ async def run_server_with_fastmcp(server_config, server_name, http_mode=False, p
 
     except KeyboardInterrupt:
         logger.info("Server execution interrupted")
-        if http_mode:
+        if http_mode or sse_mode:
             logger.warning("\nServer execution interrupted")
         return 130
     except Exception as e:
@@ -114,19 +132,23 @@ async def find_available_port(preferred_port, max_attempts=10):
 @click.command()
 @click.argument("server_name")
 @click.option("--http", is_flag=True, help="Run server over HTTP instead of stdio")
-@click.option("--port", type=int, default=DEFAULT_PORT, help=f"Port for HTTP mode (default: {DEFAULT_PORT})")
-@click.option("--host", type=str, default="127.0.0.1", help="Host address for HTTP mode (default: 127.0.0.1)")
+@click.option("--sse", is_flag=True, help="Run server over SSE instead of stdio")
+@click.option("--port", type=int, default=DEFAULT_PORT, help=f"Port for HTTP / SSE mode (default: {DEFAULT_PORT})")
+@click.option("--host", type=str, default="127.0.0.1", help="Host address for HTTP / SSE mode (default: 127.0.0.1)")
 @click.help_option("-h", "--help")
-def run(server_name, http, port, host):
-    """Execute a server from global configuration over stdio or HTTP.
+def run(server_name, http, sse, port, host):
+    """Execute a server from global configuration over stdio, HTTP, or SSE.
 
     Runs an installed MCP server from the global configuration. By default
-    runs over stdio for client communication, but can run over HTTP with --http.
+    runs over stdio for client communication, but can run over HTTP with --http
+    or over SSE with --sse.
 
     Examples:
         mcpm run mcp-server-browse                    # Run over stdio (default)
         mcpm run --http mcp-server-browse             # Run over HTTP on 127.0.0.1:6276
+        mcpm run --sse mcp-server-browse              # Run over SSE on 127.0.0.1:6276
         mcpm run --http --port 9000 filesystem       # Run over HTTP on 127.0.0.1:9000
+        mcpm run --sse --port 9000 filesystem        # Run over SSE on 127.0.0.1:9000
         mcpm run --http --host 0.0.0.0 filesystem    # Run over HTTP on 0.0.0.0:6276
 
     Note: stdio mode is typically used in MCP client configurations:
@@ -164,20 +186,31 @@ def run(server_name, http, port, host):
         if server_config.headers:
             logger.debug(f"Headers: {list(server_config.headers.keys())}")
 
-    logger.debug(f"Mode: {'HTTP' if http else 'stdio'}")
-    if http:
+    # Validate mutually exclusive options
+    if http and sse:
+        logger.error("Error: Cannot use both --http and --sse flags together")
+        sys.exit(1)
+
+    mode = "SSE" if sse else "HTTP" if http else "stdio"
+    logger.debug(f"Mode: {mode}")
+    if http or sse:
         logger.debug(f"Port: {port}")
 
     # Choose execution method
     if http:
         # Use FastMCP proxy for HTTP mode
         exit_code = asyncio.run(
-            run_server_with_fastmcp(server_config, server_name, http_mode=True, port=port, host=host)
+            run_server_with_fastmcp(server_config, server_name, http_mode=True, sse_mode=False, port=port, host=host)
+        )
+    elif sse:
+        # Use FastMCP proxy for SSE mode
+        exit_code = asyncio.run(
+            run_server_with_fastmcp(server_config, server_name, http_mode=False, sse_mode=True, port=port, host=host)
         )
     else:
         # Use FastMCP proxy for stdio mode (enables middleware and usage tracking)
         exit_code = asyncio.run(
-            run_server_with_fastmcp(server_config, server_name, http_mode=False, port=port, host=host)
+            run_server_with_fastmcp(server_config, server_name, http_mode=False, sse_mode=False, port=port, host=host)
         )
 
     sys.exit(exit_code)