From 78277db5ae7bb207c5cb0b5ba7c46918b469401e Mon Sep 17 00:00:00 2001
From: DarkFighterLuke <luca-consoli@live.it>
Date: Sat, 9 Mar 2024 18:08:21 +0100
Subject: [PATCH] Implement basic metadata and image providers for series

---
 .../Providers/SeriesImageProvider.cs          | 83 +++++++++++++++++++
 .../Providers/SeriesMetadataProvider.cs       | 83 +++++++++++++++++++
 .../TubeArchivist/Channel.cs                  | 47 +++++++++++
 3 files changed, 213 insertions(+)
 create mode 100644 Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesImageProvider.cs
 create mode 100644 Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesMetadataProvider.cs

diff --git a/Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesImageProvider.cs b/Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesImageProvider.cs
new file mode 100644
index 0000000..8b81b6d
--- /dev/null
+++ b/Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesImageProvider.cs
@@ -0,0 +1,83 @@
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Linq;
+using System.Net.Http;
+using System.Threading;
+using System.Threading.Tasks;
+using Jellyfin.Plugin.TubeArchivistMetadata.TubeArchivist;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Entities.TV;
+using MediaBrowser.Controller.Providers;
+using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Providers;
+
+namespace Jellyfin.Plugin.TubeArchivistMetadata.Providers
+{
+    /// <summary>
+    /// Image provider which interacts with TubeArchivist library.
+    /// </summary>
+    public class SeriesImageProvider : IRemoteImageProvider
+    {
+        /// <summary>
+        /// Gets the provider name.
+        /// </summary>
+        public string Name => "TubeArchivist";
+
+        /// <inheritdoc />
+        public bool Supports(BaseItem item) => item is Series;
+
+        /// <inheritdoc />
+        public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
+        {
+            return new[] { ImageType.Primary };
+        }
+
+        /// <inheritdoc />
+        public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
+        {
+            var list = new List<RemoteImageInfo>();
+            var taApi = TubeArchivistApi.GetInstance();
+            var channelTAId = item.Path.Split("/").Last();
+            var channel = await taApi.GetChannel(channelTAId).ConfigureAwait(true);
+
+            if (channel != null)
+            {
+                list.Add(new RemoteImageInfo
+                {
+                    ProviderName = Name,
+                    Type = ImageType.Primary,
+                    Url = channel.ThumbUrl
+                });
+                list.Add(new RemoteImageInfo
+                {
+                    ProviderName = Name,
+                    Type = ImageType.Art,
+                    Url = channel.TvartUrl
+                });
+                list.Add(new RemoteImageInfo
+                {
+                    ProviderName = Name,
+                    Type = ImageType.Banner,
+                    Url = channel.BannerUrl
+                });
+            }
+
+            return list;
+        }
+
+        /// <inheritdoc />
+        public async Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
+        {
+            if (Plugin.Instance == null)
+            {
+                throw new DataException("Uninitialized plugin!");
+            }
+            else
+            {
+                Console.WriteLine("Channel images: " + Plugin.Instance.Configuration.TubeArchivistUrl + url);
+                return await Plugin.Instance.HttpClient.GetAsync(new Uri(Plugin.Instance.Configuration.TubeArchivistUrl + url), cancellationToken).ConfigureAwait(false);
+            }
+        }
+    }
+}
diff --git a/Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesMetadataProvider.cs b/Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesMetadataProvider.cs
new file mode 100644
index 0000000..b44983b
--- /dev/null
+++ b/Jellyfin.Plugin.TubeArchivistMetadata/Providers/SeriesMetadataProvider.cs
@@ -0,0 +1,83 @@
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Linq;
+using System.Net.Http;
+using System.Threading;
+using System.Threading.Tasks;
+using Jellyfin.Plugin.TubeArchivistMetadata.TubeArchivist;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Entities.TV;
+using MediaBrowser.Controller.Providers;
+using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Providers;
+using Newtonsoft.Json;
+
+namespace Jellyfin.Plugin.TubeArchivistMetadata.Providers
+{
+    /// <summary>
+    /// Metadata provider which interacts with TubeArchivist library.
+    /// </summary>
+    public class SeriesMetadataProvider : IRemoteMetadataProvider<Series, SeriesInfo>
+    {
+        /// <summary>
+        /// Gets the provider name.
+        /// </summary>
+        public string Name => "TubeArchivist";
+
+        /// <inheritdoc />
+        public async Task<MetadataResult<Series>> GetMetadata(SeriesInfo info, CancellationToken cancellationToken)
+        {
+            var result = new MetadataResult<Series>();
+            var taApi = TubeArchivistApi.GetInstance();
+            var channelTAId = info.Path.Split("/").Last();
+            var channel = await taApi.GetChannel(channelTAId).ConfigureAwait(true);
+
+            if (channel != null)
+            {
+                var peopleInfo = new List<PersonInfo>();
+                PeopleHelper.AddPerson(peopleInfo, new PersonInfo
+                {
+                    Name = channel.Name,
+                    ImageUrl = channel.ThumbUrl,
+                    Type = PersonType.Actor,
+                });
+                result.HasMetadata = true;
+                result.Item = channel.ToSeries();
+                result.Provider = Name;
+                result.People = peopleInfo;
+            }
+
+            return result;
+        }
+
+        /// <inheritdoc />
+        public async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(SeriesInfo searchInfo, CancellationToken cancellationToken)
+        {
+            var results = new List<RemoteSearchResult>();
+
+            var taApi = TubeArchivistApi.GetInstance();
+            var channelTAId = searchInfo.Path.Split("/").Last();
+            var channel = await taApi.GetChannel(channelTAId).ConfigureAwait(true);
+            if (channel != null)
+            {
+                results.Add(channel.ToSearchResult());
+            }
+
+            return results;
+        }
+
+        /// <inheritdoc />
+        public async Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
+        {
+            if (Plugin.Instance == null)
+            {
+                throw new DataException("Uninitialized plugin!");
+            }
+            else
+            {
+                return await Plugin.Instance.HttpClient.GetAsync(url, cancellationToken).ConfigureAwait(false);
+            }
+        }
+    }
+}
diff --git a/Jellyfin.Plugin.TubeArchivistMetadata/TubeArchivist/Channel.cs b/Jellyfin.Plugin.TubeArchivistMetadata/TubeArchivist/Channel.cs
index 908d01f..0d947e6 100644
--- a/Jellyfin.Plugin.TubeArchivistMetadata/TubeArchivist/Channel.cs
+++ b/Jellyfin.Plugin.TubeArchivistMetadata/TubeArchivist/Channel.cs
@@ -1,3 +1,8 @@
+using System.Collections.Generic;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Entities.TV;
+using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Providers;
 using Newtonsoft.Json;
 
 namespace Jellyfin.Plugin.TubeArchivistMetadata.TubeArchivist
@@ -67,5 +72,47 @@ namespace Jellyfin.Plugin.TubeArchivistMetadata.TubeArchivist
         /// </summary>
         [JsonProperty(PropertyName = "channel_tvart_url")]
         public string TvartUrl { get; set; }
+
+        /// <summary>
+        /// Converts the TubeArchivist API channel to a Jellyfin <see cref="RemoteSearchResult"/> object.
+        /// </summary>
+        /// <returns>The channel equivalent Jellyfin <see cref="RemoteSearchResult"/> object.</returns>
+        public RemoteSearchResult ToSearchResult()
+        {
+            return new RemoteSearchResult
+            {
+                Name = Name,
+                SearchProviderName = Constants.ProviderName,
+                ImageUrl = ThumbUrl,
+                ProviderIds = new Dictionary<string, string>() { { Constants.ProviderName, Id } }
+            };
+        }
+
+        /// <summary>
+        /// Converts the TubeArchivist API channel to a Jellyfin <see cref="Series"/> object.
+        /// </summary>
+        /// <returns>The channel equivalent Jellyfin <see cref="Series"/> object.</returns>
+        public Series ToSeries()
+        {
+            return new Series
+            {
+                Name = Name,
+                Studios = new[] { Name },
+                ProviderIds = new Dictionary<string, string>()
+                {
+                    {
+                        Constants.ProviderName, Id
+                    }
+                },
+                ImageInfos = new[]
+                {
+                    new ItemImageInfo
+                    {
+                        Path = ThumbUrl,
+                        Type = ImageType.Primary
+                    }
+                }
+            };
+        }
     }
 }