parse X-MCP-Tools into ctx

Author: kerobbi

pkg/context/request.go

@@ -33,3 +33,19 @@ func GetToolsets(ctx context.Context) []string {
 	}
 	return nil
 }
+
+// toolsCtxKey is a context key for tools
+type toolsCtxKey struct{}
+
+// WithTools adds the tools to the context
+func WithTools(ctx context.Context, tools []string) context.Context {
+	return context.WithValue(ctx, toolsCtxKey{}, tools)
+}
+
+// GetTools retrieves the tools from the context
+func GetTools(ctx context.Context) []string {
+	if tools, ok := ctx.Value(toolsCtxKey{}).([]string); ok {
+		return tools
+	}
+	return nil
+}

pkg/http/handler.go

@@ -173,7 +173,7 @@ func InventoryFiltersForRequest(r *http.Request, builder *inventory.Builder) *in
 		builder = builder.WithToolsets(toolsets)
 	}
 
-	if tools := headers.ParseCommaSeparated(r.Header.Get(headers.MCPToolsHeader)); len(tools) > 0 {
+	if tools := ghcontext.GetTools(ctx); len(tools) > 0 {
 		if len(ghcontext.GetToolsets(ctx)) == 0 {
 			builder = builder.WithToolsets([]string{})
 		}

pkg/http/handler_test.go

@@ -73,10 +73,9 @@ func TestInventoryFiltersForRequest(t *testing.T) {
 		{
 			name: "tools are additive with toolsets",
 			contextSetup: func(ctx context.Context) context.Context {
-				return ghcontext.WithToolsets(ctx, []string{"repos"})
-			},
-			headers: map[string]string{
-				headers.MCPToolsHeader: "list_issues",
+				ctx = ghcontext.WithToolsets(ctx, []string{"repos"})
+				ctx = ghcontext.WithTools(ctx, []string{"list_issues"})
+				return ctx
 			},
 			expectedTools: []string{"get_file_contents", "create_repository", "list_issues"},
 		},

pkg/http/middleware/request_config.go

@@ -22,6 +22,10 @@ func WithRequestConfig(next http.Handler) http.Handler {
 			ctx = ghcontext.WithToolsets(ctx, toolsets)
 		}
 
+		if tools := headers.ParseCommaSeparated(r.Header.Get(headers.MCPToolsHeader)); len(tools) > 0 {
+			ctx = ghcontext.WithTools(ctx, tools)
+		}
+
 		next.ServeHTTP(w, r.WithContext(ctx))
 	})
 }