SrcMod/SrcMod/Shell/Modules/CompressionModule.cs

namespace SrcMod.Shell.Modules;

// Some things that can be extracted can't be compressed by SharpCompress.
// In the future I might replace it with my own, but that'll take a *really*
// long time and I'm already planning to do that for the valve compression formats,
// so I'll seethe for now.
[Module("compress")]
public static class CompressionModule
{
    [Command("gz")]
    [Command("gzip")]
    public static void CompressGZip(string source, string? destination = null,
        CompressionLevel level = CompressionLevel.Optimal)
    {
        if (destination is null)
        {
            string full = Path.GetFullPath(source),
                   name = Path.GetFileName(full),
                   folder = Program.Shell!.WorkingDirectory;
            destination ??= $"{folder}\\{name}.gz";
        }

        string absSource = Path.GetFullPath(source),
               localDest = Path.GetRelativePath(Program.Shell!.WorkingDirectory, destination);

        if (File.Exists(source))
        {
            if (File.Exists(destination)) throw new($"File already exists at \"{localDest}\"");
            string message = $"Compressing file at \"{source}\" into \"{localDest}\"...";
            Write(message);

            FileStream writer = new(localDest, FileMode.CreateNew),
                       reader = new(absSource, FileMode.Open);
            GZipStream gzip = new(writer, level);

            reader.CopyTo(gzip);

            gzip.Close();
            reader.Close();
            writer.Close();

            Console.CursorLeft = 0;
            Console.CursorTop -= (message.Length / Console.BufferWidth) + 1;
            Write(new string(' ', message.Length), newLine: false);
        }
        else if (Directory.Exists(source)) throw new("The GZip format can only compress 1 file.");
        else throw new("No file located at \"source\"");
    }

    [Command("tar")]
    [Command("tarball")]
    public static void CompressTar(string source, string? destination = null)
    {
        if (destination is null)
        {
            string full = Path.GetFullPath(source),
                   name = Path.GetFileNameWithoutExtension(full),
                   folder = Program.Shell!.WorkingDirectory;
            destination ??= $"{folder}\\{name}.tar";
        }

        string absSource = Path.GetFullPath(source),
               localDest = Path.GetRelativePath(Program.Shell!.WorkingDirectory, destination);

        if (File.Exists(source)) throw new("The Tar format cannot compress a single file.");
        else if (Directory.Exists(source))
        {
            if (File.Exists(destination)) throw new($"File already exists at \"{localDest}\"");

            Write($"Compressing folder at \"{source}\" into \"{localDest}\"...");

            FileStream writer = new(destination, FileMode.CreateNew);
            TarFile.CreateFromDirectory(absSource, writer, false);

            writer.Close();

            Console.CursorLeft = 0;
            Write(new string(' ', Console.BufferWidth), newLine: false);
            Console.SetCursorPosition(0, Console.CursorTop - 1);
            Write(new string(' ', Console.BufferWidth), newLine: false);
        }
        else throw new("No file or directory located at \"source\"");
    }

    // Rar can't be compressed.

    [Command("targz")]
    [Command("tar.gz")]
    [Command("tar-gz")]
    public static void CompressTarGzip(string source, string? destination = null,
        CompressionLevel level = CompressionLevel.Optimal)
    {
        if (destination is null)
        {
            string full = Path.GetFullPath(source),
                   name = Path.GetFileNameWithoutExtension(full),
                   folder = Program.Shell!.WorkingDirectory;
            destination ??= $"{folder}\\{name}.tar.gz";
        }

        string firstDest = Path.GetFileNameWithoutExtension(destination);

        CompressTar(source, firstDest);
        CompressGZip(firstDest, destination, level);
        File.Delete(firstDest);

        Console.CursorLeft = 0;
        Console.CursorTop--;
    }

    [Command("zip")]
    public static void CompressZip(string source, string? destination = null,
        CompressionLevel level = CompressionLevel.Optimal, string comment = "")
    {
        if (destination is null)
        {
            string full = Path.GetFullPath(source),
                   name = Path.GetFileNameWithoutExtension(full),
                   folder = Program.Shell!.WorkingDirectory;
            destination ??= $"{folder}\\{name}.zip";
        }

        string absSource = Path.GetFullPath(source),
               localDest = Path.GetRelativePath(Program.Shell!.WorkingDirectory, destination);

        if (File.Exists(source))
        {
            if (File.Exists(destination)) throw new($"File already exists at \"{localDest}\"");
            string message = $"Compressing file at \"{source}\" into \"{localDest}\"...";
            Write(message);

            FileStream writer = new(destination, FileMode.CreateNew);
            ZipArchive archive = new(writer, ZipArchiveMode.Create);
            if (!string.IsNullOrWhiteSpace(comment)) archive.Comment = comment;

            archive.CreateEntryFromFile(absSource, Path.GetFileName(absSource), level);

            archive.Dispose();
            writer.Dispose();

            Console.CursorLeft = 0;
            Console.CursorTop -= (message.Length / Console.BufferWidth) + 1;
            Write(new string(' ', message.Length), newLine: false);
        }
        else if (Directory.Exists(source))
        {
            if (File.Exists(destination)) throw new($"File already exists at \"{localDest}\"");

            Write($"Compressing folder at \"{source}\" into \"{localDest}\"...");

            FileStream writer = new(destination, FileMode.CreateNew);
            ZipArchive archive = new(writer, ZipArchiveMode.Create);
            if (!string.IsNullOrWhiteSpace(comment)) archive.Comment = comment;

            List<string> files = new(GetAllFiles(absSource)),
                         relative = new();
            for (int i = 0; i < files.Count; i++)
            {
                string f = files[i];
                if (f.Trim().ToLower() == destination.Trim().ToLower())
                {
                    files.RemoveAt(i);
                    i--;
                    continue;
                }
                relative.Add(Path.GetRelativePath(absSource, f));
            }

            int failed = 0;

            LoadingBarStart();
            for (int i = 0; i < files.Count; i++)
            {
                bool failedThisTime = false;
                try
                {
                    archive.CreateEntryFromFile(files[i], relative[i], level);
                }
                catch
                {
                    failedThisTime = true;
                    failed++;
                }
                LoadingBarSet((i + 1) / (float)files.Count, failedThisTime ? ConsoleColor.Red : ConsoleColor.DarkGreen); ;
                Console.CursorLeft = 0;
                string message = $"{relative[i]}";
                int remainder = Console.BufferWidth - message.Length;
                if (remainder >= 0) message += new string(' ', remainder);
                else message = $"...{message[(3 - remainder)..]}";

                Write(message, newLine: false);
            }

            archive.Dispose();
            writer.Dispose();

            LoadingBarEnd();

            Console.CursorLeft = 0;
            Write(new string(' ', Console.BufferWidth), newLine: false);
            Console.SetCursorPosition(0, Console.CursorTop - 2);
            Write(new string(' ', Console.BufferWidth), newLine: false);

            if (failed > 0)
            {
                Console.CursorLeft = 0;
                Write($"{failed} file{(failed == 1 ? " has" : "s have")} been ignored due to an error.",
                      ConsoleColor.DarkYellow);
            }
        }
        else throw new("No file or directory located at \"source\"");

        DateTime stamp = DateTime.Now;

        Program.Shell!.AddHistory(new()
        {
            action = delegate
            {
                if (!File.Exists(destination))
                {
                    Write("Looks like the job is already completed Boss.", ConsoleColor.DarkYellow);
                    return;
                }

                FileInfo info = new(destination);
                if ((info.LastWriteTime - stamp).TotalMilliseconds >= 10)
                    throw new("The archive has been modified and probably shouldn't be undone.");

                File.Delete(destination);
            },
            name = $"Compressed a file or folder into a zip archive located at \"{destination}\""
        });
    }

    // 7z can't be compressed.
}