fix: better episode selection logic in playlist for videos

ShokoAnime · Nov 1, 2024 · 31e56e9 · 31e56e9
1 parent 94045ad
commit 31e56e9
Showing 1 changed file with 20 additions and 5 deletions.
diff --git a/Shoko.Server/Services/GeneratedPlaylistService.cs b/Shoko.Server/Services/GeneratedPlaylistService.cs
@@ -348,13 +348,28 @@ public bool TryParsePlaylist(string[] items, out IReadOnlyList<(IReadOnlyList<IS
         yield return ([episode], videos);
     }
 
+    private static readonly EpisodeType[] _videoToEpisodeGroupPreference = [EpisodeType.Episode, EpisodeType.Special, EpisodeType.Other, EpisodeType.Credits, EpisodeType.Trailer, EpisodeType.Parody];
+
     private IEnumerable<(IReadOnlyList<IShokoEpisode> episodes, IReadOnlyList<IVideo> videos)> GetListForVideo(IVideo video)
     {
-        var episode = video.Episodes
-            .OrderBy(episode => episode.Type)
-            .ThenBy(episode => episode.EpisodeNumber)
-            .FirstOrDefault();
-        return episode is not null ? [([episode], [video])] : [];
+        var crossReferences = video.CrossReferences;
+        var seriesOrder = crossReferences
+            .OrderBy(xref => xref.Order)
+            .Select(xref => xref.AnidbAnimeID)
+            .Distinct()
+            .ToArray();
+        var episodes = crossReferences
+            .DistinctBy(xref => xref.AnidbEpisodeID)
+            .Select(xref => (xref, episode: xref.ShokoEpisode!))
+            .Where(tuple => tuple.episode is not null)
+            .GroupBy(tuple => tuple.episode.Type)
+            .OrderBy(groupBy => Array.IndexOf(_videoToEpisodeGroupPreference, groupBy.Key))
+            .First()
+            .OrderBy(tuple => Array.IndexOf(seriesOrder, tuple.xref.AnidbAnimeID))
+            .ThenBy(tuple => tuple.episode.EpisodeNumber)
+            .Select(tuple => tuple.episode)
+            .ToList() as IReadOnlyList<IShokoEpisode>;
+        return episodes is { Count: > 0 } ? [(episodes, [video])] : [];
     }
 
     public FileStreamResult GeneratePlaylist(