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

124 lines
4.4 KiB
C#
Raw Normal View History

2019-01-30 20:23:23 +00:00
using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using MediaBrowser.Common.Net;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Providers;
using Microsoft.Extensions.Logging;
2019-01-30 20:23:23 +00:00
using TvDbSharper;
using TvDbSharper.Dto;
namespace MediaBrowser.Providers.TV.TheTVDB
{
2016-05-25 00:42:12 +00:00
public class TvdbEpisodeImageProvider : IRemoteImageProvider
{
private readonly IHttpClient _httpClient;
private readonly ILogger _logger;
2019-01-30 22:28:43 +00:00
private readonly TvDbClientManager _tvDbClientManager;
public TvdbEpisodeImageProvider(IHttpClient httpClient, ILogger<TvdbEpisodeImageProvider> logger, TvDbClientManager tvDbClientManager)
{
_httpClient = httpClient;
_logger = logger;
2019-02-15 19:11:27 +00:00
_tvDbClientManager = tvDbClientManager;
}
public string Name => "TheTVDB";
2018-09-12 17:26:21 +00:00
public bool Supports(BaseItem item)
{
return item is Episode;
}
2018-09-12 17:26:21 +00:00
public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
{
return new List<ImageType>
{
ImageType.Primary
};
}
2019-01-30 20:23:23 +00:00
public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
{
var episode = (Episode)item;
2014-02-06 15:58:49 +00:00
var series = episode.Series;
var imageResult = new List<RemoteImageInfo>();
2019-02-12 18:50:19 +00:00
var language = item.GetPreferredMetadataLanguage();
2015-11-24 18:45:52 +00:00
if (series != null && TvdbSeriesProvider.IsValidSeries(series.ProviderIds))
{
// Process images
try
{
var episodeInfo = new EpisodeInfo
{
IndexNumber = episode.IndexNumber.Value,
ParentIndexNumber = episode.ParentIndexNumber.Value,
2019-12-01 19:50:24 +00:00
SeriesProviderIds = series.ProviderIds,
SeriesDisplayOrder = series.DisplayOrder
};
string episodeTvdbId = await _tvDbClientManager
.GetEpisodeTvdbId(episodeInfo, language, cancellationToken).ConfigureAwait(false);
2019-02-08 18:48:18 +00:00
if (string.IsNullOrEmpty(episodeTvdbId))
{
_logger.LogError(
"Episode {SeasonNumber}x{EpisodeNumber} not found for series {SeriesTvdbId}",
2019-08-16 18:09:30 +00:00
episodeInfo.ParentIndexNumber,
episodeInfo.IndexNumber,
series.GetProviderId(MetadataProviders.Tvdb));
return imageResult;
2019-02-08 18:48:18 +00:00
}
var episodeResult =
2019-02-20 18:35:47 +00:00
await _tvDbClientManager
.GetEpisodesAsync(Convert.ToInt32(episodeTvdbId), language, cancellationToken)
.ConfigureAwait(false);
var image = GetImageInfo(episodeResult.Data);
if (image != null)
{
imageResult.Add(image);
}
}
catch (TvDbServerException e)
2019-01-30 20:23:23 +00:00
{
_logger.LogError(e, "Failed to retrieve episode images for series {TvDbId}", series.GetProviderId(MetadataProviders.Tvdb));
}
}
return imageResult;
}
2019-01-30 20:23:23 +00:00
private RemoteImageInfo GetImageInfo(EpisodeRecord episode)
{
2019-01-30 20:23:23 +00:00
if (string.IsNullOrEmpty(episode.Filename))
{
return null;
}
return new RemoteImageInfo
{
2019-01-30 20:23:23 +00:00
Width = Convert.ToInt32(episode.ThumbWidth),
Height = Convert.ToInt32(episode.ThumbHeight),
ProviderName = Name,
Url = TvdbUtils.BannerUrl + episode.Filename,
Type = ImageType.Primary
};
}
public int Order => 0;
public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
{
return _httpClient.GetResponse(new HttpRequestOptions
{
CancellationToken = cancellationToken,
2017-04-30 02:37:51 +00:00
Url = url
});
}
}
}