using MediaBrowser.Common.Configuration; using MediaBrowser.Common.IO; using MediaBrowser.Common.Net; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Entities.Movies; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using System; using System.Globalization; using System.IO; using System.Linq; using System.Threading; using System.Threading.Tasks; using System.Xml; namespace MediaBrowser.Providers.Movies { /// /// Class FanArtMovieProvider /// class FanArtMovieProvider : FanartBaseProvider { /// /// Gets the HTTP client. /// /// The HTTP client. protected IHttpClient HttpClient { get; private set; } /// /// The _provider manager /// private readonly IProviderManager _providerManager; /// /// The us culture /// private static readonly CultureInfo UsCulture = new CultureInfo("en-US"); internal static FanArtMovieProvider Current { get; private set; } /// /// Initializes a new instance of the class. /// /// The HTTP client. /// The log manager. /// The configuration manager. /// The provider manager. /// httpClient public FanArtMovieProvider(IHttpClient httpClient, ILogManager logManager, IServerConfigurationManager configurationManager, IProviderManager providerManager) : base(logManager, configurationManager) { if (httpClient == null) { throw new ArgumentNullException("httpClient"); } HttpClient = httpClient; _providerManager = providerManager; Current = this; } public override ItemUpdateType ItemUpdateType { get { return ItemUpdateType.ImageUpdate; } } /// /// Gets a value indicating whether [refresh on version change]. /// /// true if [refresh on version change]; otherwise, false. protected override bool RefreshOnVersionChange { get { return true; } } /// /// Gets the provider version. /// /// The provider version. protected override string ProviderVersion { get { return "13"; } } public override MetadataProviderPriority Priority { get { return MetadataProviderPriority.Fifth; } } /// /// The fan art base URL /// protected string FanArtBaseUrl = "http://api.fanart.tv/webservice/movie/{0}/{1}/xml/all/1/1"; /// /// Supportses the specified item. /// /// The item. /// true if XXXX, false otherwise public override bool Supports(BaseItem item) { var trailer = item as Trailer; if (trailer != null) { return !trailer.IsLocalTrailer; } return item is Movie || item is BoxSet || item is MusicVideo; } /// /// Needses the refresh internal. /// /// The item. /// The provider info. /// true if XXXX, false otherwise protected override bool NeedsRefreshInternal(BaseItem item, BaseProviderInfo providerInfo) { if (string.IsNullOrEmpty(item.GetProviderId(MetadataProviders.Tmdb))) { return false; } if (!ConfigurationManager.Configuration.DownloadMovieImages.Art && !ConfigurationManager.Configuration.DownloadMovieImages.Logo && !ConfigurationManager.Configuration.DownloadMovieImages.Disc && !ConfigurationManager.Configuration.DownloadMovieImages.Backdrops && !ConfigurationManager.Configuration.DownloadMovieImages.Banner && !ConfigurationManager.Configuration.DownloadMovieImages.Thumb && !ConfigurationManager.Configuration.DownloadMovieImages.Primary) { return false; } if (item.HasImage(ImageType.Primary) && item.HasImage(ImageType.Art) && item.HasImage(ImageType.Logo) && item.HasImage(ImageType.Disc) && item.HasImage(ImageType.Banner) && item.HasImage(ImageType.Thumb) && item.BackdropImagePaths.Count >= ConfigurationManager.Configuration.MaxBackdrops) { return false; } return base.NeedsRefreshInternal(item, providerInfo); } protected override bool NeedsRefreshBasedOnCompareDate(BaseItem item, BaseProviderInfo providerInfo) { var id = item.GetProviderId(MetadataProviders.Tmdb); if (!string.IsNullOrEmpty(id)) { // Process images var path = GetMovieDataPath(ConfigurationManager.ApplicationPaths, id); try { var files = new DirectoryInfo(path) .EnumerateFiles("*.xml", SearchOption.TopDirectoryOnly) .Select(i => i.LastWriteTimeUtc) .ToList(); if (files.Count > 0) { return files.Max() > providerInfo.LastRefreshed; } } catch (DirectoryNotFoundException) { return true; } } return base.NeedsRefreshBasedOnCompareDate(item, providerInfo); } /// /// Gets the movie data path. /// /// The app paths. /// The TMDB id. /// System.String. internal static string GetMovieDataPath(IApplicationPaths appPaths, string tmdbId) { var dataPath = Path.Combine(GetMoviesDataPath(appPaths), tmdbId); return dataPath; } /// /// Gets the movie data path. /// /// The app paths. /// System.String. internal static string GetMoviesDataPath(IApplicationPaths appPaths) { var dataPath = Path.Combine(appPaths.DataPath, "fanart-movies"); return dataPath; } /// /// Fetches metadata and returns true or false indicating if any work that requires persistence was done /// /// The item. /// if set to true [force]. /// The cancellation token. /// Task{System.Boolean}. public override async Task FetchAsync(BaseItem item, bool force, CancellationToken cancellationToken) { cancellationToken.ThrowIfCancellationRequested(); var movieId = item.GetProviderId(MetadataProviders.Tmdb); if (!string.IsNullOrEmpty(movieId)) { var movieDataPath = GetMovieDataPath(ConfigurationManager.ApplicationPaths, movieId); var xmlPath = Path.Combine(movieDataPath, "fanart.xml"); // Only download the xml if it doesn't already exist. The prescan task will take care of getting updates if (!File.Exists(xmlPath)) { await DownloadMovieXml(movieDataPath, movieId, cancellationToken).ConfigureAwait(false); } if (File.Exists(xmlPath)) { await FetchFromXml(item, xmlPath, cancellationToken).ConfigureAwait(false); } } SetLastRefreshed(item, DateTime.UtcNow); return true; } /// /// Downloads the movie XML. /// /// The movie data path. /// The TMDB id. /// The cancellation token. /// Task. internal async Task DownloadMovieXml(string movieDataPath, string tmdbId, CancellationToken cancellationToken) { cancellationToken.ThrowIfCancellationRequested(); string url = string.Format(FanArtBaseUrl, ApiKey, tmdbId); var xmlPath = Path.Combine(movieDataPath, "fanart.xml"); Directory.CreateDirectory(movieDataPath); using (var response = await HttpClient.Get(new HttpRequestOptions { Url = url, ResourcePool = FanArtResourcePool, CancellationToken = cancellationToken }).ConfigureAwait(false)) { using (var xmlFileStream = new FileStream(xmlPath, FileMode.Create, FileAccess.Write, FileShare.Read, StreamDefaults.DefaultFileStreamBufferSize, FileOptions.Asynchronous)) { await response.CopyToAsync(xmlFileStream).ConfigureAwait(false); } } } /// /// Fetches from XML. /// /// The item. /// The XML file path. /// The cancellation token. /// Task. private async Task FetchFromXml(BaseItem item, string xmlFilePath, CancellationToken cancellationToken) { var doc = new XmlDocument(); doc.Load(xmlFilePath); var language = ConfigurationManager.Configuration.PreferredMetadataLanguage.ToLower(); cancellationToken.ThrowIfCancellationRequested(); string path; if (ConfigurationManager.Configuration.DownloadMovieImages.Disc && !item.HasImage(ImageType.Disc)) { var node = doc.SelectSingleNode("//fanart/movie/movieposters/movieposter[@lang = \"" + language + "\"]/@url") ?? doc.SelectSingleNode("//fanart/movie/movieposters/movieposter/@url"); path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { await _providerManager.SaveImage(item, path, FanArtResourcePool, ImageType.Disc, null, cancellationToken) .ConfigureAwait(false); } } cancellationToken.ThrowIfCancellationRequested(); if (ConfigurationManager.Configuration.DownloadMovieImages.Logo && !item.HasImage(ImageType.Logo)) { var node = doc.SelectSingleNode("//fanart/movie/hdmovielogos/hdmovielogo[@lang = \"" + language + "\"]/@url") ?? doc.SelectSingleNode("//fanart/movie/movielogos/movielogo[@lang = \"" + language + "\"]/@url"); if (node == null && language != "en") { //maybe just couldn't find language - try just first one node = doc.SelectSingleNode("//fanart/movie/hdmovielogos/hdmovielogo/@url") ?? doc.SelectSingleNode("//fanart/movie/movielogos/movielogo/@url"); } path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { await _providerManager.SaveImage(item, path, FanArtResourcePool, ImageType.Logo, null, cancellationToken).ConfigureAwait(false); } } cancellationToken.ThrowIfCancellationRequested(); if (ConfigurationManager.Configuration.DownloadMovieImages.Art && !item.HasImage(ImageType.Art)) { var node = doc.SelectSingleNode("//fanart/movie/hdmoviecleararts/hdmovieclearart[@lang = \"" + language + "\"]/@url") ?? doc.SelectSingleNode("//fanart/movie/hdmoviecleararts/hdmovieclearart/@url") ?? doc.SelectSingleNode("//fanart/movie/moviearts/movieart[@lang = \"" + language + "\"]/@url") ?? doc.SelectSingleNode("//fanart/movie/moviearts/movieart/@url"); path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { await _providerManager.SaveImage(item, path, FanArtResourcePool, ImageType.Art, null, cancellationToken) .ConfigureAwait(false); } } cancellationToken.ThrowIfCancellationRequested(); if (ConfigurationManager.Configuration.DownloadMovieImages.Disc && !item.HasImage(ImageType.Disc)) { var node = doc.SelectSingleNode("//fanart/movie/moviediscs/moviedisc[@lang = \"" + language + "\"]/@url") ?? doc.SelectSingleNode("//fanart/movie/moviediscs/moviedisc/@url"); path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { await _providerManager.SaveImage(item, path, FanArtResourcePool, ImageType.Disc, null, cancellationToken) .ConfigureAwait(false); } } cancellationToken.ThrowIfCancellationRequested(); if (ConfigurationManager.Configuration.DownloadMovieImages.Banner && !item.HasImage(ImageType.Banner)) { var node = doc.SelectSingleNode("//fanart/movie/moviebanners/moviebanner[@lang = \"" + language + "\"]/@url") ?? doc.SelectSingleNode("//fanart/movie/moviebanners/moviebanner/@url"); path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { await _providerManager.SaveImage(item, path, FanArtResourcePool, ImageType.Banner, null, cancellationToken) .ConfigureAwait(false); } } cancellationToken.ThrowIfCancellationRequested(); if (ConfigurationManager.Configuration.DownloadMovieImages.Thumb && !item.HasImage(ImageType.Thumb)) { var node = doc.SelectSingleNode("//fanart/movie/moviethumbs/moviethumb[@lang = \"" + language + "\"]/@url") ?? doc.SelectSingleNode("//fanart/movie/moviethumbs/moviethumb/@url"); path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { await _providerManager.SaveImage(item, path, FanArtResourcePool, ImageType.Thumb, null, cancellationToken) .ConfigureAwait(false); } } var backdropLimit = ConfigurationManager.Configuration.MaxBackdrops; if (ConfigurationManager.Configuration.DownloadMovieImages.Backdrops && item.BackdropImagePaths.Count < backdropLimit) { var nodes = doc.SelectNodes("//fanart/movie/moviebackgrounds//@url"); if (nodes != null) { var numBackdrops = item.BackdropImagePaths.Count; foreach (XmlNode node in nodes) { path = node.Value; if (!string.IsNullOrEmpty(path) && !item.ContainsImageWithSourceUrl(path)) { await _providerManager.SaveImage(item, path, FanArtResourcePool, ImageType.Backdrop, numBackdrops, cancellationToken) .ConfigureAwait(false); numBackdrops++; if (item.BackdropImagePaths.Count >= backdropLimit) break; } } } } } } }