import * as exec from "@actions/exec";
import * as io from "@actions/io";
import * as path from "path";

import { CacheFilename, CompressionMethod } from "../src/constants";
import * as tar from "../src/tar";
import * as utils from "../src/utils/actionUtils";

import fs = require("fs");

jest.mock("@actions/exec");
jest.mock("@actions/io");

const IS_WINDOWS = process.platform === "win32";
const IS_64OS = process.platform != "win32" || process.arch === "x64";

function getTempDir(): string {
    return path.join(__dirname, "_temp", "tar");
}

beforeAll(async () => {
    jest.spyOn(io, "which").mockImplementation(tool => {
        return Promise.resolve(tool);
    });

    process.env["GITHUB_WORKSPACE"] = process.cwd();
    await jest.requireActual("@actions/io").rmRF(getTempDir());
});

afterAll(async () => {
    delete process.env["GITHUB_WORKSPACE"];
    await jest.requireActual("@actions/io").rmRF(getTempDir());
});

test("zstd extract tar", async () => {
    const mkdirMock = jest.spyOn(io, "mkdirP");
    const execMock = jest.spyOn(exec, "exec");

    const archivePath = IS_WINDOWS
        ? `${process.env["windir"]}\\fakepath\\cache.tar`
        : "cache.tar";
    const workspace = process.env["GITHUB_WORKSPACE"];

    await tar.extractTar(archivePath, CompressionMethod.Zstd);

    expect(mkdirMock).toHaveBeenCalledWith(workspace);
    const tarPath = IS_WINDOWS
        ? `${process.env["windir"]}\\System32\\tar.exe`
        : "tar";
    const zstdOptions = IS_64OS ? "zstd -d --long=31" : "zstd -d --long=30";
    expect(execMock).toHaveBeenCalledTimes(1);
    expect(execMock).toHaveBeenCalledWith(
        `"${tarPath}"`,
        [
            "--use-compress-program",
            zstdOptions,
            "-xf",
            IS_WINDOWS ? archivePath.replace(/\\/g, "/") : archivePath,
            "-P",
            "-C",
            IS_WINDOWS ? workspace?.replace(/\\/g, "/") : workspace
        ],
        { cwd: undefined }
    );
});

test("gzip extract tar", async () => {
    const mkdirMock = jest.spyOn(io, "mkdirP");
    const execMock = jest.spyOn(exec, "exec");
    const archivePath = IS_WINDOWS
        ? `${process.env["windir"]}\\fakepath\\cache.tar`
        : "cache.tar";
    const workspace = process.env["GITHUB_WORKSPACE"];

    await tar.extractTar(archivePath, CompressionMethod.Gzip);

    expect(mkdirMock).toHaveBeenCalledWith(workspace);
    const tarPath = IS_WINDOWS
        ? `${process.env["windir"]}\\System32\\tar.exe`
        : "tar";
    expect(execMock).toHaveBeenCalledTimes(1);
    expect(execMock).toHaveBeenCalledWith(
        `"${tarPath}"`,
        [
            "-z",
            "-xf",
            IS_WINDOWS ? archivePath.replace(/\\/g, "/") : archivePath,
            "-P",
            "-C",
            IS_WINDOWS ? workspace?.replace(/\\/g, "/") : workspace
        ],
        { cwd: undefined }
    );
});

test("gzip extract GNU tar on windows", async () => {
    if (IS_WINDOWS) {
        jest.spyOn(fs, "existsSync").mockReturnValueOnce(false);

        const isGnuMock = jest
            .spyOn(utils, "useGnuTar")
            .mockReturnValue(Promise.resolve(true));
        const execMock = jest.spyOn(exec, "exec");
        const archivePath = `${process.env["windir"]}\\fakepath\\cache.tar`;
        const workspace = process.env["GITHUB_WORKSPACE"];

        await tar.extractTar(archivePath, CompressionMethod.Gzip);

        expect(isGnuMock).toHaveBeenCalledTimes(1);
        expect(execMock).toHaveBeenCalledTimes(1);
        expect(execMock).toHaveBeenCalledWith(
            `"tar"`,
            [
                "-z",
                "-xf",
                archivePath.replace(/\\/g, "/"),
                "-P",
                "-C",
                workspace?.replace(/\\/g, "/"),
                "--force-local"
            ],
            { cwd: undefined }
        );
    }
});

test("zstd create tar", async () => {
    const execMock = jest.spyOn(exec, "exec");

    const archiveFolder = getTempDir();
    const workspace = process.env["GITHUB_WORKSPACE"];
    const sourceDirectories = ["~/.npm/cache", `${workspace}/dist`];

    await fs.promises.mkdir(archiveFolder, { recursive: true });

    await tar.createTar(
        archiveFolder,
        sourceDirectories,
        CompressionMethod.Zstd
    );

    const tarPath = IS_WINDOWS
        ? `${process.env["windir"]}\\System32\\tar.exe`
        : "tar";
    const zstdOptions = IS_64OS ? "zstd -T0 --long=31" : "zstd -T0 --long=30";

    expect(execMock).toHaveBeenCalledTimes(1);
    expect(execMock).toHaveBeenCalledWith(
        `"${tarPath}"`,
        [
            "--use-compress-program",
            zstdOptions,
            "-cf",
            IS_WINDOWS
                ? CacheFilename.Zstd.replace(/\\/g, "/")
                : CacheFilename.Zstd,
            "-P",
            "-C",
            IS_WINDOWS ? workspace?.replace(/\\/g, "/") : workspace,
            "--files-from",
            "manifest.txt"
        ],
        {
            cwd: archiveFolder
        }
    );
});

test("gzip create tar", async () => {
    const execMock = jest.spyOn(exec, "exec");

    const archiveFolder = getTempDir();
    const workspace = process.env["GITHUB_WORKSPACE"];
    const sourceDirectories = ["~/.npm/cache", `${workspace}/dist`];

    await fs.promises.mkdir(archiveFolder, { recursive: true });

    await tar.createTar(
        archiveFolder,
        sourceDirectories,
        CompressionMethod.Gzip
    );

    const tarPath = IS_WINDOWS
        ? `${process.env["windir"]}\\System32\\tar.exe`
        : "tar";

    expect(execMock).toHaveBeenCalledTimes(1);
    expect(execMock).toHaveBeenCalledWith(
        `"${tarPath}"`,
        [
            "-z",
            "-cf",
            IS_WINDOWS
                ? CacheFilename.Gzip.replace(/\\/g, "/")
                : CacheFilename.Gzip,
            "-P",
            "-C",
            IS_WINDOWS ? workspace?.replace(/\\/g, "/") : workspace,
            "--files-from",
            "manifest.txt"
        ],
        {
            cwd: archiveFolder
        }
    );
});