chore(runner): add new token in header

Signed-off-by: Bo-Yi.Wu <appleboy.tw@gmail.com>

client/options.go

@@ -77,3 +77,21 @@ func WithUUIDHeader(uuid string) Option {
 		)
 	})
 }
+
+// WithTokenHeader add runner token in header
+func WithTokenHeader(token string) Option {
+	return OptionFunc(func(cfg *config) {
+		if token == "" {
+			return
+		}
+		cfg.opts = append(
+			cfg.opts,
+			connect.WithInterceptors(connect.UnaryInterceptorFunc(func(next connect.UnaryFunc) connect.UnaryFunc {
+				return func(ctx context.Context, req connect.AnyRequest) (connect.AnyResponse, error) {
+					req.Header().Set(core.TokenHeader, token)
+					return next(ctx, req)
+				}
+			})),
+		)
+	})
+}

cmd/daemon.go

@@ -82,6 +82,9 @@ func runDaemon(ctx context.Context, envFile string) func(cmd *cobra.Command, arg
 			if data.UUID != "" {
 				cfg.Runner.UUID = data.UUID
 			}
+			if data.Token != "" {
+				cfg.Runner.Token = data.Token
+			}
 		}
 
 		// try to connect to docker daemon
@@ -98,6 +101,7 @@ func runDaemon(ctx context.Context, envFile string) func(cmd *cobra.Command, arg
 			client.WithGRPC(cfg.Client.GRPC),
 			client.WithGRPCWeb(cfg.Client.GRPCWeb),
 			client.WithUUIDHeader(cfg.Runner.UUID),
+			client.WithTokenHeader(cfg.Runner.Token),
 		)
 
 		runner := &runtime.Runner{

config/config.go

@@ -69,6 +69,9 @@ func FromEnviron() (Config, error) {
 		if runner.UUID != "" {
 			cfg.Runner.UUID = runner.UUID
 		}
+		if runner.Token != "" {
+			cfg.Runner.Token = runner.Token
+		}
 		cfg.ForgeInstance = runner.ForgeInstance
 	}
 

core/runner.go

@@ -1,6 +1,9 @@
 package core
 
-const UUIDHeader = "x-runner-uuid"
+const (
+	UUIDHeader  = "x-runner-uuid"
+	TokenHeader = "x-runner-token"
+)
 
 // Runner struct
 type Runner struct {