From 1dab993745976e661437413f73bbb70bdfb51900 Mon Sep 17 00:00:00 2001
From: iQQBot <tianshi8650@gmail.com>
Date: Tue, 29 Apr 2025 15:48:11 +0000
Subject: [PATCH] Fix gsutil cache download issue

---
 .devcontainer/devcontainer.json   | 12 ++++----
 pkg/leeway/cache/remote/gsutil.go | 49 +++++++++++++++++++++++++++++--
 2 files changed, 52 insertions(+), 9 deletions(-)

diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 1362fdeb..c4463d50 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -2,16 +2,14 @@
 	"name": "leeway",
 	"image": "mcr.microsoft.com/devcontainers/base:ubuntu",
 	"features": {
-		"ghcr.io/devcontainers/features/node:1": {
-			"version": "lts"
-		},
-		"ghcr.io/devcontainers/features/go:1": {
-			"version": "1.24.0"
-		},
+		"ghcr.io/devcontainers/features/node:1": {},
+		"ghcr.io/devcontainers/features/go:1": {},
 		"ghcr.io/devcontainers/features/common-utils:2": {},
 		"ghcr.io/devcontainers-contrib/features/shfmt:1": {
 			"version": "3.10.0"
-		}
+		},
+		"ghcr.io/dhoeric/features/google-cloud-cli:1": {},
+		"ghcr.io/devcontainers/features/aws-cli:1": {},
 	},
 	"postStartCommand": "./.devcontainer/install-leeway.sh"
 }
diff --git a/pkg/leeway/cache/remote/gsutil.go b/pkg/leeway/cache/remote/gsutil.go
index b71d128d..7e98a968 100644
--- a/pkg/leeway/cache/remote/gsutil.go
+++ b/pkg/leeway/cache/remote/gsutil.go
@@ -97,20 +97,65 @@ func (rs *GSUtilCache) Download(ctx context.Context, dst cache.LocalCache, pkgs
 		files []string
 		dest  string
 	)
+	type urlPair struct {
+		gzURL  string
+		tarURL string
+	}
+
+	// Create a list of all possible URLs
+	var urls []string
+	packageToURLMap := make(map[cache.Package]urlPair)
 	for _, pkg := range pkgs {
 		fn, exists := dst.Location(pkg)
 		if exists {
 			continue
 		}
-
+		version, err := pkg.Version()
+		if err != nil {
+			log.WithError(err).WithField("package", pkg.FullName()).Warn("Failed to get version for package, skipping")
+			continue
+		}
 		if dest == "" {
 			dest = filepath.Dir(fn)
 		} else if dest != filepath.Dir(fn) {
 			return fmt.Errorf("gsutil only supports one target folder, not %s and %s", dest, filepath.Dir(fn))
 		}
 
-		files = append(files, fmt.Sprintf("gs://%s/%s", rs.BucketName, filepath.Base(fn)))
+		pair := urlPair{
+			gzURL:  fmt.Sprintf("gs://%s/%s.tar.gz", rs.BucketName, version),
+			tarURL: fmt.Sprintf("gs://%s/%s.tar", rs.BucketName, version),
+		}
+		packageToURLMap[pkg] = pair
+		urls = append(urls, pair.gzURL, pair.tarURL)
+	}
+	if len(urls) == 0 {
+		return nil
+	}
+
+	args := append([]string{"stat"}, urls...)
+	cmd := exec.Command("gsutil", args...)
+
+	var stdoutBuffer, stderrBuffer strings.Builder
+	cmd.Stdout = &stdoutBuffer
+	cmd.Stderr = &stderrBuffer
+
+	err := cmd.Run()
+	if err != nil && (!strings.Contains(stderrBuffer.String(), "No URLs matched")) {
+		log.Debugf("gsutil stat returned non-zero exit code: [%v], stderr: [%v]", err, stderrBuffer.String())
+		return fmt.Errorf("failed to check if files exist in remote cache: %w", err)
+	}
+
+	existingURLs := parseGSUtilStatOutput(strings.NewReader(stdoutBuffer.String()))
+	for _, urls := range packageToURLMap {
+		if _, exists := existingURLs[urls.gzURL]; exists {
+			files = append(files, urls.gzURL)
+			continue
+		}
+		if _, exists := existingURLs[urls.tarURL]; exists {
+			files = append(files, urls.tarURL)
+		}
 	}
+
 	return gsutilTransfer(dest, files)
 }