Config file and package tracking utility
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

352 lines
8.5 KiB

/*
* Copyright (C) 2022 Riyyi
*
* SPDX-License-Identifier: MIT
*/
#include <cstdio> // stderr
#include <filesystem>
#include <string>
#include <unistd.h> // geteuid, setegid, seteuid
#include <vector>
#include "dotfile.h"
#include "machine.h"
#include "macro.h"
#include "testcase.h"
#include "testsuite.h"
#include "util/file.h"
#include "util/system.h"
const bool root = !geteuid() ? true : false;
const std::filesystem::path homeDirectory = "/home/" + Machine::the().username();
const size_t homeDirectorySize = homeDirectory.string().size();
void createTestDotfiles(const std::vector<std::string>& fileNames, const std::vector<std::string>& fileContents, bool asRoot = false)
{
VERIFY(fileNames.size() == fileContents.size(), return);
if (root && !asRoot) {
setegid(Machine::the().gid());
seteuid(Machine::the().uid());
}
for (size_t i = 0; i < fileNames.size(); ++i) {
auto fileName = fileNames.at(i);
auto directory = std::filesystem::path { fileName }.parent_path();
if (!directory.empty() && !std::filesystem::exists(directory)) {
std::filesystem::create_directories(directory);
}
auto file = Util::File::create(fileName);
if (file.data().size() == 0) {
file.append(fileContents.at(i).c_str()).flush();
}
}
if (root && !asRoot) {
seteuid(0);
setegid(0);
}
}
void removeTestDotfiles(const std::vector<std::string>& files)
{
for (auto file : files) {
// Get the top-level directory
if (file.find(homeDirectory) == 0) {
file = file.substr(homeDirectorySize + 1);
}
file = file.substr(0, file.find_first_of('/'));
// Delete recursively in working directory
if (std::filesystem::exists(file)) {
std::filesystem::remove_all(file);
}
// Delete recursively in home directory
file = homeDirectory / file;
if (std::filesystem::exists(file)) {
std::filesystem::remove_all(file);
}
}
}
// -----------------------------------------
TEST_CASE(AddDotfiles)
{
std::vector<std::string> fileNames = {
homeDirectory / "__test-file-1",
homeDirectory / "__subdir/__test-file-2",
homeDirectory / "__subdir/__another-subdir/__test-file-3",
};
std::vector<std::string> fileContents = {
R"(home directory file 1
)",
R"(home directory file 2
)",
R"(home directory file 3
)",
};
createTestDotfiles(fileNames, fileContents);
Dotfile::the().add(fileNames);
for (const auto& file : fileNames) {
VERIFY(std::filesystem::exists(file), continue);
VERIFY(std::filesystem::exists(file.substr(homeDirectorySize + 1)), continue);
Util::File lhs(file);
Util::File rhs(file.substr(homeDirectorySize + 1));
EXPECT_EQ(lhs.data(), rhs.data());
}
removeTestDotfiles(fileNames);
}
TEST_CASE(AddNonExistentDotfiles)
{
stderr = Test::TestSuite::the().outputNull();
Dotfile::the().add({ homeDirectory / "__non-existent-test-file" });
stderr = Test::TestSuite::the().outputErr();
EXPECT(!std::filesystem::exists("__non-existent-test-file"));
removeTestDotfiles({ "__non-existent-test-file" });
}
TEST_CASE(PullDotfiles)
{
std::vector<std::string> fileNames = {
"__test-file-1",
"__subdir/__test-file-2",
"__subdir/__another-subdir/__test-file-3",
};
std::vector<std::string> homeFileNames;
for (const auto& file : fileNames) {
homeFileNames.push_back(homeDirectory / file);
}
std::vector<std::string> homeFileContents = {
R"(file 1 after pulling
)",
R"(file 2 after pulling
)",
R"(file 3 after pulling
)",
};
std::vector<std::string> workingDirectoryFileContents = {
R"(file 1
)",
R"(file 2
)",
R"(file 3
)",
};
createTestDotfiles(homeFileNames, homeFileContents);
createTestDotfiles(fileNames, workingDirectoryFileContents);
Dotfile::the().pull(fileNames);
for (size_t i = 0; i < fileNames.size(); ++i) {
VERIFY(std::filesystem::exists(homeFileNames.at(i)), continue);
VERIFY(std::filesystem::exists(fileNames.at(i)), continue);
Util::File lhs(homeFileNames.at(i));
Util::File rhs(fileNames.at(i));
EXPECT_EQ(lhs.data(), rhs.data());
}
removeTestDotfiles(fileNames);
}
TEST_CASE(PushDotfiles)
{
std::vector<std::string> fileNames = {
"__test-file-1",
"__subdir/__test-file-2",
"__subdir/__another-subdir/__test-file-3",
};
std::vector<std::string> fileContents = {
R"(working directory file 1
)",
R"(working directory file 2
)",
R"(working directory file 3
)",
};
createTestDotfiles(fileNames, fileContents);
Dotfile::the().push(fileNames);
for (const auto& file : fileNames) {
VERIFY(std::filesystem::exists(file), continue);
VERIFY(std::filesystem::exists(homeDirectory / file), continue);
Util::File lhs(file);
Util::File rhs(homeDirectory / file);
EXPECT_EQ(lhs.data(), rhs.data());
}
removeTestDotfiles(fileNames);
}
TEST_CASE(PushDotfilesWithExcludePath)
{
std::vector<std::string> fileNames = {
"__test-file-1",
"__subdir/__test-file-2",
"__subdir/__test-file-3",
"__another-subdir/__test-file-4.test",
};
createTestDotfiles(fileNames, { "", "", "", "" });
Dotfile::the().setExcludePaths({
{ Dotfile::ExcludeType::File, "__test-file-1" },
{ Dotfile::ExcludeType::Directory, "__subdir" },
{ Dotfile::ExcludeType::EndsWith, ".test" },
});
Dotfile::the().push(fileNames);
Dotfile::the().setExcludePaths({});
for (const auto& file : fileNames) {
EXPECT(!std::filesystem::exists(homeDirectory / file));
}
removeTestDotfiles(fileNames);
}
TEST_CASE(PushDotfilesSelectivelyComment)
{
std::vector<std::string> fileNames = {
"__test-file-1",
"__test-file-2",
"__test-file-3",
"__test-file-4",
"__test-file-5",
"__test-file-6",
"__test-file-7",
};
auto distro = Machine::the().distroId();
auto hostname = Machine::the().hostname();
auto username = Machine::the().username();
std::string placeholder = "@@@@";
std::vector<std::string> fileContents = {
"# >>> distro=" + distro + " hostname=" + hostname + " user=" + username + R"(
# this should be uncommented
# <<<
)",
"# >>> distro=" + placeholder + " hostname=" + hostname + " user=" + username + R"(
# this should remain commented
# <<<
)",
"# >>> distro=" + distro + " hostname=" + placeholder + " user=" + username + R"(
# this should remain commented
# <<<
)",
"# >>> distro=" + distro + " hostname=" + hostname + " user=" + placeholder + R"(
this should be commented
# <<<
)",
" # >>> distro=" + distro + R"(
# this should be uncommented
# <<<
)",
" # >>> user=" + username + R"(
# this should be uncommented
# <<<
)",
" # >>> hostname=" + hostname + R"(
this should remain uncommented
# <<<
)",
};
std::vector<std::string> pushedFileContents = {
"# >>> distro=" + distro + " hostname=" + hostname + " user=" + username + R"(
this should be uncommented
# <<<
)",
"# >>> distro=" + placeholder + " hostname=" + hostname + " user=" + username + R"(
# this should remain commented
# <<<
)",
"# >>> distro=" + distro + " hostname=" + placeholder + " user=" + username + R"(
# this should remain commented
# <<<
)",
"# >>> distro=" + distro + " hostname=" + hostname + " user=" + placeholder + R"(
# this should be commented
# <<<
)",
" # >>> distro=" + distro + R"(
this should be uncommented
# <<<
)",
" # >>> user=" + username + R"(
this should be uncommented
# <<<
)",
" # >>> hostname=" + hostname + R"(
this should remain uncommented
# <<<
)",
};
createTestDotfiles(fileNames, fileContents);
Dotfile::the().push(fileNames);
for (size_t i = 0; i < fileNames.size(); ++i) {
const auto& file = fileNames.at(i);
VERIFY(std::filesystem::exists(file), continue);
VERIFY(std::filesystem::exists(homeDirectory / file), continue);
Util::File lhs(homeDirectory / file);
EXPECT_EQ(lhs.data(), pushedFileContents.at(i));
}
removeTestDotfiles(fileNames);
}
TEST_CASE(AddSystemDotfiles)
{
VERIFY(geteuid() == 0, return);
Dotfile::the().setSystemDirectories({ "/etc", "/usr/lib" });
Dotfile::the().add({ "/etc/group", "/usr/lib/os-release" });
Dotfile::the().setSystemDirectories({});
EXPECT(std::filesystem::exists("etc/group"));
EXPECT(std::filesystem::exists("usr/lib/os-release"));
std::filesystem::remove_all(Dotfile::the().workingDirectory() / "etc");
std::filesystem::remove_all(Dotfile::the().workingDirectory() / "usr");
}
TEST_CASE(PullSystemDotfiles)
{
VERIFY(geteuid() == 0, return);
createTestDotfiles({ "etc/group" }, { "" }, true);
Dotfile::the().setSystemDirectories({ "/etc" });
Dotfile::the().pull({ "etc/group" });
Dotfile::the().setSystemDirectories({});
Util::File lhs("/etc/group");
Util::File rhs(Dotfile::the().workingDirectory() / "etc/group");
EXPECT_EQ(lhs.data(), rhs.data());
std::filesystem::remove_all(Dotfile::the().workingDirectory() / "etc");
}