move middleware in RegisterRoutes

Author: kerobbi

pkg/http/handler.go

@@ -76,9 +76,11 @@ func NewHTTPMcpHandler(cfg *HTTPServerConfig,
 	}
 }
 
+// RegisterRoutes registers the routes for the MCP server
 func (h *HTTPMcpHandler) RegisterRoutes(r chi.Router) {
-	r.Mount("/", h)
+	r.Use(middleware.WithRequestConfig)
 
+	r.Mount("/", h)
 	// Mount readonly and toolset routes
 	r.With(withToolset).Mount("/x/{toolset}", h)
 	r.With(withReadonly, withToolset).Mount("/x/{toolset}/readonly", h)

pkg/http/server.go

@@ -12,7 +12,6 @@ import (
 	"time"
 
 	"github.com/github/github-mcp-server/pkg/github"
-	"github.com/github/github-mcp-server/pkg/http/middleware"
 	"github.com/github/github-mcp-server/pkg/lockdown"
 	"github.com/github/github-mcp-server/pkg/translations"
 	"github.com/github/github-mcp-server/pkg/utils"
@@ -99,7 +98,6 @@ func RunHTTPServer(cfg HTTPServerConfig) error {
 	r := chi.NewRouter()
 
 	handler := NewHTTPMcpHandler(&cfg, deps, t, logger)
-	r.Use(middleware.WithRequestConfig)
 	handler.RegisterRoutes(r)
 
 	addr := fmt.Sprintf(":%d", cfg.Port)