From 2f14c2b4d03a93c30be5e415aee956f4b085045b Mon Sep 17 00:00:00 2001
From: grqx_wsl <173253225+grqx@users.noreply.github.com>
Date: Mon, 13 Jan 2025 22:11:56 +1300
Subject: [PATCH] return url results in playlists

---
 yt_dlp/extractor/nzonscreen.py | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/yt_dlp/extractor/nzonscreen.py b/yt_dlp/extractor/nzonscreen.py
index e26a43e1c2..c26d4a3786 100644
--- a/yt_dlp/extractor/nzonscreen.py
+++ b/yt_dlp/extractor/nzonscreen.py
@@ -2,8 +2,10 @@ from .common import InfoExtractor
 from ..utils import (
     float_or_none,
     int_or_none,
+    smuggle_url,
     strip_or_none,
     traverse_obj,
+    unsmuggle_url,
     url_or_none,
     urljoin,
 )
@@ -144,6 +146,8 @@ class NZOnScreenIE(InfoExtractor):
         }
 
     def _real_extract(self, url):
+        url, smuggled_data = unsmuggle_url(url, {})
+        ep_idx = smuggled_data.get('ep')
         video_id = self._match_id(url)
         webpage = self._download_webpage(url, video_id)
         title = strip_or_none((
@@ -152,10 +156,31 @@ class NZOnScreenIE(InfoExtractor):
         playlist = self._download_json(
             f'https://www.nzonscreen.com/html5/video_data/{video_id}', video_id,
             'Downloading media data')
+        playlist_len = len(playlist)
 
-        if self._yes_playlist(video_id, traverse_obj(playlist, (0, 'id'))):
+        if ep_idx is None and playlist_len > 1 and self._yes_playlist(video_id, traverse_obj(playlist, (0, 'id'))):
             return self.playlist_result(
-                [self._extract_from_api_resp(vid_info, len(playlist) == 1, title, video_id) for vid_info in playlist],
+                [self.url_result(smuggle_url(url, {'ep': idx}), NZOnScreenIE.ie_key()) for idx in range(playlist_len)],
                 playlist_id=video_id, playlist_title=title)
 
-        return self._extract_from_api_resp(playlist[0], len(playlist) == 1, title, video_id)
+        vid_info = playlist[ep_idx or 0]
+        return {
+            'alt_title': title if playlist_len == 1 else None,
+            'display_id': video_id,
+            'http_headers': {
+                'Referer': 'https://www.nzonscreen.com/',
+                'Origin': 'https://www.nzonscreen.com/',
+            },
+            'subtitles': {'en': [{
+                'url': traverse_obj(vid_info, ('h264', 'caption_url', {urljoin('https://www.nzonscreen.com')})),
+                'ext': 'vtt',
+            }]},
+            'formats': self._extract_formats(vid_info),
+            **traverse_obj(vid_info, {
+                'id': 'uuid',
+                'title': ('label', {strip_or_none}),
+                'description': ('description', {strip_or_none}),
+                'thumbnail': ('thumbnail', 'path'),
+                'duration': ('duration', {float_or_none}),
+            }),
+        }