123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237 |
- using MediaBrowser.Common.Progress;
- using MediaBrowser.Controller.Entities;
- using MediaBrowser.Controller.Entities.Audio;
- using MediaBrowser.Controller.Library;
- using MediaBrowser.Model.Dto;
- using MediaBrowser.Model.Entities;
- using MediaBrowser.Model.Logging;
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Linq;
- using System.Threading;
- using System.Threading.Tasks;
- namespace MediaBrowser.Server.Implementations.Library.Validators
- {
- /// <summary>
- /// Class ArtistsValidator
- /// </summary>
- public class ArtistsValidator
- {
- /// <summary>
- /// The _library manager
- /// </summary>
- private readonly ILibraryManager _libraryManager;
- /// <summary>
- /// The _user manager
- /// </summary>
- private readonly IUserManager _userManager;
- /// <summary>
- /// The _logger
- /// </summary>
- private readonly ILogger _logger;
- /// <summary>
- /// Initializes a new instance of the <see cref="ArtistsPostScanTask" /> class.
- /// </summary>
- /// <param name="libraryManager">The library manager.</param>
- /// <param name="userManager">The user manager.</param>
- /// <param name="logger">The logger.</param>
- public ArtistsValidator(ILibraryManager libraryManager, IUserManager userManager, ILogger logger)
- {
- _libraryManager = libraryManager;
- _userManager = userManager;
- _logger = logger;
- }
- /// <summary>
- /// Runs the specified progress.
- /// </summary>
- /// <param name="progress">The progress.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task.</returns>
- public async Task Run(IProgress<double> progress, CancellationToken cancellationToken)
- {
- var allItems = _libraryManager.RootFolder.GetRecursiveChildren();
- var allMusicArtists = allItems.OfType<MusicArtist>().ToList();
- var allSongs = allItems.OfType<Audio>().ToList();
- var innerProgress = new ActionableProgress<double>();
- innerProgress.RegisterAction(pct => progress.Report(pct * .8));
- var allArtists = await GetAllArtists(allSongs, cancellationToken, innerProgress).ConfigureAwait(false);
- progress.Report(80);
- var numComplete = 0;
- var userLibraries = _userManager.Users
- .Select(i => new Tuple<Guid, List<IHasArtist>>(i.Id, i.RootFolder.GetRecursiveChildren(i).OfType<IHasArtist>().ToList()))
- .ToList();
- var numArtists = allArtists.Count;
- foreach (var artist in allArtists)
- {
- cancellationToken.ThrowIfCancellationRequested();
- artist.ValidateImages();
- artist.ValidateBackdrops();
- var musicArtist = Artist.FindMusicArtist(artist, allMusicArtists);
- if (musicArtist != null)
- {
- MergeImages(musicArtist.Images, artist.Images);
- // Merge backdrops
- var backdrops = musicArtist.BackdropImagePaths.ToList();
- backdrops.InsertRange(0, artist.BackdropImagePaths);
- artist.BackdropImagePaths = backdrops.Distinct(StringComparer.OrdinalIgnoreCase)
- .ToList();
- }
- if (!artist.LockedFields.Contains(MetadataFields.Genres))
- {
- // Avoid implicitly captured closure
- var artist1 = artist;
- artist.Genres = allSongs.Where(i => i.HasArtist(artist1.Name))
- .SelectMany(i => i.Genres)
- .Distinct(StringComparer.OrdinalIgnoreCase)
- .ToList();
- }
- // Populate counts of items
- //SetItemCounts(artist, null, allItems.OfType<IHasArtist>());
- foreach (var lib in userLibraries)
- {
- SetItemCounts(artist, lib.Item1, lib.Item2);
- }
- numComplete++;
- double percent = numComplete;
- percent /= numArtists;
- percent *= 20;
- progress.Report(80 + percent);
- }
- progress.Report(100);
- }
- /// <summary>
- /// Sets the item counts.
- /// </summary>
- /// <param name="artist">The artist.</param>
- /// <param name="userId">The user id.</param>
- /// <param name="allItems">All items.</param>
- private void SetItemCounts(Artist artist, Guid? userId, IEnumerable<IHasArtist> allItems)
- {
- var name = artist.Name;
- var items = allItems
- .Where(i => i.HasArtist(name))
- .ToList();
- var counts = new ItemByNameCounts
- {
- TotalCount = items.Count,
- SongCount = items.OfType<Audio>().Count(),
- AlbumCount = items.OfType<MusicAlbum>().Count(),
- MusicVideoCount = items.OfType<MusicVideo>().Count()
- };
- if (userId.HasValue)
- {
- artist.UserItemCounts[userId.Value] = counts;
- }
- }
- /// <summary>
- /// Merges the images.
- /// </summary>
- /// <param name="source">The source.</param>
- /// <param name="target">The target.</param>
- private void MergeImages(Dictionary<ImageType, string> source, Dictionary<ImageType, string> target)
- {
- foreach (var key in source.Keys
- .Where(k => !target.ContainsKey(k)))
- {
- string path;
- if (source.TryGetValue(key, out path))
- {
- target[key] = path;
- }
- }
- }
- /// <summary>
- /// Gets all artists.
- /// </summary>
- /// <param name="allSongs">All songs.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <param name="progress">The progress.</param>
- /// <returns>Task{Artist[]}.</returns>
- private async Task<List<Artist>> GetAllArtists(IEnumerable<Audio> allSongs, CancellationToken cancellationToken, IProgress<double> progress)
- {
- var allArtists = allSongs
- .SelectMany(i =>
- {
- var list = new List<string>();
- if (!string.IsNullOrEmpty(i.AlbumArtist))
- {
- list.Add(i.AlbumArtist);
- }
- list.AddRange(i.Artists);
- return list;
- })
- .Distinct(StringComparer.OrdinalIgnoreCase)
- .ToList();
- var returnArtists = new List<Artist>(allArtists.Count);
- var numComplete = 0;
- var numArtists = allArtists.Count;
- foreach (var artist in allArtists)
- {
- cancellationToken.ThrowIfCancellationRequested();
- try
- {
- var artistItem = _libraryManager.GetArtist(artist);
- await artistItem.RefreshMetadata(cancellationToken).ConfigureAwait(false);
- returnArtists.Add(artistItem);
- }
- catch (IOException ex)
- {
- _logger.ErrorException("Error validating Artist {0}", ex, artist);
- }
- // Update progress
- numComplete++;
- double percent = numComplete;
- percent /= numArtists;
- progress.Report(100 * percent);
- }
- return returnArtists;
- }
- }
- }
|