Jeff Mitchell
b86430c2af
This is a precursor to another upcoming PR that will make use of these changes, but it makes that PR more manageable.
116 lines
3.2 KiB
Go
116 lines
3.2 KiB
Go
package authtoken
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/hashicorp/boundary/internal/auth/password"
|
|
"github.com/hashicorp/boundary/internal/db"
|
|
"github.com/hashicorp/boundary/internal/iam"
|
|
)
|
|
|
|
var (
|
|
defaultTokenTimeToLiveDuration = 7 * 24 * time.Hour
|
|
defaultTokenTimeToStaleDuration = 24 * time.Hour
|
|
)
|
|
|
|
// getOpts - iterate the inbound Options and return a struct
|
|
func getOpts(opt ...Option) options {
|
|
opts := getDefaultOptions()
|
|
for _, o := range opt {
|
|
o(&opts)
|
|
}
|
|
return opts
|
|
}
|
|
|
|
// Option - how Options are passed as arguments.
|
|
type Option func(*options)
|
|
|
|
// options = how options are represented
|
|
type options struct {
|
|
withTokenValue bool
|
|
withTokenTimeToLiveDuration time.Duration
|
|
withTokenTimeToStaleDuration time.Duration
|
|
withLimit int
|
|
withStatus Status
|
|
withPublicId string
|
|
withPasswordOptions []password.Option
|
|
withIamOptions []iam.Option
|
|
}
|
|
|
|
func getDefaultOptions() options {
|
|
return options{
|
|
withLimit: db.DefaultLimit,
|
|
withTokenTimeToLiveDuration: defaultTokenTimeToLiveDuration,
|
|
withTokenTimeToStaleDuration: defaultTokenTimeToStaleDuration,
|
|
}
|
|
}
|
|
|
|
// withTokenValue allows the auth token value to be included in the lookup response.
|
|
// This is purposefully not exported as it should only be used internally by the auth token repository itself.
|
|
func withTokenValue() Option {
|
|
return func(o *options) {
|
|
o.withTokenValue = true
|
|
}
|
|
}
|
|
|
|
// WithTokenTimeToLiveDuration allows setting the auth token time-to-live.
|
|
func WithTokenTimeToLiveDuration(ttl time.Duration) Option {
|
|
return func(o *options) {
|
|
if ttl > 0 {
|
|
o.withTokenTimeToLiveDuration = ttl
|
|
}
|
|
}
|
|
}
|
|
|
|
// WithTokenTimeToStaleDuration allows setting the auth token staleness duration.
|
|
func WithTokenTimeToStaleDuration(dur time.Duration) Option {
|
|
return func(o *options) {
|
|
if dur > 0 {
|
|
o.withTokenTimeToStaleDuration = dur
|
|
}
|
|
}
|
|
}
|
|
|
|
// WithLimit provides an option to provide a limit. Intentionally allowing
|
|
// negative integers. If WithLimit < 0, then unlimited results are returned.
|
|
// If WithLimit == 0, then default limits are used for results.
|
|
func WithLimit(limit int) Option {
|
|
return func(o *options) {
|
|
if limit > 0 {
|
|
o.withLimit = limit
|
|
}
|
|
}
|
|
}
|
|
|
|
// WithStatus allows setting of the auth token's Status.
|
|
func WithStatus(status Status) Option {
|
|
return func(o *options) {
|
|
o.withStatus = status
|
|
}
|
|
}
|
|
|
|
// WithPublicId allows the setting of the auth token's public id
|
|
func WithPublicId(id string) Option {
|
|
return func(o *options) {
|
|
o.withPublicId = id
|
|
}
|
|
}
|
|
|
|
// WithPasswordOptions allows passing through options for the password package.
|
|
// This is useful for things like tests where you may use testing helper
|
|
// functions in this package but they also create password resources.
|
|
func WithPasswordOptions(with ...password.Option) Option {
|
|
return func(o *options) {
|
|
o.withPasswordOptions = with
|
|
}
|
|
}
|
|
|
|
// WithIamOptions allows passing through options for the IAM package. This is
|
|
// useful for things like tests where you may use testing helper functions in
|
|
// this package but they also create IAM resources.
|
|
func WithIamOptions(with ...iam.Option) Option {
|
|
return func(o *options) {
|
|
o.withIamOptions = with
|
|
}
|
|
}
|