Clean logic (#3)

Author: vectorstain

src/mcp/server/fastmcp/server.py

@@ -790,6 +790,12 @@ async def handle_streamable_http(
         middleware: list[Middleware] = []
         required_scopes = []
 
+        # Always mount both /mcp and /mcp/ for full compatibility, regardless of default
+        # Verify that _main_path has root format -> /mcp
+        _main_path = self.settings.streamable_http_path.removesuffix("/")
+        # Format _alt_path so it ends with '/' -> /mcp/
+        _alt_path  = _main_path + "/"
+
         # Add auth endpoints if auth provider is configured
         if self._auth_server_provider:
             assert self.settings.auth
@@ -815,46 +821,29 @@ async def handle_streamable_http(
                     revocation_options=self.settings.auth.revocation_options,
                 )
             )
-            routes.append(
+            routes.extend([
                 Mount(
-                    self.settings.streamable_http_path,
+                    _main_path,
                     app=RequireAuthMiddleware(handle_streamable_http, required_scopes),
-                )
+                ),
+                Mount(
+                    _alt_path,
+                    app=RequireAuthMiddleware(handle_streamable_http, required_scopes),
+                )]
             )
         else:
             # Auth is disabled, no wrapper needed
-            routes.append(
+            routes.extend([
                 Mount(
-                    self.settings.streamable_http_path,
+                    _main_path,
                     app=handle_streamable_http,
-                )
+                ),
+                Mount(
+                    _alt_path,
+                    app=handle_streamable_http,
+                )]
             )
 
-        # Always mount both /mcp and /mcp/ for full compatibility, regardless of default
-        _main_path = self.settings.streamable_http_path
-        if _main_path.endswith("/"):
-            _alt_path = _main_path.rstrip("/")
-        else:
-            _alt_path = _main_path + "/"
-        if _alt_path != _main_path:
-            if self._auth_server_provider:
-                routes.append(
-                    Mount(
-                        _alt_path,
-                        app=RequireAuthMiddleware(
-                            handle_streamable_http,
-                            required_scopes,
-                        ),
-                    )
-                )
-            else:
-                routes.append(
-                    Mount(
-                        _alt_path,
-                        app=handle_streamable_http,
-                    )
-                )
-
         routes.extend(self._custom_starlette_routes)
 
         return Starlette(