feat: custom s3 url

Signed-off-by: Lukas Hoehl <[email protected]>

Author: hown3d

cmd/go-cache-plugin/commands.go

@@ -26,6 +26,7 @@ var flags struct {
 	CacheDir      string        `flag:"cache-dir,default=$GOCACHE_DIR,Local cache directory (required)"`
 	S3Bucket      string        `flag:"bucket,default=$GOCACHE_S3_BUCKET,S3 bucket name (required)"`
 	S3Region      string        `flag:"region,default=$GOCACHE_S3_REGION,S3 region"`
+	S3URL         string        `flag:"s3-url,default=$GOCACHE_S3_URL,S3 URL"`
 	S3PathStyle   bool          `flag:"s3-path-style,default=$GOCACHE_S3_PATH_STYLE,S3 path-style URLs (optional)"`
 	KeyPrefix     string        `flag:"prefix,default=$GOCACHE_KEY_PREFIX,S3 key prefix (optional)"`
 	MinUploadSize int64         `flag:"min-upload-size,default=$GOCACHE_MIN_SIZE,Minimum object size to upload to S3 (in bytes)"`

cmd/go-cache-plugin/setup.go

@@ -62,10 +62,19 @@ func initCacheServer(env *command.Env) (*gocache.Server, *s3util.Client, error)
 
 	vprintf("local cache directory: %s", flags.CacheDir)
 	vprintf("S3 cache bucket %q (%s)", flags.S3Bucket, region)
-	client := &s3util.Client{
-		Client: s3.NewFromConfig(cfg, func(o *s3.Options) {
+	s3opts := []func(*s3.Options){
+		func(o *s3.Options) {
 			o.UsePathStyle = flags.S3PathStyle
-		}),
+		},
+	}
+	if flags.S3URL != "" {
+		vprintf("S3 URL: %s", flags.S3URL)
+		s3opts = append(s3opts, func(o *s3.Options) {
+			o.BaseEndpoint = &flags.S3URL
+		})
+	}
+	client := &s3util.Client{
+		Client: s3.NewFromConfig(cfg, s3opts...),
 		Bucket: flags.S3Bucket,
 	}
 	cache := &gobuild.S3Cache{