From 0ac9150e59962e4ad0f9f52f67133912ad865c0f Mon Sep 17 00:00:00 2001 From: YuleiLan Date: Fri, 18 Sep 2020 16:31:19 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E8=B0=83=E6=95=B4=E4=B8=8A=E4=BC=A0?= =?UTF-8?q?=E6=8E=A5=E5=8F=A3=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apis/public/file.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/apis/public/file.go b/apis/public/file.go index 27ad51b..773868d 100644 --- a/apis/public/file.go +++ b/apis/public/file.go @@ -33,6 +33,7 @@ func UploadFile(c *gin.Context) { fileType string saveFilePath string err error + protocol string = "http" ) tag, _ = c.GetPostForm("type") fileType = c.DefaultQuery("file_type", "images") @@ -42,13 +43,17 @@ func UploadFile(c *gin.Context) { return } + if strings.HasPrefix(c.Request.Header.Get("Origin"), "https") { + protocol = "https" + } + if viper.GetBool("settings.domain.getHost") { - urlPrefix = fmt.Sprintf("http://%s/", c.Request.Host) + urlPrefix = fmt.Sprintf("%s://%s/", protocol, c.Request.Host) } else { if strings.HasSuffix(viper.GetString("settings.domain.url"), "/") { urlPrefix = viper.GetString("settings.domain.url") } else { - urlPrefix = fmt.Sprintf("http://%s/", viper.GetString("settings.domain.url")) + urlPrefix = fmt.Sprintf("%s://%s/", protocol, viper.GetString("settings.domain.url")) } }