Version v0.3.88

VERSION

@@ -1 +1 @@
-0.3.87
+0.3.88

cmd/seal/cmd_fix.go

@@ -18,6 +18,7 @@ import (
 )
 
 const summaryFlag = "summarize"
+const SealFixModeHeader = "X-Seal-Fix-Mode"
 
 func fixModeFromString(s string) phase.FixMode {
 	modes := []phase.FixMode{phase.FixModeAll, phase.FixModeRemote, phase.FixModeLocal}
@@ -179,6 +180,8 @@ func fixCommand() *cobra.Command {
 				return common.FallbackPrintableMsg(err, "failed initializing fix phase")
 			}
 
+			fixPhase.ArtifactServer.SetExtraHeaders([]api.StringPair{{Name: SealFixModeHeader, Value: fm}})
+
 			defer fixPhase.HideProgress() // should be gone when this is over, hide just in case
 
 			if !fixPhase.Config.UseSealedNames && len(silenceArray) > 0 {

internal/api/artifact_server.go

@@ -9,9 +9,14 @@ import (
 )
 
 type SealArtifactServer struct {
-	client    http.Client
-	authToken string
-	baseUrl   string
+	client       http.Client
+	authToken    string
+	baseUrl      string
+	extraHeaders []StringPair
+}
+
+func (s *SealArtifactServer) SetExtraHeaders(headers []StringPair) {
+	s.extraHeaders = headers
 }
 
 func NewArtifactServer(
@@ -65,6 +70,11 @@ func (s SealArtifactServer) Get(uri string, params []StringPair, headers []Strin
 		common.Trace("sending auth header in bulk request")
 	}
 
+	if len(s.extraHeaders) > 0 {
+		headers = append(headers, s.extraHeaders...)
+		common.Trace("adding extra headers to request", "headers", s.extraHeaders)
+	}
+
 	responseData, statusCode, err := sendSealRequest[any](s.client, "GET", reqUrl, nil, headers, params)
 	return responseData, statusCode, err
 }

internal/api/artifact_server_jfrog.go

@@ -27,6 +27,10 @@ type JFrogArtifactServer struct {
 	authHeader StringPair
 }
 
+func (s *JFrogArtifactServer) SetExtraHeaders(headers []StringPair) {
+	common.Trace("Not implemented yet")
+}
+
 func NewJFrogArtifactServer(
 	client http.Client, project string, token string, baseUrl string) *JFrogArtifactServer {
 

internal/api/artifact_server_test.go

@@ -263,3 +263,47 @@ func TestGetJsonBadObj(t *testing.T) {
 		t.Fatalf("bad err: %v", err)
 	}
 }
+
+func TestSetExtraHeaders(t *testing.T) {
+	baseurl := "http://example.com"
+	client := http.Client{}
+
+	server := NewArtifactServer(baseurl, "", "proj-id", client)
+
+	testHeaders := []StringPair{
+		{Name: "X-Custom-Header-1", Value: "Value1"},
+		{Name: "X-Custom-Header-2", Value: "Value2"},
+	}
+
+	server.SetExtraHeaders(testHeaders)
+
+	if len(server.extraHeaders) != len(testHeaders) {
+		t.Fatalf("expected %d headers, got %d", len(testHeaders), len(server.extraHeaders))
+	}
+
+	for i, header := range testHeaders {
+		if server.extraHeaders[i] != header {
+			t.Fatalf("header mismatch at index %d: expected %v, got %v", i, header, server.extraHeaders[i])
+		}
+	}
+
+	fakeRoundTripper := FakeRoundTripper{
+		statusCode: 200,
+		Validator: func(req *http.Request) {
+			for _, header := range testHeaders {
+				values := req.Header.Values(header.Name)
+				if len(values) == 0 || values[0] != header.Value {
+					t.Fatalf("header mismatch for %s: expected %s, got %v", header.Name, header.Value, values)
+				}
+			}
+		},
+	}
+
+	client.Transport = fakeRoundTripper
+	server.client = client
+
+	_, _, err := server.Get("dummy/uri", nil, nil)
+	if err != nil {
+		t.Fatalf("unexpected error during request: %v", err)
+	}
+}

internal/api/query.go

@@ -29,6 +29,7 @@ type ArtifactServer interface {
 	// will convert response from json and populate input obj
 	// if non-pointer type is passed json marshal will return error
 	GetJsonObject(url string, headers []StringPair, params []StringPair, obj any) (int, error)
+	SetExtraHeaders(headers []StringPair)
 }
 
 type PackageQueryType int