forgotten files

Author: kerobbi

pkg/http/headers/parse.go

@@ -0,0 +1,21 @@
+package headers
+
+import "strings"
+
+// ParseCommaSeparated splits a header value by comma, trims whitespace,
+// and filters out empty values
+func ParseCommaSeparated(value string) []string {
+	if value == "" {
+		return []string{}
+	}
+
+	parts := strings.Split(value, ",")
+	result := make([]string, 0, len(parts))
+	for _, p := range parts {
+		trimmed := strings.TrimSpace(p)
+		if trimmed != "" {
+			result = append(result, trimmed)
+		}
+	}
+	return result
+}

pkg/http/middleware/request_config.go

@@ -0,0 +1,41 @@
+package middleware
+
+import (
+	"net/http"
+	"slices"
+	"strings"
+
+	ghcontext "github.com/github/github-mcp-server/pkg/context"
+	"github.com/github/github-mcp-server/pkg/http/headers"
+)
+
+// WithRequestConfig is a middleware that extracts MCP-related headers and sets them in the request context
+func WithRequestConfig(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		ctx := r.Context()
+
+		// Only set from headers if not already set in ctx
+		if !ghcontext.IsReadonly(ctx) {
+			if relaxedParseBool(r.Header.Get(headers.MCPReadOnlyHeader)) {
+				ctx = ghcontext.WithReadonly(ctx, true)
+			}
+		}
+
+		if len(ghcontext.GetToolsets(ctx)) == 0 {
+			if toolsets := headers.ParseCommaSeparated(r.Header.Get(headers.MCPToolsetsHeader)); len(toolsets) > 0 {
+				ctx = ghcontext.WithToolsets(ctx, toolsets)
+			}
+		}
+
+		next.ServeHTTP(w, r.WithContext(ctx))
+	})
+}
+
+// relaxedParseBool parses a string into a boolean value, treating various
+// common false values or empty strings as false, and everything else as true.
+// It is case-insensitive and trims whitespace.
+func relaxedParseBool(s string) bool {
+	s = strings.TrimSpace(strings.ToLower(s))
+	falseValues := []string{"", "false", "0", "no", "off", "n", "f"}
+	return !slices.Contains(falseValues, s)
+}