feat: add migu live record support

Author: ihmily

app/core/platform_handlers/__init__.py

@@ -26,6 +26,7 @@
     LivemeHandler,
     LookHandler,
     MaoerFMHandler,
+    MiguHandler,
     NeteaseHandler,
     PamdaTVHandler,
     PiaopiaoHandler,
@@ -122,6 +123,7 @@ def get_platform_info(record_url: str) -> tuple:
         "3.cn": ("京东直播", "jd"),
         "faceit.com": ("faceit", "faceit"),
         "lailianjie.com": ("连接直播", "lianjie"),
+        "miguvideo.com": ("咪咕直播", "migu"),
         ".m3u8": ("自定义录制直播", "custom"),
         ".flv": ("自定义录制直播", "custom"),
     }
@@ -159,6 +161,7 @@ def get_platform_info(record_url: str) -> tuple:
     "LivemeHandler",
     "LookHandler",
     "MaoerFMHandler",
+    "MiguHandler",
     "NeteaseHandler",
     "PamdaTVHandler",
     "PiaopiaoHandler",

app/core/platform_handlers/handlers.py

@@ -1014,6 +1014,27 @@ async def get_stream_info(self, live_url: str) -> StreamData:
         return await self.live_stream.fetch_stream_url(json_data, self.record_quality)
 
 
+class MiguHandler(PlatformHandler):
+    platform = "migu"
+
+    def __init__(
+        self,
+        proxy: str | None = None,
+        cookies: str | None = None,
+        record_quality: str | None = None,
+        platform: str | None = None,
+    ) -> None:
+        super().__init__(proxy, cookies, record_quality, platform)
+        self.live_stream: streamget.MiguLiveStream | None = None
+
+    @trace_error_decorator
+    async def get_stream_info(self, live_url: str) -> StreamData:
+        if not self.live_stream:
+            self.live_stream = streamget.MiguLiveStream(proxy_addr=self.proxy, cookies=self.cookies)
+        json_data = await self.live_stream.fetch_web_stream_data(url=live_url)
+        return await self.live_stream.fetch_stream_url(json_data, self.record_quality)
+
+
 DouyinHandler.register(r"https://.*\.douyin\.com/")
 TikTokHandler.register(r"https://.*\.tiktok\.com/")
 KuaishouHandler.register(r"https://live\.kuaishou\.com/")
@@ -1061,3 +1082,4 @@ async def get_stream_info(self, live_url: str) -> StreamData:
 JDHandler.register(r"3\.cn/")
 FaceitHandler.register(r"https://.*\.faceit\.com/")
 LianJieHandler.register(r"https://.*\.lailianjie\.com/")
+MiguHandler.register(r"https://.*\.miguvideo\.com/")

app/core/stream_manager.py

@@ -74,27 +74,27 @@ def _get_filename(self, stream_info: StreamData) -> str:
             stream_info.anchor_name = utils.clean_name(stream_info.anchor_name, self._["live_room"])
 
         now = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
-        
+
         custom_template = self.user_config.get("custom_filename_template")
         if custom_template:
             filename = custom_template
             filename = filename.replace("{anchor_name}", stream_info.anchor_name or "")
             filename = filename.replace("{title}", live_title or "")
             filename = filename.replace("{time}", now)
             filename = filename.replace("{platform}", stream_info.platform or "")
-            
+
             while "__" in filename:
                 filename = filename.replace("__", "_")
-                
+
             filename = filename.strip("_")
-            
+
             if not filename:
                 full_filename = "_".join([i for i in (stream_info.anchor_name, live_title, now) if i])
             else:
                 full_filename = filename
         else:
             full_filename = "_".join([i for i in (stream_info.anchor_name, live_title, now) if i])
-            
+
         return full_filename
 
     def _get_output_dir(self, stream_info: StreamData) -> str:
@@ -139,11 +139,13 @@ def _clean_and_truncate_title(title: str) -> str | None:
         return cleaned_title
 
     def _get_record_url(self, url: str):
-        http_record_list = ["shopee"]
-        if self.platform_key in http_record_list:
-            url = url.replace("https://", "http://")
+        http_record_list = ["shopee", "migu"]
+
         if self.user_config.get("force_https_recording") and url.startswith("http://"):
             url = url.replace("http://", "https://")
+
+        if self.platform_key in http_record_list:
+            url = url.replace("https://", "http://")
         return url
 
     async def fetch_stream(self) -> StreamData:
@@ -198,13 +200,13 @@ async def start_recording(self, stream_info: StreamData):
         )
 
     async def start_ffmpeg(
-        self,
-        record_name: str,
-        live_url: str,
-        record_url: str,
-        ffmpeg_command: list,
-        save_type: str,
-        script_command: str | None = None
+            self,
+            record_name: str,
+            live_url: str,
+            record_url: str,
+            ffmpeg_command: list,
+            save_type: str,
+            script_command: str | None = None
     ) -> bool:
         """
         The child process executes ffmpeg for recording
@@ -286,11 +288,11 @@ async def start_ffmpeg(
                 if not self.recording.is_recording:
                     logger.success(f"Live recording has stopped: {record_name}")
                 else:
-                    
+
                     logger.success(f"Live recording completed: {record_name}")
                     msg_manager = MessagePusher(self.settings)
                     user_config = self.settings.user_config
-                    
+
                     if (self.app.recording_enabled and MessagePusher.should_push_message(
                             self.settings, self.recording, check_manually_stopped=True, message_type='end') and
                             not self.recording.notified_live_end):
@@ -453,13 +455,13 @@ async def _do_converts_mp4(self, converts_file_path: str, is_original_delete: bo
             logger.error(f"An unknown error occurred: {e}")
 
     async def custom_script_execute(
-        self,
-        script_command: str,
-        record_name: str,
-        save_file_path: str,
-        save_type: str,
-        split_video_by_time: bool,
-        converts_to_mp4: bool
+            self,
+            script_command: str,
+            record_name: str,
+            save_file_path: str,
+            save_type: str,
+            split_video_by_time: bool,
+            converts_to_mp4: bool
     ):
         from ..process_manager import BackgroundService
 
@@ -542,3 +544,4 @@ def get_headers_params(live_url, platform_key):
             "blued": "referer:https://app.blued.cn",
         }
         return record_headers.get(platform_key)
+