using System; using System.IO; using System.Linq; using Emby.Naming.Common; using Emby.Naming.Video; using Jellyfin.Data.Enums; using Jellyfin.Extensions; using MediaBrowser.Controller.Drawing; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Providers; using MediaBrowser.Controller.Resolvers; using MediaBrowser.Model.Entities; namespace Emby.Server.Implementations.Library.Resolvers { /// /// Class PhotoResolver. /// public class PhotoResolver : ItemResolver { private readonly IImageProcessor _imageProcessor; private readonly NamingOptions _namingOptions; private readonly IDirectoryService _directoryService; private static readonly string[] _ignoreFiles = new[] { "folder", "thumb", "landscape", "fanart", "backdrop", "poster", "cover", "logo", "default" }; /// /// Initializes a new instance of the class. /// /// The image processor. /// The naming options. /// The directory service. public PhotoResolver(IImageProcessor imageProcessor, NamingOptions namingOptions, IDirectoryService directoryService) { _imageProcessor = imageProcessor; _namingOptions = namingOptions; _directoryService = directoryService; } /// /// Resolves the specified args. /// /// The args. /// Trailer. protected override Photo? Resolve(ItemResolveArgs args) { if (!args.IsDirectory) { // Must be an image file within a photo collection var collectionType = args.CollectionType; if (collectionType == CollectionType.photos || (collectionType == CollectionType.homevideos && args.LibraryOptions.EnablePhotos)) { if (IsImageFile(args.Path, _imageProcessor)) { var filename = Path.GetFileNameWithoutExtension(args.Path.AsSpan()); // Make sure the image doesn't belong to a video file var files = _directoryService.GetFiles(Path.GetDirectoryName(args.Path) ?? throw new InvalidOperationException("Path can't be a root directory.")); foreach (var file in files) { if (IsOwnedByMedia(_namingOptions, file.FullName, filename)) { return null; } } return new Photo { Path = args.Path }; } } } return null; } internal static bool IsOwnedByMedia(NamingOptions namingOptions, string file, ReadOnlySpan imageFilename) { return VideoResolver.IsVideoFile(file, namingOptions) && IsOwnedByResolvedMedia(file, imageFilename); } internal static bool IsOwnedByResolvedMedia(ReadOnlySpan file, ReadOnlySpan imageFilename) => imageFilename.StartsWith(Path.GetFileNameWithoutExtension(file), StringComparison.OrdinalIgnoreCase); internal static bool IsImageFile(string path, IImageProcessor imageProcessor) { ArgumentNullException.ThrowIfNull(path); var extension = Path.GetExtension(path.AsSpan()).TrimStart('.'); if (!imageProcessor.SupportedInputFormats.Contains(extension, StringComparison.OrdinalIgnoreCase)) { return false; } var filename = Path.GetFileNameWithoutExtension(path); if (_ignoreFiles.Any(i => filename.StartsWith(i, StringComparison.OrdinalIgnoreCase))) { return false; } return true; } } }