mirror of
https://github.com/CachyOS/New-Cli-Installer.git
synced 2025-01-23 22:42:31 +08:00
82 lines
1.8 KiB
C++
82 lines
1.8 KiB
C++
#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);
|
|
}
|