From 41be79eeb1c8783b78b011d7625da7cdd204a064 Mon Sep 17 00:00:00 2001
From: scampower3 <81431263+scampower3@users.noreply.github.com>
Date: Wed, 22 May 2024 21:22:11 +0800
Subject: [PATCH] Adds a option to include specials for missing episode
 provider (#144)

---
 Jellyfin.Plugin.Tvdb/Configuration/PluginConfiguration.cs  | 5 +++++
 Jellyfin.Plugin.Tvdb/Configuration/config.html             | 6 ++++++
 .../Providers/TvdbMissingEpisodeProvider.cs                | 7 +++++++
 3 files changed, 18 insertions(+)

diff --git a/Jellyfin.Plugin.Tvdb/Configuration/PluginConfiguration.cs b/Jellyfin.Plugin.Tvdb/Configuration/PluginConfiguration.cs
index 2c475bb..e2630fe 100644
--- a/Jellyfin.Plugin.Tvdb/Configuration/PluginConfiguration.cs
+++ b/Jellyfin.Plugin.Tvdb/Configuration/PluginConfiguration.cs
@@ -54,5 +54,10 @@ public int MetadataUpdateInHours
         /// Gets or sets the fallback languages.
         /// </summary>
         public string FallbackLanguages { get; set; } = string.Empty;
+
+        /// <summary>
+        /// Gets or sets a value indicating whether to include missing specials.
+        /// </summary>
+        public bool IncludeMissingSpecials { get; set; } = true;
     }
 }
diff --git a/Jellyfin.Plugin.Tvdb/Configuration/config.html b/Jellyfin.Plugin.Tvdb/Configuration/config.html
index 5e7dbc2..732dfa0 100644
--- a/Jellyfin.Plugin.Tvdb/Configuration/config.html
+++ b/Jellyfin.Plugin.Tvdb/Configuration/config.html
@@ -49,6 +49,10 @@ <h2 class="sectionTitle">TheTVDB Settings:</h2>
                             If the preferred metadata language is not available, the plugin will attempt to retrieve metadata in these languages (in the order listed). Separate language codes with commas (e.g., en, fr, de, ja).
                         </div>
                     </div>
+                    <label class="checkboxContainer">
+                        <input is="emby-checkbox" type="checkbox" id="includeMissingSpecials" />
+                        <span>Include missing specials</span>
+                    </label>
                     <br />
                     <div>
                         <button is="emby-button" type="submit" data-theme="b" class="raised button-submit block">
@@ -72,6 +76,7 @@ <h2 class="sectionTitle">TheTVDB Settings:</h2>
                     document.getElementById('cacheDurationInDays').value = config.CacheDurationInDays;
                     document.getElementById('metadataUpdateInHours').value = config.MetadataUpdateInHours;
                     document.getElementById('fallbackLanguages').value = config.FallbackLanguages;
+                    document.getElementById('includeMissingSpecials').checked = config.IncludeMissingSpecials || true;
                     Dashboard.hideLoadingMsg();
                 });
             },
@@ -85,6 +90,7 @@ <h2 class="sectionTitle">TheTVDB Settings:</h2>
                     config.CacheDurationInDays = document.getElementById('cacheDurationInDays').value;
                     config.MetadataUpdateInHours = document.getElementById('metadataUpdateInHours').value;
                     config.FallbackLanguages = document.getElementById('fallbackLanguages').value;
+                    config.IncludeMissingSpecials = document.getElementById('includeMissingSpecials').checked;
                     ApiClient.updatePluginConfiguration(TvdbPluginConfiguration.uniquePluginId, config).then(function (result) {
                         Dashboard.processPluginConfigurationUpdateResult(result);
                     });
diff --git a/Jellyfin.Plugin.Tvdb/Providers/TvdbMissingEpisodeProvider.cs b/Jellyfin.Plugin.Tvdb/Providers/TvdbMissingEpisodeProvider.cs
index 0559eb0..0a89efd 100644
--- a/Jellyfin.Plugin.Tvdb/Providers/TvdbMissingEpisodeProvider.cs
+++ b/Jellyfin.Plugin.Tvdb/Providers/TvdbMissingEpisodeProvider.cs
@@ -62,6 +62,8 @@ public TvdbMissingEpisodeProvider(
             _logger = logger;
         }
 
+        private static bool IncludeMissingSpecials => TvdbPlugin.Instance?.Configuration.IncludeMissingSpecials ?? false;
+
         private static bool EpisodeExists(EpisodeBaseRecord episodeRecord, IReadOnlyList<Episode> existingEpisodes)
         {
             return existingEpisodes.Any(episode => EpisodeEquals(episode, episodeRecord));
@@ -161,6 +163,11 @@ private async Task HandleSeries(Series series)
 
             var allEpisodes = await GetAllEpisodes(tvdbId, series.DisplayOrder, series.GetPreferredMetadataLanguage()).ConfigureAwait(false);
 
+            if (!IncludeMissingSpecials)
+            {
+                allEpisodes = allEpisodes.Where(e => e.SeasonNumber != 0).ToList();
+            }
+
             var allSeasons = allEpisodes
                 .Where(ep => ep.SeasonNumber.HasValue)
                 .Select(ep => ep.SeasonNumber!.Value)