jellyfin/MediaBrowser.Server.Implementations/Library/Resolvers/Audio/MusicAlbumResolver.cs

174 lines
6.6 KiB
C#
Raw Normal View History

2014-07-26 17:30:15 +00:00
using MediaBrowser.Common.IO;
using MediaBrowser.Controller.Entities.Audio;
2013-09-18 02:43:34 +00:00
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library;
2014-02-13 05:11:54 +00:00
using MediaBrowser.Controller.Providers;
using MediaBrowser.Controller.Resolvers;
using MediaBrowser.Model.Entities;
2014-07-26 17:30:15 +00:00
using MediaBrowser.Model.Logging;
2014-11-16 20:44:08 +00:00
using MediaBrowser.Naming.Audio;
2014-12-15 05:16:23 +00:00
using MediaBrowser.Naming.Common;
using System;
using System.Collections.Generic;
using System.IO;
namespace MediaBrowser.Server.Implementations.Library.Resolvers.Audio
{
/// <summary>
/// Class MusicAlbumResolver
/// </summary>
public class MusicAlbumResolver : ItemResolver<MusicAlbum>
{
2014-07-26 17:30:15 +00:00
private readonly ILogger _logger;
private readonly IFileSystem _fileSystem;
2014-11-16 20:44:08 +00:00
private readonly ILibraryManager _libraryManager;
2014-07-26 17:30:15 +00:00
2014-11-16 20:44:08 +00:00
public MusicAlbumResolver(ILogger logger, IFileSystem fileSystem, ILibraryManager libraryManager)
2014-07-26 17:30:15 +00:00
{
_logger = logger;
_fileSystem = fileSystem;
2014-11-16 20:44:08 +00:00
_libraryManager = libraryManager;
2014-07-26 17:30:15 +00:00
}
/// <summary>
/// Gets the priority.
/// </summary>
/// <value>The priority.</value>
public override ResolverPriority Priority
{
2014-12-20 06:06:27 +00:00
get
{
// Behind special folder resolver
return ResolverPriority.Second;
}
}
/// <summary>
/// Resolves the specified args.
/// </summary>
/// <param name="args">The args.</param>
/// <returns>MusicAlbum.</returns>
protected override MusicAlbum Resolve(ItemResolveArgs args)
{
if (!args.IsDirectory) return null;
2014-12-20 06:06:27 +00:00
// Avoid mis-identifying top folders
if (args.Parent.IsRoot) return null;
2014-12-15 05:16:23 +00:00
if (args.HasParent<MusicAlbum>()) return null;
var collectionType = args.GetCollectionType();
2014-12-20 06:06:27 +00:00
var isMusicMediaFolder = string.Equals(collectionType, CollectionType.Music, StringComparison.OrdinalIgnoreCase);
2014-07-30 03:31:35 +00:00
2014-03-06 05:17:13 +00:00
// If there's a collection type and it's not music, don't allow it.
2014-07-30 03:31:35 +00:00
if (!isMusicMediaFolder)
{
return null;
}
2014-06-29 17:58:04 +00:00
2014-11-16 20:44:08 +00:00
return IsMusicAlbum(args) ? new MusicAlbum() : null;
}
/// <summary>
/// Determine if the supplied file data points to a music album
/// </summary>
/// <param name="path">The path.</param>
2014-02-13 05:11:54 +00:00
/// <param name="directoryService">The directory service.</param>
2014-07-26 17:30:15 +00:00
/// <param name="logger">The logger.</param>
/// <param name="fileSystem">The file system.</param>
2014-11-16 20:44:08 +00:00
/// <param name="libraryManager">The library manager.</param>
/// <returns><c>true</c> if [is music album] [the specified data]; otherwise, <c>false</c>.</returns>
2014-11-16 20:44:08 +00:00
public static bool IsMusicAlbum(string path, IDirectoryService directoryService, ILogger logger, IFileSystem fileSystem,
ILibraryManager libraryManager)
{
2014-11-16 20:44:08 +00:00
return ContainsMusic(directoryService.GetFileSystemEntries(path), true, directoryService, logger, fileSystem, libraryManager);
}
/// <summary>
/// Determine if the supplied resolve args should be considered a music album
/// </summary>
/// <param name="args">The args.</param>
/// <returns><c>true</c> if [is music album] [the specified args]; otherwise, <c>false</c>.</returns>
2014-11-16 20:44:08 +00:00
private bool IsMusicAlbum(ItemResolveArgs args)
{
// Args points to an album if parent is an Artist folder or it directly contains music
if (args.IsDirectory)
{
//if (args.Parent is MusicArtist) return true; //saves us from testing children twice
2014-11-16 20:44:08 +00:00
if (ContainsMusic(args.FileSystemChildren, true, args.DirectoryService, _logger, _fileSystem, _libraryManager)) return true;
}
return false;
}
/// <summary>
/// Determine if the supplied list contains what we should consider music
/// </summary>
/// <param name="list">The list.</param>
2014-06-29 17:58:04 +00:00
/// <param name="allowSubfolders">if set to <c>true</c> [allow subfolders].</param>
/// <param name="directoryService">The directory service.</param>
2014-07-26 17:30:15 +00:00
/// <param name="logger">The logger.</param>
/// <param name="fileSystem">The file system.</param>
2014-11-16 20:44:08 +00:00
/// <param name="libraryManager">The library manager.</param>
/// <returns><c>true</c> if the specified list contains music; otherwise, <c>false</c>.</returns>
2014-08-27 03:25:39 +00:00
private static bool ContainsMusic(IEnumerable<FileSystemInfo> list,
bool allowSubfolders,
IDirectoryService directoryService,
ILogger logger,
2014-11-16 20:44:08 +00:00
IFileSystem fileSystem,
ILibraryManager libraryManager)
{
2014-07-26 17:30:15 +00:00
var discSubfolderCount = 0;
2014-12-28 06:21:39 +00:00
var notMultiDisc = false;
2014-07-26 17:30:15 +00:00
2014-06-25 15:12:39 +00:00
foreach (var fileSystemInfo in list)
{
2014-06-25 15:12:39 +00:00
if ((fileSystemInfo.Attributes & FileAttributes.Directory) == FileAttributes.Directory)
{
2014-12-28 06:21:39 +00:00
if (allowSubfolders)
2014-06-29 17:58:04 +00:00
{
2014-12-28 06:21:39 +00:00
var path = fileSystemInfo.FullName;
var isMultiDisc = IsMultiDiscFolder(path);
var hasMusic = ContainsMusic(directoryService.GetFileSystemEntries(path), false, directoryService, logger, fileSystem, libraryManager);
if (isMultiDisc && hasMusic)
{
logger.Debug("Found multi-disc folder: " + path);
discSubfolderCount++;
}
else if (hasMusic)
{
// If there are folders underneath with music that are not multidisc, then this can't be a multi-disc album
notMultiDisc = true;
}
2014-06-29 17:58:04 +00:00
}
2014-06-25 15:12:39 +00:00
}
var fullName = fileSystemInfo.FullName;
2013-09-18 02:43:34 +00:00
2014-11-16 20:44:08 +00:00
if (libraryManager.IsAudioFile(fullName))
2014-06-25 15:12:39 +00:00
{
return true;
}
}
2014-12-28 06:21:39 +00:00
if (notMultiDisc)
2014-06-29 17:58:04 +00:00
{
2014-12-28 06:21:39 +00:00
return false;
2014-06-29 17:58:04 +00:00
}
2014-12-28 06:21:39 +00:00
return discSubfolderCount > 0 && discSubfolderCount > 10;
2014-06-29 17:58:04 +00:00
}
2014-12-28 06:21:39 +00:00
private static bool IsMultiDiscFolder(string path)
2014-06-29 17:58:04 +00:00
{
2014-11-28 17:41:47 +00:00
var parser = new AlbumParser(new ExtendedNamingOptions(), new Naming.Logging.NullLogger());
2014-11-16 20:44:08 +00:00
var result = parser.ParseMultiPart(path);
return result.IsMultiPart;
2014-06-29 17:58:04 +00:00
}
}
}