497 lines
17 KiB
C#
497 lines
17 KiB
C#
using MediaBrowser.Common.IO;
|
|
using MediaBrowser.Controller;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.Movies;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.Providers;
|
|
using MediaBrowser.Controller.Resolvers;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Logging;
|
|
using MediaBrowser.Naming.Common;
|
|
using MediaBrowser.Naming.IO;
|
|
using MediaBrowser.Naming.Video;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
|
|
namespace MediaBrowser.Server.Implementations.Library.Resolvers.Movies
|
|
{
|
|
/// <summary>
|
|
/// Class MovieResolver
|
|
/// </summary>
|
|
public class MovieResolver : BaseVideoResolver<Video>, IMultiItemResolver
|
|
{
|
|
private readonly IServerApplicationPaths _applicationPaths;
|
|
private readonly ILogger _logger;
|
|
private readonly IFileSystem _fileSystem;
|
|
|
|
public MovieResolver(ILibraryManager libraryManager, IServerApplicationPaths applicationPaths, ILogger logger, IFileSystem fileSystem)
|
|
: base(libraryManager)
|
|
{
|
|
_applicationPaths = applicationPaths;
|
|
_logger = logger;
|
|
_fileSystem = fileSystem;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the priority.
|
|
/// </summary>
|
|
/// <value>The priority.</value>
|
|
public override ResolverPriority Priority
|
|
{
|
|
get
|
|
{
|
|
// Give plugins a chance to catch iso's first
|
|
// Also since we have to loop through child files looking for videos,
|
|
// see if we can avoid some of that by letting other resolvers claim folders first
|
|
// Also run after series resolver
|
|
return ResolverPriority.Third;
|
|
}
|
|
}
|
|
|
|
public MultiItemResolverResult ResolveMultiple(Folder parent,
|
|
List<FileSystemInfo> files,
|
|
string collectionType,
|
|
IDirectoryService directoryService)
|
|
{
|
|
if (IsInvalid(parent, collectionType, files))
|
|
{
|
|
return null;
|
|
}
|
|
|
|
if (string.Equals(collectionType, CollectionType.MusicVideos, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return ResolveVideos<MusicVideo>(parent, files, directoryService, collectionType);
|
|
}
|
|
|
|
if (string.Equals(collectionType, CollectionType.HomeVideos, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return ResolveVideos<Video>(parent, files, directoryService, collectionType);
|
|
}
|
|
|
|
if (string.IsNullOrEmpty(collectionType))
|
|
{
|
|
// Owned items should just use the plain video type
|
|
if (parent == null)
|
|
{
|
|
return ResolveVideos<Video>(parent, files, directoryService, collectionType);
|
|
}
|
|
|
|
return ResolveVideos<Video>(parent, files, directoryService, collectionType);
|
|
}
|
|
|
|
if (string.Equals(collectionType, CollectionType.Movies, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return ResolveVideos<Movie>(parent, files, directoryService, collectionType);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
private MultiItemResolverResult ResolveVideos<T>(Folder parent, IEnumerable<FileSystemInfo> fileSystemEntries, IDirectoryService directoryService, string collectionType)
|
|
where T : Video, new()
|
|
{
|
|
var files = new List<FileSystemInfo>();
|
|
var videos = new List<BaseItem>();
|
|
var leftOver = new List<FileSystemInfo>();
|
|
|
|
// Loop through each child file/folder and see if we find a video
|
|
foreach (var child in fileSystemEntries)
|
|
{
|
|
if ((child.Attributes & FileAttributes.Directory) == FileAttributes.Directory)
|
|
{
|
|
leftOver.Add(child);
|
|
}
|
|
else
|
|
{
|
|
files.Add(child);
|
|
}
|
|
}
|
|
|
|
var resolver = new VideoListResolver(new ExtendedNamingOptions(), new Naming.Logging.NullLogger());
|
|
var resolverResult = resolver.Resolve(files.Select(i => new PortableFileInfo
|
|
{
|
|
FullName = i.FullName,
|
|
Type = FileInfoType.File
|
|
|
|
}).ToList(), false).ToList();
|
|
|
|
var result = new MultiItemResolverResult
|
|
{
|
|
ExtraFiles = leftOver,
|
|
Items = videos
|
|
};
|
|
|
|
var isInMixedFolder = resolverResult.Count > 0;
|
|
|
|
foreach (var video in resolverResult)
|
|
{
|
|
var firstVideo = video.Files.First();
|
|
|
|
var videoItem = new T
|
|
{
|
|
Path = video.Files[0].Path,
|
|
IsInMixedFolder = isInMixedFolder,
|
|
ProductionYear = video.Year,
|
|
Name = video.Name,
|
|
AdditionalParts = video.Files.Skip(1).Select(i => i.Path).ToList()
|
|
};
|
|
|
|
SetVideoType(videoItem, firstVideo);
|
|
Set3DFormat(videoItem, firstVideo);
|
|
|
|
result.Items.Add(videoItem);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Resolves the specified args.
|
|
/// </summary>
|
|
/// <param name="args">The args.</param>
|
|
/// <returns>Video.</returns>
|
|
protected override Video Resolve(ItemResolveArgs args)
|
|
{
|
|
var collectionType = args.GetCollectionType();
|
|
|
|
if (IsInvalid(args.Parent, collectionType, args.FileSystemChildren))
|
|
{
|
|
return null;
|
|
}
|
|
|
|
// Find movies with their own folders
|
|
if (args.IsDirectory)
|
|
{
|
|
if (string.Equals(collectionType, CollectionType.MusicVideos, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return FindMovie<MusicVideo>(args.Path, args.Parent, args.FileSystemChildren.ToList(), args.DirectoryService, collectionType);
|
|
}
|
|
|
|
if (string.Equals(collectionType, CollectionType.HomeVideos, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return FindMovie<Video>(args.Path, args.Parent, args.FileSystemChildren.ToList(), args.DirectoryService, collectionType);
|
|
}
|
|
|
|
if (string.IsNullOrEmpty(collectionType))
|
|
{
|
|
// Owned items should just use the plain video type
|
|
if (args.Parent == null)
|
|
{
|
|
return FindMovie<Video>(args.Path, args.Parent, args.FileSystemChildren.ToList(), args.DirectoryService, collectionType);
|
|
}
|
|
|
|
return FindMovie<Movie>(args.Path, args.Parent, args.FileSystemChildren.ToList(), args.DirectoryService, collectionType);
|
|
}
|
|
|
|
if (string.Equals(collectionType, CollectionType.Movies, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return FindMovie<Movie>(args.Path, args.Parent, args.FileSystemChildren.ToList(), args.DirectoryService, collectionType);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
// Owned items will be caught by the plain video resolver
|
|
if (args.Parent == null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
Video item = null;
|
|
|
|
if (string.Equals(collectionType, CollectionType.MusicVideos, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
item = ResolveVideo<MusicVideo>(args, false);
|
|
}
|
|
|
|
// To find a movie file, the collection type must be movies or boxsets
|
|
else if (string.Equals(collectionType, CollectionType.Movies, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
item = ResolveVideo<Movie>(args, true);
|
|
}
|
|
|
|
else if (string.Equals(collectionType, CollectionType.HomeVideos, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
item = ResolveVideo<Video>(args, false);
|
|
}
|
|
else if (string.IsNullOrEmpty(collectionType))
|
|
{
|
|
item = ResolveVideo<Video>(args, false);
|
|
}
|
|
|
|
if (item != null)
|
|
{
|
|
item.IsInMixedFolder = true;
|
|
}
|
|
|
|
return item;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets the initial item values.
|
|
/// </summary>
|
|
/// <param name="item">The item.</param>
|
|
/// <param name="args">The args.</param>
|
|
protected override void SetInitialItemValues(Video item, ItemResolveArgs args)
|
|
{
|
|
base.SetInitialItemValues(item, args);
|
|
|
|
SetProviderIdFromPath(item);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets the provider id from path.
|
|
/// </summary>
|
|
/// <param name="item">The item.</param>
|
|
private void SetProviderIdFromPath(Video item)
|
|
{
|
|
//we need to only look at the name of this actual item (not parents)
|
|
var justName = item.IsInMixedFolder ? Path.GetFileName(item.Path) : Path.GetFileName(item.ContainingFolderPath);
|
|
|
|
var id = justName.GetAttributeValue("tmdbid");
|
|
|
|
if (!string.IsNullOrEmpty(id))
|
|
{
|
|
item.SetProviderId(MetadataProviders.Tmdb, id);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Finds a movie based on a child file system entries
|
|
/// </summary>
|
|
/// <typeparam name="T"></typeparam>
|
|
/// <param name="path">The path.</param>
|
|
/// <param name="parent">The parent.</param>
|
|
/// <param name="fileSystemEntries">The file system entries.</param>
|
|
/// <param name="directoryService">The directory service.</param>
|
|
/// <param name="collectionType">Type of the collection.</param>
|
|
/// <returns>Movie.</returns>
|
|
private T FindMovie<T>(string path, Folder parent, List<FileSystemInfo> fileSystemEntries, IDirectoryService directoryService, string collectionType)
|
|
where T : Video, new()
|
|
{
|
|
var multiDiscFolders = new List<FileSystemInfo>();
|
|
|
|
// Search for a folder rip
|
|
foreach (var child in fileSystemEntries)
|
|
{
|
|
var filename = child.Name;
|
|
|
|
if ((child.Attributes & FileAttributes.Directory) == FileAttributes.Directory)
|
|
{
|
|
if (IsDvdDirectory(filename))
|
|
{
|
|
var movie = new T
|
|
{
|
|
Path = path,
|
|
VideoType = VideoType.Dvd
|
|
};
|
|
Set3DFormat(movie);
|
|
return movie;
|
|
}
|
|
if (IsBluRayDirectory(filename))
|
|
{
|
|
var movie = new T
|
|
{
|
|
Path = path,
|
|
VideoType = VideoType.BluRay
|
|
};
|
|
Set3DFormat(movie);
|
|
return movie;
|
|
}
|
|
|
|
multiDiscFolders.Add(child);
|
|
}
|
|
else if (IsDvdFile(filename))
|
|
{
|
|
var movie = new T
|
|
{
|
|
Path = path,
|
|
VideoType = VideoType.Dvd
|
|
};
|
|
Set3DFormat(movie);
|
|
return movie;
|
|
}
|
|
}
|
|
|
|
var result = ResolveVideos<T>(parent, fileSystemEntries, directoryService, collectionType);
|
|
|
|
var supportsMultiVersion = !string.Equals(collectionType, CollectionType.HomeVideos) &&
|
|
!string.Equals(collectionType, CollectionType.MusicVideos);
|
|
|
|
// Test for multi-editions
|
|
if (result.Items.Count > 1 && supportsMultiVersion)
|
|
{
|
|
var filenamePrefix = Path.GetFileName(path);
|
|
|
|
if (!string.IsNullOrWhiteSpace(filenamePrefix))
|
|
{
|
|
if (result.Items.All(i => _fileSystem.GetFileNameWithoutExtension(i.Path).StartsWith(filenamePrefix + " - ", StringComparison.OrdinalIgnoreCase)))
|
|
{
|
|
var movie = (T)result.Items[0];
|
|
movie.Name = filenamePrefix;
|
|
movie.LocalAlternateVersions = result.Items.Skip(1).Select(i => i.Path).ToList();
|
|
movie.IsInMixedFolder = false;
|
|
|
|
_logger.Debug("Multi-version video found: " + movie.Path);
|
|
|
|
return movie;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (result.Items.Count == 1)
|
|
{
|
|
var movie = (T)result.Items[0];
|
|
movie.IsInMixedFolder = false;
|
|
return movie;
|
|
}
|
|
|
|
if (result.Items.Count == 0 && multiDiscFolders.Count > 0)
|
|
{
|
|
return GetMultiDiscMovie<T>(multiDiscFolders, directoryService);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the multi disc movie.
|
|
/// </summary>
|
|
/// <typeparam name="T"></typeparam>
|
|
/// <param name="multiDiscFolders">The folders.</param>
|
|
/// <param name="directoryService">The directory service.</param>
|
|
/// <returns>``0.</returns>
|
|
private T GetMultiDiscMovie<T>(List<FileSystemInfo> multiDiscFolders, IDirectoryService directoryService)
|
|
where T : Video, new()
|
|
{
|
|
var videoTypes = new List<VideoType>();
|
|
|
|
var folderPaths = multiDiscFolders.Select(i => i.FullName).Where(i =>
|
|
{
|
|
var subFileEntries = directoryService.GetFileSystemEntries(i)
|
|
.ToList();
|
|
|
|
var subfolders = subFileEntries
|
|
.Where(e => (e.Attributes & FileAttributes.Directory) == FileAttributes.Directory)
|
|
.Select(d => d.Name)
|
|
.ToList();
|
|
|
|
if (subfolders.Any(IsDvdDirectory))
|
|
{
|
|
videoTypes.Add(VideoType.Dvd);
|
|
return true;
|
|
}
|
|
if (subfolders.Any(IsBluRayDirectory))
|
|
{
|
|
videoTypes.Add(VideoType.BluRay);
|
|
return true;
|
|
}
|
|
|
|
var subFiles = subFileEntries
|
|
.Where(e => (e.Attributes & FileAttributes.Directory) != FileAttributes.Directory)
|
|
.Select(d => d.Name);
|
|
|
|
if (subFiles.Any(IsDvdFile))
|
|
{
|
|
videoTypes.Add(VideoType.Dvd);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
|
|
}).OrderBy(i => i).ToList();
|
|
|
|
// If different video types were found, don't allow this
|
|
if (videoTypes.Distinct().Count() > 1)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
if (folderPaths.Count == 0)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
var resolver = new StackResolver(new ExtendedNamingOptions(), new Naming.Logging.NullLogger());
|
|
|
|
var result = resolver.ResolveDirectories(folderPaths);
|
|
|
|
if (result.Stacks.Count != 1)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
return new T
|
|
{
|
|
Path = folderPaths[0],
|
|
|
|
AdditionalParts = folderPaths.Skip(1).ToList(),
|
|
|
|
VideoType = videoTypes[0],
|
|
|
|
Name = result.Stacks[0].Name
|
|
};
|
|
}
|
|
|
|
private bool IsInvalid(Folder parent, string collectionType, IEnumerable<FileSystemInfo> files)
|
|
{
|
|
if (parent != null)
|
|
{
|
|
if (parent.IsRoot)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
// Don't do any resolving within a series structure
|
|
if (string.IsNullOrEmpty(collectionType))
|
|
{
|
|
if (HasParent<Series>(parent) || HasParent<Season>(parent))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
// Since the looping is expensive, this is an optimization to help us avoid it
|
|
if (files.Select(i => i.Name).Contains("series.xml", StringComparer.OrdinalIgnoreCase))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
var validCollectionTypes = new[]
|
|
{
|
|
string.Empty,
|
|
CollectionType.Movies,
|
|
CollectionType.HomeVideos,
|
|
CollectionType.MusicVideos,
|
|
CollectionType.Movies
|
|
};
|
|
|
|
return !validCollectionTypes.Contains(collectionType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
|
|
}
|
|
|
|
private bool HasParent<T>(Folder parent)
|
|
where T : Folder
|
|
{
|
|
if (parent != null)
|
|
{
|
|
var item = parent as T;
|
|
|
|
// Just in case the user decided to nest episodes.
|
|
// Not officially supported but in some cases we can handle it.
|
|
if (item == null)
|
|
{
|
|
item = parent.Parents.OfType<T>().FirstOrDefault();
|
|
}
|
|
|
|
return item != null;
|
|
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
}
|