Fetch video count in playlist metadata (#801)

Tyrrrz · Jul 25, 2024 · d4b1422 · d4b1422
1 parent 9f1bffd
commit d4b1422
Show file tree

Hide file tree

Showing 6 changed files with 56 additions and 1 deletion.
diff --git a/YoutubeExplode.Tests/PlaylistSpecs.cs b/YoutubeExplode.Tests/PlaylistSpecs.cs
@@ -31,6 +31,7 @@ public async Task I_can_get_the_metadata_of_a_playlist()
         playlist
             .Description.Should()
             .Contain("Digital Analytics Fundamentals course on Analytics Academy");
+        playlist.VideosCount.Should().Be(22);
         playlist.Thumbnails.Should().NotBeEmpty();
     }
 

diff --git a/YoutubeExplode/Bridge/IPlaylistData.cs b/YoutubeExplode/Bridge/IPlaylistData.cs
@@ -12,5 +12,7 @@ internal interface IPlaylistData
 
     string? Description { get; }
 
+    int? VideosCount { get; }
+
     IReadOnlyList<ThumbnailData> Thumbnails { get; }
 }
diff --git a/YoutubeExplode/Bridge/PlaylistBrowseResponse.cs b/YoutubeExplode/Bridge/PlaylistBrowseResponse.cs
@@ -101,6 +101,28 @@ internal partial class PlaylistBrowseResponse(JsonElement content) : IPlaylistDa
             ?.GetPropertyOrNull("value")
             ?.GetStringOrNull();
 
+    [Lazy]
+    public int? VideosCount =>
+        SidebarPrimary
+            ?.GetPropertyOrNull("stats")
+            ?.EnumerateArrayOrNull()
+            ?.FirstOrNull()
+            ?.GetPropertyOrNull("runs")
+            ?.EnumerateArrayOrNull()
+            ?.FirstOrNull()
+            ?.GetPropertyOrNull("text")
+            ?.GetStringOrNull()
+            ?.ParseIntOrNull()
+        ?? SidebarPrimary
+            ?.GetPropertyOrNull("stats")
+            ?.EnumerateArrayOrNull()
+            ?.FirstOrNull()
+            ?.GetPropertyOrNull("simpleText")
+            ?.GetStringOrNull()
+            ?.Split(' ')
+            ?.FirstOrDefault()
+            ?.ParseIntOrNull();
+
     [Lazy]
     public IReadOnlyList<ThumbnailData> Thumbnails =>
         SidebarPrimary

diff --git a/YoutubeExplode/Bridge/PlaylistNextResponse.cs b/YoutubeExplode/Bridge/PlaylistNextResponse.cs
@@ -34,6 +34,25 @@ internal partial class PlaylistNextResponse(JsonElement content) : IPlaylistData
 
     public string? Description => null;
 
+    [Lazy]
+    public int? VideosCount =>
+        ContentRoot
+            ?.GetPropertyOrNull("totalVideosText")
+            ?.GetPropertyOrNull("runs")
+            ?.EnumerateArrayOrNull()
+            ?.FirstOrNull()
+            ?.GetPropertyOrNull("text")
+            ?.GetStringOrNull()
+            ?.ParseIntOrNull()
+        ?? ContentRoot
+            ?.GetPropertyOrNull("videoCountText")
+            ?.GetPropertyOrNull("runs")
+            ?.EnumerateArrayOrNull()
+            ?.ElementAtOrNull(2)
+            ?.GetPropertyOrNull("text")
+            ?.GetStringOrNull()
+            ?.ParseIntOrNull();
+
     [Lazy]
     public IReadOnlyList<ThumbnailData> Thumbnails => Videos.FirstOrDefault()?.Thumbnails ?? [];
 

diff --git a/YoutubeExplode/Playlists/Playlist.cs b/YoutubeExplode/Playlists/Playlist.cs
@@ -12,6 +12,7 @@ public class Playlist(
     string title,
     Author? author,
     string description,
+    int? videosCount,
     IReadOnlyList<Thumbnail> thumbnails
 ) : IPlaylist
 {
@@ -32,6 +33,14 @@ IReadOnlyList<Thumbnail> thumbnails
     /// </summary>
     public string Description { get; } = description;
 
+    /// <summary>
+    /// Count of total videos.
+    /// </summary>
+    /// <remarks>
+    /// May be null in case of playlists with infinite videos (e.g. mixes).
+    /// </remarks>
+    public int? VideosCount { get; } = videosCount;
+
     /// <inheritdoc />
     public IReadOnlyList<Thumbnail> Thumbnails { get; } = thumbnails;
 

diff --git a/YoutubeExplode/Playlists/PlaylistClient.cs b/YoutubeExplode/Playlists/PlaylistClient.cs
@@ -42,6 +42,8 @@ channelId is not null && channelTitle is not null
         // System playlists have no description
         var description = response.Description ?? "";
 
+        var videosCount = response.VideosCount;
+
         var thumbnails = response
             .Thumbnails.Select(t =>
             {
@@ -63,7 +65,7 @@ channelId is not null && channelTitle is not null
             })
             .ToArray();
 
-        return new Playlist(playlistId, title, author, description, thumbnails);
+        return new Playlist(playlistId, title, author, description, videosCount, thumbnails);
     }
 
     /// <summary>