using Jellyfin.Plugin.MetaShark.Api; using Jellyfin.Plugin.MetaShark.Model; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using Microsoft.Extensions.Logging; using StringMetric; using System; using System.Collections.Generic; using System.Diagnostics.CodeAnalysis; using System.Globalization; using System.Linq; using System.Net.Http; using System.Text; using System.Text.RegularExpressions; using System.Threading; using System.Threading.Tasks; using System.Web; using TMDbLib.Objects.General; namespace Jellyfin.Plugin.MetaShark.Providers { public abstract class BaseProvider { /// /// Gets the provider name. /// public const string DoubanProviderName = "Douban"; /// /// Gets the provider id. /// public const string DoubanProviderId = "DoubanID"; /// /// Name of the provider. /// public const string TmdbProviderName = "TheMovieDb"; protected readonly Configuration.PluginConfiguration _config; protected readonly ILogger _logger; protected readonly IHttpClientFactory _httpClientFactory; protected readonly DoubanApi _doubanApi; protected readonly TmdbApi _tmdbApi; protected readonly OmdbApi _omdbApi; protected readonly ILibraryManager _libraryManager; protected Regex regMetaSourcePrefix = new Regex(@"^\[.+\]", RegexOptions.Compiled); public string Pattern { get { return this._config.Pattern; } } protected BaseProvider(IHttpClientFactory httpClientFactory, ILogger logger, ILibraryManager libraryManager, DoubanApi doubanApi, TmdbApi tmdbApi, OmdbApi omdbApi) { this._doubanApi = doubanApi; this._tmdbApi = tmdbApi; this._omdbApi = omdbApi; this._libraryManager = libraryManager; this._logger = logger; this._httpClientFactory = httpClientFactory; this._config = Plugin.Instance == null ? new Configuration.PluginConfiguration() : Plugin.Instance.Configuration; } protected async Task GuestByDoubanAsync(ItemLookupInfo info, CancellationToken cancellationToken) { // ParseName is required here. // Caller provides the filename with extension stripped and NOT the parsed filename var parsedName = this._libraryManager.ParseName(info.Name); this.Log($"GuestByDouban of [name]: {info.Name} year: {info.Year} search name: {parsedName.Name}"); var result = await this._doubanApi.SearchAsync(parsedName.Name, cancellationToken).ConfigureAwait(false); var jw = new JaroWinkler(); foreach (var item in result) { if (jw.Similarity(parsedName.Name, item.Name) < 0.8) { continue; } if (parsedName.Year == null || parsedName.Year == 0) { this.Log($"GuestByDouban of [name] found Sid: \"{item.Sid}\""); return item.Sid; } if (parsedName.Year == item.Year) { this.Log($"GuestByDouban of [name] found Sid: \"{item.Sid}\""); return item.Sid; } } return null; } protected async Task GuestSeasonByDoubanAsync(string name, int? year, CancellationToken cancellationToken) { if (year == null || year == 0) { return null; } this.Log($"GuestSeasonByDouban of [name]: {name} year: {year}"); var result = await this._doubanApi.SearchAsync(name, cancellationToken).ConfigureAwait(false); var jw = new JaroWinkler(); foreach (var item in result) { this.Log($"GuestSeasonByDouban: name: {name} item.Name: {item.Name} score: {jw.Similarity(name, item.Name)} "); if (jw.Similarity(name, item.Name) < 0.8) { continue; } if (year == item.Year) { this.Log($"GuestSeasonByDouban of [name] found Sid: \"{item.Sid}\""); return item.Sid; } } return null; } protected async Task GuestByTmdbAsync(ItemLookupInfo info, CancellationToken cancellationToken) { // ParseName is required here. // Caller provides the filename with extension stripped and NOT the parsed filename var parsedName = this._libraryManager.ParseName(info.Name); this.Log($"GuestByTmdb of [name]: {info.Name} search name: {parsedName.Name}"); var jw = new JaroWinkler(); switch (info) { case MovieInfo: var movieResults = await this._tmdbApi.SearchMovieAsync(parsedName.Name, parsedName.Year ?? 0, info.MetadataLanguage, cancellationToken).ConfigureAwait(false); foreach (var item in movieResults) { if (jw.Similarity(parsedName.Name, item.Title) > 0.8) { this.Log($"GuestByTmdb of [name] found tmdb id: \"{item.Id}\""); return item.Id.ToString(CultureInfo.InvariantCulture); } } break; case SeriesInfo: var seriesResults = await this._tmdbApi.SearchSeriesAsync(parsedName.Name, info.MetadataLanguage, cancellationToken).ConfigureAwait(false); foreach (var item in seriesResults) { if (jw.Similarity(parsedName.Name, item.Name) > 0.8) { this.Log($"GuestByTmdb of [name] found tmdb id: \"{item.Id}\""); return item.Id.ToString(CultureInfo.InvariantCulture); } } break; } return null; } protected string AppendMetaSourcePrefix(string name, string source) { if (string.IsNullOrEmpty(name)) { return name; } return $"[{source}]{name}"; } protected string RemoveMetaSourcePrefix(string name) { if (string.IsNullOrEmpty(name)) { return name; } return regMetaSourcePrefix.Replace(name, string.Empty); } protected string GetProxyImageUrl(string url) { var encodedUrl = HttpUtility.UrlEncode(url); return $"/plugin/metashark/proxy/image/?url={encodedUrl}"; } protected void Log(string? message, params object?[] args) { this._logger.LogInformation($"[MetaShark] {message}", args); } /// /// Adjusts the image's language code preferring the 5 letter language code eg. en-US. /// /// The image's actual language code. /// The requested language code. /// The language code. protected string AdjustImageLanguage(string imageLanguage, string requestLanguage) { if (!string.IsNullOrEmpty(imageLanguage) && !string.IsNullOrEmpty(requestLanguage) && requestLanguage.Length > 2 && imageLanguage.Length == 2 && requestLanguage.StartsWith(imageLanguage, StringComparison.OrdinalIgnoreCase)) { return requestLanguage; } return imageLanguage; } /// /// Maps the TMDB provided roles for crew members to Jellyfin roles. /// /// Crew member to map against the Jellyfin person types. /// The Jellyfin person type. [SuppressMessage("Microsoft.Maintainability", "CA1309: Use ordinal StringComparison", Justification = "AFAIK we WANT InvariantCulture comparisons here and not Ordinal")] public string MapCrewToPersonType(Crew crew) { if (crew.Department.Equals("production", StringComparison.InvariantCultureIgnoreCase) && crew.Job.Contains("director", StringComparison.InvariantCultureIgnoreCase)) { return PersonType.Director; } if (crew.Department.Equals("production", StringComparison.InvariantCultureIgnoreCase) && crew.Job.Contains("producer", StringComparison.InvariantCultureIgnoreCase)) { return PersonType.Producer; } if (crew.Department.Equals("writing", StringComparison.InvariantCultureIgnoreCase)) { return PersonType.Writer; } return string.Empty; } /// /// Normalizes a language string for use with TMDb's include image language parameter. /// /// The preferred language as either a 2 letter code with or without country code. /// The comma separated language string. public static string GetImageLanguagesParam(string preferredLanguage) { var languages = new List(); if (!string.IsNullOrEmpty(preferredLanguage)) { preferredLanguage = NormalizeLanguage(preferredLanguage); languages.Add(preferredLanguage); if (preferredLanguage.Length == 5) // like en-US { // Currently, TMDB supports 2-letter language codes only // They are planning to change this in the future, thus we're // supplying both codes if we're having a 5-letter code. languages.Add(preferredLanguage.Substring(0, 2)); } } languages.Add("null"); if (!string.Equals(preferredLanguage, "en", StringComparison.OrdinalIgnoreCase)) { languages.Add("en"); } return string.Join(',', languages); } /// /// Normalizes a language string for use with TMDb's language parameter. /// /// The language code. /// The normalized language code. public static string NormalizeLanguage(string language) { if (string.IsNullOrEmpty(language)) { return language; } // They require this to be uppercase // Everything after the hyphen must be written in uppercase due to a way TMDB wrote their api. // See here: https://www.themoviedb.org/talk/5119221d760ee36c642af4ad?page=3#56e372a0c3a3685a9e0019ab var parts = language.Split('-'); if (parts.Length == 2) { language = parts[0] + "-" + parts[1].ToUpperInvariant(); } return language; } } }