using MediaBrowser.Common.Extensions; using MediaBrowser.Common.Net; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Entities.Audio; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using System; using System.IO; using System.Text; using System.Threading; using System.Threading.Tasks; using System.Xml; namespace MediaBrowser.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 "17"; } } /// /// 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; } var comparisonData = Guid.Empty; var artistMusicBrainzId = item.Parent.GetProviderId(MetadataProviders.Musicbrainz); if (!string.IsNullOrEmpty(artistMusicBrainzId)) { var artistXmlPath = FanArtArtistProvider.GetArtistDataPath(ConfigurationManager.CommonApplicationPaths, artistMusicBrainzId); artistXmlPath = Path.Combine(artistXmlPath, "fanart.xml"); comparisonData = GetComparisonData(new FileInfo(artistXmlPath)); } // Refresh anytime the parent mbz id changes if (providerInfo.Data != comparisonData) { return true; } return base.NeedsRefreshInternal(item, providerInfo); } /// /// Gets the comparison data. /// /// Guid. private Guid GetComparisonData(FileInfo artistXmlFileInfo) { return artistXmlFileInfo.Exists ? (artistXmlFileInfo.FullName + artistXmlFileInfo.LastWriteTimeUtc.Ticks).GetMD5() : Guid.Empty; } /// /// 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 artistMusicBrainzId = item.Parent.GetProviderId(MetadataProviders.Musicbrainz); BaseProviderInfo data; if (!item.ProviderData.TryGetValue(Id, out data)) { data = new BaseProviderInfo(); item.ProviderData[Id] = data; } var comparisonData = Guid.Empty; if (!string.IsNullOrEmpty(artistMusicBrainzId)) { var artistXmlPath = FanArtArtistProvider.GetArtistDataPath(ConfigurationManager.CommonApplicationPaths, artistMusicBrainzId); artistXmlPath = Path.Combine(artistXmlPath, "fanart.xml"); var artistXmlFileInfo = new FileInfo(artistXmlPath); comparisonData = GetComparisonData(artistXmlFileInfo); if (artistXmlFileInfo.Exists) { var album = (MusicAlbum)item; var releaseEntryId = item.GetProviderId(MetadataProviders.Musicbrainz); // Fanart uses the release group id so we'll have to get that now using the release entry id if (string.IsNullOrEmpty(album.MusicBrainzReleaseGroupId)) { album.MusicBrainzReleaseGroupId = await GetReleaseGroupId(releaseEntryId, cancellationToken).ConfigureAwait(false); } var doc = new XmlDocument(); doc.Load(artistXmlPath); cancellationToken.ThrowIfCancellationRequested(); if (ConfigurationManager.Configuration.DownloadMusicAlbumImages.Disc && !item.HasImage(ImageType.Disc)) { // Try try with the release entry Id, if that doesn't produce anything try the release group id var node = doc.SelectSingleNode("//fanart/music/albums/album[@id=\"" + releaseEntryId + "\"]/cdart/@url"); if (node == null && !string.IsNullOrEmpty(album.MusicBrainzReleaseGroupId)) { node = doc.SelectSingleNode("//fanart/music/albums/album[@id=\"" + album.MusicBrainzReleaseGroupId + "\"]/cdart/@url"); } var path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { item.SetImage(ImageType.Disc, await _providerManager.DownloadAndSaveImage(item, path, DiscFile, ConfigurationManager.Configuration.SaveLocalMeta, FanArtResourcePool, cancellationToken).ConfigureAwait(false)); } } if (ConfigurationManager.Configuration.DownloadMusicAlbumImages.Primary && !item.HasImage(ImageType.Primary)) { // Try try with the release entry Id, if that doesn't produce anything try the release group id var node = doc.SelectSingleNode("//fanart/music/albums/album[@id=\"" + releaseEntryId + "\"]/albumcover/@url"); if (node == null && !string.IsNullOrEmpty(album.MusicBrainzReleaseGroupId)) { node = doc.SelectSingleNode("//fanart/music/albums/album[@id=\"" + album.MusicBrainzReleaseGroupId + "\"]/albumcover/@url"); } var path = node != null ? node.Value : null; if (!string.IsNullOrEmpty(path)) { item.SetImage(ImageType.Primary, await _providerManager.DownloadAndSaveImage(item, path, PrimaryFile, ConfigurationManager.Configuration.SaveLocalMeta, FanArtResourcePool, cancellationToken).ConfigureAwait(false)); } } } } data.Data = comparisonData; SetLastRefreshed(item, DateTime.UtcNow); return true; } /// /// The _last music brainz request /// private DateTime _lastRequestDate = DateTime.MinValue; /// /// 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 - _lastRequestDate).TotalMilliseconds; // MusicBrainz is extremely adamant about limiting to one request per second if (diff > 0) { await Task.Delay(Convert.ToInt32(diff), cancellationToken).ConfigureAwait(false); } _lastRequestDate = DateTime.Now; var doc = new XmlDocument(); using (var xml = await HttpClient.Get(new HttpRequestOptions { Url = url, CancellationToken = cancellationToken, UserAgent = Environment.MachineName }).ConfigureAwait(false)) { using (var oReader = new StreamReader(xml, Encoding.UTF8)) { doc.Load(oReader); } } return doc; } finally { _lastRequestDate = 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; } } }