micro/auth/service/service.go

235 lines
5.5 KiB
Go
Raw Normal View History

package service
import (
"context"
"strings"
"time"
"github.com/micro/go-micro/v2/auth"
2020-05-20 11:59:01 +01:00
"github.com/micro/go-micro/v2/auth/rules"
pb "github.com/micro/go-micro/v2/auth/service/proto"
"github.com/micro/go-micro/v2/auth/token"
"github.com/micro/go-micro/v2/auth/token/jwt"
"github.com/micro/go-micro/v2/client"
)
// svc is the service implementation of the Auth interface
type svc struct {
options auth.Options
auth pb.AuthService
2020-05-26 15:52:21 +01:00
rules pb.RulesService
jwt token.Provider
}
func (s *svc) String() string {
return "service"
}
func (s *svc) Init(opts ...auth.Option) {
for _, o := range opts {
o(&s.options)
}
2020-05-11 17:57:39 +01:00
s.auth = pb.NewAuthService("go.micro.auth", s.options.Client)
2020-05-26 15:52:21 +01:00
s.rules = pb.NewRulesService("go.micro.auth", s.options.Client)
// if we have a JWT public key passed as an option,
// we can decode tokens with the type "JWT" locally
// and not have to make an RPC call
if key := s.options.PublicKey; len(key) > 0 {
s.jwt = jwt.NewTokenProvider(token.WithPublicKey(key))
}
}
func (s *svc) Options() auth.Options {
return s.options
}
// Generate a new account
2020-04-01 17:20:02 +01:00
func (s *svc) Generate(id string, opts ...auth.GenerateOption) (*auth.Account, error) {
options := auth.NewGenerateOptions(opts...)
rsp, err := s.auth.Generate(context.TODO(), &pb.GenerateRequest{
Id: id,
Type: options.Type,
Secret: options.Secret,
Scopes: options.Scopes,
Metadata: options.Metadata,
Provider: options.Provider,
}, s.callOpts()...)
if err != nil {
return nil, err
}
return serializeAccount(rsp.Account), nil
}
// Grant access to a resource
2020-05-20 11:59:01 +01:00
func (s *svc) Grant(rule *auth.Rule) error {
2020-05-21 16:41:55 +01:00
access := pb.Access_UNKNOWN
if rule.Access == auth.AccessGranted {
access = pb.Access_GRANTED
} else if rule.Access == auth.AccessDenied {
access = pb.Access_DENIED
}
2020-05-26 15:52:21 +01:00
_, err := s.rules.Create(context.TODO(), &pb.CreateRequest{
2020-05-20 11:59:01 +01:00
Rule: &pb.Rule{
Id: rule.ID,
2020-05-21 14:56:17 +01:00
Scope: rule.Scope,
2020-05-20 11:59:01 +01:00
Priority: rule.Priority,
2020-05-21 16:41:55 +01:00
Access: access,
2020-05-20 11:59:01 +01:00
Resource: &pb.Resource{
Type: rule.Resource.Type,
Name: rule.Resource.Name,
Endpoint: rule.Resource.Endpoint,
},
},
}, s.callOpts()...)
2020-05-21 16:41:55 +01:00
return err
}
// Revoke access to a resource
2020-05-20 11:59:01 +01:00
func (s *svc) Revoke(rule *auth.Rule) error {
2020-05-26 15:52:21 +01:00
_, err := s.rules.Delete(context.TODO(), &pb.DeleteRequest{
2020-05-21 12:07:22 +01:00
Id: rule.ID,
}, s.callOpts()...)
2020-05-21 16:41:55 +01:00
return err
}
2020-05-26 15:52:21 +01:00
func (s *svc) Rules(opts ...auth.RulesOption) ([]*auth.Rule, error) {
var options auth.RulesOptions
for _, o := range opts {
o(&options)
2020-04-07 16:24:51 +01:00
}
2020-05-26 15:52:21 +01:00
if options.Context == nil {
options.Context = context.TODO()
}
callOpts := append(s.callOpts(), client.WithCache(time.Second*30))
rsp, err := s.rules.List(options.Context, &pb.ListRequest{}, callOpts...)
2020-05-26 15:52:21 +01:00
if err != nil {
return nil, err
}
2020-05-26 15:52:21 +01:00
rules := make([]*auth.Rule, len(rsp.Rules))
for i, r := range rsp.Rules {
rules[i] = serializeRule(r)
}
2020-05-26 15:52:21 +01:00
return rules, nil
2020-05-20 11:59:01 +01:00
}
// Verify an account has access to a resource
2020-05-20 16:49:52 +01:00
func (s *svc) Verify(acc *auth.Account, res *auth.Resource, opts ...auth.VerifyOption) error {
2020-05-21 16:41:55 +01:00
var options auth.VerifyOptions
2020-05-20 16:49:52 +01:00
for _, o := range opts {
o(&options)
}
2020-05-26 15:52:21 +01:00
rs, err := s.Rules(auth.RulesContext(options.Context))
if err != nil {
return err
}
2020-05-20 16:49:52 +01:00
2020-05-26 15:52:21 +01:00
return rules.Verify(rs, acc, res)
}
// Inspect a token
func (s *svc) Inspect(token string) (*auth.Account, error) {
2020-04-01 14:25:00 +01:00
// try to decode JWT locally and fall back to srv if an error occurs
if len(strings.Split(token, ".")) == 3 && s.jwt != nil {
2020-04-07 16:24:51 +01:00
return s.jwt.Inspect(token)
}
2020-04-07 16:24:51 +01:00
// the token is not a JWT or we do not have the keys to decode it,
// fall back to the auth service
rsp, err := s.auth.Inspect(context.TODO(), &pb.InspectRequest{Token: token}, s.callOpts()...)
if err != nil {
return nil, err
}
return serializeAccount(rsp.Account), nil
}
// Token generation using an account ID and secret
2020-04-01 14:25:00 +01:00
func (s *svc) Token(opts ...auth.TokenOption) (*auth.Token, error) {
options := auth.NewTokenOptions(opts...)
rsp, err := s.auth.Token(context.Background(), &pb.TokenRequest{
2020-04-01 14:25:00 +01:00
Id: options.ID,
Secret: options.Secret,
RefreshToken: options.RefreshToken,
TokenExpiry: int64(options.Expiry.Seconds()),
}, s.callOpts()...)
if err != nil {
return nil, err
}
return serializeToken(rsp.Token), nil
}
func serializeToken(t *pb.Token) *auth.Token {
return &auth.Token{
2020-04-01 14:25:00 +01:00
AccessToken: t.AccessToken,
RefreshToken: t.RefreshToken,
Created: time.Unix(t.Created, 0),
Expiry: time.Unix(t.Expiry, 0),
}
}
func serializeAccount(a *pb.Account) *auth.Account {
return &auth.Account{
ID: a.Id,
Secret: a.Secret,
2020-05-21 16:41:55 +01:00
Issuer: a.Issuer,
Metadata: a.Metadata,
Scopes: a.Scopes,
}
}
2020-05-13 17:35:57 +01:00
2020-05-26 15:52:21 +01:00
func serializeRule(r *pb.Rule) *auth.Rule {
var access auth.Access
if r.Access == pb.Access_GRANTED {
access = auth.AccessGranted
} else {
access = auth.AccessDenied
}
return &auth.Rule{
ID: r.Id,
Scope: r.Scope,
Access: access,
Priority: r.Priority,
Resource: &auth.Resource{
Type: r.Resource.Type,
Name: r.Resource.Name,
Endpoint: r.Resource.Endpoint,
},
}
}
2020-05-13 17:35:57 +01:00
func (s *svc) callOpts() []client.CallOption {
return []client.CallOption{
client.WithAddress(s.options.Addrs...),
}
}
2020-05-13 17:35:57 +01:00
// NewAuth returns a new instance of the Auth service
func NewAuth(opts ...auth.Option) auth.Auth {
options := auth.NewOptions(opts...)
if options.Client == nil {
options.Client = client.DefaultClient
}
if len(options.Addrs) == 0 {
options.Addrs = []string{"127.0.0.1:8010"}
}
2020-05-13 17:35:57 +01:00
2020-05-26 15:52:21 +01:00
return &svc{
2020-05-13 17:35:57 +01:00
auth: pb.NewAuthService("go.micro.auth", options.Client),
2020-05-26 15:52:21 +01:00
rules: pb.NewRulesService("go.micro.auth", options.Client),
2020-05-13 17:35:57 +01:00
options: options,
}
}