| | | 1 | | using System; |
| | | 2 | | using System.IO; |
| | | 3 | | using System.Linq; |
| | | 4 | | using Emby.Naming.Common; |
| | | 5 | | using Emby.Naming.Video; |
| | | 6 | | using Jellyfin.Data.Enums; |
| | | 7 | | using Jellyfin.Extensions; |
| | | 8 | | using MediaBrowser.Controller.Drawing; |
| | | 9 | | using MediaBrowser.Controller.Entities; |
| | | 10 | | using MediaBrowser.Controller.Library; |
| | | 11 | | using MediaBrowser.Controller.Providers; |
| | | 12 | | using MediaBrowser.Controller.Resolvers; |
| | | 13 | | using MediaBrowser.Model.Entities; |
| | | 14 | | |
| | | 15 | | namespace Emby.Server.Implementations.Library.Resolvers |
| | | 16 | | { |
| | | 17 | | /// <summary> |
| | | 18 | | /// Class PhotoResolver. |
| | | 19 | | /// </summary> |
| | | 20 | | public class PhotoResolver : ItemResolver<Photo> |
| | | 21 | | { |
| | | 22 | | private readonly IImageProcessor _imageProcessor; |
| | | 23 | | private readonly NamingOptions _namingOptions; |
| | | 24 | | private readonly IDirectoryService _directoryService; |
| | | 25 | | |
| | 0 | 26 | | private static readonly string[] _ignoreFiles = new[] |
| | 0 | 27 | | { |
| | 0 | 28 | | "folder", |
| | 0 | 29 | | "thumb", |
| | 0 | 30 | | "landscape", |
| | 0 | 31 | | "fanart", |
| | 0 | 32 | | "backdrop", |
| | 0 | 33 | | "poster", |
| | 0 | 34 | | "cover", |
| | 0 | 35 | | "logo", |
| | 0 | 36 | | "default" |
| | 0 | 37 | | }; |
| | | 38 | | |
| | | 39 | | /// <summary> |
| | | 40 | | /// Initializes a new instance of the <see cref="PhotoResolver"/> class. |
| | | 41 | | /// </summary> |
| | | 42 | | /// <param name="imageProcessor">The image processor.</param> |
| | | 43 | | /// <param name="namingOptions">The naming options.</param> |
| | | 44 | | /// <param name="directoryService">The directory service.</param> |
| | 21 | 45 | | public PhotoResolver(IImageProcessor imageProcessor, NamingOptions namingOptions, IDirectoryService directorySer |
| | | 46 | | { |
| | 21 | 47 | | _imageProcessor = imageProcessor; |
| | 21 | 48 | | _namingOptions = namingOptions; |
| | 21 | 49 | | _directoryService = directoryService; |
| | 21 | 50 | | } |
| | | 51 | | |
| | | 52 | | /// <summary> |
| | | 53 | | /// Resolves the specified args. |
| | | 54 | | /// </summary> |
| | | 55 | | /// <param name="args">The args.</param> |
| | | 56 | | /// <returns>Trailer.</returns> |
| | | 57 | | protected override Photo? Resolve(ItemResolveArgs args) |
| | | 58 | | { |
| | 62 | 59 | | if (!args.IsDirectory) |
| | | 60 | | { |
| | | 61 | | // Must be an image file within a photo collection |
| | 12 | 62 | | var collectionType = args.CollectionType; |
| | | 63 | | |
| | 12 | 64 | | if (collectionType == CollectionType.photos |
| | 12 | 65 | | || (collectionType == CollectionType.homevideos && args.LibraryOptions.EnablePhotos)) |
| | | 66 | | { |
| | 0 | 67 | | if (IsImageFile(args.Path, _imageProcessor)) |
| | | 68 | | { |
| | 0 | 69 | | var filename = Path.GetFileNameWithoutExtension(args.Path.AsSpan()); |
| | | 70 | | |
| | | 71 | | // Make sure the image doesn't belong to a video file |
| | 0 | 72 | | var files = _directoryService.GetFiles(Path.GetDirectoryName(args.Path) |
| | 0 | 73 | | ?? throw new InvalidOperationException("Path can't be a root directory.")); |
| | | 74 | | |
| | 0 | 75 | | foreach (var file in files) |
| | | 76 | | { |
| | 0 | 77 | | if (IsOwnedByMedia(_namingOptions, file.FullName, filename)) |
| | | 78 | | { |
| | 0 | 79 | | return null; |
| | | 80 | | } |
| | | 81 | | } |
| | | 82 | | |
| | 0 | 83 | | return new Photo |
| | 0 | 84 | | { |
| | 0 | 85 | | Path = args.Path |
| | 0 | 86 | | }; |
| | | 87 | | } |
| | | 88 | | } |
| | | 89 | | } |
| | | 90 | | |
| | 62 | 91 | | return null; |
| | 0 | 92 | | } |
| | | 93 | | |
| | | 94 | | internal static bool IsOwnedByMedia(NamingOptions namingOptions, string file, ReadOnlySpan<char> imageFilename) |
| | | 95 | | { |
| | 0 | 96 | | return VideoResolver.IsVideoFile(file, namingOptions) && IsOwnedByResolvedMedia(file, imageFilename); |
| | | 97 | | } |
| | | 98 | | |
| | | 99 | | internal static bool IsOwnedByResolvedMedia(ReadOnlySpan<char> file, ReadOnlySpan<char> imageFilename) |
| | 0 | 100 | | => imageFilename.StartsWith(Path.GetFileNameWithoutExtension(file), StringComparison.OrdinalIgnoreCase); |
| | | 101 | | |
| | | 102 | | internal static bool IsImageFile(string path, IImageProcessor imageProcessor) |
| | | 103 | | { |
| | 0 | 104 | | ArgumentNullException.ThrowIfNull(path); |
| | | 105 | | |
| | 0 | 106 | | var extension = Path.GetExtension(path.AsSpan()).TrimStart('.'); |
| | 0 | 107 | | if (!imageProcessor.SupportedInputFormats.Contains(extension, StringComparison.OrdinalIgnoreCase)) |
| | | 108 | | { |
| | 0 | 109 | | return false; |
| | | 110 | | } |
| | | 111 | | |
| | 0 | 112 | | var filename = Path.GetFileNameWithoutExtension(path); |
| | | 113 | | |
| | 0 | 114 | | if (_ignoreFiles.Any(i => filename.StartsWith(i, StringComparison.OrdinalIgnoreCase))) |
| | | 115 | | { |
| | 0 | 116 | | return false; |
| | | 117 | | } |
| | | 118 | | |
| | 0 | 119 | | return true; |
| | | 120 | | } |
| | | 121 | | } |
| | | 122 | | } |