fix(component): fix missing propagation of original request header instill-artifact component (#1056)

Because

- We need to propagate headers with the `instill-` prefix to the
instill-artifact component.

This commit

- Fixes missing propagation of original request header instill-artifact
component

Author: donch1989

pkg/component/data/instillartifact/v0/client.go

@@ -49,6 +49,11 @@ func getRequestMetadata(vars map[string]any) metadata.MD {
 		"Instill-Auth-Type", "user",
 	)
 
+	originalHeader := util.GetOriginalHeader(vars)
+	for k, v := range originalHeader {
+		md.Set(k, v.(string))
+	}
+
 	if requester := util.GetInstillRequesterUID(vars); requester != "" {
 		md.Set("Instill-Requester-Uid", requester)
 	}

pkg/component/internal/util/helper.go

@@ -215,6 +215,13 @@ func GetInstillRequesterUID(vars map[string]any) string {
 	return vars["__PIPELINE_REQUESTER_UID"].(string)
 }
 
+func GetOriginalHeader(vars map[string]any) map[string]any {
+	if v, ok := vars["__ORIGINAL_HEADER"]; ok {
+		return v.(map[string]any)
+	}
+	return nil
+}
+
 func ConvertDataFrameToMarkdownTable(rows [][]string) string {
 	var sb strings.Builder
 

pkg/recipe/variable.go

@@ -50,6 +50,9 @@ type SystemVariables struct {
 	MgmtBackend         string `json:"__MGMT_BACKEND"`
 	ArtifactBackend     string `json:"__ARTIFACT_BACKEND"`
 	AgentBackend        string `json:"__AGENT_BACKEND"`
+
+	// OriginalHeader contains the original context header from the request.
+	OriginalHeader map[string]string `json:"__ORIGINAL_HEADER"`
 }
 
 // TODO: GenerateSystemVariables will be refactored for better code structure.

pkg/service/pipeline.go

@@ -19,6 +19,7 @@ import (
 	"go.temporal.io/sdk/client"
 	"go.temporal.io/sdk/temporal"
 	"go.uber.org/zap"
+	"google.golang.org/grpc/metadata"
 	"google.golang.org/protobuf/types/known/anypb"
 	"google.golang.org/protobuf/types/known/structpb"
 
@@ -1709,7 +1710,18 @@ func (s *service) triggerPipeline(
 				PipelineRequesterUID: requester.NsUID,
 				PipelineRequesterID:  requester.NsID,
 				HeaderAuthorization:  resourcex.GetRequestSingleHeader(ctx, "authorization"),
-				ExpiryRule:           triggerParams.expiryRule,
+				OriginalHeader: func() map[string]string {
+					md, ok := metadata.FromIncomingContext(ctx)
+					if !ok {
+						return nil
+					}
+					header := make(map[string]string)
+					for k, v := range md {
+						header[k] = v[0]
+					}
+					return header
+				}(),
+				ExpiryRule: triggerParams.expiryRule,
 			},
 			Streaming: isStreaming,
 			Mode:      mgmtpb.Mode_MODE_SYNC,
@@ -1805,7 +1817,18 @@ func (s *service) triggerAsyncPipeline(ctx context.Context, params triggerParams
 				PipelineRequesterUID: requester.NsUID,
 				PipelineRequesterID:  requester.NsID,
 				HeaderAuthorization:  resourcex.GetRequestSingleHeader(ctx, "authorization"),
-				ExpiryRule:           params.expiryRule,
+				OriginalHeader: func() map[string]string {
+					md, ok := metadata.FromIncomingContext(ctx)
+					if !ok {
+						return nil
+					}
+					header := make(map[string]string)
+					for k, v := range md {
+						header[k] = v[0]
+					}
+					return header
+				}(),
+				ExpiryRule: params.expiryRule,
 			},
 			Streaming: isStreaming,
 			Mode:      mgmtpb.Mode_MODE_ASYNC,