feat(EmbyServerHandler、JellyfinHandler): 添加 HTTPStrm 最终 URL 获取功能，减少客户端重定向次数

Author: AkimioJR

config/config.yaml.example

@@ -41,6 +41,7 @@ ClientFilter:                               # 客户端过滤器
 HTTPStrm:                                   # HTTPStrm 相关配置（Strm 文件内容是 标准 HTTP URL）
   Enable: True                              # 是否开启 HttpStrm 重定向
   TransCode: False                          # False：强制关闭转码 True：保持原有转码设置
+  FinalURL: True                            # 对 URL 进行重定向判断，找到非重定向地址再重定向给客户端，减少客户端重定向次数（适用于 Strm 内容是局域网地址但是想要在公网之中播放）
   PrefixList:                               # EmbyServer 中 Strm 文件的前缀（符合该前缀的 Strm 文件且被正确识别为 HTTP 协议都会路由到该规则下）
     - /media/strm/http
     - /media/strm/https

internal/config/type.go

@@ -56,6 +56,7 @@ type ClientFilterSetting struct {
 type HTTPStrmSetting struct {
 	Enable     bool
 	TransCode  bool // false->强制关闭转码 true->保持原有转码设置
+	FinalURL   bool // 对 URL 进行重定向判断，找到非重定向地址再重定向给客户端，减少客户端重定向次数
 	PrefixList []string
 }
 

internal/handler/emby.go

@@ -238,8 +238,19 @@ func (embyServerHandler *EmbyServerHandler) VideosHandler(ctx *gin.Context) {
 			switch strmFileType {
 			case constants.HTTPStrm:
 				if *mediasource.Protocol == emby.HTTP {
-					logging.Info("HTTPStrm 重定向至：", *mediasource.Path)
-					ctx.Redirect(http.StatusFound, *mediasource.Path)
+					redirectURL := *mediasource.Path
+					if config.HTTPStrm.FinalURL {
+						logging.Debug("HTTPStrm 启用获取最终 URL，开始尝试获取最终 URL")
+						if finalURL, err := getFinalURL(redirectURL); err != nil {
+							logging.Warning("获取最终 URL 失败，使用原始 URL：", err)
+						} else {
+							redirectURL = finalURL
+						}
+					} else {
+						logging.Debug("HTTPStrm 未启用获取最终 URL，直接使用原始 URL")
+					}
+					logging.Info("HTTPStrm 重定向至：", redirectURL)
+					ctx.Redirect(http.StatusFound, redirectURL)
 				}
 				return
 			case constants.AlistStrm: // 无需判断 *mediasource.Container 是否以Strm结尾，当 AlistStrm 存储的位置有对应的文件时，*mediasource.Container 会被设置为文件后缀

internal/handler/jellyfin.go

@@ -210,8 +210,19 @@ func (jellyfinHandler *JellyfinHandler) VideosHandler(ctx *gin.Context) {
 			switch strmFileType {
 			case constants.HTTPStrm:
 				if *mediasource.Protocol == jellyfin.HTTP {
-					logging.Infof("HTTPStrm 重定向至：%s", *mediasource.Path)
-					ctx.Redirect(http.StatusFound, *mediasource.Path)
+					redirectURL := *mediasource.Path
+					if config.HTTPStrm.FinalURL {
+						logging.Debug("HTTPStrm 启用获取最终 URL，开始尝试获取最终 URL")
+						if finalURL, err := getFinalURL(redirectURL); err != nil {
+							logging.Warning("获取最终 URL 失败，使用原始 URL：", err)
+						} else {
+							redirectURL = finalURL
+						}
+					} else {
+						logging.Debug("HTTPStrm 未启用获取最终 URL，直接使用原始 URL")
+					}
+					logging.Info("HTTPStrm 重定向至：", redirectURL)
+					ctx.Redirect(http.StatusFound, redirectURL)
 				}
 				return
 			case constants.AlistStrm: // 无需判断 *mediasource.Container 是否以Strm结尾，当 AlistStrm 存储的位置有对应的文件时，*mediasource.Container 会被设置为文件后缀

internal/handler/utils.go

@@ -6,12 +6,15 @@ import (
 	"MediaWarp/internal/logging"
 	"bytes"
 	"compress/gzip"
+	"errors"
 	"fmt"
 	"io"
 	"net/http"
 	"net/http/httputil"
+	"net/url"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/andybalholm/brotli"
 	"github.com/gin-gonic/gin"
@@ -135,3 +138,74 @@ func updateBody(rw *http.Response, content []byte) error {
 
 	return nil
 }
+
+const (
+	MaxRedirectAttempts = 10               // 最大重定向次数限制
+	RedirectTimeout     = 10 * time.Second // 最大超时时间
+
+)
+
+var (
+	ErrInvalidLocationHeader = errors.New("重定向 Location 头无效")
+	ErrMaxRedirectsExceeded  = fmt.Errorf("超过最大重定向次数限制（%d）", MaxRedirectAttempts)
+)
+
+// 获取URL的最终目标地址（自动跟踪重定向）
+func getFinalURL(rawURL string) (string, error) {
+
+	parsedURL, err := url.Parse(rawURL) // 验证并解析输入URL
+	if err != nil {
+		return "", fmt.Errorf("非法 URL： %w", err)
+	}
+	if parsedURL.Scheme == "" {
+		return "", fmt.Errorf("URL 缺少协议头： %s", parsedURL)
+	}
+
+	// 创建自定义HTTP客户端配置
+	client := &http.Client{
+		Timeout: RedirectTimeout,
+		CheckRedirect: func(req *http.Request, via []*http.Request) error {
+			// 禁止自动重定向，以便手动处理
+			return http.ErrUseLastResponse
+		},
+	}
+
+	currentURL := parsedURL.String()
+	visited := make(map[string]struct{}, MaxRedirectAttempts)
+	redirectChain := make([]string, 0, MaxRedirectAttempts+1)
+
+	// 跟踪重定向链
+	for i := 0; i <= MaxRedirectAttempts; i++ {
+		// 检测循环重定向
+		if _, exists := visited[currentURL]; exists {
+			return "", fmt.Errorf("检测到循环重定向，重定向链: %s", strings.Join(redirectChain, " -> "))
+		}
+		visited[currentURL] = struct{}{}
+		redirectChain = append(redirectChain, currentURL)
+
+		// 创建HEAD请求（更高效，只获取头部信息）
+		resp, err := client.Head(currentURL)
+		if err != nil {
+			return "", fmt.Errorf("发送 HTTP 请求失败：%w", err)
+		}
+		defer resp.Body.Close()
+
+		// 检查是否需要重定向 (3xx 状态码)
+		if resp.StatusCode >= http.StatusMultipleChoices && resp.StatusCode < http.StatusBadRequest {
+			location, err := resp.Location()
+			if err != nil {
+				return "", ErrInvalidLocationHeader
+			}
+
+			// 处理相对路径重定向
+			currentURL = location.String()
+			continue
+		}
+
+		// 返回最终的非重定向URL
+		logging.Debug("重定向链：", strings.Join(redirectChain, " -> "))
+		return resp.Request.URL.String(), nil
+	}
+
+	return "", ErrMaxRedirectsExceeded
+}