Get the first chunk in advance

so that the returned `url` "works" in mpv etc

i can download the first one as part of the normal self._extract_m3u8_formats, and pass the url as the first chunk of my chunking set up
then the chunks stuff still works, and i can return an url that will play in mpv
...except that it will be 5 minutes ahead because it's a "live"stream
oh well its better than nothing lmao

1 files changed, 18 insertions, 10 deletions

diff --git a/yt_dlp_plugins/extractor/radiko_hacks.py b/yt_dlp_plugins/extractor/radiko_hacks.py
index 418aa70..1d2d1df 100644
--- a/yt_dlp_plugins/extractor/radiko_hacks.py
+++ b/yt_dlp_plugins/extractor/radiko_hacks.py
@@ -10,7 +10,7 @@ from yt_dlp.utils import (
 
 # "hacks" as in great jank/schizo shit that works anyway
 
-def _generate_as_live_fragments(self, playlist_base_url, start_at, end_at, domain, headers={}):
+def _generate_as_live_fragments(self, playlist_base_url, start_at, end_at, domain, headers={}, first_chunk=None):
 	playlist = []
 	chunk_length = 300  # max the api allows
 
@@ -26,27 +26,35 @@ def _generate_as_live_fragments(self, playlist_base_url, start_at, end_at, domai
 			"l": chunk_length,
 		})
 
-		chunk_fragments, real_chunk_length = _get_chunk_playlist(self, chunk_url, domain, chunk_num, headers)
+		chunk_fragments, real_chunk_length = _get_chunk_playlist(self, chunk_url, domain, chunk_num, headers, first_chunk)
 
 		cursor += round(real_chunk_length)
 		chunk_num += 1
+		first_chunk = None
 
 		for frag in chunk_fragments:
 			yield frag
 
 
-def _get_chunk_playlist(self, chunk_url, src_id, chunk_num, headers={}):
+def _get_chunk_playlist(self, chunk_url, src_id, chunk_num, headers={}, first_chunk=None):
 	EXTINF_duration = re.compile(r"^#EXTINF:([\d.]+),", flags=re.MULTILINE)
 
 	playlist = ""
 	chunk_id = join_nonempty(src_id, chunk_num)
-	base_format = self._extract_m3u8_formats(
-		chunk_url, chunk_id, fatal=False, headers=headers,
-#		note=f"Preparing {src_id} chunk {chunk_num}"
-		note=False,
-		errnote=f"Failed to get {src_id} chunk {chunk_num} base format",
-	)
-	m3u8_url = traverse_obj(base_format, (..., "url",), get_all=False)
+
+	if first_chunk:
+		m3u8_url = first_chunk
+	else:
+		self.write_debug(f"Preparing {src_id} chunk {chunk_num}")
+		base_formats = self._extract_m3u8_formats(
+			chunk_url, chunk_id, fatal=False, headers=headers,
+	#		note=f"Preparing {src_id} chunk {chunk_num}"
+			note=False,
+			errnote=f"Failed to get {src_id} chunk {chunk_num} base format",
+		)
+		m3u8_url = traverse_obj(base_formats, (..., "url",), get_all=False)
+
+	self.write_debug(f"Getting {src_id} chunk {chunk_num} playlist")
 	playlist = self._download_webpage(m3u8_url, chunk_id, note=False, errnote=f"Failed to get {src_id} chunk {chunk_num} playlist")
 	#note=f"Getting {src_id} chunk {chunk_num} fragments")