optimize episode number detection

kookxiang · Aug 26, 2024 · 21cb52c · 21cb52c
1 parent 5147c02
commit 21cb52c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/Jellyfin.Plugin.Bangumi.Test/Episode.cs b/Jellyfin.Plugin.Bangumi.Test/Episode.cs
@@ -382,6 +382,21 @@ public async Task FixEpisodeIndexWithNumberInName()
         Assert.IsNotNull(episodeData, "episode data should not be null");
         Assert.IsNotNull(episodeData.Item, "episode data should not be null");
         Assert.AreEqual(22, episodeData.Item.IndexNumber, "should fix episode index automatically");
+
+        episodeData = await _provider.GetMetadata(new EpisodeInfo
+        {
+            IndexNumber = 0,
+            Path = FakePath.CreateFile("Detective Conan/[SBSUB][CONAN][5][WEBRIP][1080P][HEVC_AAC][CHS_CHT_JP](951D8C84).mkv"),
+            SeriesProviderIds = new Dictionary<string, string>
+            {
+                {
+                    Constants.ProviderName, "899"
+                }
+            }
+        }, _token);
+        Assert.IsNotNull(episodeData, "episode data should not be null");
+        Assert.IsNotNull(episodeData.Item, "episode data should not be null");
+        Assert.AreEqual(5, episodeData.Item.IndexNumber, "should fix episode index automatically");
     }
 
     [TestMethod]

diff --git a/Jellyfin.Plugin.Bangumi/Providers/EpisodeProvider.cs b/Jellyfin.Plugin.Bangumi/Providers/EpisodeProvider.cs
@@ -39,7 +39,8 @@ public class EpisodeProvider(BangumiApi api, ILogger<EpisodeProvider> log, ILibr
         new(@"EP?([\d\.]{2,})", RegexOptions.IgnoreCase),
         new(@"\[([\d\.]{2,})"),
         new(@"#([\d\.]{2,})"),
-        new(@"(\d{2,})")
+        new(@"(\d{2,})"),
+        new(@"\[([\d\.]+)\]")
     };
 
     private static readonly Regex OpeningEpisodeFileNameRegex = new(@"(NC)?OP([^a-zA-Z]|$)");