#include "utils.hpp"
#include "initcpio.hpp"

#include <cassert>
#include <fstream>
#include <string>
#include <string_view>
#include <vector>
#include <filesystem>

#include <fmt/core.h>

namespace fs = std::filesystem;

static constexpr auto MKINITCPIO_STR = R"(
# MODULES
MODULES=()

# BINARIES
BINARIES=()

# FILES
FILES=()

# HOOKS
HOOKS=(base udev autodetect modconf block filesystems keyboard fsck)
)";

static constexpr auto MKINITCPIO_TEST = R"(
# MODULES
MODULES=(radeon crc32c-intel)

# BINARIES
BINARIES=()

# FILES
FILES=()

# HOOKS
HOOKS=(base udev autodetect modconf block filesystems keyboard fsck btrfs usr lvm2 zfs)
)";


int main() {
    static constexpr std::string_view filename{"/tmp/mkinitcpio.conf"};

    // Open mkinitcpio file for writing.
    std::ofstream mhinitcpio_file{filename.data()};
    assert(mhinitcpio_file.is_open());

    // Setup mkinitcpio file.
    mhinitcpio_file << MKINITCPIO_STR;
    mhinitcpio_file.close();

    auto initcpio = detail::Initcpio{filename};

    // Insert data.
    initcpio.append_module("radeon");
    initcpio.append_hook("btrfs");
    initcpio.append_module("crc32c-intel");
    initcpio.hooks.insert(initcpio.hooks.end(), {"usr", "lvm2", "zfs"});

    // Write data.
    assert(initcpio.write());

    // Checking insertion of equal items
    initcpio.append_module("radeon");
    initcpio.append_hook("btrfs");
    initcpio.append_module("crc32c-intel");

    // Write data.
    assert(initcpio.write());

    // Check if file is equal to test data.
    // "\n# MODULES\nMODULES=(crc32c-intel)\n\n# BINARIES\nBINARIES=()\n\n# FILES\nFILES=()\n\n# HOOKS\nHOOKS=(base usr lvm2 zfs)"\n
    const auto& file_content = utils::read_whole_file(filename);
    assert(file_content == MKINITCPIO_TEST);

    // Cleanup.
    fs::remove(filename);
}