traefikoidc/internal/providers/aws_cognito.go

package providers

import (
	"net/url"
)

// AWSCognitoProvider encapsulates AWS Cognito-specific OIDC logic.
type AWSCognitoProvider struct {
	*BaseProvider
}

// NewAWSCognitoProvider creates a new instance of the AWSCognitoProvider.
func NewAWSCognitoProvider() *AWSCognitoProvider {
	return &AWSCognitoProvider{
		BaseProvider: NewBaseProvider(),
	}
}

// GetType returns the provider's type.
func (p *AWSCognitoProvider) GetType() ProviderType {
	return ProviderTypeAWSCognito
}

// GetCapabilities returns the specific capabilities of the AWS Cognito provider.
func (p *AWSCognitoProvider) GetCapabilities() ProviderCapabilities {
	return ProviderCapabilities{
		SupportsRefreshTokens:      true,
		RequiresOfflineAccessScope: false, // Cognito doesn't use offline_access scope
		RequiresPromptConsent:      false,
		PreferredTokenValidation:   "id", // Cognito typically uses ID tokens
	}
}

// BuildAuthParams configures AWS Cognito-specific authentication parameters.
func (p *AWSCognitoProvider) BuildAuthParams(baseParams url.Values, scopes []string) (*AuthParams, error) {
	// AWS Cognito supports standard OIDC parameters
	baseParams.Set("response_type", "code")

	// Remove offline_access scope as Cognito doesn't use it
	var filteredScopes []string
	for _, scope := range scopes {
		if scope != "offline_access" {
			filteredScopes = append(filteredScopes, scope)
		}
	}

	// Ensure openid scope is present
	hasOpenID := false
	for _, scope := range filteredScopes {
		if scope == "openid" {
			hasOpenID = true
			break
		}
	}
	if !hasOpenID {
		filteredScopes = append(filteredScopes, "openid")
	}

	// Default Cognito scopes if none specified
	if len(filteredScopes) == 1 && filteredScopes[0] == "openid" {
		filteredScopes = append(filteredScopes, "email", "profile")
	}

	return &AuthParams{
		URLValues: baseParams,
		Scopes:    filteredScopes,
	}, nil
}

// AWS Cognito requires user pool and domain configuration.
func (p *AWSCognitoProvider) ValidateConfig() error {
	return p.BaseProvider.ValidateConfig()
}