Merge remote-tracking branch 'origin/main' into cdoxsey/identity-manager-refactor-1

internal/identity/legacymanager/config.go

@@ -0,0 +1,87 @@
+package legacymanager
+
+import (
+	"time"
+
+	"github.com/pomerium/pomerium/internal/events"
+	"github.com/pomerium/pomerium/pkg/grpc/databroker"
+)
+
+var (
+	defaultSessionRefreshGracePeriod     = 1 * time.Minute
+	defaultSessionRefreshCoolOffDuration = 10 * time.Second
+)
+
+type config struct {
+	authenticator                 Authenticator
+	dataBrokerClient              databroker.DataBrokerServiceClient
+	sessionRefreshGracePeriod     time.Duration
+	sessionRefreshCoolOffDuration time.Duration
+	now                           func() time.Time
+	eventMgr                      *events.Manager
+	enabled                       bool
+}
+
+func newConfig(options ...Option) *config {
+	cfg := new(config)
+	WithSessionRefreshGracePeriod(defaultSessionRefreshGracePeriod)(cfg)
+	WithSessionRefreshCoolOffDuration(defaultSessionRefreshCoolOffDuration)(cfg)
+	WithNow(time.Now)(cfg)
+	WithEnabled(true)(cfg)
+	for _, option := range options {
+		option(cfg)
+	}
+	return cfg
+}
+
+// An Option customizes the configuration used for the identity manager.
+type Option func(*config)
+
+// WithAuthenticator sets the authenticator in the config.
+func WithAuthenticator(authenticator Authenticator) Option {
+	return func(cfg *config) {
+		cfg.authenticator = authenticator
+	}
+}
+
+// WithDataBrokerClient sets the databroker client in the config.
+func WithDataBrokerClient(dataBrokerClient databroker.DataBrokerServiceClient) Option {
+	return func(cfg *config) {
+		cfg.dataBrokerClient = dataBrokerClient
+	}
+}
+
+// WithSessionRefreshGracePeriod sets the session refresh grace period used by the manager.
+func WithSessionRefreshGracePeriod(dur time.Duration) Option {
+	return func(cfg *config) {
+		cfg.sessionRefreshGracePeriod = dur
+	}
+}
+
+// WithSessionRefreshCoolOffDuration sets the session refresh cool-off duration used by the manager.
+func WithSessionRefreshCoolOffDuration(dur time.Duration) Option {
+	return func(cfg *config) {
+		cfg.sessionRefreshCoolOffDuration = dur
+	}
+}
+
+// WithNow customizes the time.Now function used by the manager.
+func WithNow(now func() time.Time) Option {
+	return func(cfg *config) {
+		cfg.now = now
+	}
+}
+
+// WithEventManager passes an event manager to record events
+func WithEventManager(mgr *events.Manager) Option {
+	return func(cfg *config) {
+		cfg.eventMgr = mgr
+	}
+}
+
+// WithEnabled sets the enabled option in the config.
+func WithEnabled(enabled bool) Option {
+	return func(cfg *config) {
+		cfg.enabled = enabled
+	}
+}