jellyfin/MediaBrowser.Providers/TV/TheTVDB/TvDbClientManager.cs

245 lines
9.6 KiB
C#
Raw Normal View History

2019-01-30 22:28:43 +00:00
using System;
2019-02-07 22:56:57 +00:00
using System.Collections.Generic;
2019-02-08 18:48:18 +00:00
using System.Linq;
2019-02-12 18:50:19 +00:00
using System.Reflection;
2019-02-07 19:28:19 +00:00
using System.Threading;
using System.Threading.Tasks;
2019-02-08 18:48:18 +00:00
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
2019-02-07 19:28:19 +00:00
using Microsoft.Extensions.Caching.Memory;
2019-01-30 22:28:43 +00:00
using TvDbSharper;
2019-02-07 19:28:19 +00:00
using TvDbSharper.Dto;
2019-01-30 22:28:43 +00:00
2019-02-11 19:50:23 +00:00
namespace MediaBrowser.Providers.TV.TheTVDB
2019-01-30 22:28:43 +00:00
{
2019-02-15 19:11:27 +00:00
public class TvDbClientManager
2019-01-30 22:28:43 +00:00
{
2019-02-07 19:57:42 +00:00
private readonly SemaphoreSlim _cacheWriteLock = new SemaphoreSlim(1, 1);
2019-02-15 19:11:27 +00:00
private readonly IMemoryCache _cache;
2019-02-20 10:03:04 +00:00
private readonly TvDbClient _tvDbClient;
private DateTime _tokenCreatedAt;
2019-02-12 18:50:19 +00:00
private const string DefaultLanguage = "en";
2019-01-30 22:28:43 +00:00
2019-02-15 19:11:27 +00:00
public TvDbClientManager(IMemoryCache memoryCache)
2019-01-30 22:28:43 +00:00
{
2019-02-15 19:11:27 +00:00
_cache = memoryCache;
2019-02-20 10:03:04 +00:00
_tvDbClient = new TvDbClient();
_tvDbClient.Authentication.AuthenticateAsync(TvdbUtils.TvdbApiKey);
_tokenCreatedAt = DateTime.Now;
2019-01-30 22:28:43 +00:00
}
public TvDbClient TvDbClient
{
get
{
// Refresh if necessary
2019-02-20 10:03:04 +00:00
if (_tokenCreatedAt > DateTime.Now.Subtract(TimeSpan.FromHours(20)))
2019-01-30 22:28:43 +00:00
{
try
{
2019-02-20 10:03:04 +00:00
_tvDbClient.Authentication.RefreshTokenAsync();
2019-01-30 22:28:43 +00:00
}
catch
{
2019-02-20 10:03:04 +00:00
_tvDbClient.Authentication.AuthenticateAsync(TvdbUtils.TvdbApiKey);
2019-01-30 22:28:43 +00:00
}
2019-02-20 10:03:04 +00:00
_tokenCreatedAt = DateTime.Now;
2019-01-30 22:28:43 +00:00
}
2019-02-12 18:50:19 +00:00
2019-02-20 10:03:04 +00:00
return _tvDbClient;
2019-01-30 22:28:43 +00:00
}
}
2019-02-07 19:28:19 +00:00
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<SeriesSearchResult[]>> GetSeriesByNameAsync(string name, string language,
CancellationToken cancellationToken)
2019-02-07 19:28:19 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("series", name, language);
return TryGetValue(cacheKey, language,() => TvDbClient.Search.SearchSeriesByNameAsync(name, cancellationToken));
2019-02-07 19:28:19 +00:00
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<Series>> GetSeriesByIdAsync(int tvdbId, string language,
CancellationToken cancellationToken)
2019-02-07 19:28:19 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("series", tvdbId, language);
return TryGetValue(cacheKey, language,() => TvDbClient.Series.GetAsync(tvdbId, cancellationToken));
2019-02-07 20:59:57 +00:00
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<EpisodeRecord>> GetEpisodesAsync(int episodeTvdbId, string language,
CancellationToken cancellationToken)
2019-02-07 20:59:57 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("episode", episodeTvdbId, language);
return TryGetValue(cacheKey, language,() => TvDbClient.Episodes.GetAsync(episodeTvdbId, cancellationToken));
2019-02-07 20:59:57 +00:00
}
2019-02-12 18:50:19 +00:00
public async Task<List<EpisodeRecord>> GetAllEpisodesAsync(int tvdbId, string language,
CancellationToken cancellationToken)
2019-02-07 22:56:57 +00:00
{
// Traverse all episode pages and join them together
var episodes = new List<EpisodeRecord>();
2019-02-20 18:35:47 +00:00
var episodePage = await GetEpisodesPageAsync(tvdbId, new EpisodeQuery(), language, cancellationToken)
.ConfigureAwait(false);
2019-02-07 22:56:57 +00:00
episodes.AddRange(episodePage.Data);
2019-02-08 18:48:18 +00:00
if (!episodePage.Links.Next.HasValue || !episodePage.Links.Last.HasValue)
{
return episodes;
}
int next = episodePage.Links.Next.Value;
int last = episodePage.Links.Last.Value;
2019-02-07 22:56:57 +00:00
for (var page = next; page <= last; ++page)
{
2019-02-20 18:35:47 +00:00
episodePage = await GetEpisodesPageAsync(tvdbId, page, new EpisodeQuery(), language, cancellationToken)
.ConfigureAwait(false);
2019-02-07 22:56:57 +00:00
episodes.AddRange(episodePage.Data);
}
return episodes;
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<SeriesSearchResult[]>> GetSeriesByImdbIdAsync(string imdbId, string language,
CancellationToken cancellationToken)
2019-02-07 20:59:57 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("series", imdbId, language);
return TryGetValue(cacheKey, language,() => TvDbClient.Search.SearchSeriesByImdbIdAsync(imdbId, cancellationToken));
2019-02-07 20:59:57 +00:00
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<SeriesSearchResult[]>> GetSeriesByZap2ItIdAsync(string zap2ItId, string language,
CancellationToken cancellationToken)
2019-02-07 20:59:57 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("series", zap2ItId, language);
return TryGetValue( cacheKey, language,() => TvDbClient.Search.SearchSeriesByZap2ItIdAsync(zap2ItId, cancellationToken));
2019-02-07 20:59:57 +00:00
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<Actor[]>> GetActorsAsync(int tvdbId, string language,
CancellationToken cancellationToken)
2019-02-07 20:59:57 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("actors", tvdbId, language);
return TryGetValue(cacheKey, language,() => TvDbClient.Series.GetActorsAsync(tvdbId, cancellationToken));
2019-02-07 20:59:57 +00:00
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<Image[]>> GetImagesAsync(int tvdbId, ImagesQuery imageQuery, string language,
CancellationToken cancellationToken)
2019-02-07 20:59:57 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("images", tvdbId, language, imageQuery);
2019-02-12 18:50:19 +00:00
return TryGetValue(cacheKey, language,() => TvDbClient.Series.GetImagesAsync(tvdbId, imageQuery, cancellationToken));
}
public Task<TvDbResponse<Language[]>> GetLanguagesAsync(CancellationToken cancellationToken)
{
2019-02-12 18:50:19 +00:00
return TryGetValue("languages", null,() => TvDbClient.Languages.GetAllAsync(cancellationToken));
2019-02-07 19:57:42 +00:00
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<EpisodesSummary>> GetSeriesEpisodeSummaryAsync(int tvdbId, string language,
CancellationToken cancellationToken)
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey("seriesepisodesummary", tvdbId, language);
return TryGetValue(cacheKey, language,
() => TvDbClient.Series.GetEpisodesSummaryAsync(tvdbId, cancellationToken));
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<EpisodeRecord[]>> GetEpisodesPageAsync(int tvdbId, int page, EpisodeQuery episodeQuery,
string language, CancellationToken cancellationToken)
2019-02-07 22:56:57 +00:00
{
2019-02-12 20:04:45 +00:00
var cacheKey = GenerateKey(language, tvdbId, episodeQuery);
2019-02-12 18:50:19 +00:00
return TryGetValue(cacheKey, language,
2019-02-07 22:56:57 +00:00
() => TvDbClient.Series.GetEpisodesAsync(tvdbId, page, episodeQuery, cancellationToken));
}
2019-02-12 18:50:19 +00:00
public Task<string> GetEpisodeTvdbId(EpisodeInfo searchInfo, string language,
CancellationToken cancellationToken)
2019-02-08 18:48:18 +00:00
{
searchInfo.SeriesProviderIds.TryGetValue(MetadataProviders.Tvdb.ToString(),
out var seriesTvdbId);
var episodeQuery = new EpisodeQuery();
// Prefer SxE over premiere date as it is more robust
if (searchInfo.IndexNumber.HasValue && searchInfo.ParentIndexNumber.HasValue)
{
episodeQuery.AiredEpisode = searchInfo.IndexNumber.Value;
episodeQuery.AiredSeason = searchInfo.ParentIndexNumber.Value;
}
else if (searchInfo.PremiereDate.HasValue)
{
// tvdb expects yyyy-mm-dd format
episodeQuery.FirstAired = searchInfo.PremiereDate.Value.ToString("yyyy-MM-dd");
}
2019-02-13 10:23:54 +00:00
return GetEpisodeTvdbId(Convert.ToInt32(seriesTvdbId), episodeQuery, language, cancellationToken);
2019-02-08 18:48:18 +00:00
}
2019-02-12 18:50:19 +00:00
public async Task<string> GetEpisodeTvdbId(int seriesTvdbId, EpisodeQuery episodeQuery,
2019-02-13 10:23:54 +00:00
string language,
CancellationToken cancellationToken)
2019-02-08 18:48:18 +00:00
{
2019-02-20 18:35:47 +00:00
var episodePage =
await GetEpisodesPageAsync(Convert.ToInt32(seriesTvdbId), episodeQuery, language, cancellationToken)
.ConfigureAwait(false);
2019-02-08 18:48:18 +00:00
return episodePage.Data.FirstOrDefault()?.Id.ToString();
}
2019-02-12 18:50:19 +00:00
public Task<TvDbResponse<EpisodeRecord[]>> GetEpisodesPageAsync(int tvdbId, EpisodeQuery episodeQuery,
string language, CancellationToken cancellationToken)
{
2019-02-12 18:50:19 +00:00
return GetEpisodesPageAsync(tvdbId, 1, episodeQuery, language, cancellationToken);
}
2019-02-12 18:50:19 +00:00
private async Task<T> TryGetValue<T>(string key, string language, Func<Task<T>> resultFactory)
2019-02-07 19:57:42 +00:00
{
if (_cache.TryGetValue(key, out T cachedValue))
2019-02-07 19:28:19 +00:00
{
2019-02-07 19:57:42 +00:00
return cachedValue;
}
await _cacheWriteLock.WaitAsync().ConfigureAwait(false);
try
2019-02-07 19:57:42 +00:00
{
if (_cache.TryGetValue(key, out cachedValue))
{
return cachedValue;
}
2019-02-20 10:03:04 +00:00
_tvDbClient.AcceptedLanguage = TvdbUtils.NormalizeLanguage(language) ?? DefaultLanguage;
2019-02-20 18:35:47 +00:00
var result = await resultFactory.Invoke().ConfigureAwait(false);
_cache.Set(key, result, TimeSpan.FromHours(1));
2019-02-07 19:57:42 +00:00
return result;
2019-02-07 19:28:19 +00:00
}
finally
{
_cacheWriteLock.Release();
}
2019-02-07 19:28:19 +00:00
}
2019-02-12 18:50:19 +00:00
2019-02-12 20:04:45 +00:00
private static string GenerateKey(params object[] objects)
2019-02-12 18:50:19 +00:00
{
var key = string.Empty;
foreach (var obj in objects)
{
var objType = obj.GetType();
if (objType.IsPrimitive || objType == typeof(string))
{
key += obj + ";";
2019-02-12 18:50:19 +00:00
}
else
{
foreach (PropertyInfo propertyInfo in objType.GetProperties())
{
var currentValue = propertyInfo.GetValue(obj, null);
if (currentValue == null)
{
continue;
}
2019-02-13 10:23:54 +00:00
key += propertyInfo.Name + "=" + currentValue + ";";
2019-02-12 18:50:19 +00:00
}
}
}
return key;
}
2019-01-30 22:28:43 +00:00
}
}