fixed: 修复抖音无法获取监控信息的问题 (#630)

* fixed: 修复抖音无法获取监控信息的问题 * 修复配置中的抖音cookie会被覆盖的问题 --------- Co-authored-by: hucaloof <[email protected]> Co-authored-by: kira1928 <[email protected]>
hr3lxphr6j · Dec 28, 2023 · 981d95a · 981d95a
1 parent 055765c
commit 981d95a
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/src/live/douyin/douyin.go b/src/live/douyin/douyin.go
@@ -47,6 +47,10 @@ func createRandomCookie() string {
 	return utils.GenRandomString(21, randomCookieChars)
 }
 
+func createRandomOdintt() string {
+	return utils.GenRandomString(160, randomCookieChars)
+}
+
 type Live struct {
 	internal.BaseLive
 	responseCookies             map[string]string
@@ -66,12 +70,13 @@ func (l *Live) getLiveRoomWebPageResponse() (body string, err error) {
 	}
 
 	// proxy, _ := url.Parse("http://localhost:8888")
-	// resp, err := requests.NewSession(&http.Client{
-	// 	Transport: &http.Transport{
-	// 		Proxy: http.ProxyURL(proxy),
-	// 	},
-	// }).Get(
-	resp, err := requests.Get(
+	requestSession := requests.NewSession(&http.Client{
+		// Transport: &http.Transport{
+		// 	Proxy: http.ProxyURL(proxy),
+		// },
+	})
+	req, err := requests.NewRequest(
+		http.MethodGet,
 		l.Url.String(),
 		live.CommonUserAgent,
 		requests.Cookies(cookieKVs),
@@ -82,6 +87,12 @@ func (l *Live) getLiveRoomWebPageResponse() (body string, err error) {
 	if err != nil {
 		return
 	}
+	cookieWithOdinTt := fmt.Sprintf("odin_tt=%s; %s", createRandomOdintt(), req.Header.Get("Cookie"))
+	req.Header.Set("Cookie", cookieWithOdinTt)
+	resp, err := requestSession.Do(req)
+	if err != nil {
+		return
+	}
 	switch code := resp.StatusCode; code {
 	case http.StatusOK:
 		for _, cookie := range resp.Cookies() {