From fdbfea31089c6a2eab53b74e4a49e8b34bd36f97 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=82=A5=E7=BE=8A?= <youshandefeiyang@gmail.com>
Date: Fri, 31 May 2024 03:00:28 +0800
Subject: [PATCH] fix huya

---
 Golang/liveurls/huya.go | 66 ++++++++++++++++++++++++-----------------
 1 file changed, 39 insertions(+), 27 deletions(-)

diff --git a/Golang/liveurls/huya.go b/Golang/liveurls/huya.go
index 6b96336..2a3ba26 100644
--- a/Golang/liveurls/huya.go
+++ b/Golang/liveurls/huya.go
@@ -22,6 +22,8 @@ import (
 	"strconv"
 	"strings"
 	"time"
+
+	"github.com/tidwall/gjson"
 )
 
 type Huya struct {
@@ -128,23 +130,34 @@ func processAntiCode(antiCode string, uid int, streamName string) string {
 	return q.Encode()
 }
 
-func format(liveArr map[string]any, uid int) map[string]any {
+func format(jsonStr string, uid int) map[string]any {
 	streamInfo := map[string]any{"flv": make(map[string]string), "hls": make(map[string]string)}
 	cdnType := map[string]string{"HY": "hycdn", "AL": "alicdn", "TX": "txcdn", "HW": "hwcdn", "HS": "hscdn", "WS": "wscdn"}
-	for _, s := range liveArr["roomInfo"].(map[string]any)["tLiveInfo"].(map[string]any)["tLiveStreamInfo"].(map[string]any)["vStreamInfo"].(map[string]any)["value"].([]any) {
-		sMap := s.(map[string]any)
-		if sMap["sFlvUrl"] != nil {
-			streamInfo["flv"].(map[string]string)[cdnType[sMap["sCdnType"].(string)]] = sMap["sFlvUrl"].(string) + "/" + sMap["sStreamName"].(string) + "." + sMap["sFlvUrlSuffix"].(string) + "?" + processAntiCode(sMap["sFlvAntiCode"].(string), uid, sMap["sStreamName"].(string))
+	ojsonStr := gjson.Get(jsonStr, "roomInfo.tLiveInfo.tLiveStreamInfo.vStreamInfo").String()
+	fmt.Println(gjson.Get(ojsonStr, "value"))
+	qreg := regexp.MustCompile(`(?i){"_proto"[\s\S]*?"value":([\s\S]*),"_classname"`)
+	qres := qreg.FindStringSubmatch(ojsonStr)
+	gjson.Parse(qres[1]).ForEach(func(_, value gjson.Result) bool {
+		sFlvUrl := value.Get("sFlvUrl").String()
+		sFlvUrlSuffix := value.Get("sFlvUrlSuffix").String()
+		sHlsUrl := value.Get("sHlsUrl").String()
+		sHlsUrlSuffix := value.Get("sHlsUrlSuffix").String()
+		sStreamName := value.Get("sStreamName").String()
+		sCdnType := value.Get("sCdnType").String()
+		sFlvAntiCode := value.Get("sFlvAntiCode").String()
+		sHlsAntiCode := value.Get("sHlsAntiCode").String()
+		if sFlvUrl != "" {
+			streamInfo["flv"].(map[string]string)[cdnType[sCdnType]] = sFlvUrl + "/" + sStreamName + "." + sFlvUrlSuffix + "?" + processAntiCode(sFlvAntiCode, uid, sStreamName)
 		}
-		if sMap["sHlsUrl"] != nil {
-			streamInfo["hls"].(map[string]string)[cdnType[sMap["sCdnType"].(string)]] = sMap["sHlsUrl"].(string) + "/" + sMap["sStreamName"].(string) + "." + sMap["sHlsUrlSuffix"].(string) + "?" + processAntiCode(sMap["sHlsAntiCode"].(string), uid, sMap["sStreamName"].(string))
+		if sHlsUrl != "" {
+			streamInfo["hls"].(map[string]string)[cdnType[sCdnType]] = sHlsUrl + "/" + sStreamName + "." + sHlsUrlSuffix + "?" + processAntiCode(sHlsAntiCode, uid, sStreamName)
 		}
-	}
+		return true
+	})
 	return streamInfo
 }
 
 func (h *Huya) GetLiveUrl() any {
-	var liveArr map[string]any
 	liveurl := "https://m.huya.com/" + h.Rid
 	client := &http.Client{}
 	r, _ := http.NewRequest("GET", liveurl, nil)
@@ -154,41 +167,40 @@ func (h *Huya) GetLiveUrl() any {
 	defer resp.Body.Close()
 	body, _ := io.ReadAll(resp.Body)
 	str := string(body)
-	freg := regexp.MustCompile(`(?i)<script> window.HNF_GLOBAL_INIT = (.*) </script>`)
+	freg := regexp.MustCompile(`(?i)<script>[\s\S]window.HNF_GLOBAL_INIT = ([\s\S]*?) </script>`)
 	res := freg.FindStringSubmatch(str)
+	if len(res) > 1 {
+		jsonStr := res[1]
+		liveStatus := gjson.Get(jsonStr, "roomInfo.eLiveStatus").Int()
+		var mediaurl any
 
-	json.Unmarshal([]byte(res[1]), &liveArr)
-	var mediaurl any
-	if roomInfo, ok := liveArr["roomInfo"].(map[string]any); ok {
-		if liveStatus, ok := roomInfo["eLiveStatus"].(float64); ok && liveStatus == 2 {
-			realurl := format(liveArr, uid)
+		if liveStatus == 2 {
+			realurl := format(jsonStr, uid)
 			if h.Type == "display" {
 				return realurl
 			}
 			for k, v := range realurl {
-				switch k {
-				case h.Media:
+				if k == h.Media {
 					if urlarr, ok := v.(map[string]string); ok {
 						for k, v := range urlarr {
-							switch k {
-							case h.Cdn:
+							if k == h.Cdn {
 								mediaurl = strings.Replace(v, "http://", "https://", 1)
 							}
 						}
 					}
 				}
 			}
-
-		} else if liveStatus, ok := roomInfo["eLiveStatus"].(float64); ok && liveStatus == 3 {
-			if roomProfile, ok := liveArr["roomProfile"].(map[string]any); ok {
-				if liveLineUrl, ok := roomProfile["liveLineUrl"].(string); ok {
-					decodedLiveLineUrl, _ := base64.StdEncoding.DecodeString(liveLineUrl)
-					mediaurl = "https:" + string(decodedLiveLineUrl)
-				}
+		} else if liveStatus == 3 {
+			liveLineUrl := gjson.Get(jsonStr, "roomProfile.liveLineUrl").String()
+			if liveLineUrl != "" {
+				decodedLiveLineUrl, _ := base64.StdEncoding.DecodeString(liveLineUrl)
+				mediaurl = "https:" + string(decodedLiveLineUrl)
 			}
 		} else {
 			mediaurl = nil
 		}
+		return mediaurl
 	}
-	return mediaurl
+
+	return nil
 }