package main

import (
	"context"
	"fmt"
	"log"

	"github.com/modelcontextprotocol/go-sdk/mcp"
)

func createAuthMiddleware(token string) mcp.Middleware {
	log.Println("Use auth token")
	return func(next mcp.MethodHandler) mcp.MethodHandler {
		return func(ctx context.Context, method string, req mcp.Request) (mcp.Result, error) {
			authHeader := req.GetExtra().Header.Get("Authorization")
			if authHeader != "Bearer "+token {
				return nil, fmt.Errorf("unauthorized")
			}
			return next(ctx, method, req)
		}
	}
}