feat: support enable without token

porter.go

@@ -31,12 +31,13 @@ const (
 )
 
 type Porter struct {
-	server       *grpc.Server
-	wrapper      wrapper
-	logger       log.Logger
-	app          *kratos.App
-	consulConfig *capi.Config
-	serverConfig *ServerConfig
+	server        *grpc.Server
+	requireAsUser bool
+	wrapper       wrapper
+	logger        log.Logger
+	app           *kratos.App
+	consulConfig  *capi.Config
+	serverConfig  *ServerConfig
 }
 
 type PorterConfig struct {
@@ -66,6 +67,12 @@ func WithPorterConsulConfig(config *capi.Config) PorterOption {
 	}
 }
 
+func WithAsUser() PorterOption {
+	return func(p *Porter) {
+		p.requireAsUser = true
+	}
+}
+
 func (p *Porter) Run() error {
 	return p.app.Run()
 }
@@ -104,11 +111,12 @@ func NewPorter(ctx context.Context, config PorterConfig, handler Handler, option
 		return nil, err
 	}
 	c := wrapper{
-		Handler: handler,
-		Config:  config,
-		Logger:  p.logger,
-		Token:   nil,
-		Client:  client,
+		Handler:      handler,
+		Config:       config,
+		Logger:       p.logger,
+		requireToken: p.requireAsUser,
+		Token:        nil,
+		Client:       client,
 	}
 	p.wrapper = c
 	p.server = NewServer(
@@ -179,6 +187,9 @@ func WellKnownToString(e protoreflect.Enum) string {
 }
 
 func (p *Porter) AsUser(ctx context.Context, userID int64) (*LibrarianClient, error) {
+	if !p.requireAsUser {
+		return nil, errors.New("init porter with `WithAsUser` option to use this method")
+	}
 	if p.wrapper.Token == nil {
 		return nil, errors.New("porter not enabled")
 	}

wrapper.go

@@ -18,11 +18,12 @@ import (
 )
 
 type wrapper struct {
-	Handler Handler
-	Config  PorterConfig
-	Logger  log.Logger
-	Token   *tokenInfo
-	Client  sephirah.LibrarianSephirahServiceClient
+	Handler      Handler
+	Config       PorterConfig
+	Logger       log.Logger
+	requireToken bool
+	Token        *tokenInfo
+	Client       sephirah.LibrarianSephirahServiceClient
 }
 
 type tokenInfo struct {
@@ -49,15 +50,20 @@ func (s *wrapper) EnablePorter(ctx context.Context, req *pb.EnablePorterRequest)
 			return nil, fmt.Errorf("porter already enabled by %d", s.Token.enabler)
 		}
 	}
-	ctx = metadata.AppendToOutgoingContext(ctx, "authorization", "Bearer "+req.GetRefreshToken())
-	resp, err := s.Client.RefreshToken(ctx, new(sephirah.RefreshTokenRequest))
-	if err != nil {
-		return nil, err
-	}
-	s.Token = &tokenInfo{
-		enabler:      req.GetSephirahId(),
-		AccessToken:  resp.GetAccessToken(),
-		refreshToken: resp.GetRefreshToken(),
+	if s.requireToken {
+		ctx = metadata.AppendToOutgoingContext(ctx, "authorization", "Bearer "+req.GetRefreshToken())
+		resp, err := s.Client.RefreshToken(ctx, new(sephirah.RefreshTokenRequest))
+		if err != nil {
+			return nil, err
+		}
+		s.Token = &tokenInfo{
+			enabler:      req.GetSephirahId(),
+			AccessToken:  resp.GetAccessToken(),
+			refreshToken: resp.GetRefreshToken(),
+		}
+	} else {
+		s.Token = new(tokenInfo)
+		s.Token.enabler = req.GetSephirahId()
 	}
 	return &pb.EnablePorterResponse{}, nil
 }