MusicArtistResolver.cs 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. #nullable disable
  2. using System;
  3. using System.Linq;
  4. using System.Threading.Tasks;
  5. using MediaBrowser.Controller.Configuration;
  6. using MediaBrowser.Controller.Entities.Audio;
  7. using MediaBrowser.Controller.Library;
  8. using MediaBrowser.Controller.Resolvers;
  9. using MediaBrowser.Model.Entities;
  10. using MediaBrowser.Model.IO;
  11. using Microsoft.Extensions.Logging;
  12. namespace Emby.Server.Implementations.Library.Resolvers.Audio
  13. {
  14. /// <summary>
  15. /// Class MusicArtistResolver.
  16. /// </summary>
  17. public class MusicArtistResolver : ItemResolver<MusicArtist>
  18. {
  19. private readonly ILogger<MusicAlbumResolver> _logger;
  20. private readonly IFileSystem _fileSystem;
  21. private readonly ILibraryManager _libraryManager;
  22. private readonly IServerConfigurationManager _config;
  23. /// <summary>
  24. /// Initializes a new instance of the <see cref="MusicArtistResolver"/> class.
  25. /// </summary>
  26. /// <param name="logger">The logger for the created <see cref="MusicAlbumResolver"/> instances.</param>
  27. /// <param name="fileSystem">The file system.</param>
  28. /// <param name="libraryManager">The library manager.</param>
  29. /// <param name="config">The configuration manager.</param>
  30. public MusicArtistResolver(
  31. ILogger<MusicAlbumResolver> logger,
  32. IFileSystem fileSystem,
  33. ILibraryManager libraryManager,
  34. IServerConfigurationManager config)
  35. {
  36. _logger = logger;
  37. _fileSystem = fileSystem;
  38. _libraryManager = libraryManager;
  39. _config = config;
  40. }
  41. /// <summary>
  42. /// Gets the priority.
  43. /// </summary>
  44. /// <value>The priority.</value>
  45. public override ResolverPriority Priority => ResolverPriority.Second;
  46. /// <summary>
  47. /// Resolves the specified args.
  48. /// </summary>
  49. /// <param name="args">The args.</param>
  50. /// <returns>MusicArtist.</returns>
  51. protected override MusicArtist Resolve(ItemResolveArgs args)
  52. {
  53. if (!args.IsDirectory)
  54. {
  55. return null;
  56. }
  57. // Don't allow nested artists
  58. if (args.HasParent<MusicArtist>() || args.HasParent<MusicAlbum>())
  59. {
  60. return null;
  61. }
  62. var collectionType = args.GetCollectionType();
  63. var isMusicMediaFolder = string.Equals(collectionType, CollectionType.Music, StringComparison.OrdinalIgnoreCase);
  64. // If there's a collection type and it's not music, it can't be a series
  65. if (!isMusicMediaFolder)
  66. {
  67. return null;
  68. }
  69. if (args.ContainsFileSystemEntryByName("artist.nfo"))
  70. {
  71. return new MusicArtist();
  72. }
  73. // Avoid mis-identifying top folders
  74. if (args.Parent.IsRoot)
  75. {
  76. return null;
  77. }
  78. var directoryService = args.DirectoryService;
  79. var albumResolver = new MusicAlbumResolver(_logger, _fileSystem, _libraryManager);
  80. // If we contain an album assume we are an artist folder
  81. var directories = args.FileSystemChildren.Where(i => i.IsDirectory);
  82. var result = Parallel.ForEach(directories, (fileSystemInfo, state) =>
  83. {
  84. if (albumResolver.IsMusicAlbum(fileSystemInfo.FullName, directoryService))
  85. {
  86. // stop once we see a music album
  87. state.Stop();
  88. }
  89. });
  90. return !result.IsCompleted ? new MusicArtist() : null;
  91. }
  92. }
  93. }