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.IO; 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; using Jellyfin.Plugin.MetaShark.Configuration; using Jellyfin.Plugin.MetaShark.Core; using Jellyfin.Plugin.MetaShark.Parser; 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 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 PluginConfiguration config { get { return Plugin.Instance?.Configuration ?? new PluginConfiguration(); } } 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; } protected async Task GuessByDoubanAsync(ItemLookupInfo info, CancellationToken cancellationToken) { // ParseName is required here. // Caller provides the filename with extension stripped and NOT the parsed filename BTNamePareser pareser = new BTNamePareser(); var search_info = pareser.Match(info.Name, this._logger); string searchName = search_info.ChineseName != null ? search_info.ChineseName : search_info.EnglishName; if (info.Year == null && search_info.Year != null) { info.Year = int.Parse(search_info.Year); } this.Log($"GuessByDouban of [name]: {info.Name} year: {info.Year} search name: {searchName}"); var result = await this._doubanApi.SearchAsync(searchName, cancellationToken).ConfigureAwait(false); var jw = new JaroWinkler(); foreach (var item in result) { if (info is MovieInfo && item.Category != "电影") { continue; } if (info is SeriesInfo && item.Category != "电视剧") { continue; } //英文关键词搜,结果是中文的情况,不适用相似匹配 if (jw.Similarity(searchName, item.Name) > 0.8 || jw.Similarity(searchName, item.OriginalName) > 0.8) { this.Log($"GuessByDouban of [name] found Sid: {item.Sid}"); return item.Sid; } if (item.Name.Contains(searchName) && (info.Year != null && info.Year == item.Year)) { this.Log($"GuessByDouban of [name] found Sid: {item.Sid}"); return item.Sid; } if (searchName.IsChinese() != item.Name.IsChinese() && searchName.IsChinese() != item.OriginalName.IsChinese()) { this.Log($"GuessByDouban of [name] found tmdb id: \"{item.Sid}\""); return item.Sid; } } return null; } protected async Task GuestDoubanSeasonByYearAsync(string name, int? year, CancellationToken cancellationToken) { if (year == null || year == 0) { return null; } this.Log($"GuestDoubanSeasonByYear of [name]: {name} year: {year}"); var result = await this._doubanApi.SearchAsync(name, cancellationToken).ConfigureAwait(false); var jw = new JaroWinkler(); foreach (var item in result) { if (item.Category != "电视剧") { continue; } // this.Log($"GuestDoubanSeasonByYear name: {name} douban_name: {item.Name} douban_sid: {item.Sid} douban_year: {item.Year} score: {score} "); if (jw.Similarity(name, item.Name) > 0.8) { this.Log($"GuestDoubanSeasonByYear of [name] found Sid: {item.Sid}"); return item.Sid; } if ((name.IsChinese() != item.Name.IsChinese() && name.IsChinese() != item.OriginalName.IsChinese()) || year == item.Year) { this.Log($"GuestDoubanSeasonByYear of [name] found Sid: \"{item.Sid}\""); return item.Sid; } } return null; } // 通过季数,搜索结果按年份排序后,取对应季数索引项(不适合每季标题差异太大的,如葫芦兄弟和葫芦小金刚) protected async Task GuestDoubanSeasonByNumberAsync(string name, int? seasonNumber, CancellationToken cancellationToken) { if (seasonNumber == null || seasonNumber == 0) { return null; } this.Log($"GuestDoubanSeasonByNumber of [name]: {name} seasonNumber: {seasonNumber}"); var result = await this._doubanApi.SearchAsync(name, cancellationToken).ConfigureAwait(false); var jw = new JaroWinkler(); var matchList = new List(); foreach (var item in result) { if (item.Category != "电视剧") { continue; } var score = jw.Similarity(name, item.Name); if (score < 0.8) { continue; } // this.Log($"GuestDoubanSeasonByNumber name: {name} douban_name: {item.Name} douban_sid: {item.Sid} douban_year: {item.Year} score: {score} "); matchList.Add(item); } matchList.Sort((x, y) => x.Year.CompareTo(y.Year)); if (matchList.Count >= seasonNumber) { var matchItem = matchList[seasonNumber.Value - 1]; var sid = matchItem.Sid; this.Log($"GuestDoubanSeasonByNumber of [name] found Sid: {sid}"); return 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 BTNamePareser pareser = new BTNamePareser(); var search_info = pareser.Match(info.Name, this._logger); string searchName = search_info.ChineseName != null ? search_info.ChineseName : search_info.EnglishName; if (info.Year == null && search_info.Year != null) { info.Year = int.Parse(search_info.Year); } this.Log($"GuestByTmdb of [name]: {info.Name} search name: {searchName}"); var jw = new JaroWinkler(); switch (info) { case MovieInfo: var movieResults = await this._tmdbApi.SearchMovieAsync(searchName, info.Year ?? 0, info.MetadataLanguage, cancellationToken).ConfigureAwait(false); foreach (var item in movieResults) { if (jw.Similarity(searchName, item.Title) > 0.8 || jw.Similarity(searchName, item.OriginalTitle) > 0.8) { this.Log($"GuestByTmdb of [name] found tmdb id: \"{item.Id}\""); return item.Id.ToString(CultureInfo.InvariantCulture); } if (searchName.IsChinese() != item.Title.IsChinese() && searchName.IsChinese() != item.OriginalTitle.IsChinese()) { 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(searchName, info.MetadataLanguage, cancellationToken).ConfigureAwait(false); foreach (var item in seriesResults) { if (jw.Similarity(searchName, item.Name) > 0.8 || jw.Similarity(searchName, item.OriginalName) > 0.8) { this.Log($"GuestByTmdb of [name] found tmdb id: \"{item.Id}\""); return item.Id.ToString(CultureInfo.InvariantCulture); } if (searchName.IsChinese() != item.Name.IsChinese() && searchName.IsChinese() != item.OriginalName.IsChinese()) { this.Log($"GuestByTmdb of [name] found tmdb id: \"{item.Id}\""); return item.Id.ToString(CultureInfo.InvariantCulture); } } break; } return null; } // 通过imdb获取tmdbId protected async Task GetTmdbIdByImdbAsync(string imdb, string language, CancellationToken cancellationToken) { if (string.IsNullOrEmpty(imdb)) { return null; } // 通过imdb获取TMDB id (豆瓣的imdb id可能是旧的,需要先从omdb接口获取最新的imdb id var omdbItem = await this._omdbApi.GetByImdbID(imdb, cancellationToken).ConfigureAwait(false); if (omdbItem != null) { var findResult = await this._tmdbApi.FindByExternalIdAsync(omdbItem.ImdbID, TMDbLib.Objects.Find.FindExternalSource.Imdb, language, cancellationToken).ConfigureAwait(false); if (findResult?.MovieResults != null && findResult.MovieResults.Count > 0) { var tmdbId = findResult.MovieResults[0].Id; this.Log($"Found tmdb [id]: {tmdbId} by imdb id: {imdb}"); return $"{tmdbId}"; } if (findResult?.TvResults != null && findResult.TvResults.Count > 0) { var tmdbId = findResult.TvResults[0].Id; this.Log($"Found tmdb [id]: {tmdbId} by imdb id: {imdb}"); return $"{tmdbId}"; } } return null; } 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; } } }