From 221cdf3611ae317f9c5a7071a8bee28fac0b176e Mon Sep 17 00:00:00 2001
From: KirCute_ECT <951206789@qq.com>
Date: Wed, 25 Dec 2024 21:13:23 +0800
Subject: [PATCH] feat(s3): support custom host presign (#7699 close #7696)

---
 drivers/s3/driver.go | 8 ++++++--
 drivers/s3/meta.go   | 1 +
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/drivers/s3/driver.go b/drivers/s3/driver.go
index 2b72d78980f..82c050a1fe8 100644
--- a/drivers/s3/driver.go
+++ b/drivers/s3/driver.go
@@ -99,8 +99,12 @@ func (d *S3) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*mo
 	var link model.Link
 	var err error
 	if d.CustomHost != "" {
-		err = req.Build()
-		link.URL = req.HTTPRequest.URL.String()
+		if d.EnableCustomHostPresign {
+			link.URL, err = req.Presign(time.Hour * time.Duration(d.SignURLExpire))
+		} else {
+			err = req.Build()
+			link.URL = req.HTTPRequest.URL.String()
+		}
 		if d.RemoveBucket {
 			link.URL = strings.Replace(link.URL, "/"+d.Bucket, "", 1)
 		}
diff --git a/drivers/s3/meta.go b/drivers/s3/meta.go
index 4436c61508e..4de4b60a690 100644
--- a/drivers/s3/meta.go
+++ b/drivers/s3/meta.go
@@ -14,6 +14,7 @@ type Addition struct {
 	SecretAccessKey          string `json:"secret_access_key" required:"true"`
 	SessionToken             string `json:"session_token"`
 	CustomHost               string `json:"custom_host"`
+	EnableCustomHostPresign  bool   `json:"enable_custom_host_presign"`
 	SignURLExpire            int    `json:"sign_url_expire" type:"number" default:"4"`
 	Placeholder              string `json:"placeholder"`
 	ForcePathStyle           bool   `json:"force_path_style"`