From 9a1254b745e90840a879cabe6751dbf4af0d3cfd Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Tue, 11 Jun 2024 11:57:22 +0530
Subject: [PATCH] [cli] Log selected headers on download failure

---
 cli/internal/api/client.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/cli/internal/api/client.go b/cli/internal/api/client.go
index 1127e6c2c2..c0b8089b42 100644
--- a/cli/internal/api/client.go
+++ b/cli/internal/api/client.go
@@ -76,7 +76,10 @@ func NewClient(p Params) *Client {
 			AddRetryCondition(func(r *resty.Response, err error) bool {
 				shouldRetry := r.StatusCode() == 429 || r.StatusCode() >= 500
 				if shouldRetry {
-					log.Printf("retrying download due to %d code", r.StatusCode())
+					amxRequestID := r.Header().Get("X-Amz-Request-Id")
+					cfRayID := r.Header().Get("CF-Ray")
+					wasabiRefID := r.Header().Get("X-Wasabi-Cm-Reference-Id")
+					log.Printf("Retry scheduled. error statusCode: %d, X-Amz-Request-Id: %s, CF-Ray: %s, X-Wasabi-Cm-Reference-Id: %s", r.StatusCode(), amxRequestID, cfRayID, wasabiRefID)
 				}
 				return shouldRetry
 			}),