123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444 |
- using MediaBrowser.Common.Configuration;
- using MediaBrowser.Common.IO;
- using MediaBrowser.Common.Net;
- using MediaBrowser.Controller.Channels;
- using MediaBrowser.Controller.Configuration;
- using MediaBrowser.Controller.Entities;
- using MediaBrowser.Controller.Entities.Movies;
- using MediaBrowser.Controller.Providers;
- using MediaBrowser.Model.Channels;
- using MediaBrowser.Model.Dto;
- using MediaBrowser.Model.Entities;
- using MediaBrowser.Model.Providers;
- using MediaBrowser.Providers.Music;
- 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;
- namespace MediaBrowser.Providers.Movies
- {
- public class FanartMovieImageProvider : IRemoteImageProvider, IHasChangeMonitor, IHasOrder
- {
- private readonly CultureInfo _usCulture = new CultureInfo("en-US");
- private readonly IServerConfigurationManager _config;
- private readonly IHttpClient _httpClient;
- private readonly IFileSystem _fileSystem;
- private const string FanArtBaseUrl = "http://api.fanart.tv/webservice/movie/{0}/{1}/xml/all/1/1";
- internal static FanartMovieImageProvider Current;
- public FanartMovieImageProvider(IServerConfigurationManager config, IHttpClient httpClient, IFileSystem fileSystem)
- {
- _config = config;
- _httpClient = httpClient;
- _fileSystem = fileSystem;
- Current = this;
- }
- public string Name
- {
- get { return ProviderName; }
- }
- public static string ProviderName
- {
- get { return "FanArt"; }
- }
- public bool Supports(IHasImages item)
- {
- var trailer = item as Trailer;
- if (trailer != null)
- {
- return !trailer.IsLocalTrailer;
- }
- return item is Movie || item is BoxSet || item is MusicVideo;
- }
- public IEnumerable<ImageType> GetSupportedImages(IHasImages item)
- {
- return new List<ImageType>
- {
- ImageType.Primary,
- ImageType.Thumb,
- ImageType.Art,
- ImageType.Logo,
- ImageType.Disc,
- ImageType.Banner,
- ImageType.Backdrop
- };
- }
- public async Task<IEnumerable<RemoteImageInfo>> GetImages(IHasImages item, CancellationToken cancellationToken)
- {
- var baseItem = (BaseItem)item;
- var list = new List<RemoteImageInfo>();
- var movieId = baseItem.GetProviderId(MetadataProviders.Tmdb);
- if (!string.IsNullOrEmpty(movieId))
- {
- await EnsureMovieXml(movieId, cancellationToken).ConfigureAwait(false);
- var xmlPath = GetFanartXmlPath(movieId);
- try
- {
- AddImages(list, xmlPath, cancellationToken);
- }
- catch (FileNotFoundException)
- {
- // No biggie. Don't blow up
- }
- }
- var language = item.GetPreferredMetadataLanguage();
- var isLanguageEn = string.Equals(language, "en", StringComparison.OrdinalIgnoreCase);
- // Sort first by width to prioritize HD versions
- return list.OrderByDescending(i => i.Width ?? 0)
- .ThenByDescending(i =>
- {
- if (string.Equals(language, 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);
- }
- private void AddImages(List<RemoteImageInfo> list, string xmlPath, CancellationToken cancellationToken)
- {
- using (var streamReader = new StreamReader(xmlPath, Encoding.UTF8))
- {
- // Use XmlReader for best performance
- using (var reader = XmlReader.Create(streamReader, new XmlReaderSettings
- {
- CheckCharacters = false,
- IgnoreProcessingInstructions = true,
- IgnoreComments = true,
- ValidationType = ValidationType.None
- }))
- {
- reader.MoveToContent();
- // Loop through each element
- while (reader.Read())
- {
- cancellationToken.ThrowIfCancellationRequested();
- if (reader.NodeType == XmlNodeType.Element)
- {
- switch (reader.Name)
- {
- case "movie":
- {
- using (var subReader = reader.ReadSubtree())
- {
- AddImages(list, subReader, cancellationToken);
- }
- break;
- }
- default:
- reader.Skip();
- break;
- }
- }
- }
- }
- }
- }
- private void AddImages(List<RemoteImageInfo> list, XmlReader reader, CancellationToken cancellationToken)
- {
- reader.MoveToContent();
- while (reader.Read())
- {
- if (reader.NodeType == XmlNodeType.Element)
- {
- switch (reader.Name)
- {
- case "hdmoviecleararts":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Art, 1000, 562);
- }
- break;
- }
- case "hdmovielogos":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Logo, 800, 310);
- }
- break;
- }
- case "moviediscs":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Disc, 1000, 1000);
- }
- break;
- }
- case "movieposters":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Primary, 1000, 1426);
- }
- break;
- }
- case "movielogos":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Logo, 400, 155);
- }
- break;
- }
- case "moviearts":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Art, 500, 281);
- }
- break;
- }
- case "moviethumbs":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Thumb, 1000, 562);
- }
- break;
- }
- case "moviebanners":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Banner, 1000, 185);
- }
- break;
- }
- case "moviebackgrounds":
- {
- using (var subReader = reader.ReadSubtree())
- {
- PopulateImageCategory(list, subReader, cancellationToken, ImageType.Backdrop, 1920, 1080);
- }
- break;
- }
- default:
- {
- using (reader.ReadSubtree())
- {
- }
- break;
- }
- }
- }
- }
- }
- private void PopulateImageCategory(List<RemoteImageInfo> list, XmlReader reader, CancellationToken cancellationToken, ImageType type, int width, int height)
- {
- reader.MoveToContent();
- while (reader.Read())
- {
- cancellationToken.ThrowIfCancellationRequested();
- if (reader.NodeType == XmlNodeType.Element)
- {
- switch (reader.Name)
- {
- case "hdmovielogo":
- case "moviedisc":
- case "hdmovieclearart":
- case "movieposter":
- case "movielogo":
- case "movieart":
- case "moviethumb":
- case "moviebanner":
- case "moviebackground":
- {
- var url = reader.GetAttribute("url");
- if (!string.IsNullOrEmpty(url))
- {
- var likesString = reader.GetAttribute("likes");
- int likes;
- var info = new RemoteImageInfo
- {
- RatingType = RatingType.Likes,
- Type = type,
- Width = width,
- Height = height,
- ProviderName = Name,
- Url = url,
- Language = reader.GetAttribute("lang")
- };
- if (!string.IsNullOrEmpty(likesString) && int.TryParse(likesString, NumberStyles.Any, _usCulture, out likes))
- {
- info.CommunityRating = likes;
- }
- list.Add(info);
- }
- break;
- }
- default:
- reader.Skip();
- break;
- }
- }
- }
- }
- public int Order
- {
- get { return 1; }
- }
- public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
- {
- return _httpClient.GetResponse(new HttpRequestOptions
- {
- CancellationToken = cancellationToken,
- Url = url,
- ResourcePool = FanartArtistProvider.Current.FanArtResourcePool
- });
- }
- public bool HasChanged(IHasMetadata item, IDirectoryService directoryService, DateTime date)
- {
- if (!_config.Configuration.EnableFanArtUpdates)
- {
- return false;
- }
- var id = item.GetProviderId(MetadataProviders.Tmdb);
- if (!string.IsNullOrEmpty(id))
- {
- // Process images
- var xmlPath = GetFanartXmlPath(id);
- var fileInfo = new FileInfo(xmlPath);
- return !fileInfo.Exists || _fileSystem.GetLastWriteTimeUtc(fileInfo) > date;
- }
- return false;
- }
- /// <summary>
- /// Gets the movie data path.
- /// </summary>
- /// <param name="appPaths">The app paths.</param>
- /// <param name="tmdbId">The TMDB id.</param>
- /// <returns>System.String.</returns>
- internal static string GetMovieDataPath(IApplicationPaths appPaths, string tmdbId)
- {
- var dataPath = Path.Combine(GetMoviesDataPath(appPaths), tmdbId);
- return dataPath;
- }
- /// <summary>
- /// Gets the movie data path.
- /// </summary>
- /// <param name="appPaths">The app paths.</param>
- /// <returns>System.String.</returns>
- internal static string GetMoviesDataPath(IApplicationPaths appPaths)
- {
- var dataPath = Path.Combine(appPaths.CachePath, "fanart-movies");
- return dataPath;
- }
- public string GetFanartXmlPath(string tmdbId)
- {
- var movieDataPath = GetMovieDataPath(_config.ApplicationPaths, tmdbId);
- return Path.Combine(movieDataPath, "fanart.xml");
- }
- /// <summary>
- /// Downloads the movie XML.
- /// </summary>
- /// <param name="tmdbId">The TMDB id.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task.</returns>
- internal async Task DownloadMovieXml(string tmdbId, CancellationToken cancellationToken)
- {
- cancellationToken.ThrowIfCancellationRequested();
- var url = string.Format(FanArtBaseUrl, FanartArtistProvider.ApiKey, tmdbId);
- var xmlPath = GetFanartXmlPath(tmdbId);
- Directory.CreateDirectory(Path.GetDirectoryName(xmlPath));
- using (var response = await _httpClient.Get(new HttpRequestOptions
- {
- Url = url,
- ResourcePool = FanartArtistProvider.Current.FanArtResourcePool,
- CancellationToken = cancellationToken
- }).ConfigureAwait(false))
- {
- using (var xmlFileStream = _fileSystem.GetFileStream(xmlPath, FileMode.Create, FileAccess.Write, FileShare.Read, true))
- {
- await response.CopyToAsync(xmlFileStream).ConfigureAwait(false);
- }
- }
- }
- private readonly Task _cachedTask = Task.FromResult(true);
- internal Task EnsureMovieXml(string tmdbId, CancellationToken cancellationToken)
- {
- var path = GetFanartXmlPath(tmdbId);
- var fileInfo = _fileSystem.GetFileSystemInfo(path);
- if (fileInfo.Exists)
- {
- if ((DateTime.UtcNow - _fileSystem.GetLastWriteTimeUtc(fileInfo)).TotalDays <= 3)
- {
- return _cachedTask;
- }
- }
- return DownloadMovieXml(tmdbId, cancellationToken);
- }
- }
- }
|