148 lines
2.8 KiB
Go
148 lines
2.8 KiB
Go
package jwt
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/micro/go-micro/v2/auth"
|
|
"github.com/micro/go-micro/v2/auth/rules"
|
|
"github.com/micro/go-micro/v2/auth/token"
|
|
jwtToken "github.com/micro/go-micro/v2/auth/token/jwt"
|
|
)
|
|
|
|
// NewAuth returns a new instance of the Auth service
|
|
func NewAuth(opts ...auth.Option) auth.Auth {
|
|
j := new(jwt)
|
|
j.Init(opts...)
|
|
return j
|
|
}
|
|
|
|
type rule struct {
|
|
role string
|
|
resource *auth.Resource
|
|
}
|
|
|
|
type jwt struct {
|
|
options auth.Options
|
|
jwt token.Provider
|
|
rules []*auth.Rule
|
|
|
|
sync.Mutex
|
|
}
|
|
|
|
func (j *jwt) String() string {
|
|
return "jwt"
|
|
}
|
|
|
|
func (j *jwt) Init(opts ...auth.Option) {
|
|
j.Lock()
|
|
defer j.Unlock()
|
|
|
|
for _, o := range opts {
|
|
o(&j.options)
|
|
}
|
|
|
|
j.jwt = jwtToken.NewTokenProvider(
|
|
token.WithPrivateKey(j.options.PrivateKey),
|
|
token.WithPublicKey(j.options.PublicKey),
|
|
)
|
|
}
|
|
|
|
func (j *jwt) Options() auth.Options {
|
|
j.Lock()
|
|
defer j.Unlock()
|
|
return j.options
|
|
}
|
|
|
|
func (j *jwt) Generate(id string, opts ...auth.GenerateOption) (*auth.Account, error) {
|
|
options := auth.NewGenerateOptions(opts...)
|
|
account := &auth.Account{
|
|
ID: id,
|
|
Type: options.Type,
|
|
Roles: options.Roles,
|
|
Scopes: options.Scopes,
|
|
Provider: options.Provider,
|
|
Metadata: options.Metadata,
|
|
}
|
|
|
|
// generate a JWT secret which can be provided to the Token() method
|
|
// and exchanged for an access token
|
|
secret, err := j.jwt.Generate(account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
account.Secret = secret.Token
|
|
|
|
// return the account
|
|
return account, nil
|
|
}
|
|
|
|
func (j *jwt) Grant(rule *auth.Rule) error {
|
|
j.Lock()
|
|
defer j.Unlock()
|
|
j.rules = append(j.rules, rule)
|
|
return nil
|
|
}
|
|
|
|
func (j *jwt) Revoke(rule *auth.Rule) error {
|
|
j.Lock()
|
|
defer j.Unlock()
|
|
|
|
rules := []*auth.Rule{}
|
|
for _, r := range rules {
|
|
if r.ID != rule.ID {
|
|
rules = append(rules, r)
|
|
}
|
|
}
|
|
|
|
j.rules = rules
|
|
return nil
|
|
}
|
|
|
|
func (j *jwt) Verify(acc *auth.Account, res *auth.Resource) error {
|
|
j.Lock()
|
|
defer j.Unlock()
|
|
return rules.Verify(j.options.Namespace, j.rules, acc, res)
|
|
}
|
|
|
|
func (j *jwt) Rules() ([]*auth.Rule, error) {
|
|
j.Lock()
|
|
defer j.Unlock()
|
|
return j.rules, nil
|
|
}
|
|
|
|
func (j *jwt) Inspect(token string) (*auth.Account, error) {
|
|
return j.jwt.Inspect(token)
|
|
}
|
|
|
|
func (j *jwt) Token(opts ...auth.TokenOption) (*auth.Token, error) {
|
|
options := auth.NewTokenOptions(opts...)
|
|
|
|
secret := options.RefreshToken
|
|
if len(options.Secret) > 0 {
|
|
secret = options.Secret
|
|
}
|
|
|
|
account, err := j.jwt.Inspect(secret)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
access, err := j.jwt.Generate(account, token.WithExpiry(options.Expiry))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
refresh, err := j.jwt.Generate(account, token.WithExpiry(options.Expiry+time.Hour))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &auth.Token{
|
|
Created: access.Created,
|
|
Expiry: access.Expiry,
|
|
AccessToken: access.Token,
|
|
RefreshToken: refresh.Token,
|
|
}, nil
|
|
}
|