From 9c2a6fa44928200f281c94ce23537801a4c9e6d3 Mon Sep 17 00:00:00 2001
From: ChocoLZS <chocoielzs@gmail.com>
Date: Wed, 21 Aug 2024 11:43:45 +0800
Subject: [PATCH] feat: add support for uliza in pia-live

---
 yt_dlp/extractor/_extractors.py    |   3 +-
 yt_dlp/extractor/pialive.py        |  55 ++++++++++++++
 yt_dlp/extractor/piaulizaportal.py | 118 +++++++++++++++++++++--------
 3 files changed, 144 insertions(+), 32 deletions(-)
 create mode 100644 yt_dlp/extractor/pialive.py

diff --git a/yt_dlp/extractor/_extractors.py b/yt_dlp/extractor/_extractors.py
index 9b73fcd75e..d9d2609078 100644
--- a/yt_dlp/extractor/_extractors.py
+++ b/yt_dlp/extractor/_extractors.py
@@ -1510,8 +1510,9 @@ from .pgatour import PGATourIE
 from .philharmoniedeparis import PhilharmonieDeParisIE
 from .phoenix import PhoenixIE
 from .photobucket import PhotobucketIE
+from .pialive import PiaLiveIE
 from .piapro import PiaproIE
-from .piaulizaportal import PIAULIZAPortalIE
+from .piaulizaportal import PIAULIZAPortalAPIIE, PIAULIZAPortalIE
 from .picarto import (
     PicartoIE,
     PicartoVodIE,
diff --git a/yt_dlp/extractor/pialive.py b/yt_dlp/extractor/pialive.py
new file mode 100644
index 0000000000..d662f372da
--- /dev/null
+++ b/yt_dlp/extractor/pialive.py
@@ -0,0 +1,55 @@
+from .common import InfoExtractor
+from .piaulizaportal import PIAULIZAPortalAPIIE
+from ..networking import Request
+from ..utils import multipart_encode, smuggle_url
+
+
+class PiaLiveIE(InfoExtractor):
+    PLAYER_ROOT_URL = 'https://player.pia-live.jp'
+    PIA_LIVE_API_URL = 'https://api.pia-live.jp'
+    _VALID_URL = r'https?://player\.pia-live\.jp/stream/(?P<id>[\w-]+)'
+
+    def _real_extract(self, url):
+        video_key = self._match_id(url)
+        webpage = self._download_webpage(url, video_key)
+        program_code = self._search_regex(r"const programCode = '(.*?)';", webpage, 'program code')
+
+        prod_configure = self._download_webpage(
+            self.PLAYER_ROOT_URL + self._search_regex(r'<script [^>]*\bsrc="(/statics/js/s_prod[^"]+)"', webpage, 'prod configure page'),
+            program_code,
+            headers={'Referer': self.PLAYER_ROOT_URL},
+            note='Fetching prod configure page', errnote='Unable to fetch prod configure page',
+        )
+
+        api_key = self._search_regex(r"const APIKEY = '(.*?)';", prod_configure, 'api key')
+        payload, content_type = multipart_encode({
+            'play_url': video_key,
+            'api_key': api_key,
+        })
+        player_tag_list = self._download_json(
+            Request(
+                url=f'{self.PIA_LIVE_API_URL}/perf/player-tag-list/{program_code}',
+                method='POST',
+                data=payload,
+                headers={
+                    'Content-Type': content_type,
+                },
+            ),
+            program_code,
+        )
+
+        player_data_url = self._search_regex(PIAULIZAPortalAPIIE.SCRIPT_TAG_REGEX_PATTERN,
+                                             player_tag_list['data']['movie_one_tag'], 'player data url')
+
+        return {
+            **self.url_result(
+                smuggle_url(
+                    player_data_url,
+                    {'video_id': program_code, 'referer': self.PLAYER_ROOT_URL, 'info_dict': {
+                        'id': program_code,
+                        'title': self._html_extract_title(webpage),
+                    }},
+                ),
+                ie=PIAULIZAPortalAPIIE.ie_key(),
+            ),
+        }
diff --git a/yt_dlp/extractor/piaulizaportal.py b/yt_dlp/extractor/piaulizaportal.py
index 1eb6d92b72..b42f89c4d7 100644
--- a/yt_dlp/extractor/piaulizaportal.py
+++ b/yt_dlp/extractor/piaulizaportal.py
@@ -1,11 +1,78 @@
 from .common import InfoExtractor
-from ..utils import (
-    ExtractorError,
-    int_or_none,
-    parse_qs,
-    time_seconds,
-    traverse_obj,
-)
+from ..utils import ExtractorError, int_or_none, parse_qs, smuggle_url, time_seconds, traverse_obj, unsmuggle_url
+
+
+class PIAULIZAPortalAPIIE(InfoExtractor):
+    IE_DESC = 'https://player-api.p.uliza.jp - PIA ULIZA m3u8'
+    SCRIPT_TAG_REGEX_PATTERN = r'<script [^>]*\bsrc="(https://player-api\.p\.uliza\.jp/v1/players/[^"]+)"'
+
+    _VALID_URL = r'https://player-api\.p\.uliza\.jp/v1/players/(?P<id>.*)'
+    _TESTS = [
+        {
+            'url': 'https://player-api.p.uliza.jp/v1/players/timeshift-disabled/pia/admin?type=normal&playerobjectname=ulizaPlayer&name=livestream01_dvr&repeatable=true',
+            'info_dict': {
+                'id': 'timeshift-disabled/pia/admin?type=normal&playerobjectname=ulizaPlayer&name=livestream01_dvr&repeatable=true',
+                'title': 'livestream01_dvr',
+                'live_status': 'was_live',
+            },
+            'params': {
+                'skip_download': True,
+                'ignore_no_formats_error': True,
+            },
+        },
+        {
+            'url': 'https://player-api.p.uliza.jp/v1/players/uliza_jp_gallery_normal/promotion/admin?type=presentation&name=cookings&targetid=player1',
+            'info_dict': {
+                'id': 'uliza_jp_gallery_normal/promotion/admin?type=presentation&name=cookings&targetid=player1',
+                'title': 'cookings',
+                'live_status': 'not_live',
+            },
+            'params': {
+                'skip_download': True,
+                'ignore_no_formats_error': True,
+            },
+        },
+        {
+            'url': 'https://player-api.p.uliza.jp/v1/players/default-player/pia/admin?type=normal&name=pia_movie_uliza_fix&targetid=ulizahtml5&repeatable=true',
+            'info_dict': {
+                'id': 'default-player/pia/admin?type=normal&name=pia_movie_uliza_fix&targetid=ulizahtml5&repeatable=true',
+                'title': 'pia_movie_uliza_fix',
+                'live_status': 'not_live',
+            },
+            'params': {
+                'skip_download': True,
+                'ignore_no_formats_error': True,
+            },
+        },
+    ]
+
+    def _real_extract(self, url):
+        url, smuggled_data = unsmuggle_url(url, {})
+        video_id = smuggled_data.get('video_id') or self._search_regex(r'&name=([^&]+)', self._match_id(url), 'video id')
+        player_data = self._download_webpage(
+            url,
+            video_id,
+            headers={'Referer': smuggled_data.get('referer') or 'https://player-api.p.uliza.jp/'},
+            note='Fetching player data', errnote='Unable to fetch player data',
+        )
+
+        formats = self._extract_m3u8_formats(
+            self._search_regex(
+                r'["\'](https://vms-api\.p\.uliza\.jp/v1/prog-index\.m3u8[^"\']+)', player_data,
+                'm3u8 url', default=None),
+            video_id, fatal=False)
+        m3u8_type = self._search_regex(
+            r'/hls/(dvr|video)/', traverse_obj(formats, (0, 'url')), 'm3u8 type', default=None)
+        return {
+            'id': video_id,
+            'title': smuggled_data.get('video_title') or video_id,
+            'formats': formats,
+            'live_status': {
+                'video': 'is_live',
+                'dvr': 'was_live',  # short-term archives
+            }.get(m3u8_type, 'not_live'),  # VOD or long-term archives
+            **smuggled_data.get('info_dict', {}),
+        }
 
 
 class PIAULIZAPortalIE(InfoExtractor):
@@ -44,27 +111,16 @@ class PIAULIZAPortalIE(InfoExtractor):
 
         webpage = self._download_webpage(url, video_id)
 
-        player_data = self._download_webpage(
-            self._search_regex(
-                r'<script [^>]*\bsrc="(https://player-api\.p\.uliza\.jp/v1/players/[^"]+)"',
-                webpage, 'player data url'),
-            video_id, headers={'Referer': 'https://ulizaportal.jp/'},
-            note='Fetching player data', errnote='Unable to fetch player data')
-
-        formats = self._extract_m3u8_formats(
-            self._search_regex(
-                r'["\'](https://vms-api\.p\.uliza\.jp/v1/prog-index\.m3u8[^"\']+)', player_data,
-                'm3u8 url', default=None),
-            video_id, fatal=False)
-        m3u8_type = self._search_regex(
-            r'/hls/(dvr|video)/', traverse_obj(formats, (0, 'url')), 'm3u8 type', default=None)
-
-        return {
-            'id': video_id,
-            'title': self._html_extract_title(webpage),
-            'formats': formats,
-            'live_status': {
-                'video': 'is_live',
-                'dvr': 'was_live',  # short-term archives
-            }.get(m3u8_type, 'not_live'),  # VOD or long-term archives
-        }
+        player_data_url = self._search_regex(
+            PIAULIZAPortalAPIIE.SCRIPT_TAG_REGEX_PATTERN,
+            webpage, 'player data url')
+        return self.url_result(
+            smuggle_url(
+                player_data_url,
+                {'video_id': video_id, 'referer': 'https://ulizaportal.jp/', 'info_dict': {
+                    'id': video_id,
+                    'title': self._html_extract_title(webpage),
+                }},
+            ),
+            ie=PIAULIZAPortalAPIIE.ie_key(),
+        )