Kavita/API/Services/ArchiveService.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.IO.Compression;
using System.Linq;
using System.Threading.Tasks;
using System.Xml.Serialization;
using API.Archive;
using API.Data.Metadata;
using API.Extensions;
using API.Services.Tasks;
using Kavita.Common;
using Microsoft.Extensions.Logging;
using SharpCompress.Archives;
using SharpCompress.Common;

namespace API.Services
{
    public interface IArchiveService
    {
        void ExtractArchive(string archivePath, string extractPath);
        int GetNumberOfPagesFromArchive(string archivePath);
        string GetCoverImage(string archivePath, string fileName, string outputDirectory);
        bool IsValidArchive(string archivePath);
        ComicInfo GetComicInfo(string archivePath);
        ArchiveLibrary CanOpen(string archivePath);
        bool ArchiveNeedsFlattening(ZipArchive archive);
        Task<Tuple<byte[], string>> CreateZipForDownload(IEnumerable<string> files, string tempFolder);
        string FindCoverImageFilename(string archivePath, IList<string> entryNames);
    }

    /// <summary>
    /// Responsible for manipulating Archive files. Used by <see cref="CacheService"/> and <see cref="ScannerService"/>
    /// </summary>
    // ReSharper disable once ClassWithVirtualMembersNeverInherited.Global
    public class ArchiveService : IArchiveService
    {
        private readonly ILogger<ArchiveService> _logger;
        private readonly IDirectoryService _directoryService;
        private readonly IImageService _imageService;
        private const string ComicInfoFilename = "comicinfo";

        public ArchiveService(ILogger<ArchiveService> logger, IDirectoryService directoryService, IImageService imageService)
        {
            _logger = logger;
            _directoryService = directoryService;
            _imageService = imageService;
        }

        /// <summary>
        /// Checks if a File can be opened. Requires up to 2 opens of the filestream.
        /// </summary>
        /// <param name="archivePath"></param>
        /// <returns></returns>
        public virtual ArchiveLibrary CanOpen(string archivePath)
        {
            if (string.IsNullOrEmpty(archivePath) || !(File.Exists(archivePath) && Parser.Parser.IsArchive(archivePath) || Parser.Parser.IsEpub(archivePath))) return ArchiveLibrary.NotSupported;

            var ext = _directoryService.FileSystem.Path.GetExtension(archivePath).ToUpper();
            if (ext.Equals(".CBR") || ext.Equals(".RAR")) return ArchiveLibrary.SharpCompress;

            try
            {
                using var a2 = ZipFile.OpenRead(archivePath);
                return ArchiveLibrary.Default;
            }
            catch (Exception)
            {
                try
                {
                    using var a1 = ArchiveFactory.Open(archivePath);
                    return ArchiveLibrary.SharpCompress;
                }
                catch (Exception)
                {
                    return ArchiveLibrary.NotSupported;
                }
            }
        }

        public int GetNumberOfPagesFromArchive(string archivePath)
        {
            if (!IsValidArchive(archivePath))
            {
                _logger.LogError("Archive {ArchivePath} could not be found", archivePath);
                return 0;
            }

            try
            {
                var libraryHandler = CanOpen(archivePath);
                switch (libraryHandler)
                {
                    case ArchiveLibrary.Default:
                    {
                        using var archive = ZipFile.OpenRead(archivePath);
                        return archive.Entries.Count(e => !Parser.Parser.HasBlacklistedFolderInPath(e.FullName) && Parser.Parser.IsImage(e.FullName));
                    }
                    case ArchiveLibrary.SharpCompress:
                    {
                        using var archive = ArchiveFactory.Open(archivePath);
                        return archive.Entries.Count(entry => !entry.IsDirectory &&
                                                              !Parser.Parser.HasBlacklistedFolderInPath(Path.GetDirectoryName(entry.Key) ?? string.Empty)
                                                              && Parser.Parser.IsImage(entry.Key));
                    }
                    case ArchiveLibrary.NotSupported:
                        _logger.LogWarning("[GetNumberOfPagesFromArchive] This archive cannot be read: {ArchivePath}. Defaulting to 0 pages", archivePath);
                        return 0;
                    default:
                        _logger.LogWarning("[GetNumberOfPagesFromArchive] There was an exception when reading archive stream: {ArchivePath}. Defaulting to 0 pages", archivePath);
                        return 0;
                }
            }
            catch (Exception ex)
            {
                _logger.LogWarning(ex, "[GetNumberOfPagesFromArchive] There was an exception when reading archive stream: {ArchivePath}. Defaulting to 0 pages", archivePath);
                return 0;
            }
        }

        /// <summary>
        /// Finds the first instance of a folder entry and returns it
        /// </summary>
        /// <param name="entryFullNames"></param>
        /// <returns>Entry name of match, null if no match</returns>
        public static string FindFolderEntry(IEnumerable<string> entryFullNames)
        {
            var result = entryFullNames
                .OrderByNatural(Path.GetFileNameWithoutExtension)
                .Where(path => !(Path.EndsInDirectorySeparator(path) || Parser.Parser.HasBlacklistedFolderInPath(path) || path.StartsWith(Parser.Parser.MacOsMetadataFileStartsWith)))
                .FirstOrDefault(Parser.Parser.IsCoverImage);

            return string.IsNullOrEmpty(result) ? null : result;
        }

        /// <summary>
        /// Returns first entry that is an image and is not in a blacklisted folder path. Uses <see cref="OrderByNatural"/> for ordering files
        /// </summary>
        /// <param name="entryFullNames"></param>
        /// <returns>Entry name of match, null if no match</returns>
        public static string? FirstFileEntry(IEnumerable<string> entryFullNames, string archiveName)
        {
            // First check if there are any files that are not in a nested folder before just comparing by filename. This is needed
            // because NaturalSortComparer does not work with paths and doesn't seem 001.jpg as before chapter 1/001.jpg.
            var fullNames = entryFullNames
                .OrderByNatural(c => c.GetFullPathWithoutExtension())
                .Where(path => !(Path.EndsInDirectorySeparator(path) || Parser.Parser.HasBlacklistedFolderInPath(path) || path.StartsWith(Parser.Parser.MacOsMetadataFileStartsWith)) && Parser.Parser.IsImage(path))
                .ToList();
            if (fullNames.Count == 0) return null;

            var nonNestedFile = fullNames.Where(entry => (Path.GetDirectoryName(entry) ?? string.Empty).Equals(archiveName))
                .OrderByNatural(c => c.GetFullPathWithoutExtension())
                .FirstOrDefault();

            if (!string.IsNullOrEmpty(nonNestedFile)) return nonNestedFile;

            // Check the first folder and sort within that to see if we can find a file, else fallback to first file with basic sort.
            // Get first folder, then sort within that
            var firstDirectoryFile = fullNames.OrderByNatural(Path.GetDirectoryName).FirstOrDefault();
            if (!string.IsNullOrEmpty(firstDirectoryFile))
            {
                var firstDirectory = Path.GetDirectoryName(firstDirectoryFile);
                if (!string.IsNullOrEmpty(firstDirectory))
                {
                    var firstDirectoryResult = fullNames.Where(f => firstDirectory.Equals(Path.GetDirectoryName(f)))
                        .OrderByNatural(Path.GetFileNameWithoutExtension)
                        .FirstOrDefault();

                    if (!string.IsNullOrEmpty(firstDirectoryResult)) return firstDirectoryResult;
                }
            }

            var result = fullNames
                .OrderByNatural(Path.GetFileNameWithoutExtension)
                .FirstOrDefault();

            return string.IsNullOrEmpty(result) ? null : result;
        }


        /// <summary>
        /// Generates byte array of cover image.
        /// Given a path to a compressed file <see cref="Parser.Parser.ArchiveFileExtensions"/>, will ensure the first image (respects directory structure) is returned unless
        /// a folder/cover.(image extension) exists in the the compressed file (if duplicate, the first is chosen)
        ///
        /// This skips over any __MACOSX folder/file iteration.
        /// </summary>
        /// <remarks>This always creates a thumbnail</remarks>
        /// <param name="archivePath"></param>
        /// <param name="fileName">File name to use based on context of entity.</param>
        /// <returns></returns>
        public string GetCoverImage(string archivePath, string fileName, string outputDirectory)
        {
            if (archivePath == null || !IsValidArchive(archivePath)) return string.Empty;
            try
            {
                var libraryHandler = CanOpen(archivePath);
                switch (libraryHandler)
                {
                    case ArchiveLibrary.Default:
                    {
                        using var archive = ZipFile.OpenRead(archivePath);
                        var entryNames = archive.Entries.Select(e => e.FullName).ToList();

                        var entryName = FindCoverImageFilename(archivePath, entryNames);
                        var entry = archive.Entries.Single(e => e.FullName == entryName);

                        using var stream = entry.Open();
                        return _imageService.WriteCoverThumbnail(stream, fileName, outputDirectory);
                    }
                    case ArchiveLibrary.SharpCompress:
                    {
                        using var archive = ArchiveFactory.Open(archivePath);
                        var entryNames = archive.Entries.Where(archiveEntry => !archiveEntry.IsDirectory).Select(e => e.Key).ToList();

                        var entryName = FindCoverImageFilename(archivePath, entryNames);
                        var entry = archive.Entries.Single(e => e.Key == entryName);

                        using var stream = entry.OpenEntryStream();
                        return _imageService.WriteCoverThumbnail(stream, fileName, outputDirectory);
                    }
                    case ArchiveLibrary.NotSupported:
                        _logger.LogWarning("[GetCoverImage] This archive cannot be read: {ArchivePath}. Defaulting to no cover image", archivePath);
                        return string.Empty;
                    default:
                        _logger.LogWarning("[GetCoverImage] There was an exception when reading archive stream: {ArchivePath}. Defaulting to no cover image", archivePath);
                        return string.Empty;
                }
            }
            catch (Exception ex)
            {
                _logger.LogWarning(ex, "[GetCoverImage] There was an exception when reading archive stream: {ArchivePath}. Defaulting to no cover image", archivePath);
            }

            return string.Empty;
        }

        /// <summary>
        /// Given a list of image paths (assume within an archive), find the filename that corresponds to the cover
        /// </summary>
        /// <param name="archivePath"></param>
        /// <param name="entryNames"></param>
        /// <returns></returns>
        public string FindCoverImageFilename(string archivePath, IList<string> entryNames)
        {
            var entryName = FindFolderEntry(entryNames) ?? FirstFileEntry(entryNames, Path.GetFileName(archivePath));
            return entryName;
        }

        /// <summary>
        /// Given an archive stream, will assess whether directory needs to be flattened so that the extracted archive files are directly
        /// under extract path and not nested in subfolders. See <see cref="DirectoryInfoExtensions"/> Flatten method.
        /// </summary>
        /// <param name="archive">An opened archive stream</param>
        /// <returns></returns>
        public bool ArchiveNeedsFlattening(ZipArchive archive)
        {
            // Sometimes ZipArchive will list the directory and others it will just keep it in the FullName
            return archive.Entries.Count > 0 &&
                   !Path.HasExtension(archive.Entries.ElementAt(0).FullName) ||
                   archive.Entries.Any(e => e.FullName.Contains(Path.AltDirectorySeparatorChar) && !Parser.Parser.HasBlacklistedFolderInPath(e.FullName));
        }

        // TODO: Refactor CreateZipForDownload to return the temp file so we can stream it from temp
        /// <summary>
        ///
        /// </summary>
        /// <param name="files"></param>
        /// <param name="tempFolder">Temp folder name to use for preparing the files. Will be created and deleted</param>
        /// <returns></returns>
        /// <exception cref="KavitaException"></exception>
        public async Task<Tuple<byte[], string>> CreateZipForDownload(IEnumerable<string> files, string tempFolder)
        {
            var dateString = DateTime.Now.ToShortDateString().Replace("/", "_");

            var tempLocation = Path.Join(_directoryService.TempDirectory, $"{tempFolder}_{dateString}");
            _directoryService.ExistOrCreate(tempLocation);
            if (!_directoryService.CopyFilesToDirectory(files, tempLocation))
            {
                throw new KavitaException("Unable to copy files to temp directory archive download.");
            }

            var zipPath = Path.Join(_directoryService.TempDirectory, $"kavita_{tempFolder}_{dateString}.zip");
            try
            {
                ZipFile.CreateFromDirectory(tempLocation, zipPath);
            }
            catch (AggregateException ex)
            {
                _logger.LogError(ex, "There was an issue creating temp archive");
                throw new KavitaException("There was an issue creating temp archive");
            }


            var fileBytes = await _directoryService.ReadFileAsync(zipPath);

            _directoryService.ClearAndDeleteDirectory(tempLocation); // NOTE: For sending back just zip, just schedule this to be called after the file is returned or let next temp storage cleanup take care of it
            (new FileInfo(zipPath)).Delete();

            return Tuple.Create(fileBytes, zipPath);
        }


        /// <summary>
        /// Test if the archive path exists and an archive
        /// </summary>
        /// <param name="archivePath"></param>
        /// <returns></returns>
        public bool IsValidArchive(string archivePath)
        {
            if (!File.Exists(archivePath))
            {
                _logger.LogWarning("Archive {ArchivePath} could not be found", archivePath);
                return false;
            }

            if (Parser.Parser.IsArchive(archivePath) || Parser.Parser.IsEpub(archivePath)) return true;

            _logger.LogWarning("Archive {ArchivePath} is not a valid archive", archivePath);
            return false;
        }

        private static bool ValidComicInfoArchiveEntry(string fullName, string name)
        {
            var filenameWithoutExtension = Path.GetFileNameWithoutExtension(name).ToLower();
            return !Parser.Parser.HasBlacklistedFolderInPath(fullName)
                && filenameWithoutExtension.Equals(ComicInfoFilename, StringComparison.InvariantCultureIgnoreCase)
                && !filenameWithoutExtension.StartsWith(Parser.Parser.MacOsMetadataFileStartsWith)
                && Parser.Parser.IsXml(name);
        }

        /// <summary>
        /// This can be null if nothing is found or any errors occur during access
        /// </summary>
        /// <param name="archivePath"></param>
        /// <returns></returns>
        public ComicInfo? GetComicInfo(string archivePath)
        {
            if (!IsValidArchive(archivePath)) return null;

            try
            {
                if (!File.Exists(archivePath)) return null;

                var libraryHandler = CanOpen(archivePath);
                switch (libraryHandler)
                {
                    case ArchiveLibrary.Default:
                    {
                        using var archive = ZipFile.OpenRead(archivePath);

                        var entry = archive.Entries.FirstOrDefault(x => ValidComicInfoArchiveEntry(x.FullName, x.Name));
                        if (entry != null)
                        {
                            using var stream = entry.Open();
                            var serializer = new XmlSerializer(typeof(ComicInfo));
                            var info = (ComicInfo) serializer.Deserialize(stream);
                            ComicInfo.CleanComicInfo(info);
                            return info;
                        }

                        break;
                    }
                    case ArchiveLibrary.SharpCompress:
                    {
                        using var archive = ArchiveFactory.Open(archivePath);
                        var entry = archive.Entries.FirstOrDefault(entry =>
                            ValidComicInfoArchiveEntry(Path.GetDirectoryName(entry.Key), entry.Key));

                        if (entry != null)
                        {
                            using var stream = entry.OpenEntryStream();
                            var serializer = new XmlSerializer(typeof(ComicInfo));
                            var info = (ComicInfo) serializer.Deserialize(stream);
                            ComicInfo.CleanComicInfo(info);
                            return info;
                        }

                        break;
                    }
                    case ArchiveLibrary.NotSupported:
                        _logger.LogWarning("[GetComicInfo] This archive cannot be read: {ArchivePath}", archivePath);
                        return null;
                    default:
                        _logger.LogWarning(
                            "[GetComicInfo] There was an exception when reading archive stream: {ArchivePath}",
                            archivePath);
                        return null;
                }
            }
            catch (Exception ex)
            {
                _logger.LogWarning(ex, "[GetComicInfo] There was an exception when reading archive stream: {Filepath}", archivePath);
            }

            return null;
        }


        private void ExtractArchiveEntities(IEnumerable<IArchiveEntry> entries, string extractPath)
        {
            _directoryService.ExistOrCreate(extractPath);
            foreach (var entry in entries)
            {
                entry.WriteToDirectory(extractPath, new ExtractionOptions()
                {
                    ExtractFullPath = true, // Don't flatten, let the flatterner ensure correct order of nested folders
                    Overwrite = false
                });
            }
        }

        private void ExtractArchiveEntries(ZipArchive archive, string extractPath)
        {
            // TODO: In cases where we try to extract, but there are InvalidPathChars, we need to inform the user (throw exception, let middleware inform user)
            var needsFlattening = ArchiveNeedsFlattening(archive);
            if (!archive.HasFiles() && !needsFlattening) return;

            archive.ExtractToDirectory(extractPath, true);
            if (!needsFlattening) return;

            _logger.LogDebug("Extracted archive is nested in root folder, flattening...");
            _directoryService.Flatten(extractPath);
        }

        /// <summary>
        /// Extracts an archive to a temp cache directory. Returns path to new directory. If temp cache directory already exists,
        /// will return that without performing an extraction. Returns empty string if there are any invalidations which would
        /// prevent operations to perform correctly (missing archivePath file, empty archive, etc).
        /// </summary>
        /// <param name="archivePath">A valid file to an archive file.</param>
        /// <param name="extractPath">Path to extract to</param>
        /// <returns></returns>
        public void ExtractArchive(string archivePath, string extractPath)
        {
            if (!IsValidArchive(archivePath)) return;

            if (Directory.Exists(extractPath)) return;

            var sw = Stopwatch.StartNew();

            try
            {
                var libraryHandler = CanOpen(archivePath);
                switch (libraryHandler)
                {
                    case ArchiveLibrary.Default:
                    {
                        using var archive = ZipFile.OpenRead(archivePath);
                        ExtractArchiveEntries(archive, extractPath);
                        break;
                    }
                    case ArchiveLibrary.SharpCompress:
                    {
                        using var archive = ArchiveFactory.Open(archivePath);
                        ExtractArchiveEntities(archive.Entries.Where(entry => !entry.IsDirectory
                                                                              && !Parser.Parser.HasBlacklistedFolderInPath(Path.GetDirectoryName(entry.Key) ?? string.Empty)
                                                                              && Parser.Parser.IsImage(entry.Key)), extractPath);
                        break;
                    }
                    case ArchiveLibrary.NotSupported:
                        _logger.LogWarning("[ExtractArchive] This archive cannot be read: {ArchivePath}", archivePath);
                        return;
                    default:
                        _logger.LogWarning("[ExtractArchive] There was an exception when reading archive stream: {ArchivePath}", archivePath);
                        return;
                }

            }
            catch (Exception e)
            {
                _logger.LogWarning(e, "[ExtractArchive] There was a problem extracting {ArchivePath} to {ExtractPath}",archivePath, extractPath);
                return;
            }
            _logger.LogDebug("Extracted archive to {ExtractPath} in {ElapsedMilliseconds} milliseconds", extractPath, sw.ElapsedMilliseconds);
        }
    }
}