mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 13:57:14 -04:00
442 lines
18 KiB
C#
442 lines
18 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using System.Text.RegularExpressions;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
using MediaBrowser.Common.Net;
|
||
using MediaBrowser.Controller.Entities;
|
||
using MediaBrowser.Controller.Library;
|
||
using MediaBrowser.Controller.Providers;
|
||
using MediaBrowser.Model.Entities;
|
||
using MediaBrowser.Model.Globalization;
|
||
using MediaBrowser.Model.Providers;
|
||
using Microsoft.Extensions.Logging;
|
||
using TvDbSharper;
|
||
using TvDbSharper.Dto;
|
||
using Series = MediaBrowser.Controller.Entities.TV.Series;
|
||
|
||
namespace MediaBrowser.Providers.Plugins.TheTvdb
|
||
{
|
||
public class TvdbSeriesProvider : IRemoteMetadataProvider<Series, SeriesInfo>, IHasOrder
|
||
{
|
||
internal static TvdbSeriesProvider Current { get; private set; }
|
||
private readonly IHttpClient _httpClient;
|
||
private readonly ILogger _logger;
|
||
private readonly ILibraryManager _libraryManager;
|
||
private readonly ILocalizationManager _localizationManager;
|
||
private readonly TvdbClientManager _tvdbClientManager;
|
||
|
||
public TvdbSeriesProvider(IHttpClient httpClient, ILogger<TvdbSeriesProvider> logger, ILibraryManager libraryManager, ILocalizationManager localizationManager, TvdbClientManager tvdbClientManager)
|
||
{
|
||
_httpClient = httpClient;
|
||
_logger = logger;
|
||
_libraryManager = libraryManager;
|
||
_localizationManager = localizationManager;
|
||
Current = this;
|
||
_tvdbClientManager = tvdbClientManager;
|
||
}
|
||
|
||
public async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(SeriesInfo searchInfo, CancellationToken cancellationToken)
|
||
{
|
||
if (IsValidSeries(searchInfo.ProviderIds))
|
||
{
|
||
var metadata = await GetMetadata(searchInfo, cancellationToken).ConfigureAwait(false);
|
||
|
||
if (metadata.HasMetadata)
|
||
{
|
||
return new List<RemoteSearchResult>
|
||
{
|
||
new RemoteSearchResult
|
||
{
|
||
Name = metadata.Item.Name,
|
||
PremiereDate = metadata.Item.PremiereDate,
|
||
ProductionYear = metadata.Item.ProductionYear,
|
||
ProviderIds = metadata.Item.ProviderIds,
|
||
SearchProviderName = Name
|
||
}
|
||
};
|
||
}
|
||
}
|
||
|
||
return await FindSeries(searchInfo.Name, searchInfo.Year, searchInfo.MetadataLanguage, cancellationToken).ConfigureAwait(false);
|
||
}
|
||
|
||
public async Task<MetadataResult<Series>> GetMetadata(SeriesInfo itemId, CancellationToken cancellationToken)
|
||
{
|
||
var result = new MetadataResult<Series>
|
||
{
|
||
QueriedById = true
|
||
};
|
||
|
||
if (!IsValidSeries(itemId.ProviderIds))
|
||
{
|
||
result.QueriedById = false;
|
||
await Identify(itemId).ConfigureAwait(false);
|
||
}
|
||
|
||
cancellationToken.ThrowIfCancellationRequested();
|
||
|
||
if (IsValidSeries(itemId.ProviderIds))
|
||
{
|
||
result.Item = new Series();
|
||
result.HasMetadata = true;
|
||
|
||
await FetchSeriesData(result, itemId.MetadataLanguage, itemId.ProviderIds, cancellationToken)
|
||
.ConfigureAwait(false);
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
private async Task FetchSeriesData(MetadataResult<Series> result, string metadataLanguage, Dictionary<string, string> seriesProviderIds, CancellationToken cancellationToken)
|
||
{
|
||
var series = result.Item;
|
||
|
||
if (seriesProviderIds.TryGetValue(MetadataProviders.Tvdb.ToString(), out var tvdbId) && !string.IsNullOrEmpty(tvdbId))
|
||
{
|
||
series.SetProviderId(MetadataProviders.Tvdb, tvdbId);
|
||
}
|
||
|
||
if (seriesProviderIds.TryGetValue(MetadataProviders.Imdb.ToString(), out var imdbId) && !string.IsNullOrEmpty(imdbId))
|
||
{
|
||
series.SetProviderId(MetadataProviders.Imdb, imdbId);
|
||
tvdbId = await GetSeriesByRemoteId(imdbId, MetadataProviders.Imdb.ToString(), metadataLanguage,
|
||
cancellationToken).ConfigureAwait(false);
|
||
}
|
||
|
||
if (seriesProviderIds.TryGetValue(MetadataProviders.Zap2It.ToString(), out var zap2It) && !string.IsNullOrEmpty(zap2It))
|
||
{
|
||
series.SetProviderId(MetadataProviders.Zap2It, zap2It);
|
||
tvdbId = await GetSeriesByRemoteId(zap2It, MetadataProviders.Zap2It.ToString(), metadataLanguage,
|
||
cancellationToken).ConfigureAwait(false);
|
||
}
|
||
|
||
try
|
||
{
|
||
var seriesResult =
|
||
await _tvdbClientManager
|
||
.GetSeriesByIdAsync(Convert.ToInt32(tvdbId), metadataLanguage, cancellationToken)
|
||
.ConfigureAwait(false);
|
||
MapSeriesToResult(result, seriesResult.Data, metadataLanguage);
|
||
}
|
||
catch (TvDbServerException e)
|
||
{
|
||
_logger.LogError(e, "Failed to retrieve series with id {TvdbId}", tvdbId);
|
||
return;
|
||
}
|
||
|
||
cancellationToken.ThrowIfCancellationRequested();
|
||
|
||
result.ResetPeople();
|
||
|
||
try
|
||
{
|
||
var actorsResult = await _tvdbClientManager
|
||
.GetActorsAsync(Convert.ToInt32(tvdbId), metadataLanguage, cancellationToken).ConfigureAwait(false);
|
||
MapActorsToResult(result, actorsResult.Data);
|
||
}
|
||
catch (TvDbServerException e)
|
||
{
|
||
_logger.LogError(e, "Failed to retrieve actors for series {TvdbId}", tvdbId);
|
||
}
|
||
}
|
||
|
||
private async Task<string> GetSeriesByRemoteId(string id, string idType, string language, CancellationToken cancellationToken)
|
||
{
|
||
|
||
TvDbResponse<SeriesSearchResult[]> result = null;
|
||
|
||
try
|
||
{
|
||
if (string.Equals(idType, MetadataProviders.Zap2It.ToString(), StringComparison.OrdinalIgnoreCase))
|
||
{
|
||
result = await _tvdbClientManager.GetSeriesByZap2ItIdAsync(id, language, cancellationToken)
|
||
.ConfigureAwait(false);
|
||
}
|
||
else
|
||
{
|
||
result = await _tvdbClientManager.GetSeriesByImdbIdAsync(id, language, cancellationToken)
|
||
.ConfigureAwait(false);
|
||
}
|
||
}
|
||
catch (TvDbServerException e)
|
||
{
|
||
_logger.LogError(e, "Failed to retrieve series with remote id {RemoteId}", id);
|
||
}
|
||
|
||
return result?.Data.First().Id.ToString();
|
||
}
|
||
|
||
/// <summary>
|
||
/// Check whether a dictionary of provider IDs includes an entry for a valid TV metadata provider.
|
||
/// </summary>
|
||
/// <param name="seriesProviderIds">The dictionary to check.</param>
|
||
/// <returns>True, if the dictionary contains a valid TV provider ID, otherwise false.</returns>
|
||
internal static bool IsValidSeries(Dictionary<string, string> seriesProviderIds)
|
||
{
|
||
return seriesProviderIds.ContainsKey(MetadataProviders.Tvdb.ToString()) ||
|
||
seriesProviderIds.ContainsKey(MetadataProviders.Imdb.ToString()) ||
|
||
seriesProviderIds.ContainsKey(MetadataProviders.Zap2It.ToString());
|
||
}
|
||
|
||
/// <summary>
|
||
/// Finds the series.
|
||
/// </summary>
|
||
/// <param name="name">The name.</param>
|
||
/// <param name="year">The year.</param>
|
||
/// <param name="language">The language.</param>
|
||
/// <param name="cancellationToken">The cancellation token.</param>
|
||
/// <returns>Task{System.String}.</returns>
|
||
private async Task<IEnumerable<RemoteSearchResult>> FindSeries(string name, int? year, string language, CancellationToken cancellationToken)
|
||
{
|
||
var results = await FindSeriesInternal(name, language, cancellationToken).ConfigureAwait(false);
|
||
|
||
if (results.Count == 0)
|
||
{
|
||
var parsedName = _libraryManager.ParseName(name);
|
||
var nameWithoutYear = parsedName.Name;
|
||
|
||
if (!string.IsNullOrWhiteSpace(nameWithoutYear) && !string.Equals(nameWithoutYear, name, StringComparison.OrdinalIgnoreCase))
|
||
{
|
||
results = await FindSeriesInternal(nameWithoutYear, language, cancellationToken).ConfigureAwait(false);
|
||
}
|
||
}
|
||
|
||
return results.Where(i =>
|
||
{
|
||
if (year.HasValue && i.ProductionYear.HasValue)
|
||
{
|
||
// Allow one year tolerance
|
||
return Math.Abs(year.Value - i.ProductionYear.Value) <= 1;
|
||
}
|
||
|
||
return true;
|
||
});
|
||
}
|
||
|
||
private async Task<List<RemoteSearchResult>> FindSeriesInternal(string name, string language, CancellationToken cancellationToken)
|
||
{
|
||
var comparableName = GetComparableName(name);
|
||
var list = new List<Tuple<List<string>, RemoteSearchResult>>();
|
||
TvDbResponse<SeriesSearchResult[]> result;
|
||
try
|
||
{
|
||
result = await _tvdbClientManager.GetSeriesByNameAsync(comparableName, language, cancellationToken)
|
||
.ConfigureAwait(false);
|
||
}
|
||
catch (TvDbServerException e)
|
||
{
|
||
_logger.LogError(e, "No series results found for {Name}", comparableName);
|
||
return new List<RemoteSearchResult>();
|
||
}
|
||
|
||
foreach (var seriesSearchResult in result.Data)
|
||
{
|
||
var tvdbTitles = new List<string>
|
||
{
|
||
GetComparableName(seriesSearchResult.SeriesName)
|
||
};
|
||
tvdbTitles.AddRange(seriesSearchResult.Aliases.Select(GetComparableName));
|
||
|
||
DateTime.TryParse(seriesSearchResult.FirstAired, out var firstAired);
|
||
var remoteSearchResult = new RemoteSearchResult
|
||
{
|
||
Name = tvdbTitles.FirstOrDefault(),
|
||
ProductionYear = firstAired.Year,
|
||
SearchProviderName = Name,
|
||
ImageUrl = TvdbUtils.BannerUrl + seriesSearchResult.Banner
|
||
|
||
};
|
||
try
|
||
{
|
||
var seriesSesult =
|
||
await _tvdbClientManager.GetSeriesByIdAsync(seriesSearchResult.Id, language, cancellationToken)
|
||
.ConfigureAwait(false);
|
||
remoteSearchResult.SetProviderId(MetadataProviders.Imdb, seriesSesult.Data.ImdbId);
|
||
remoteSearchResult.SetProviderId(MetadataProviders.Zap2It, seriesSesult.Data.Zap2itId);
|
||
}
|
||
catch (TvDbServerException e)
|
||
{
|
||
_logger.LogError(e, "Unable to retrieve series with id {TvdbId}", seriesSearchResult.Id);
|
||
}
|
||
|
||
remoteSearchResult.SetProviderId(MetadataProviders.Tvdb, seriesSearchResult.Id.ToString());
|
||
list.Add(new Tuple<List<string>, RemoteSearchResult>(tvdbTitles, remoteSearchResult));
|
||
}
|
||
|
||
return list
|
||
.OrderBy(i => i.Item1.Contains(comparableName, StringComparer.OrdinalIgnoreCase) ? 0 : 1)
|
||
.ThenBy(i => list.IndexOf(i))
|
||
.Select(i => i.Item2)
|
||
.ToList();
|
||
}
|
||
|
||
/// <summary>
|
||
/// The remove
|
||
/// </summary>
|
||
const string remove = "\"'!`?";
|
||
/// <summary>
|
||
/// The spacers
|
||
/// </summary>
|
||
const string spacers = "/,.:;\\(){}[]+-_=–*"; // (there are two types of dashes, short and long)
|
||
|
||
/// <summary>
|
||
/// Gets the name of the comparable.
|
||
/// </summary>
|
||
/// <param name="name">The name.</param>
|
||
/// <returns>System.String.</returns>
|
||
private string GetComparableName(string name)
|
||
{
|
||
name = name.ToLowerInvariant();
|
||
name = name.Normalize(NormalizationForm.FormKD);
|
||
var sb = new StringBuilder();
|
||
foreach (var c in name)
|
||
{
|
||
if (c >= 0x2B0 && c <= 0x0333)
|
||
{
|
||
// skip char modifier and diacritics
|
||
}
|
||
else if (remove.IndexOf(c) > -1)
|
||
{
|
||
// skip chars we are removing
|
||
}
|
||
else if (spacers.IndexOf(c) > -1)
|
||
{
|
||
sb.Append(" ");
|
||
}
|
||
else if (c == '&')
|
||
{
|
||
sb.Append(" and ");
|
||
}
|
||
else
|
||
{
|
||
sb.Append(c);
|
||
}
|
||
}
|
||
sb.Replace(", the", string.Empty).Replace("the ", " ").Replace(" the ", " ");
|
||
|
||
return Regex.Replace(sb.ToString().Trim(), @"\s+", " ");
|
||
}
|
||
|
||
private void MapSeriesToResult(MetadataResult<Series> result, TvDbSharper.Dto.Series tvdbSeries, string metadataLanguage)
|
||
{
|
||
Series series = result.Item;
|
||
series.SetProviderId(MetadataProviders.Tvdb, tvdbSeries.Id.ToString());
|
||
series.Name = tvdbSeries.SeriesName;
|
||
series.Overview = (tvdbSeries.Overview ?? string.Empty).Trim();
|
||
result.ResultLanguage = metadataLanguage;
|
||
series.AirDays = TVUtils.GetAirDays(tvdbSeries.AirsDayOfWeek);
|
||
series.AirTime = tvdbSeries.AirsTime;
|
||
series.CommunityRating = (float?)tvdbSeries.SiteRating;
|
||
series.SetProviderId(MetadataProviders.Imdb, tvdbSeries.ImdbId);
|
||
series.SetProviderId(MetadataProviders.Zap2It, tvdbSeries.Zap2itId);
|
||
if (Enum.TryParse(tvdbSeries.Status, true, out SeriesStatus seriesStatus))
|
||
{
|
||
series.Status = seriesStatus;
|
||
}
|
||
|
||
if (DateTime.TryParse(tvdbSeries.FirstAired, out var date))
|
||
{
|
||
// dates from tvdb are UTC but without offset or Z
|
||
series.PremiereDate = date;
|
||
series.ProductionYear = date.Year;
|
||
}
|
||
|
||
if (!string.IsNullOrEmpty(tvdbSeries.Runtime) && double.TryParse(tvdbSeries.Runtime, out double runtime))
|
||
{
|
||
series.RunTimeTicks = TimeSpan.FromMinutes(runtime).Ticks;
|
||
}
|
||
|
||
foreach (var genre in tvdbSeries.Genre)
|
||
{
|
||
series.AddGenre(genre);
|
||
}
|
||
|
||
if (!string.IsNullOrEmpty(tvdbSeries.Network))
|
||
{
|
||
series.AddStudio(tvdbSeries.Network);
|
||
}
|
||
|
||
if (result.Item.Status.HasValue && result.Item.Status.Value == SeriesStatus.Ended)
|
||
{
|
||
try
|
||
{
|
||
var episodeSummary = _tvdbClientManager
|
||
.GetSeriesEpisodeSummaryAsync(tvdbSeries.Id, metadataLanguage, CancellationToken.None).Result.Data;
|
||
var maxSeasonNumber = episodeSummary.AiredSeasons.Select(s => Convert.ToInt32(s)).Max();
|
||
var episodeQuery = new EpisodeQuery
|
||
{
|
||
AiredSeason = maxSeasonNumber
|
||
};
|
||
var episodesPage =
|
||
_tvdbClientManager.GetEpisodesPageAsync(tvdbSeries.Id, episodeQuery, metadataLanguage, CancellationToken.None).Result.Data;
|
||
result.Item.EndDate = episodesPage.Select(e =>
|
||
{
|
||
DateTime.TryParse(e.FirstAired, out var firstAired);
|
||
return firstAired;
|
||
}).Max();
|
||
}
|
||
catch (TvDbServerException e)
|
||
{
|
||
_logger.LogError(e, "Failed to find series end date for series {TvdbId}", tvdbSeries.Id);
|
||
}
|
||
}
|
||
}
|
||
|
||
private static void MapActorsToResult(MetadataResult<Series> result, IEnumerable<Actor> actors)
|
||
{
|
||
foreach (Actor actor in actors)
|
||
{
|
||
var personInfo = new PersonInfo
|
||
{
|
||
Type = PersonType.Actor,
|
||
Name = (actor.Name ?? string.Empty).Trim(),
|
||
Role = actor.Role,
|
||
ImageUrl = TvdbUtils.BannerUrl + actor.Image,
|
||
SortOrder = actor.SortOrder
|
||
};
|
||
|
||
if (!string.IsNullOrWhiteSpace(personInfo.Name))
|
||
{
|
||
result.AddPerson(personInfo);
|
||
}
|
||
}
|
||
}
|
||
|
||
public string Name => "TheTVDB";
|
||
|
||
public async Task Identify(SeriesInfo info)
|
||
{
|
||
if (!string.IsNullOrWhiteSpace(info.GetProviderId(MetadataProviders.Tvdb)))
|
||
{
|
||
return;
|
||
}
|
||
|
||
var srch = await FindSeries(info.Name, info.Year, info.MetadataLanguage, CancellationToken.None)
|
||
.ConfigureAwait(false);
|
||
|
||
var entry = srch.FirstOrDefault();
|
||
|
||
if (entry != null)
|
||
{
|
||
var id = entry.GetProviderId(MetadataProviders.Tvdb);
|
||
info.SetProviderId(MetadataProviders.Tvdb, id);
|
||
}
|
||
}
|
||
|
||
public int Order => 0;
|
||
|
||
public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
|
||
{
|
||
return _httpClient.GetResponse(new HttpRequestOptions
|
||
{
|
||
CancellationToken = cancellationToken,
|
||
Url = url,
|
||
BufferContent = false
|
||
});
|
||
}
|
||
}
|
||
}
|