From 94d6358b232e5e4bb352544d210817b64eb9e4b4 Mon Sep 17 00:00:00 2001
From: Mikal Stordal <mikalstordal@gmail.com>
Date: Sat, 14 Dec 2024 19:36:49 +0100
Subject: [PATCH] feat: expose direct props to get tmdb entities through the
 abstraction

---
 .../DataModels/Shoko/IShokoEpisode.cs         | 10 ++++
 .../DataModels/Shoko/IShokoSeries.cs          | 12 ++++-
 .../Shoko.Plugin.Abstractions.csproj          |  2 +-
 Shoko.Server/Models/SVR_AnimeEpisode.cs       | 46 ++++++++++---------
 Shoko.Server/Models/SVR_AnimeSeries.cs        | 10 ++--
 5 files changed, 54 insertions(+), 26 deletions(-)

diff --git a/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoEpisode.cs b/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoEpisode.cs
index 09a962a5f..d91f508bc 100644
--- a/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoEpisode.cs
+++ b/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoEpisode.cs
@@ -27,6 +27,16 @@ public interface IShokoEpisode : IEpisode
     /// </summary>
     IEpisode AnidbEpisode { get; }
 
+    /// <summary>
+    /// A direct link to all tmdb episodes linked to the shoko episode.
+    /// </summary>
+    IReadOnlyList<IEpisode> TmdbEpisodes { get; }
+
+    /// <summary>
+    /// A direct link to all tmdb movies linked to the shoko episode.
+    /// </summary>
+    IReadOnlyList<IMovie> TmdbMovies { get; }
+
     /// <summary>
     /// All episodes linked to this shoko episode.
     /// </summary>
diff --git a/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoSeries.cs b/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoSeries.cs
index d09a60e63..4e73aa9f7 100644
--- a/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoSeries.cs
+++ b/Shoko.Plugin.Abstractions/DataModels/Shoko/IShokoSeries.cs
@@ -23,10 +23,20 @@ public interface IShokoSeries : ISeries
     int TopLevelGroupID { get; }
 
     /// <summary>
-    /// A direct link to the anidb series metadata.
+    /// A direct link to the anidb anime metadata.
     /// </summary>
     ISeries AnidbAnime { get; }
 
+    /// <summary>
+    /// A direct link to all tmdb shows linked to the shoko series.
+    /// </summary>
+    IReadOnlyList<ISeries> TmdbShows { get; }
+
+    /// <summary>
+    /// A direct link to all tmdb movies linked to the shoko series.
+    /// </summary>
+    IReadOnlyList<IMovie> TmdbMovies { get; }
+
     /// <summary>
     /// All series linked to this shoko series.
     /// </summary>
diff --git a/Shoko.Plugin.Abstractions/Shoko.Plugin.Abstractions.csproj b/Shoko.Plugin.Abstractions/Shoko.Plugin.Abstractions.csproj
index b605d43cc..d80d20b0e 100644
--- a/Shoko.Plugin.Abstractions/Shoko.Plugin.Abstractions.csproj
+++ b/Shoko.Plugin.Abstractions/Shoko.Plugin.Abstractions.csproj
@@ -13,7 +13,7 @@
         <RepositoryUrl>https://github.com/ShokoAnime/ShokoServer</RepositoryUrl>
         <PackageTags>plugins, shoko, anime, metadata, tagging</PackageTags>
         <PackageReleaseNotes>Renamer Rewrite</PackageReleaseNotes>
-        <Version>4.1.0-beta2</Version>
+        <Version>4.1.0-beta3</Version>
         <Configurations>Debug;Release;Benchmarks</Configurations>
         <Platforms>AnyCPU;x64</Platforms>
         <IncludeSourceRevisionInInformationalVersion>false</IncludeSourceRevisionInInformationalVersion>
diff --git a/Shoko.Server/Models/SVR_AnimeEpisode.cs b/Shoko.Server/Models/SVR_AnimeEpisode.cs
index 705672d2f..4a60fe2e3 100644
--- a/Shoko.Server/Models/SVR_AnimeEpisode.cs
+++ b/Shoko.Server/Models/SVR_AnimeEpisode.cs
@@ -330,6 +330,31 @@ IReadOnlyList<AnimeTitle> IWithTitles.Titles
 
     IReadOnlyList<IShokoEpisode> IEpisode.ShokoEpisodes => [this];
 
+    IReadOnlyList<IVideoCrossReference> IEpisode.CrossReferences =>
+        RepoFactory.CrossRef_File_Episode.GetByEpisodeID(AniDB_EpisodeID);
+
+    IReadOnlyList<IVideo> IEpisode.VideoList =>
+        RepoFactory.CrossRef_File_Episode.GetByEpisodeID(AniDB_EpisodeID)
+            .DistinctBy(xref => xref.Hash)
+            .Select(xref => xref.VideoLocal)
+            .WhereNotNull()
+            .ToList();
+
+    #endregion
+
+    #region IShokoEpisode Implementation
+
+    int IShokoEpisode.AnidbEpisodeID => AniDB_EpisodeID;
+
+    IShokoSeries? IShokoEpisode.Series => AnimeSeries;
+
+    IEpisode IShokoEpisode.AnidbEpisode => AniDB_Episode ??
+        throw new NullReferenceException($"Unable to find AniDB Episode {AniDB_EpisodeID} for AnimeEpisode {AnimeEpisodeID}");
+
+    IReadOnlyList<IEpisode> IShokoEpisode.TmdbEpisodes => TmdbEpisodes;
+
+    IReadOnlyList<IMovie> IShokoEpisode.TmdbMovies => TmdbMovies;
+
     IReadOnlyList<IEpisode> IShokoEpisode.LinkedEpisodes
     {
         get
@@ -362,26 +387,5 @@ IReadOnlyList<IMovie> IShokoEpisode.LinkedMovies
         }
     }
 
-    IReadOnlyList<IVideoCrossReference> IEpisode.CrossReferences =>
-        RepoFactory.CrossRef_File_Episode.GetByEpisodeID(AniDB_EpisodeID);
-
-    IReadOnlyList<IVideo> IEpisode.VideoList =>
-        RepoFactory.CrossRef_File_Episode.GetByEpisodeID(AniDB_EpisodeID)
-            .DistinctBy(xref => xref.Hash)
-            .Select(xref => xref.VideoLocal)
-            .WhereNotNull()
-            .ToList();
-
-    #endregion
-
-    #region IShokoEpisode Implementation
-
-    int IShokoEpisode.AnidbEpisodeID => AniDB_EpisodeID;
-
-    IShokoSeries? IShokoEpisode.Series => AnimeSeries;
-
-    IEpisode IShokoEpisode.AnidbEpisode => AniDB_Episode ??
-        throw new NullReferenceException($"Unable to find AniDB Episode {AniDB_EpisodeID} for AnimeEpisode {AnimeEpisodeID}");
-
     #endregion
 }
diff --git a/Shoko.Server/Models/SVR_AnimeSeries.cs b/Shoko.Server/Models/SVR_AnimeSeries.cs
index bce80c171..8a9471601 100644
--- a/Shoko.Server/Models/SVR_AnimeSeries.cs
+++ b/Shoko.Server/Models/SVR_AnimeSeries.cs
@@ -648,15 +648,18 @@ public override string ToString()
 
     int IShokoSeries.TopLevelGroupID => TopLevelAnimeGroup.AnimeGroupID;
 
-    ISeries IShokoSeries.AnidbAnime => AniDB_Anime ??
-        throw new NullReferenceException($"Unable to find AniDB anime with id {AniDB_ID} in IShokoSeries.AnidbAnime");
-
     IShokoGroup IShokoSeries.ParentGroup => AnimeGroup;
 
     IShokoGroup IShokoSeries.TopLevelGroup => TopLevelAnimeGroup;
 
     IReadOnlyList<IShokoGroup> IShokoSeries.AllParentGroups => AllGroupsAbove;
 
+    ISeries IShokoSeries.AnidbAnime => AniDB_Anime ??
+        throw new NullReferenceException($"Unable to find AniDB anime with id {AniDB_ID} in IShokoSeries.AnidbAnime");
+
+    IReadOnlyList<ISeries> IShokoSeries.TmdbShows => TmdbShows;
+
+    IReadOnlyList<IMovie> IShokoSeries.TmdbMovies => TmdbMovies;
 
     IReadOnlyList<ISeries> IShokoSeries.LinkedSeries
     {
@@ -675,6 +678,7 @@ IReadOnlyList<ISeries> IShokoSeries.LinkedSeries
             return seriesList;
         }
     }
+
     IReadOnlyList<IMovie> IShokoSeries.LinkedMovies
     {
         get