86 lines
2.1 KiB
Go
86 lines
2.1 KiB
Go
package configser
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"strings"
|
|
"videoweb/app/api/proto"
|
|
"videoweb/app/appg"
|
|
"videoweb/pub/common/ipaddr"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// ClientIPFromCtx 从 gin 请求上下文解析客户端 IP。
|
|
func ClientIPFromCtx(ctx context.Context) string {
|
|
c, ok := ctx.(*gin.Context)
|
|
if !ok || c == nil {
|
|
return ""
|
|
}
|
|
return ipaddr.ResolveClientIP(c)
|
|
}
|
|
|
|
// IsOverseasIP 与 m3u8 TS 域名判断一致：按 v4/v6 库识别是否海外。
|
|
func IsOverseasIP(ip string) bool {
|
|
ip = strings.TrimSpace(ip)
|
|
if ip == "" {
|
|
return false
|
|
}
|
|
detector := appg.DetectorV6
|
|
if ipaddr.IsIPv4(ip) {
|
|
detector = appg.DetectorV4
|
|
}
|
|
if detector == nil {
|
|
return false
|
|
}
|
|
ok, _ := detector.IsOverseasIP(ip)
|
|
return ok
|
|
}
|
|
|
|
// GetAppImageCdn 海外 IP 且配置了 imgAppUsCdn 时用海外图床，否则用 imgAppCdn。
|
|
func GetAppImageCdn(ip string) string {
|
|
cfg := GetConfigInfo()
|
|
if IsOverseasIP(ip) {
|
|
if us := strings.TrimSpace(cfg.ImgAppUsCdn); us != "" {
|
|
return us
|
|
}
|
|
}
|
|
return strings.TrimSpace(cfg.ImgCdn)
|
|
}
|
|
|
|
// ImageCdnFromCtx 按当前请求 IP 选择图片 CDN。
|
|
func ImageCdnFromCtx(ctx context.Context) string {
|
|
return GetAppImageCdn(ClientIPFromCtx(ctx))
|
|
}
|
|
|
|
// GetRequestConfig 返回当前请求应使用的配置副本，海外 IP 且海外图床非空时覆盖 ImgCdn。
|
|
func GetRequestConfig(ctx context.Context) proto.ConfigInfo {
|
|
cfg := GetConfigInfo()
|
|
if cdn := GetAppImageCdn(ClientIPFromCtx(ctx)); cdn != "" {
|
|
cfg.ImgCdn = cdn
|
|
}
|
|
return cfg
|
|
}
|
|
|
|
// RewriteImageCdn 将已拼好的国内图床域名替换为当前请求应使用的图床，用于缓存数据。
|
|
func RewriteImageCdn(ctx context.Context, data any) {
|
|
if data == nil {
|
|
return
|
|
}
|
|
from := strings.TrimRight(strings.TrimSpace(GetConfigInfo().ImgCdn), "/")
|
|
to := strings.TrimRight(GetAppImageCdn(ClientIPFromCtx(ctx)), "/")
|
|
if from == "" || to == "" || from == to {
|
|
return
|
|
}
|
|
raw, err := json.Marshal(data)
|
|
if err != nil {
|
|
return
|
|
}
|
|
replaced := bytes.ReplaceAll(raw, []byte(from), []byte(to))
|
|
if bytes.Equal(raw, replaced) {
|
|
return
|
|
}
|
|
_ = json.Unmarshal(replaced, data)
|
|
}
|