89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package services
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
|
|
"server/models"
|
|
|
|
beego "github.com/beego/beego/v2/server/web"
|
|
)
|
|
|
|
// PublicRequestBaseURL 根据请求拼出对外访问的根（用于把 /uploads/... 拼成完整下载地址）
|
|
func PublicRequestBaseURL(c *beego.Controller) (scheme, host string) {
|
|
scheme = "http"
|
|
if c.Ctx.Request.TLS != nil {
|
|
scheme = "https"
|
|
}
|
|
if strings.EqualFold(strings.TrimSpace(c.Ctx.Request.Header.Get("X-Forwarded-Proto")), "https") {
|
|
scheme = "https"
|
|
}
|
|
host = strings.TrimSpace(c.Ctx.Request.Host)
|
|
return scheme, host
|
|
}
|
|
|
|
// ResolveSoftwareDownloadURL 优先使用自定义 download_url；否则根据 file_id 读附件 src 拼完整 URL
|
|
func ResolveSoftwareDownloadURLs(scheme, host string, downloadURLs *string) map[string]string {
|
|
result := map[string]string{}
|
|
if downloadURLs == nil || strings.TrimSpace(*downloadURLs) == "" {
|
|
return result
|
|
}
|
|
raw := strings.TrimSpace(*downloadURLs)
|
|
var values map[string]string
|
|
if err := json.Unmarshal([]byte(raw), &values); err != nil {
|
|
return result
|
|
}
|
|
for k, v := range values {
|
|
platform := strings.ToLower(strings.TrimSpace(k))
|
|
url := strings.TrimSpace(v)
|
|
if platform == "" || url == "" {
|
|
continue
|
|
}
|
|
result[platform] = ResolvePublicURL(scheme, host, url)
|
|
}
|
|
return result
|
|
}
|
|
|
|
// ResolvePublicURL 把相对路径拼成公开 URL；http(s) 地址原样返回。
|
|
func ResolvePublicURL(scheme, host, src string) string {
|
|
src = strings.TrimSpace(src)
|
|
if src == "" {
|
|
return ""
|
|
}
|
|
if strings.HasPrefix(strings.ToLower(src), "http://") || strings.HasPrefix(strings.ToLower(src), "https://") {
|
|
return src
|
|
}
|
|
if host == "" {
|
|
return src
|
|
}
|
|
if !strings.HasPrefix(src, "/") {
|
|
src = "/" + src
|
|
}
|
|
if scheme == "" {
|
|
scheme = "http"
|
|
}
|
|
return scheme + "://" + host + src
|
|
}
|
|
|
|
// ResolveSoftwareDownloadURL 优先使用自定义 download_url；否则根据 file_id 读附件 src 拼完整 URL
|
|
func ResolveSoftwareDownloadURL(scheme, host string, downloadURL *string, fileID *uint64) string {
|
|
if downloadURL != nil {
|
|
u := strings.TrimSpace(*downloadURL)
|
|
if u != "" {
|
|
return u
|
|
}
|
|
}
|
|
if fileID == nil || *fileID == 0 {
|
|
return ""
|
|
}
|
|
var f models.SystemFile
|
|
err := models.Orm.QueryTable(new(models.SystemFile)).
|
|
Filter("id", *fileID).
|
|
Filter("delete_time__isnull", true).
|
|
One(&f)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return ResolvePublicURL(scheme, host, f.Src)
|
|
}
|