2020-02-03 11:16:02 +03:00
|
|
|
package auth
|
|
|
|
|
2020-03-09 20:16:31 +03:00
|
|
|
import (
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/micro/go-micro/v2/auth/provider"
|
2020-03-23 19:19:30 +03:00
|
|
|
"github.com/micro/go-micro/v2/store"
|
2020-03-09 20:16:31 +03:00
|
|
|
)
|
2020-03-07 14:06:57 +03:00
|
|
|
|
2020-02-03 11:16:02 +03:00
|
|
|
type Options struct {
|
2020-03-31 14:44:34 +03:00
|
|
|
// ID is the services auth ID
|
|
|
|
ID string
|
2020-03-31 20:17:01 +03:00
|
|
|
// RefreshToken is used to generate new tokens
|
|
|
|
RefreshToken string
|
2020-03-31 14:44:34 +03:00
|
|
|
// Token is the services token used to authenticate itself
|
|
|
|
Token *Token
|
2020-02-26 01:15:44 +03:00
|
|
|
// Public key base64 encoded
|
|
|
|
PublicKey string
|
|
|
|
// Private key base64 encoded
|
|
|
|
PrivateKey string
|
2020-03-07 14:06:57 +03:00
|
|
|
// Provider is an auth provider
|
|
|
|
Provider provider.Provider
|
|
|
|
// LoginURL is the relative url path where a user can login
|
|
|
|
LoginURL string
|
2020-03-23 19:19:30 +03:00
|
|
|
// Store to back auth
|
|
|
|
Store store.Store
|
2020-02-03 11:16:02 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
type Option func(o *Options)
|
|
|
|
|
2020-03-23 19:19:30 +03:00
|
|
|
// Store to back auth
|
|
|
|
func Store(s store.Store) Option {
|
2020-02-10 11:26:28 +03:00
|
|
|
return func(o *Options) {
|
2020-03-23 19:19:30 +03:00
|
|
|
o.Store = s
|
2020-02-10 11:26:28 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-03 11:16:02 +03:00
|
|
|
// PublicKey is the JWT public key
|
|
|
|
func PublicKey(key string) Option {
|
|
|
|
return func(o *Options) {
|
2020-02-26 01:15:44 +03:00
|
|
|
o.PublicKey = key
|
2020-02-03 11:16:02 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// PrivateKey is the JWT private key
|
|
|
|
func PrivateKey(key string) Option {
|
|
|
|
return func(o *Options) {
|
2020-02-26 01:15:44 +03:00
|
|
|
o.PrivateKey = key
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-31 14:44:34 +03:00
|
|
|
// Credentials sets the auth credentials
|
2020-03-31 20:17:01 +03:00
|
|
|
func Credentials(id, refresh string) Option {
|
2020-02-26 01:15:44 +03:00
|
|
|
return func(o *Options) {
|
2020-03-31 14:44:34 +03:00
|
|
|
o.ID = id
|
2020-03-31 20:17:01 +03:00
|
|
|
o.RefreshToken = refresh
|
2020-02-03 11:16:02 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-07 14:06:57 +03:00
|
|
|
// Provider set the auth provider
|
|
|
|
func Provider(p provider.Provider) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.Provider = p
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// LoginURL sets the auth LoginURL
|
|
|
|
func LoginURL(url string) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.LoginURL = url
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-03 11:16:02 +03:00
|
|
|
type GenerateOptions struct {
|
2020-02-26 01:15:44 +03:00
|
|
|
// Metadata associated with the account
|
2020-02-03 11:16:02 +03:00
|
|
|
Metadata map[string]string
|
2020-02-26 01:15:44 +03:00
|
|
|
// Roles/scopes associated with the account
|
2020-03-23 19:19:30 +03:00
|
|
|
Roles []string
|
2020-03-30 11:51:37 +03:00
|
|
|
// Namespace the account belongs too
|
|
|
|
Namespace string
|
2020-02-03 11:16:02 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
type GenerateOption func(o *GenerateOptions)
|
|
|
|
|
2020-03-23 19:19:30 +03:00
|
|
|
// WithMetadata for the generated account
|
|
|
|
func WithMetadata(md map[string]string) GenerateOption {
|
2020-02-03 11:16:02 +03:00
|
|
|
return func(o *GenerateOptions) {
|
|
|
|
o.Metadata = md
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-23 19:19:30 +03:00
|
|
|
// WithRoles for the generated account
|
2020-03-24 13:18:34 +03:00
|
|
|
func WithRoles(rs ...string) GenerateOption {
|
2020-02-03 11:16:02 +03:00
|
|
|
return func(o *GenerateOptions) {
|
|
|
|
o.Roles = rs
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-30 11:51:37 +03:00
|
|
|
// WithNamespace for the generated account
|
|
|
|
func WithNamespace(n string) GenerateOption {
|
|
|
|
return func(o *GenerateOptions) {
|
|
|
|
o.Namespace = n
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-03 11:16:02 +03:00
|
|
|
// NewGenerateOptions from a slice of options
|
|
|
|
func NewGenerateOptions(opts ...GenerateOption) GenerateOptions {
|
|
|
|
var options GenerateOptions
|
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
2020-03-23 19:19:30 +03:00
|
|
|
return options
|
|
|
|
}
|
|
|
|
|
2020-03-31 12:06:13 +03:00
|
|
|
type TokenOptions struct {
|
2020-03-23 19:19:30 +03:00
|
|
|
// TokenExpiry is the time the token should live for
|
|
|
|
TokenExpiry time.Duration
|
|
|
|
}
|
|
|
|
|
2020-03-31 12:06:13 +03:00
|
|
|
type TokenOption func(o *TokenOptions)
|
2020-03-23 19:19:30 +03:00
|
|
|
|
|
|
|
// WithTokenExpiry for the token
|
2020-03-31 12:06:13 +03:00
|
|
|
func WithTokenExpiry(ex time.Duration) TokenOption {
|
|
|
|
return func(o *TokenOptions) {
|
2020-03-23 19:19:30 +03:00
|
|
|
o.TokenExpiry = ex
|
2020-03-09 20:16:31 +03:00
|
|
|
}
|
2020-03-23 19:19:30 +03:00
|
|
|
}
|
|
|
|
|
2020-03-31 12:06:13 +03:00
|
|
|
// NewTokenOptions from a slice of options
|
|
|
|
func NewTokenOptions(opts ...TokenOption) TokenOptions {
|
|
|
|
var options TokenOptions
|
2020-03-23 19:19:30 +03:00
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
|
|
|
|
|
|
|
// set defualt expiry of token
|
|
|
|
if options.TokenExpiry == 0 {
|
|
|
|
options.TokenExpiry = time.Minute
|
|
|
|
}
|
|
|
|
|
2020-02-03 11:16:02 +03:00
|
|
|
return options
|
|
|
|
}
|