From 7d8c19b9212d4136b420ef3604bbfe1d6180446a Mon Sep 17 00:00:00 2001
From: xjasonlyu <xjasonlyu@gmail.com>
Date: Tue, 28 Jun 2022 13:31:10 +0800
Subject: [PATCH] Add collection support

---
 .../Configuration/PluginConfiguration.cs         |  6 ++++++
 .../Configuration/configPage.html                |  9 +++++++++
 ...LoggerExtensions.cs => JellyfinExtensions.cs} | 16 +++++++++++++++-
 .../Providers/MovieProvider.cs                   |  3 +++
 4 files changed, 33 insertions(+), 1 deletion(-)
 rename Jellyfin.Plugin.JavTube/Extensions/{LoggerExtensions.cs => JellyfinExtensions.cs} (71%)

diff --git a/Jellyfin.Plugin.JavTube/Configuration/PluginConfiguration.cs b/Jellyfin.Plugin.JavTube/Configuration/PluginConfiguration.cs
index cba9705..6ca438f 100644
--- a/Jellyfin.Plugin.JavTube/Configuration/PluginConfiguration.cs
+++ b/Jellyfin.Plugin.JavTube/Configuration/PluginConfiguration.cs
@@ -18,6 +18,12 @@ public enum TranslationEngine
 
 public class PluginConfiguration : BasePluginConfiguration
 {
+    #region Collection
+
+    public bool EnableCollection { get; set; } = false;
+
+    #endregion
+
     #region Rating
 
     public bool EnableRating { get; set; } = true;
diff --git a/Jellyfin.Plugin.JavTube/Configuration/configPage.html b/Jellyfin.Plugin.JavTube/Configuration/configPage.html
index 0759517..c66279f 100644
--- a/Jellyfin.Plugin.JavTube/Configuration/configPage.html
+++ b/Jellyfin.Plugin.JavTube/Configuration/configPage.html
@@ -35,6 +35,13 @@
                     <div class="fieldDescription">Access Token of JavTube Server.</div>
                 </div>
 
+                <div class="checkboxContainer checkboxContainer-withDescripton">
+                    <label class="emby-checkbox-label">
+                        <input id="chkEnableCollection" is="emby-checkbox" name="chkEnableCollection" type="checkbox"/>
+                        <span>Enable Collection</span>
+                    </label>
+                </div>
+
                 <div class="checkboxContainer checkboxContainer-withDescripton">
                     <label class="emby-checkbox-label">
                         <input id="chkEnableRating" is="emby-checkbox" name="chkEnableRating" type="checkbox"/>
@@ -143,6 +150,7 @@
             ApiClient.getPluginConfiguration(JavTubePluginConfig.pluginUniqueId).then(function (config) {
                 $('#txtServer', page).val(config.Server).change();
                 $('#txtToken', page).val(config.Token).change();
+                page.querySelector('#chkEnableCollection').checked = config.EnableCollection;
                 page.querySelector('#chkEnableRating').checked = config.EnableRating;
                 page.querySelector('#chkEnableTrailer').checked = config.EnableTrailer;
                 $('#sltTranslationMode', page).val(config.TranslationMode).change();
@@ -161,6 +169,7 @@
             ApiClient.getPluginConfiguration(JavTubePluginConfig.pluginUniqueId).then(function (config) {
                 config.Server = $('#txtServer', form).val();
                 config.Token = $('#txtToken', form).val();
+                config.EnableCollection = $('#chkEnableCollection').prop('checked');
                 config.EnableRating = $('#chkEnableRating').prop('checked');
                 config.EnableTrailer = $('#chkEnableTrailer').prop('checked');
                 config.TranslationMode = $('#sltTranslationMode', form).val();
diff --git a/Jellyfin.Plugin.JavTube/Extensions/LoggerExtensions.cs b/Jellyfin.Plugin.JavTube/Extensions/JellyfinExtensions.cs
similarity index 71%
rename from Jellyfin.Plugin.JavTube/Extensions/LoggerExtensions.cs
rename to Jellyfin.Plugin.JavTube/Extensions/JellyfinExtensions.cs
index 333e01a..75dc2bb 100644
--- a/Jellyfin.Plugin.JavTube/Extensions/LoggerExtensions.cs
+++ b/Jellyfin.Plugin.JavTube/Extensions/JellyfinExtensions.cs
@@ -1,12 +1,24 @@
 #if !__EMBY__
 #pragma warning disable CA2254
 
+using MediaBrowser.Controller.Entities.Movies;
 using Microsoft.Extensions.Logging;
 
 namespace Jellyfin.Plugin.JavTube.Extensions;
 
-internal static class LoggerExtensions
+internal static class JellyfinExtensions
 {
+    #region MovieExtensions
+
+    public static void AddCollection(this Movie movie, string name)
+    {
+        movie.CollectionName = name;
+    }
+
+    #endregion
+
+    #region LoggerExtensions
+
     public static void Debug(this ILogger logger, string message, params object[] args)
     {
         logger.LogDebug(message, args);
@@ -26,6 +38,8 @@ internal static class LoggerExtensions
     {
         logger.LogError(message, args);
     }
+
+    #endregion
 }
 
 #pragma warning restore CA2254
diff --git a/Jellyfin.Plugin.JavTube/Providers/MovieProvider.cs b/Jellyfin.Plugin.JavTube/Providers/MovieProvider.cs
index 008899c..c27ecab 100644
--- a/Jellyfin.Plugin.JavTube/Providers/MovieProvider.cs
+++ b/Jellyfin.Plugin.JavTube/Providers/MovieProvider.cs
@@ -91,6 +91,9 @@ public class MovieProvider : BaseProvider, IRemoteMetadataProvider<Movie, MovieI
         if (Configuration.EnableRating)
             result.Item.CommunityRating = m.Score > 0 ? (float)Math.Round(m.Score * 2, 1) : null;
 
+        if (Configuration.EnableCollection && !string.IsNullOrWhiteSpace(m.Series))
+            result.Item.AddCollection(m.Series);
+
         // Add Studio.
         if (!string.IsNullOrWhiteSpace(m.Maker))
             result.Item.AddStudio(m.Maker);