mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-08 04:18:06 +02:00
356 lines
12 KiB
C#
356 lines
12 KiB
C#
using MediaBrowser.Common.IO;
|
|
using MediaBrowser.Common.Net;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.Providers;
|
|
using MediaBrowser.Model.Dto;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Providers;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using System.Xml;
|
|
using CommonIO;
|
|
|
|
namespace MediaBrowser.Providers.TV
|
|
{
|
|
public class TvdbSeriesImageProvider : IRemoteImageProvider, IHasOrder, IHasItemChangeMonitor
|
|
{
|
|
private readonly IServerConfigurationManager _config;
|
|
private readonly IHttpClient _httpClient;
|
|
private readonly CultureInfo _usCulture = new CultureInfo("en-US");
|
|
private readonly IFileSystem _fileSystem;
|
|
|
|
public TvdbSeriesImageProvider(IServerConfigurationManager config, IHttpClient httpClient, IFileSystem fileSystem)
|
|
{
|
|
_config = config;
|
|
_httpClient = httpClient;
|
|
_fileSystem = fileSystem;
|
|
}
|
|
|
|
public string Name
|
|
{
|
|
get { return ProviderName; }
|
|
}
|
|
|
|
public static string ProviderName
|
|
{
|
|
get { return "TheTVDB"; }
|
|
}
|
|
|
|
public bool Supports(IHasImages item)
|
|
{
|
|
return item is Series;
|
|
}
|
|
|
|
public IEnumerable<ImageType> GetSupportedImages(IHasImages item)
|
|
{
|
|
return new List<ImageType>
|
|
{
|
|
ImageType.Primary,
|
|
ImageType.Banner,
|
|
ImageType.Backdrop
|
|
};
|
|
}
|
|
|
|
public async Task<IEnumerable<RemoteImageInfo>> GetImages(IHasImages item, CancellationToken cancellationToken)
|
|
{
|
|
if (TvdbSeriesProvider.IsValidSeries(item.ProviderIds))
|
|
{
|
|
var language = item.GetPreferredMetadataLanguage();
|
|
|
|
var seriesDataPath = await TvdbSeriesProvider.Current.EnsureSeriesInfo(item.ProviderIds, language, cancellationToken).ConfigureAwait(false);
|
|
|
|
var path = Path.Combine(seriesDataPath, "banners.xml");
|
|
|
|
try
|
|
{
|
|
var seriesOffset = TvdbSeriesProvider.GetSeriesOffset(item.ProviderIds);
|
|
if (seriesOffset != null && seriesOffset.Value != 0)
|
|
return TvdbSeasonImageProvider.GetImages(path, language, seriesOffset.Value + 1, cancellationToken);
|
|
|
|
return GetImages(path, language, cancellationToken);
|
|
}
|
|
catch (FileNotFoundException)
|
|
{
|
|
// No tvdb data yet. Don't blow up
|
|
}
|
|
catch (DirectoryNotFoundException)
|
|
{
|
|
// No tvdb data yet. Don't blow up
|
|
}
|
|
}
|
|
|
|
return new RemoteImageInfo[] { };
|
|
}
|
|
|
|
private IEnumerable<RemoteImageInfo> GetImages(string xmlPath, string preferredLanguage, CancellationToken cancellationToken)
|
|
{
|
|
var settings = new XmlReaderSettings
|
|
{
|
|
CheckCharacters = false,
|
|
IgnoreProcessingInstructions = true,
|
|
IgnoreComments = true,
|
|
ValidationType = ValidationType.None
|
|
};
|
|
|
|
var list = new List<RemoteImageInfo>();
|
|
|
|
using (var streamReader = new StreamReader(xmlPath, Encoding.UTF8))
|
|
{
|
|
// Use XmlReader for best performance
|
|
using (var reader = XmlReader.Create(streamReader, settings))
|
|
{
|
|
reader.MoveToContent();
|
|
|
|
// Loop through each element
|
|
while (reader.Read())
|
|
{
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
if (reader.NodeType == XmlNodeType.Element)
|
|
{
|
|
switch (reader.Name)
|
|
{
|
|
case "Banner":
|
|
{
|
|
using (var subtree = reader.ReadSubtree())
|
|
{
|
|
AddImage(subtree, list);
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
reader.Skip();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var isLanguageEn = string.Equals(preferredLanguage, "en", StringComparison.OrdinalIgnoreCase);
|
|
|
|
return list.OrderByDescending(i =>
|
|
{
|
|
if (string.Equals(preferredLanguage, i.Language, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return 3;
|
|
}
|
|
if (!isLanguageEn)
|
|
{
|
|
if (string.Equals("en", i.Language, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return 2;
|
|
}
|
|
}
|
|
if (string.IsNullOrEmpty(i.Language))
|
|
{
|
|
return isLanguageEn ? 3 : 2;
|
|
}
|
|
return 0;
|
|
})
|
|
.ThenByDescending(i => i.CommunityRating ?? 0)
|
|
.ThenByDescending(i => i.VoteCount ?? 0)
|
|
.ToList();
|
|
}
|
|
|
|
private void AddImage(XmlReader reader, List<RemoteImageInfo> images)
|
|
{
|
|
reader.MoveToContent();
|
|
|
|
string bannerType = null;
|
|
string url = null;
|
|
int? bannerSeason = null;
|
|
int? width = null;
|
|
int? height = null;
|
|
string language = null;
|
|
double? rating = null;
|
|
int? voteCount = null;
|
|
string thumbnailUrl = null;
|
|
|
|
while (reader.Read())
|
|
{
|
|
if (reader.NodeType == XmlNodeType.Element)
|
|
{
|
|
switch (reader.Name)
|
|
{
|
|
case "Rating":
|
|
{
|
|
var val = reader.ReadElementContentAsString() ?? string.Empty;
|
|
|
|
double rval;
|
|
|
|
if (double.TryParse(val, NumberStyles.Any, _usCulture, out rval))
|
|
{
|
|
rating = rval;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case "RatingCount":
|
|
{
|
|
var val = reader.ReadElementContentAsString() ?? string.Empty;
|
|
|
|
int rval;
|
|
|
|
if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
|
|
{
|
|
voteCount = rval;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case "Language":
|
|
{
|
|
language = reader.ReadElementContentAsString() ?? string.Empty;
|
|
break;
|
|
}
|
|
|
|
case "ThumbnailPath":
|
|
{
|
|
thumbnailUrl = reader.ReadElementContentAsString() ?? string.Empty;
|
|
break;
|
|
}
|
|
|
|
case "BannerType":
|
|
{
|
|
bannerType = reader.ReadElementContentAsString() ?? string.Empty;
|
|
|
|
break;
|
|
}
|
|
|
|
case "BannerPath":
|
|
{
|
|
url = reader.ReadElementContentAsString() ?? string.Empty;
|
|
break;
|
|
}
|
|
|
|
case "BannerType2":
|
|
{
|
|
var bannerType2 = reader.ReadElementContentAsString() ?? string.Empty;
|
|
|
|
// Sometimes the resolution is stuffed in here
|
|
var resolutionParts = bannerType2.Split('x');
|
|
|
|
if (resolutionParts.Length == 2)
|
|
{
|
|
int rval;
|
|
|
|
if (int.TryParse(resolutionParts[0], NumberStyles.Integer, _usCulture, out rval))
|
|
{
|
|
width = rval;
|
|
}
|
|
|
|
if (int.TryParse(resolutionParts[1], NumberStyles.Integer, _usCulture, out rval))
|
|
{
|
|
height = rval;
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case "Season":
|
|
{
|
|
var val = reader.ReadElementContentAsString();
|
|
|
|
if (!string.IsNullOrWhiteSpace(val))
|
|
{
|
|
bannerSeason = int.Parse(val);
|
|
}
|
|
break;
|
|
}
|
|
|
|
|
|
default:
|
|
reader.Skip();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!string.IsNullOrEmpty(url) && !bannerSeason.HasValue)
|
|
{
|
|
var imageInfo = new RemoteImageInfo
|
|
{
|
|
RatingType = RatingType.Score,
|
|
CommunityRating = rating,
|
|
VoteCount = voteCount,
|
|
Url = TVUtils.BannerUrl + url,
|
|
ProviderName = Name,
|
|
Language = language,
|
|
Width = width,
|
|
Height = height
|
|
};
|
|
|
|
if (!string.IsNullOrEmpty(thumbnailUrl))
|
|
{
|
|
imageInfo.ThumbnailUrl = TVUtils.BannerUrl + thumbnailUrl;
|
|
}
|
|
|
|
if (string.Equals(bannerType, "poster", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
imageInfo.Type = ImageType.Primary;
|
|
images.Add(imageInfo);
|
|
}
|
|
else if (string.Equals(bannerType, "series", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
imageInfo.Type = ImageType.Banner;
|
|
images.Add(imageInfo);
|
|
}
|
|
else if (string.Equals(bannerType, "fanart", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
imageInfo.Type = ImageType.Backdrop;
|
|
images.Add(imageInfo);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
public int Order
|
|
{
|
|
get { return 0; }
|
|
}
|
|
|
|
public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
|
|
{
|
|
return _httpClient.GetResponse(new HttpRequestOptions
|
|
{
|
|
CancellationToken = cancellationToken,
|
|
Url = url,
|
|
ResourcePool = TvdbSeriesProvider.Current.TvDbResourcePool
|
|
});
|
|
}
|
|
|
|
public bool HasChanged(IHasMetadata item, MetadataStatus status, IDirectoryService directoryService)
|
|
{
|
|
if (!TvdbSeriesProvider.Current.GetTvDbOptions().EnableAutomaticUpdates)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (TvdbSeriesProvider.IsValidSeries(item.ProviderIds))
|
|
{
|
|
// Process images
|
|
var imagesXmlPath = Path.Combine(TvdbSeriesProvider.GetSeriesDataPath(_config.ApplicationPaths, item.ProviderIds), "banners.xml");
|
|
|
|
var fileInfo = _fileSystem.GetFileInfo(imagesXmlPath);
|
|
|
|
return fileInfo.Exists && _fileSystem.GetLastWriteTimeUtc(fileInfo) > (status.DateLastMetadataRefresh ?? DateTime.MinValue);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|