use jellyfin builtin rating compare

kookxiang · Jan 22, 2023 · 0dfa6b1 · 0dfa6b1
1 parent d8c2916
commit 0dfa6b1
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/Jellyfin.Plugin.Bangumi/Configuration/PluginConfiguration.cs b/Jellyfin.Plugin.Bangumi/Configuration/PluginConfiguration.cs
@@ -16,7 +16,7 @@ public class PluginConfiguration : BasePluginConfiguration
 
     public bool ReportPlaybackStatusToBangumi { get; set; } = true;
 
-    public bool SkipNSFWPlaybackReport { get; set; } = false;
+    public bool SkipNSFWPlaybackReport { get; set; } = true;
 
     public bool ReportManualStatusChangeToBangumi { get; set; } = false;
 

diff --git a/Jellyfin.Plugin.Bangumi/PlaybackScrobbler.cs b/Jellyfin.Plugin.Bangumi/PlaybackScrobbler.cs
@@ -12,23 +12,30 @@
 using MediaBrowser.Controller.Library;
 using MediaBrowser.Controller.Plugins;
 using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Globalization;
 using Microsoft.Extensions.Logging;
 using CollectionType = Jellyfin.Plugin.Bangumi.Model.CollectionType;
 
 namespace Jellyfin.Plugin.Bangumi;
 
 public class PlaybackScrobbler : IServerEntryPoint
 {
+    // https://github.com/jellyfin/jellyfin/blob/master/Emby.Server.Implementations/Localization/Ratings/jp.csv
+    // https://github.com/jellyfin/jellyfin/blob/master/Emby.Server.Implementations/Localization/Ratings/us.csv
+    private const int RatingNSFW = 10;
+
     private static readonly Dictionary<Guid, HashSet<string>> Store = new();
     private readonly BangumiApi _api;
+    private readonly ILocalizationManager _localizationManager;
     private readonly ILogger<PlaybackScrobbler> _log;
 
     private readonly OAuthStore _store;
     private readonly IUserDataManager _userDataManager;
 
-    public PlaybackScrobbler(IUserManager userManager, IUserDataManager userDataManager, OAuthStore store, BangumiApi api, ILogger<PlaybackScrobbler> log)
+    public PlaybackScrobbler(IUserManager userManager, IUserDataManager userDataManager, ILocalizationManager localizationManager, OAuthStore store, BangumiApi api, ILogger<PlaybackScrobbler> log)
     {
         _userDataManager = userDataManager;
+        _localizationManager = localizationManager;
         _store = store;
         _api = api;
         _log = log;
@@ -143,7 +150,17 @@ private async Task ReportPlaybackStatus(BaseItem item, Guid userId, bool played)
                         subjectId = episode.ParentId;
                 }
 
-                if (item.OfficialRating is "X" or "XXX" && Configuration.SkipNSFWPlaybackReport)
+                var ratingLevel = _localizationManager.GetRatingLevel(item.OfficialRating);
+                if (ratingLevel == null)
+                {
+                    foreach (var parent in item.GetParents())
+                    {
+                        ratingLevel = _localizationManager.GetRatingLevel(parent.OfficialRating);
+                        if (ratingLevel != null) break;
+                    }
+                }
+
+                if (ratingLevel >= RatingNSFW && Configuration.SkipNSFWPlaybackReport)
                 {
                     _log.LogInformation("item #{Name} marked as NSFW, skipped", item.Name);
                     return;