jellyfin-plugin-metashark/Jellyfin.Plugin.MetaShark/Providers/EpisodeProvider.cs

using Jellyfin.Plugin.MetaShark.Api;
using Jellyfin.Plugin.MetaShark.Core;
using Jellyfin.Plugin.MetaShark.Model;
using MediaBrowser.Common.Net;
using MediaBrowser.Common.Plugins;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Providers;
using Microsoft.Extensions.Logging;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net.Http;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading;
using System.Threading.Tasks;

namespace Jellyfin.Plugin.MetaShark.Providers
{
    public class EpisodeProvider : BaseProvider, IRemoteMetadataProvider<Episode, EpisodeInfo>
    {

        private static readonly Regex[] EpisodeFileNameRegex =
        {
            new(@"\[([\d\.]{2,})\]"),
            new(@"- ?([\d\.]{2,})"),
            new(@"EP?([\d\.]{2,})", RegexOptions.IgnoreCase),
            new(@"\[([\d\.]{2,})"),
            new(@"#([\d\.]{2,})"),
            new(@"(\d{2,})")
        };

        public EpisodeProvider(IHttpClientFactory httpClientFactory, ILoggerFactory loggerFactory, ILibraryManager libraryManager, DoubanApi doubanApi, TmdbApi tmdbApi, OmdbApi omdbApi)
            : base(httpClientFactory, loggerFactory.CreateLogger<EpisodeProvider>(), libraryManager, doubanApi, tmdbApi, omdbApi)
        {
        }

        public string Name => Plugin.PluginName;


        /// <inheritdoc />
        public async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(EpisodeInfo info, CancellationToken cancellationToken)
        {
            this.Log($"GetEpisodeSearchResults of [name]: {info.Name}");
            return await Task.FromResult(Enumerable.Empty<RemoteSearchResult>());
        }

        /// <inheritdoc />
        public async Task<MetadataResult<Episode>> GetMetadata(EpisodeInfo info, CancellationToken cancellationToken)
        {
            // 重新识别时，info的IndexNumber和ParentIndexNumber是从文件路径解析出来的，假如命名不规范，就会导致解析出错误值
            // 刷新元数据不覆盖时，IndexNumber和ParentIndexNumber是从当前的元数据获取
            this.Log($"GetEpisodeMetadata of [name]: {info.Name} number: {info.IndexNumber} ParentIndexNumber: {info.ParentIndexNumber}");
            var result = new MetadataResult<Episode>();


            // 剧集信息只有tmdb有
            info.SeriesProviderIds.TryGetValue(MetadataProvider.Tmdb.ToString(), out var seriesTmdbId);
            var seasonNumber = info.ParentIndexNumber;
            var episodeNumber = info.IndexNumber;
            var indexNumberEnd = info.IndexNumberEnd;
            // 修正anime命名格式导致的seasonNumber错误（从season元数据读取)
            var parent = _libraryManager.FindByPath(Path.GetDirectoryName(info.Path), true);
            if (parent is Season season)
            {
                this.Log("FixSeasionNumber: old: {0} new: {1}", seasonNumber, season.IndexNumber);
                seasonNumber = season.IndexNumber;
            }
            // 没有season级目录或目录不命名不规范时，会为null
            if (seasonNumber is null or 0)
            {
                seasonNumber = 1;
            }
            // 修正anime命名格式导致的episodeNumber错误
            var fileName = Path.GetFileName(info.Path) ?? string.Empty;
            var guessInfo = this.GuessEpisodeNumber(fileName);
            this.Log("GuessEpisodeNumber: fileName: {0} seasonNumber: {1} episodeNumber: {2}", fileName, guessInfo.seasonNumber, guessInfo.episodeNumber);
            if (guessInfo.seasonNumber.HasValue && guessInfo.seasonNumber != seasonNumber)
            {
                seasonNumber = guessInfo.seasonNumber.Value;
            }
            if (guessInfo.episodeNumber.HasValue && guessInfo.episodeNumber != episodeNumber)
            {
                episodeNumber = guessInfo.episodeNumber;

                result.HasMetadata = true;
                result.Item = new Episode
                {
                    ParentIndexNumber = seasonNumber,
                    IndexNumber = episodeNumber
                };
            }



            if (episodeNumber is null or 0 || seasonNumber is null or 0 || string.IsNullOrEmpty(seriesTmdbId))
            {
                this.Log("Lack meta message. episodeNumber: {0} seasonNumber: {1} seriesTmdbId:{2}", episodeNumber, seasonNumber, seriesTmdbId);
                return result;
            }

            // 利用season缓存取剧集信息会更快
            var seasonResult = await this._tmdbApi
                .GetSeasonAsync(seriesTmdbId.ToInt(), seasonNumber.Value, info.MetadataLanguage, info.MetadataLanguage, cancellationToken)
                .ConfigureAwait(false);
            if (seasonResult == null || seasonResult.Episodes.Count < episodeNumber.Value)
            {
                this.Log("Can‘t found episode data from tmdb. Name: {0} seriesTmdbId: {1} seasonNumber: {2} episodeNumber: {3}", info.Name, seriesTmdbId, seasonNumber, episodeNumber);
                return result;
            }

            var episodeResult = seasonResult.Episodes[episodeNumber.Value - 1];

            result.HasMetadata = true;
            result.QueriedById = true;

            if (!string.IsNullOrEmpty(episodeResult.Overview))
            {
                // if overview is non-empty, we can assume that localized data was returned
                result.ResultLanguage = info.MetadataLanguage;
            }

            var item = new Episode
            {
                IndexNumber = episodeNumber,
                ParentIndexNumber = seasonNumber,
                IndexNumberEnd = info.IndexNumberEnd
            };


            item.PremiereDate = episodeResult.AirDate;
            item.ProductionYear = episodeResult.AirDate?.Year;
            item.Name = episodeResult.Name;
            item.Overview = episodeResult.Overview;
            item.CommunityRating = Convert.ToSingle(episodeResult.VoteAverage);

            result.Item = item;

            return result;
        }

        /// <inheritdoc />
        public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
        {
            this.Log("GetImageResponse url: {0}", url);
            return _httpClientFactory.CreateClient().GetAsync(new Uri(url), cancellationToken);
        }

        public GuessInfo GuessEpisodeNumber(string fileName, double max = double.PositiveInfinity)
        {
            var guessInfo = new GuessInfo();

            var parseResult = AnitomySharp.AnitomySharp.Parse(fileName);
            var animeSpecialType = parseResult.FirstOrDefault(x => x.Category == AnitomySharp.Element.ElementCategory.ElementAnimeType && x.Value == "SP");
            if (animeSpecialType != null)
            {
                guessInfo.seasonNumber = 0;
            }
            var animeEpisode = parseResult.FirstOrDefault(x => x.Category == AnitomySharp.Element.ElementCategory.ElementEpisodeNumber);
            if (animeEpisode != null)
            {
                guessInfo.episodeNumber = animeEpisode.Value.ToInt();
            }

            if (!guessInfo.episodeNumber.HasValue)
            {
                foreach (var regex in EpisodeFileNameRegex)
                {
                    if (!regex.IsMatch(fileName))
                        continue;
                    if (!int.TryParse(regex.Match(fileName).Groups[1].Value.Trim('.'), out var index))
                        continue;
                    guessInfo.episodeNumber = index;
                    break;
                }
            }

            if (guessInfo.episodeNumber > 1000)
            {
                // 可能解析了分辨率，忽略返回
                guessInfo.episodeNumber = null;
            }

            return guessInfo;
        }

    }
}