using MediaBrowser.Common.Net; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Entities.Audio; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using MediaBrowser.Model.Net; using System; using System.IO; using System.Text; using System.Threading; using System.Threading.Tasks; using System.Xml; namespace MediaBrowser.Controller.Providers.Music { /// /// Class FanArtAlbumProvider /// public class FanArtAlbumProvider : FanartBaseProvider { /// /// The _provider manager /// private readonly IProviderManager _providerManager; /// /// The _music brainz resource pool /// private readonly SemaphoreSlim _musicBrainzResourcePool = new SemaphoreSlim(1, 1); /// /// Gets the HTTP client. /// /// The HTTP client. protected IHttpClient HttpClient { get; private set; } internal static FanArtAlbumProvider Current { get; private set; } /// /// Initializes a new instance of the class. /// /// The HTTP client. /// The log manager. /// The configuration manager. /// The provider manager. public FanArtAlbumProvider(IHttpClient httpClient, ILogManager logManager, IServerConfigurationManager configurationManager, IProviderManager providerManager) : base(logManager, configurationManager) { _providerManager = providerManager; HttpClient = httpClient; Current = this; } /// /// Supportses the specified item. /// /// The item. /// true if XXXX, false otherwise public override bool Supports(BaseItem item) { return item is MusicAlbum; } /// /// 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 "12"; } } /// /// 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.Musicbrainz))) { return false; } if (!ConfigurationManager.Configuration.DownloadMusicAlbumImages.Disc && !ConfigurationManager.Configuration.DownloadMusicAlbumImages.Primary) { return false; } return base.NeedsRefreshInternal(item, providerInfo); } /// /// 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 album = (MusicAlbum)item; if (string.IsNullOrEmpty(album.MusicBrainzReleaseGroupId)) { album.MusicBrainzReleaseGroupId = await GetReleaseGroupId(item.GetProviderId(MetadataProviders.Musicbrainz), cancellationToken).ConfigureAwait(false); } // If still empty there's nothing more we can do if (string.IsNullOrEmpty(album.MusicBrainzReleaseGroupId)) { SetLastRefreshed(item, DateTime.UtcNow); return true; } var url = string.Format("http://api.fanart.tv/webservice/album/{0}/{1}/xml/all/1/1", APIKey, album.MusicBrainzReleaseGroupId); var doc = new XmlDocument(); var status = ProviderRefreshStatus.Success; using (var xml = await HttpClient.Get(new HttpRequestOptions { Url = url, ResourcePool = FanArtResourcePool, CancellationToken = cancellationToken, EnableResponseCache = true }).ConfigureAwait(false)) { doc.Load(xml); } cancellationToken.ThrowIfCancellationRequested(); if (doc.HasChildNodes) { if (ConfigurationManager.Configuration.DownloadMusicAlbumImages.Disc && !item.ResolveArgs.ContainsMetaFileByName(DISC_FILE)) { var node = doc.SelectSingleNode("//fanart/music/albums/album/cdart/@url"); var path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { Logger.Debug("FanArtProvider getting Disc for " + item.Name); try { item.SetImage(ImageType.Disc, await _providerManager.DownloadAndSaveImage(item, path, DISC_FILE, ConfigurationManager.Configuration.SaveLocalMeta, FanArtResourcePool, cancellationToken).ConfigureAwait(false)); } catch (HttpException) { status = ProviderRefreshStatus.CompletedWithErrors; } } } if (ConfigurationManager.Configuration.DownloadMusicAlbumImages.Primary && !item.ResolveArgs.ContainsMetaFileByName(PRIMARY_FILE)) { var node = doc.SelectSingleNode("//fanart/music/albums/album/albumcover/@url"); var path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { Logger.Debug("FanArtProvider getting albumcover for " + item.Name); try { item.SetImage(ImageType.Primary, await _providerManager.DownloadAndSaveImage(item, path, PRIMARY_FILE, ConfigurationManager.Configuration.SaveLocalMeta, FanArtResourcePool, cancellationToken).ConfigureAwait(false)); } catch (HttpException) { status = ProviderRefreshStatus.CompletedWithErrors; } } } } SetLastRefreshed(item, DateTime.UtcNow, status); return true; } /// /// The _last music brainz request /// private DateTime _lastMusicBrainzRequest = DateTime.MinValue; private readonly SemaphoreSlim _musicBrainzSemaphore = new SemaphoreSlim(1, 1); /// /// Gets the music brainz response. /// /// The URL. /// The cancellation token. /// Task{XmlDocument}. internal async Task GetMusicBrainzResponse(string url, CancellationToken cancellationToken) { await _musicBrainzResourcePool.WaitAsync(cancellationToken).ConfigureAwait(false); try { var diff = 1500 - (DateTime.Now - _lastMusicBrainzRequest).TotalMilliseconds; // MusicBrainz is extremely adamant about limiting to one request per second if (diff > 0) { await Task.Delay(Convert.ToInt32(diff), cancellationToken).ConfigureAwait(false); } _lastMusicBrainzRequest = DateTime.Now; var doc = new XmlDocument(); using (var xml = await HttpClient.Get(new HttpRequestOptions { Url = url, CancellationToken = cancellationToken, ResourcePool = _musicBrainzSemaphore, UserAgent = "MediaBrowserServer/www.mediabrowser3.com", EnableResponseCache = true }).ConfigureAwait(false)) { using (var oReader = new StreamReader(xml, Encoding.UTF8)) { doc.Load(oReader); } } return doc; } finally { _lastMusicBrainzRequest = DateTime.Now; _musicBrainzResourcePool.Release(); } } /// /// Gets the release group id internal. /// /// The release entry id. /// The cancellation token. /// Task{System.String}. private async Task GetReleaseGroupId(string releaseEntryId, CancellationToken cancellationToken) { var url = string.Format("http://www.musicbrainz.org/ws/2/release-group/?query=reid:{0}", releaseEntryId); var doc = await GetMusicBrainzResponse(url, cancellationToken).ConfigureAwait(false); var ns = new XmlNamespaceManager(doc.NameTable); ns.AddNamespace("mb", "http://musicbrainz.org/ns/mmd-2.0#"); var node = doc.SelectSingleNode("//mb:release-group-list/mb:release-group/@id", ns); return node != null ? node.Value : null; } } }