gin-vue-admin/server/mcp/context.go

package mcpTool

import (
	"context"
	"net/http"
	"strings"

	"github.com/flipped-aurora/gin-vue-admin/server/global"
)

type mcpContextKey string

const authTokenContextKey mcpContextKey = "mcp-auth-token"

func WithHTTPRequestContext(ctx context.Context, r *http.Request) context.Context {
	token := extractIncomingAuthToken(r.Header)
	return context.WithValue(ctx, authTokenContextKey, token)
}

func configuredAuthHeader() string {
	if header := strings.TrimSpace(global.GVA_CONFIG.MCP.AuthHeader); header != "" {
		return header
	}
	return "x-token"
}

func ConfiguredAuthHeader() string {
	return configuredAuthHeader()
}

func authTokenFromContext(ctx context.Context) string {
	token, _ := ctx.Value(authTokenContextKey).(string)
	return strings.TrimSpace(token)
}

func extractIncomingAuthToken(headers http.Header) string {
	candidates := []string{
		configuredAuthHeader(),
		"x-token",
		"token",
		"authorization",
	}

	seen := make(map[string]struct{}, len(candidates))
	for _, name := range candidates {
		key := strings.ToLower(strings.TrimSpace(name))
		if key == "" {
			continue
		}
		if _, ok := seen[key]; ok {
			continue
		}
		seen[key] = struct{}{}

		value := strings.TrimSpace(headers.Get(name))
		if value == "" {
			continue
		}
		if key == "authorization" {
			return strings.TrimSpace(strings.TrimPrefix(value, "Bearer "))
		}
		return value
	}

	return ""
}