Support filtering by file size in GetByFilename

ShokoAnime · May 12, 2024 · 7209b1d · 7209b1d
1 parent ee7cf7a
commit 7209b1d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/Shoko.Server/Plex/TVShow/SVR_Episode.cs b/Shoko.Server/Plex/TVShow/SVR_Episode.cs
@@ -1,4 +1,4 @@
-using System.IO;
+using System.IO;
 using Shoko.Models.Plex.TVShow;
 using Shoko.Server.Models;
 using Shoko.Server.Repositories;
@@ -15,7 +15,7 @@ public SVR_Episode(PlexHelper helper)
     }
 
     public SVR_AnimeEpisode AnimeEpisode =>
-        RepoFactory.AnimeEpisode.GetByFilename(Path.GetFileName(Media[0].Part[0].File));
+        RepoFactory.AnimeEpisode.GetByFilename(Path.GetFileName(Media[0].Part[0].File), Media[0].Part[0].Size);
 
     public void Unscrobble()
     {

diff --git a/Shoko.Server/Repositories/Cached/AnimeEpisodeRepository.cs b/Shoko.Server/Repositories/Cached/AnimeEpisodeRepository.cs
@@ -1,4 +1,4 @@
-using System;
+using System;
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
@@ -56,8 +56,9 @@ public SVR_AnimeEpisode GetByAniDBEpisodeID(int epid)
     /// Get the AnimeEpisode 
     /// </summary>
     /// <param name="name">The filename of the anime to search for.</param>
+    /// <param name="size">The size of the file in bytes</param>
     /// <returns>the AnimeEpisode given the file information</returns>
-    public SVR_AnimeEpisode GetByFilename(string name)
+    public SVR_AnimeEpisode GetByFilename(string name, long size = -1)
     {
         if (string.IsNullOrEmpty(name))
         {
@@ -68,6 +69,7 @@ public SVR_AnimeEpisode GetByFilename(string name)
             .Where(v => name.Equals(v?.FilePath?.Split(Path.DirectorySeparatorChar).LastOrDefault(),
                 StringComparison.InvariantCultureIgnoreCase))
             .Select(a => RepoFactory.VideoLocal.GetByID(a.VideoLocalID)).Where(a => a != null)
+            .Where(a => size == -1 || a.FileSize == size)
             .SelectMany(a => GetByHash(a.Hash)).ToArray();
         var ep = eps.FirstOrDefault(a => a.AniDB_Episode.EpisodeType == (int)EpisodeType.Episode);
         return ep ?? eps.FirstOrDefault();