087d8085b7<!-- Provide a brief summary of your changes --> ## Motivation and Context <!-- Why is this change needed? What problem does it solve? --> Follow up from #658 addressing the feedback from @tadasant ## How Has This Been Tested? <!-- Have you tested this in a real application? Which scenarios were tested? --> ## Breaking Changes <!-- Will users need to update their code or configurations? --> ## Types of changes <!-- What types of changes does your code introduce? Put an `x` in all the boxes that apply: --> - [ ] Bug fix (non-breaking change which fixes an issue) - [ ] New feature (non-breaking change which adds functionality) - [ ] Breaking change (fix or feature that would cause existing functionality to change) - [ ] Documentation update ## Checklist <!-- Go over all the following points, and put an `x` in all the boxes that apply. --> - [ ] I have read the [MCP Documentation](https://modelcontextprotocol.io) - [ ] My code follows the repository's style guidelines - [ ] New and existing tests pass locally - [ ] I have added appropriate error handling - [ ] I have added or updated documentation as needed ## Additional context <!-- Add any other context, implementation notes, or design decisions --> Signed-off-by: Radoslav Dimitrov <radoslav@stacklok.com>
87 lines
2.7 KiB
Go
87 lines
2.7 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/danielgtaylor/huma/v2"
|
|
v0 "github.com/modelcontextprotocol/registry/internal/api/handlers/v0"
|
|
"github.com/modelcontextprotocol/registry/internal/auth"
|
|
"github.com/modelcontextprotocol/registry/internal/config"
|
|
)
|
|
|
|
// NoneHandler handles anonymous authentication
|
|
type NoneHandler struct {
|
|
config *config.Config
|
|
jwtManager *auth.JWTManager
|
|
}
|
|
|
|
// NewNoneHandler creates a new anonymous authentication handler
|
|
func NewNoneHandler(cfg *config.Config) *NoneHandler {
|
|
return &NoneHandler{
|
|
config: cfg,
|
|
jwtManager: auth.NewJWTManager(cfg),
|
|
}
|
|
}
|
|
|
|
// RegisterNoneEndpoint registers the anonymous authentication endpoint
|
|
// WARNING: This endpoint is intended for local development and automated tests only.
|
|
// It should NOT be enabled in production environments as it bypasses normal authentication.
|
|
func RegisterNoneEndpoint(api huma.API, pathPrefix string, cfg *config.Config) {
|
|
if !cfg.EnableAnonymousAuth {
|
|
return
|
|
}
|
|
|
|
handler := NewNoneHandler(cfg)
|
|
|
|
// Anonymous token endpoint for development/testing only
|
|
huma.Register(api, huma.Operation{
|
|
OperationID: "get-anonymous-token" + strings.ReplaceAll(pathPrefix, "/", "-"),
|
|
Method: http.MethodPost,
|
|
Path: pathPrefix + "/auth/none",
|
|
Summary: "Get anonymous Registry JWT (Development/Testing Only)",
|
|
Description: "Get a short-lived Registry JWT token for publishing and editing servers in the io.modelcontextprotocol.anonymous/* namespace. This endpoint is intended for local development and automated testing only.",
|
|
Tags: []string{"auth"},
|
|
}, func(ctx context.Context, _ *struct{}) (*v0.Response[auth.TokenResponse], error) {
|
|
response, err := handler.GetAnonymousToken(ctx)
|
|
if err != nil {
|
|
return nil, huma.Error500InternalServerError("Failed to generate token", err)
|
|
}
|
|
|
|
return &v0.Response[auth.TokenResponse]{
|
|
Body: *response,
|
|
}, nil
|
|
})
|
|
}
|
|
|
|
// GetAnonymousToken generates an anonymous Registry JWT token
|
|
func (h *NoneHandler) GetAnonymousToken(ctx context.Context) (*auth.TokenResponse, error) {
|
|
// Build permissions for anonymous namespace only
|
|
permissions := []auth.Permission{
|
|
{
|
|
Action: auth.PermissionActionPublish,
|
|
ResourcePattern: "io.modelcontextprotocol.anonymous/*",
|
|
},
|
|
{
|
|
Action: auth.PermissionActionEdit,
|
|
ResourcePattern: "io.modelcontextprotocol.anonymous/*",
|
|
},
|
|
}
|
|
|
|
// Create JWT claims for anonymous user
|
|
claims := auth.JWTClaims{
|
|
AuthMethod: auth.MethodNone,
|
|
AuthMethodSubject: "anonymous",
|
|
Permissions: permissions,
|
|
}
|
|
|
|
// Generate Registry JWT token
|
|
tokenResponse, err := h.jwtManager.GenerateTokenResponse(ctx, claims)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to generate JWT token: %w", err)
|
|
}
|
|
|
|
return tokenResponse, nil
|
|
}
|