#nullable disable #pragma warning disable CS1591 using System; using System.IO; using System.Linq; using Jellyfin.Extensions; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Library; using MediaBrowser.Model.Entities; namespace Emby.Server.Implementations.Library.Resolvers.Books { public class BookResolver : MediaBrowser.Controller.Resolvers.ItemResolver { private readonly string[] _validExtensions = { ".azw", ".azw3", ".cb7", ".cbr", ".cbt", ".cbz", ".epub", ".mobi", ".pdf" }; public override Book Resolve(ItemResolveArgs args) { var collectionType = args.GetCollectionType(); // Only process items that are in a collection folder containing books if (!string.Equals(collectionType, CollectionType.Books, StringComparison.OrdinalIgnoreCase)) { return null; } if (args.IsDirectory) { return GetBook(args); } var extension = Path.GetExtension(args.Path); if (extension != null && _validExtensions.Contains(extension, StringComparison.OrdinalIgnoreCase)) { // It's a book return new Book { Path = args.Path, IsInMixedFolder = true }; } return null; } private Book GetBook(ItemResolveArgs args) { var bookFiles = args.FileSystemChildren.Where(f => { var fileExtension = Path.GetExtension(f.FullName) ?? string.Empty; return _validExtensions.Contains( fileExtension, StringComparer.OrdinalIgnoreCase); }).ToList(); // Don't return a Book if there is more (or less) than one document in the directory if (bookFiles.Count != 1) { return null; } return new Book { Path = bookFiles[0].FullName }; } } }