From 1efe835c69ffcd7bd7ddee77f859347f818b0fd6 Mon Sep 17 00:00:00 2001
From: JellyBrick <shlee1503@naver.com>
Date: Thu, 19 Oct 2023 05:40:05 +0900
Subject: [PATCH] fix: fixed an issue where only the first 100 songs in a
 playlist were downloaded (#1329)

---
 src/plugins/downloader/back.ts | 28 +++++++++++++++++++---------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/src/plugins/downloader/back.ts b/src/plugins/downloader/back.ts
index 63cd7cde..fbd8a160 100644
--- a/src/plugins/downloader/back.ts
+++ b/src/plugins/downloader/back.ts
@@ -361,7 +361,7 @@ async function iterableStreamToTargetFile(
 
     ffmpeg.setProgress(({ ratio }) => {
       sendFeedback(`Converting: ${Math.floor(ratio * 100)}%`, ratio);
-      increasePlaylistProgress(0.15 + ratio * 0.85);
+      increasePlaylistProgress(0.15 + (ratio * 0.85));
     });
 
     const safeVideoNameWithExtension = `${safeVideoName}.${extension}`;
@@ -468,8 +468,12 @@ export async function downloadPlaylist(givenUrl?: string | URL) {
   console.log(`trying to get playlist ID: '${playlistId}'`);
   sendFeedback('Getting playlist info…');
   let playlist: Playlist;
+  const items: YTNodes.MusicResponsiveListItem[] = [];
   try {
     playlist = await yt.music.getPlaylist(playlistId);
+    if (playlist?.items) {
+      items.push(...playlist.items.as(YTNodes.MusicResponsiveListItem));
+    }
   } catch (error: unknown) {
     sendError(
       Error(
@@ -485,13 +489,6 @@ export async function downloadPlaylist(givenUrl?: string | URL) {
     sendError(new Error('Playlist is empty'));
   }
 
-  const items = playlist.items!.as(YTNodes.MusicResponsiveListItem);
-  if (items.length === 1) {
-    sendFeedback('Playlist has only one item, downloading it directly');
-    await downloadSongFromId(items.at(0)!.id!);
-    return;
-  }
-
   const normalPlaylistTitle = playlist.header?.title?.text;
   const playlistTitle =
     normalPlaylistTitle ??
@@ -502,6 +499,19 @@ export async function downloadPlaylist(givenUrl?: string | URL) {
     'NO_TITLE';
   const isAlbum = !normalPlaylistTitle;
 
+  while (playlist.has_continuation) {
+    playlist = await playlist.getContinuation();
+    if (playlist?.items) {
+      items.push(...playlist.items.as(YTNodes.MusicResponsiveListItem));
+    }
+  }
+
+  if (items.length === 1) {
+    sendFeedback('Playlist has only one item, downloading it directly');
+    await downloadSongFromId(items.at(0)!.id!);
+    return;
+  }
+
   let safePlaylistTitle = filenamify(playlistTitle, { replacement: ' ' });
   if (!is.macOS()) {
     safePlaylistTitle = safePlaylistTitle.normalize('NFC');
@@ -542,7 +552,7 @@ export async function downloadPlaylist(givenUrl?: string | URL) {
 
   const increaseProgress = (itemPercentage: number) => {
     const currentProgress = (counter - 1) / (items.length ?? 1);
-    const newProgress = currentProgress + progressStep * itemPercentage;
+    const newProgress = currentProgress + (progressStep * itemPercentage);
     win.setProgressBar(newProgress);
   };