From d73bce492877b9768b40f32b6f12e4c9d8dd0d86 Mon Sep 17 00:00:00 2001
From: hui1601 <hui1601@naver.com>
Date: Sat, 8 Jun 2024 23:48:01 +0900
Subject: [PATCH] [soopglobal] Add extractor

---
 yt_dlp/extractor/_extractors.py |   4 +
 yt_dlp/extractor/soopglobal.py  | 125 ++++++++++++++++++++++++++++++++
 2 files changed, 129 insertions(+)
 create mode 100644 yt_dlp/extractor/soopglobal.py

diff --git a/yt_dlp/extractor/_extractors.py b/yt_dlp/extractor/_extractors.py
index e9cd38a65..4d92012d9 100644
--- a/yt_dlp/extractor/_extractors.py
+++ b/yt_dlp/extractor/_extractors.py
@@ -1874,6 +1874,10 @@
     SonyLIVIE,
     SonyLIVSeriesIE,
 )
+from .soopglobal import (
+    SoopGlobalLiveIE,
+    SoopGlobalVodIE,
+)
 from .soundcloud import (
     SoundcloudEmbedIE,
     SoundcloudIE,
diff --git a/yt_dlp/extractor/soopglobal.py b/yt_dlp/extractor/soopglobal.py
new file mode 100644
index 000000000..f6e1f7635
--- /dev/null
+++ b/yt_dlp/extractor/soopglobal.py
@@ -0,0 +1,125 @@
+import uuid
+
+from yt_dlp import int_or_none, traverse_obj
+from yt_dlp.compat import functools
+from yt_dlp.extractor.common import InfoExtractor
+from yt_dlp.utils import UserNotLive, bool_or_none, parse_iso8601
+
+
+class SoopGlobalLiveIE(InfoExtractor):
+    IE_NAME = 'soopglobal:live'
+    _VALID_URL = r'https?://www\.sooplive\.com/(?P<id>[\w]+$)'
+    _TESTS = [{
+        'url': 'https://www.sooplive.com/soopbowl',
+        'info_dict': {
+            'id': 'soopbowl',
+            'ext': 'mp4',
+            'title': str,
+            'thumbnail': r're:^https?://.*\.jpg$',
+            'channel': 'SoopBowl',
+            'channel_id': 'soopbowl',
+            'concurrent_view_count': int,
+            'channel_follower_count': int,
+            'timestamp': 1717852526,
+            'upload_date': '20240608',
+            'live_status': 'is_live',
+            'view_count': int,
+            'age_limit': False,
+        },
+    }]
+
+    def _real_extract(self, url):
+        channel_id = self._match_id(url)
+        client_id = str(uuid.uuid4())
+        live_detail = self._download_json(
+            f'https://api.sooplive.com/stream/info/{channel_id}', channel_id,
+            headers={'client-id': client_id},
+            note='Downloading live info', errnote='Unable to download live info')
+        if not live_detail.get('isStream'):
+            raise UserNotLive(video_id=channel_id)
+
+        age_limit = 0
+        if traverse_obj(live_detail, ('data', 'isAdult', {bool_or_none})):
+            age_limit = 19
+
+        live_statistic = self._download_json(
+            f'https://api.sooplive.com/stream/info/{channel_id}/live', channel_id,
+            headers={'client-id': client_id},
+            note='Downloading live statistics', errnote='Unable to download live statistics')
+
+        channel_info = self._download_json(
+            f'https://api.sooplive.com/channel/info/{channel_id}', channel_id,
+            headers={'client-id': client_id},
+            note='Downloading channel information', errnote='Unable to download channel information')
+
+        formats, subtitles = self._extract_m3u8_formats_and_subtitles(
+            f'https://api.sooplive.com/media/live/{channel_id}/master.m3u8', channel_id,
+            headers={'client-id': client_id},
+            note='Downloading live stream', errnote='Unable to download live stream')
+
+        return {
+            'id': channel_id,
+            'channel_id': channel_id,
+            'is_live': True,
+            'formats': formats,
+            'subtitles': subtitles,
+            'view_count': live_statistic.get('viewer'),
+            'age_limit': age_limit,
+            **traverse_obj(channel_info.get('streamerChannelInfo'), {
+                'channel': ('nickname', {str}),
+                'channel_id': ('channelId', {str}),
+                'channel_follower_count': ('totalFollowerCount', {int_or_none}),
+            }),
+            **traverse_obj(live_detail.get('data'), {
+                'title': ('title', {str}),
+                'timestamp': ('streamStartDate', {functools.partial(parse_iso8601)}),
+                'concurrent_view_count': ('totalStreamCumulativeViewer', {int_or_none}),
+                'thumbnail': ('thumbnailUrl', {str}),
+            }),
+        }
+
+
+class SoopGlobalVodIE(InfoExtractor):
+    IE_NAME = 'soopglobal:vod'
+    _VALID_URL = r'https?://www\.sooplive\.com/video/(?P<id>[\d]+)'
+    _TESTS = [{
+        'url': 'https://www.sooplive.com/video/607',
+        'info_dict': {
+            'id': '607',
+            'ext': 'mp4',
+            'title': str,
+            'thumbnail': r're:^https?://.*\.jpg$',
+            'channel': '샤미요',
+            'channel_id': 'shamiyo',
+            'timestamp': 1717051284,
+            'upload_date': '20240530',
+            'view_count': int,
+            'age_limit': False,
+        },
+        'params': {'skip_download': 'm3u8'},
+    }]
+
+    def _real_extract(self, url):
+        video_id = self._match_id(url)
+        client_id = str(uuid.uuid4())
+        video_info = self._download_json(
+            f'https://api.sooplive.com/vod/info/{video_id}', video_id,
+            headers={'client-id': client_id},
+            note='Downloading video info', errnote='Unable to download video info')
+        channel_id = video_info.get('channelId')
+        formats, subtitles = self._extract_m3u8_formats_and_subtitles(
+            f'https://api.sooplive.com/media/vod/{channel_id}/{video_id}/master.m3u8', video_id,
+            headers={'client-id': client_id},
+            note='Downloading video stream', errnote='Unable to download video stream')
+        return {
+            'id': video_id,
+            'channel': video_info.get('nickName'),
+            'channel_id': channel_id,
+            'title': video_info.get('titleName'),
+            'thumbnail': video_info.get('thumb'),
+            'timestamp': parse_iso8601(video_info.get('createDate')),
+            'view_count': video_info.get('readCnt'),
+            'age_limit': 0 if not video_info.get('isAdult') else 19,
+            'formats': formats,
+            'subtitles': subtitles,
+        }