From b61ff5ed0882cac78bea05107dc27b93865bced8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=81=A24=2E435km/s?=
 <54017765+CoolZxp@users.noreply.github.com>
Date: Tue, 6 May 2025 11:22:19 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E9=80=82=E9=85=8DTwitCasting,TwitCasti?=
 =?UTF-8?q?ng=E6=B7=BB=E5=8A=A0Cookie,TwitCasting=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E7=94=BB=E8=B4=A8=E9=80=89=E6=8B=A9=20(#1282)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* feat: 适配TwitCasting,TwitCasting添加Cookie,TwitCasting添加画质选择

* Update __init__.py

---------

Co-authored-by: ForgQi <34411314+ForgQi@users.noreply.github.com>
---
 app/ui/plugins/twitcasting.tsx | 35 ++++++++++++++
 biliup/__init__.py             |  2 +-
 biliup/plugins/twitcasting.py  | 88 +++++++++++++++++++++++-----------
 3 files changed, 97 insertions(+), 28 deletions(-)

diff --git a/app/ui/plugins/twitcasting.tsx b/app/ui/plugins/twitcasting.tsx
index 9c331d2..6688023 100644
--- a/app/ui/plugins/twitcasting.tsx
+++ b/app/ui/plugins/twitcasting.tsx
@@ -23,6 +23,25 @@ const TwitCasting: React.FC<Props> = props => {
   return (
     <>
       <Collapse.Panel header="TwitCasting" itemKey="twitcasting">
+        <Form.Select
+          field="twitcasting_quality"
+          extraText={
+            <div style={{ fontSize: '14px' }}>
+              TwitCasting自选画质，没有选中的画质则会自动选择更低级别清晰度，如更低级别画质依旧没有则选择最清晰的。
+            </div>
+          }
+          label="画质等级（twitcasting_quality）"
+          style={{ width: '100%' }}
+          fieldStyle={{
+            alignSelf: 'stretch',
+            padding: 0,
+          }}
+          showClear={true}
+        >
+          <Select.Option value="high">高画质（high）</Select.Option>
+          <Select.Option value="medium">中画质（medium）</Select.Option>
+          <Select.Option value="low">低画质（low）</Select.Option>
+        </Form.Select>
         <Form.Switch
           field="twitcasting_danmaku"
           extraText="录制TwitCasting弹幕，默认关闭"
@@ -32,6 +51,22 @@ const TwitCasting: React.FC<Props> = props => {
             padding: 0,
           }}
         />
+      <Form.Input
+          field="user.twitcasting_cookie"
+          extraText={
+            <div className="semi-form-field-extra">
+              Cookie格式:
+              <br />
+              <code style={{ color: 'blue' }}>tc_id=xxxxxxx; tc_ss=xxxxxxx;</code>
+            </div>
+          }
+          label="TwitCasting Cookie（twitcasting_cookie）"
+          style={{ width: '100%' }}
+          fieldStyle={{
+            alignSelf: 'stretch',
+            padding: 0,
+          }}
+        />
         <Form.Input
           field="twitcasting_password"
           label="TwitCasting直播间密码（twitcasting_password）"
diff --git a/biliup/__init__.py b/biliup/__init__.py
index ed8bbb7..93770da 100644
--- a/biliup/__init__.py
+++ b/biliup/__init__.py
@@ -2,7 +2,7 @@ import logging
 import platform
 import sys
 
-__version__ = "0.4.96"
+__version__ = "0.4.97"
 
 LOG_CONF = {
     'version': 1,
diff --git a/biliup/plugins/twitcasting.py b/biliup/plugins/twitcasting.py
index b33ed5f..f798a1e 100644
--- a/biliup/plugins/twitcasting.py
+++ b/biliup/plugins/twitcasting.py
@@ -15,48 +15,82 @@ class Twitcasting(DownloadBase):
     def __init__(self, fname, url, suffix='flv'):
         super().__init__(fname, url, suffix)
         self.twitcasting_danmaku = config.get('twitcasting_danmaku', False)
-        self.twitcasting_password = config.get('twitcasting_password', '')
+        self.twitcasting_password = config.get('user.twitcasting_password')
+        self.twitcasting_quality = config.get('twitcasting_quality')
+        self.twitcasting_cookie = config.get('user', {}).get('twitcasting_cookie')
         self.fake_headers['referer'] = "https://twitcasting.tv/"
-        if self.twitcasting_password:
-            self.fake_headers[
-                'cookie'] = f"wpass={hashlib.md5(self.twitcasting_password.encode(encoding='UTF-8')).hexdigest()}"
+
         # TODO 传递过于繁琐
-        self.movie_id = None
+        self._movie_id = None
 
     async def acheck_stream(self, is_check=False):
-        # with requests.Session() as s:
-        biliup.common.util.client.headers = self.fake_headers
+        cookies = {}
+        if self.twitcasting_cookie:
+            cookies = dict(item.strip().split("=", 1) for item in self.twitcasting_cookie.split(";"))
+        if self.twitcasting_password:
+            cookies["wpass"] = hashlib.md5(self.twitcasting_password.encode(encoding='UTF-8')).hexdigest()
+        if cookies:
+            self.fake_headers['cookie'] = "; ".join([f"{k}={v}" for k, v in cookies.items()])
 
-        uploader_id = match1(self.url, r'twitcasting.tv/([^/?]+)')
-        response = await biliup.common.util.client.get(f'https://twitcasting.tv/streamserver.php?target={uploader_id}&mode=client&player=pc_web',
-                                                       timeout=5)
-        if response.status_code != 200:
-            logger.warning(f"{Twitcasting.__name__}: {self.url}: 获取错误，本次跳过")
-            return False
-        room_info = response.json()
-        if not room_info:
-            logger.warning(f"{Twitcasting.__name__}: {self.url}: 直播间地址错误")
-            return False
-        if not room_info['movie']['live']:
-            logger.debug(f"{Twitcasting.__name__}: {self.url}: 未开播")
-            return False
-
-        self.movie_id = room_info['movie']['id']
-
-        room_html = (await biliup.common.util.client.get(f'https://twitcasting.tv/{uploader_id}', timeout=5)).text
+        room_html = (await biliup.common.util.client.get(self.url, headers=self.fake_headers)).text
         if 'Enter the secret word to access' in room_html:
             logger.warning(f"{Twitcasting.__name__}: {self.url}: 直播间需要密码")
             return False
-        self.room_title = match1(room_html, r'<meta name="twitter:title" content="([^"]*)"')
+
         # 尺寸不合适
         # self.live_cover_url = match1(room_html, r'<meta property="og:image" content="([^"]*)"')
-        self.raw_stream_url = f"https://twitcasting.tv/{uploader_id}/metastream.m3u8?mode=source"
+        self.room_title = match1(room_html, r'<meta name="twitter:title" content="([^"]*)"')
+        uploader_id = match1(room_html, r'<meta name="twitter:creator" content="([^"]*)"')
+
+        response = await biliup.common.util.client.get(
+            f'https://twitcasting.tv/streamserver.php?target={uploader_id}&mode=client&player=pc_web',
+            headers=self.fake_headers)
+        if response.status_code != 200:
+            logger.warning(f"{Twitcasting.__name__}: {self.url}: 获取错误，本次跳过")
+            return False
+        stream_info = response.json()
+        if not stream_info:
+            logger.warning(f"{Twitcasting.__name__}: {self.url}: 直播间地址错误")
+            return False
+        if not stream_info['movie']['live']:
+            logger.debug(f"{Twitcasting.__name__}: {self.url}: 未开播")
+            return False
+
+        self._movie_id = stream_info['movie']['id']
+
+        if not stream_info.get("tc-hls", {}).get("streams"):
+            logger.error(f"{Twitcasting.__name__}: {self.url}: 未获取到到直播流 => {stream_info}")
+            return False
+
+        stream_url = None
+        quality_levels = ["high", "medium", "low"]
+        streams = stream_info["tc-hls"]["streams"]
+        if self.twitcasting_quality:
+            quality_levels_filter = quality_levels[quality_levels.index(self.twitcasting_quality):]
+            for quality_level in quality_levels_filter:
+                if quality_level in streams:
+                    stream_url = streams[quality_level]
+                    break
+        if not stream_url:
+            for quality_level in quality_levels:
+                if quality_level in streams:
+                    stream_url = streams[quality_level]
+                    break
+        if not stream_url:
+            stream_url = next(iter(streams.values()))
+
+        if not stream_url:
+            logger.error(f"{Twitcasting.__name__}: {self.url}: 未查找到直播流 => {stream_info}")
+            return False
+
+        self.raw_stream_url = stream_url
+
         return True
 
     def danmaku_init(self):
         if self.twitcasting_danmaku:
             self.danmaku = DanmakuClient(self.url, self.gen_download_filename(), {
-                'movie_id': self.movie_id,
+                'movie_id': self._movie_id,
                 'password': self.twitcasting_password,
             })