mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 03:48:06 +02:00
122 lines
4.3 KiB
C#
122 lines
4.3 KiB
C#
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
|
|
{
|
|
/// <summary>
|
|
/// Class PhotoResolver.
|
|
/// </summary>
|
|
public class PhotoResolver : ItemResolver<Photo>
|
|
{
|
|
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"
|
|
};
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="PhotoResolver"/> class.
|
|
/// </summary>
|
|
/// <param name="imageProcessor">The image processor.</param>
|
|
/// <param name="namingOptions">The naming options.</param>
|
|
/// <param name="directoryService">The directory service.</param>
|
|
public PhotoResolver(IImageProcessor imageProcessor, NamingOptions namingOptions, IDirectoryService directoryService)
|
|
{
|
|
_imageProcessor = imageProcessor;
|
|
_namingOptions = namingOptions;
|
|
_directoryService = directoryService;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Resolves the specified args.
|
|
/// </summary>
|
|
/// <param name="args">The args.</param>
|
|
/// <returns>Trailer.</returns>
|
|
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<char> imageFilename)
|
|
{
|
|
return VideoResolver.IsVideoFile(file, namingOptions) && IsOwnedByResolvedMedia(file, imageFilename);
|
|
}
|
|
|
|
internal static bool IsOwnedByResolvedMedia(ReadOnlySpan<char> file, ReadOnlySpan<char> 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;
|
|
}
|
|
}
|
|
}
|