修复快手 long-video 解析

parser/kuaishou.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"regexp"
+	"strings"
 
 	"github.com/go-resty/resty/v2"
 	"github.com/tidwall/gjson"
@@ -14,10 +15,30 @@ type kuaiShou struct{}
 
 func (k kuaiShou) parseShareUrl(shareUrl string) (*VideoParseInfo, error) {
 	client := resty.New()
-	res, err := client.R().
+	// disable redirects in the HTTP client, get params before redirects
+	client.SetRedirectPolicy(resty.NoRedirectPolicy())
+	shareRes, err := client.R().
 		SetHeader(HttpHeaderUserAgent, DefaultUserAgent).
 		SetHeader("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7").
 		Get(shareUrl)
+	// 非 resty.ErrAutoRedirectDisabled 错误时，返回错误
+	if !errors.Is(err, resty.ErrAutoRedirectDisabled) {
+		return nil, err
+	}
+
+	locationRes, err := shareRes.RawResponse.Location()
+	if err != nil {
+		return nil, err
+	}
+
+	// /fw/long-video/ 返回结果不一样, 统一替换为 /fw/photo/ 请求
+	locationUrl := locationRes.String()
+	locationUrl = strings.ReplaceAll(locationUrl, "/fw/long-video/", "/fw/photo/")
+
+	res, err := client.R().
+		SetHeader(HttpHeaderUserAgent, DefaultUserAgent).
+		SetHeader("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7").
+		Get(locationUrl)
 	if err != nil {
 		return nil, err
 	}