// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.IO;
using System.IO.Compression;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;

namespace Microsoft.Arcade.Common
{
    public class ZipArchiveManager : IZipArchiveManager
    {
        public ZipArchive OpenArchive(string archivePath, ZipArchiveMode mode)
            => ZipFile.Open(archivePath, mode);

        public async Task AddResourceFileToArchive<TAssembly>(string archivePath, string resourceName, string targetFileName = null)
        {
            using Stream fileStream = GetResourceFileContent<TAssembly>(resourceName);
            await AddContentToArchive(archivePath, targetFileName ?? resourceName, fileStream);
        }

        public void ArchiveDirectory(string directoryPath, string archivePath, bool includeBaseDirectory)
            => ZipFile.CreateFromDirectory(directoryPath, archivePath, CompressionLevel.Fastest, includeBaseDirectory);

        public void ArchiveFile(string filePath, string archivePath)
        {
            bool archiveExists = File.Exists(archivePath);
            string entryName = Path.GetFileName(filePath);

            using FileStream fs = File.OpenWrite(archivePath);
            using ZipArchive zip = new(fs, archiveExists ? ZipArchiveMode.Update : ZipArchiveMode.Create, false);
                
            // Overwrite previous file, if any (otherwise dual records were created)
            if (archiveExists)
            {
                zip.Entries.FirstOrDefault(e => e.FullName == entryName)?.Delete();
            }

            zip.CreateEntryFromFile(filePath, entryName);
        }

        public Task AddContentToArchive(string archivePath, string targetFilename, string content)
            => AddContentToArchive(archivePath, targetFilename, new MemoryStream(Encoding.UTF8.GetBytes(content)));

        public async Task AddContentToArchive(string archivePath, string targetFilename, Stream content)
        {
            using FileStream archiveStream = new(archivePath, FileMode.Open);
            using ZipArchive archive = new(archiveStream, ZipArchiveMode.Update);

            // Overwrite previous file, if any (otherwise dual records were created)
            archive.Entries.FirstOrDefault(e => e.FullName == targetFilename)?.Delete();

            ZipArchiveEntry entry = archive.CreateEntry(targetFilename);
            using Stream targetStream = entry.Open();
            await content.CopyToAsync(targetStream);
        }

        public static Stream GetResourceFileContent<TAssembly>(string resourceFileName)
        {
            Assembly assembly = typeof(TAssembly).Assembly;
            return assembly.GetManifestResourceStream($"{assembly.GetName().Name}.{resourceFileName}");
        }
    }
}