mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-30 19:06:33 +02:00
b7896b3153
Consolidate all logic specific to the stateless authenticate flow into a a new Stateless type in a new package internal/authenticateflow. This is in preparation for adding a new Stateful type implementing the older stateful authenticate flow (from Pomerium v0.20 and previous). This change is intended as a pure refactoring of existing logic, with no changes in functionality.
47 lines
1.5 KiB
Go
47 lines
1.5 KiB
Go
package authenticate
|
|
|
|
import (
|
|
"github.com/pomerium/pomerium/config"
|
|
"github.com/pomerium/pomerium/internal/authenticateflow"
|
|
"github.com/pomerium/pomerium/internal/identity"
|
|
identitypb "github.com/pomerium/pomerium/pkg/grpc/identity"
|
|
)
|
|
|
|
type authenticateConfig struct {
|
|
getIdentityProvider func(options *config.Options, idpID string) (identity.Authenticator, error)
|
|
profileTrimFn func(*identitypb.Profile)
|
|
authEventFn authenticateflow.AuthEventFn
|
|
}
|
|
|
|
// An Option customizes the Authenticate config.
|
|
type Option func(*authenticateConfig)
|
|
|
|
func getAuthenticateConfig(options ...Option) *authenticateConfig {
|
|
cfg := new(authenticateConfig)
|
|
WithGetIdentityProvider(defaultGetIdentityProvider)(cfg)
|
|
for _, option := range options {
|
|
option(cfg)
|
|
}
|
|
return cfg
|
|
}
|
|
|
|
// WithGetIdentityProvider sets the getIdentityProvider function in the config.
|
|
func WithGetIdentityProvider(getIdentityProvider func(options *config.Options, idpID string) (identity.Authenticator, error)) Option {
|
|
return func(cfg *authenticateConfig) {
|
|
cfg.getIdentityProvider = getIdentityProvider
|
|
}
|
|
}
|
|
|
|
// WithProfileTrimFn sets the profileTrimFn function in the config
|
|
func WithProfileTrimFn(profileTrimFn func(*identitypb.Profile)) Option {
|
|
return func(cfg *authenticateConfig) {
|
|
cfg.profileTrimFn = profileTrimFn
|
|
}
|
|
}
|
|
|
|
// WithOnAuthenticationEventHook sets the authEventFn function in the config
|
|
func WithOnAuthenticationEventHook(fn authenticateflow.AuthEventFn) Option {
|
|
return func(cfg *authenticateConfig) {
|
|
cfg.authEventFn = fn
|
|
}
|
|
}
|