From 78d04e140abbacbe1af41832d23ae115a609987d Mon Sep 17 00:00:00 2001
From: DirtyRacer <DirtyRacer@ya.ru>
Date: Wed, 6 Nov 2024 23:01:26 +0700
Subject: [PATCH] add colletion filter by scene type

---
 .../Configuration/PluginConfiguration.cs             | 12 ++++++++++++
 .../Configuration/configPage.html                    | 11 +++++++++++
 .../ScheduledTasks/AddCollection.cs                  |  8 ++++++++
 3 files changed, 31 insertions(+)

diff --git a/Jellyfin.Plugin.ThePornDB/Configuration/PluginConfiguration.cs b/Jellyfin.Plugin.ThePornDB/Configuration/PluginConfiguration.cs
index 7ee06f5..d543d3a 100644
--- a/Jellyfin.Plugin.ThePornDB/Configuration/PluginConfiguration.cs
+++ b/Jellyfin.Plugin.ThePornDB/Configuration/PluginConfiguration.cs
@@ -19,6 +19,14 @@ public enum TagStyle
         Disabled = 2,
     }
 
+    public enum CollectionType
+    {
+        Scene = 0,
+        Movie = 1,
+        JAV = 2,
+        All = 3,
+    }
+
     public enum StudioStyle
     {
         Site = 0,
@@ -74,6 +82,8 @@ public PluginConfiguration()
 
             this.CollectionMinSize = 0;
             this.AddCollectionToCollections = true;
+            this.CollectionType = CollectionType.All;
+
             this.StudioStyle = StudioStyle.All;
 
             this.UseCustomTitle = false;
@@ -113,6 +123,8 @@ public PluginConfiguration()
 
         public bool AddCollectionToCollections { get; set; }
 
+        public CollectionType CollectionType { get; set; }
+
         public StudioStyle StudioStyle { get; set; }
 
         public bool UseCustomTitle { get; set; }
diff --git a/Jellyfin.Plugin.ThePornDB/Configuration/configPage.html b/Jellyfin.Plugin.ThePornDB/Configuration/configPage.html
index eaf840f..e39e978 100644
--- a/Jellyfin.Plugin.ThePornDB/Configuration/configPage.html
+++ b/Jellyfin.Plugin.ThePornDB/Configuration/configPage.html
@@ -56,6 +56,15 @@ <h1>ThePornDB</h1>
                             <span>Add Collection to Collections</span>
                         </label>
                     </div>
+                    <div class="selectContainer">
+                        <label class="selectLabel" for="CollectionType">Collection Type</label>
+                        <select is="emby-select" id="CollectionType" name="CollectionType" class="emby-select-withcolor emby-select">
+                            <option value="All">All</option>
+                            <option value="Scene">Scene</option>
+                            <option value="Movie">Movie</option>
+                            <option value="JAV">JAV</option>
+                        </select>
+                    </div>
                     <div class="selectContainer">
                         <label class="selectLabel" for="StudioStyle">Studio Style</label>
                         <select is="emby-select" id="StudioStyle" name="StudioStyle" class="emby-select-withcolor emby-select">
@@ -154,6 +163,7 @@ <h1>ThePornDB</h1>
                     $('#AddCollectionOnSite').prop('checked', config.AddCollectionOnSite);
                     $('#CollectionMinSize').val(config.CollectionMinSize).change();
                     $('#AddCollectionToCollections').prop('checked', config.AddCollectionToCollections);
+                    $('#CollectionType').val(config.CollectionType).change();
                     $('#StudioStyle').val(config.StudioStyle).change();
                     $('#UseCustomTitle').prop('checked', config.UseCustomTitle);
                     $('#CustomTitle').val(config.CustomTitle).change();
@@ -181,6 +191,7 @@ <h1>ThePornDB</h1>
                     config.AddCollectionOnSite = $('#AddCollectionOnSite').prop('checked');
                     config.CollectionMinSize = $('#CollectionMinSize').val();
                     config.AddCollectionToCollections = $('#AddCollectionToCollections').prop('checked');
+                    config.CollectionType = $('#CollectionType').val();
                     config.StudioStyle = $('#StudioStyle').val();
                     config.UseCustomTitle = $('#UseCustomTitle').prop('checked');
                     config.CustomTitle = $('#CustomTitle').val();
diff --git a/Jellyfin.Plugin.ThePornDB/ScheduledTasks/AddCollection.cs b/Jellyfin.Plugin.ThePornDB/ScheduledTasks/AddCollection.cs
index 781a87d..d0fb341 100644
--- a/Jellyfin.Plugin.ThePornDB/ScheduledTasks/AddCollection.cs
+++ b/Jellyfin.Plugin.ThePornDB/ScheduledTasks/AddCollection.cs
@@ -7,6 +7,9 @@
 using MediaBrowser.Controller.Entities;
 using MediaBrowser.Controller.Library;
 using MediaBrowser.Model.Tasks;
+using ThePornDB.Configuration;
+using ThePornDB.Helpers;
+using ThePornDB.Providers;
 
 #if __EMBY__
 #else
@@ -45,6 +48,11 @@ public async Task ExecuteAsync(IProgress<double> progress, CancellationToken can
             progress?.Report(0);
 
             var items = this.libraryManager.GetItemList(new InternalItemsQuery()).Where(o => o.ProviderIds.ContainsKey(Plugin.Instance.Name));
+            if (Plugin.Instance.Configuration.CollectionType != CollectionType.All)
+            {
+                var (prefixID, _, _) = Base.GetSettings((SceneType)Plugin.Instance.Configuration.CollectionType);
+                items = items.Where(o => o.ProviderIds[Plugin.Instance.Name].StartsWith(prefixID, StringComparison.OrdinalIgnoreCase));
+            }
 
             if (Plugin.Instance.Configuration.UseUnmatchedTag)
             {