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 Microsoft.AspNetCore.Http; 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 readonly IHttpContextAccessor _httpContextAccessor; protected Regex regMetaSourcePrefix = new Regex(@"^\[.+\]", RegexOptions.Compiled); protected PluginConfiguration config { get { return Plugin.Instance?.Configuration ?? new PluginConfiguration(); } } protected BaseProvider(IHttpClientFactory httpClientFactory, ILogger logger, ILibraryManager libraryManager, IHttpContextAccessor httpContextAccessor, 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._httpContextAccessor = httpContextAccessor; } protected async Task GuessByDoubanAsync(ItemLookupInfo info, CancellationToken cancellationToken) { // ParseName is required here. // Caller provides the filename with extension stripped and NOT the parsed filename var fileName = GetNotParsedName(info); var parseResult = NameParser.Parse(fileName); var searchName = !string.IsNullOrEmpty(parseResult.ChineseName) ? parseResult.ChineseName : parseResult.Name; info.Year = parseResult.Year; // 默认parser对anime年份会解析出错,以anitomy为准 this.Log($"GuessByDouban of [name]: {info.Name} [file_name]: {fileName} [year]: {info.Year} [search name]: {searchName}"); List result; DoubanSubject? item; // 假如存在年份,先通过suggest接口查找,减少搜索页访问次数,避免封禁(suggest没法区分电影或电视剧,排序也比搜索页差些) if (config.EnableDoubanAvoidRiskControl) { if (info.Year != null && info.Year > 0) { result = await this._doubanApi.SearchBySuggestAsync(searchName, cancellationToken).ConfigureAwait(false); item = result.Where(x => x.Year == info.Year && x.Name == searchName).FirstOrDefault(); if (item != null) { this.Log($"GuessByDouban of [name]: {searchName} found Sid: {item.Sid} (suggest)"); return item.Sid; } item = result.Where(x => x.Year == info.Year).FirstOrDefault(); if (item != null) { this.Log($"GuessByDouban of [name]: {searchName} found Sid: {item.Sid} (suggest)"); return item.Sid; } } } // 通过搜索页面查找 result = await this._doubanApi.SearchAsync(searchName, cancellationToken).ConfigureAwait(false); var cat = info is MovieInfo ? "电影" : "电视剧"; // 优先返回对应年份的电影 if (info.Year != null && info.Year > 0) { item = result.Where(x => x.Category == cat && x.Year == info.Year).FirstOrDefault(); if (item != null) { this.Log($"GuessByDouban of [name]: {searchName} found Sid: {item.Sid}"); return item.Sid; } } // 不存在年份时,返回第一个 item = result.Where(x => x.Category == cat).FirstOrDefault(); if (item != null) { this.Log($"GuessByDouban of [name]: {searchName} found Sid: {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}"); // 先通过suggest接口查找,减少搜索页访问次数,避免封禁(suggest没法区分电影或电视剧,排序也比搜索页差些) if (config.EnableDoubanAvoidRiskControl) { var suggestResult = await this._doubanApi.SearchBySuggestAsync(name, cancellationToken).ConfigureAwait(false); var suggestItem = suggestResult.Where(x => x.Year == year && x.Name == name).FirstOrDefault(); if (suggestItem != null) { this.Log($"GuestDoubanSeasonByYear of [name] found Sid: \"{suggestItem.Sid}\" (suggest)"); return suggestItem.Sid; } suggestItem = suggestResult.Where(x => x.Year == year).FirstOrDefault(); if (suggestItem != null) { this.Log($"GuestDoubanSeasonByYear of [name] found Sid: \"{suggestItem.Sid}\" (suggest)"); return suggestItem.Sid; } } // 通过搜索页面查找 var result = await this._doubanApi.SearchAsync(name, cancellationToken).ConfigureAwait(false); var item = result.Where(x => x.Category == "电视剧" && x.Year == year).FirstOrDefault(); if (item != null && !string.IsNullOrEmpty(item.Sid)) { this.Log($"GuestDoubanSeasonByYear 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 fileName = GetNotParsedName(info); var parseResult = NameParser.Parse(fileName); var searchName = !string.IsNullOrEmpty(parseResult.ChineseName) ? parseResult.ChineseName : parseResult.Name; info.Year = parseResult.Year; // 默认parser对anime年份会解析出错,以anitomy为准 this.Log($"GuestByTmdb of [name]: {info.Name} [file_name]: {fileName} [year]: {info.Year} [search name]: {searchName}"); switch (info) { case MovieInfo: var movieResults = await this._tmdbApi.SearchMovieAsync(searchName, info.Year ?? 0, info.MetadataLanguage, cancellationToken).ConfigureAwait(false); var movieItem = movieResults.FirstOrDefault(); if (movieItem != null) { // bt种子都是英文名,但电影是中日韩泰印法地区时,都不适用相似匹配,去掉限制 return movieItem.Id.ToString(CultureInfo.InvariantCulture); } break; case SeriesInfo: var seriesResults = await this._tmdbApi.SearchSeriesAsync(searchName, info.MetadataLanguage, cancellationToken).ConfigureAwait(false); var seriesItem = seriesResults.FirstOrDefault(); if (seriesItem != null) { // bt种子都是英文名,但电影是中日韩泰印法地区时,都不适用相似匹配,去掉限制 return seriesItem.Id.ToString(CultureInfo.InvariantCulture); } break; } return null; } protected async Task GetTmdbIdByImdbAsync(string imdb, string language, CancellationToken cancellationToken) { if (string.IsNullOrEmpty(imdb)) { return null; } // 豆瓣的imdb id可能是旧的,需要先从omdb接口获取最新的imdb id var omdbItem = await this._omdbApi.GetByImdbID(imdb, cancellationToken).ConfigureAwait(false); if (!string.IsNullOrEmpty(omdbItem?.ImdbID)) { imdb = omdbItem.ImdbID; } // 通过imdb获取tmdbId var findResult = await this._tmdbApi.FindByExternalIdAsync(imdb, 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; } /// /// 浏览器来源请求,返回代理地址(no-referer对于background-image不生效),其他客户端请求,返回原始图片地址 /// protected string GetProxyImageUrl(string url) { var fromWeb = false; if (_httpContextAccessor.HttpContext != null) { var clientInfo = _httpContextAccessor.HttpContext.Request.Headers.FirstOrDefault(x => x.Key == "X-Emby-Authorization").Value.FirstOrDefault() ?? string.Empty; fromWeb = clientInfo.Contains("Jellyfin Web"); } if (fromWeb) { var encodedUrl = HttpUtility.UrlEncode(url); return $"/plugin/metashark/proxy/image/?url={encodedUrl}"; } else { return url; } } 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; } protected string GetNotParsedName(ItemLookupInfo info) { var directoryName = Path.GetFileName(Path.GetDirectoryName(info.Path)); if (directoryName != null && directoryName.StartsWith(info.Name)) { return directoryName; } return Path.GetFileNameWithoutExtension(info.Path) ?? info.Name; } } }