transmission/tests/libtransmission/subprocess-test-program.cc

68 lines
1.6 KiB
C++

// This file Copyright (C) 2017-2022 Mnemosyne LLC.
// It may be used under GPLv2 (SPDX: GPL-2.0-only), GPLv3 (SPDX: GPL-3.0-only),
// or any future license endorsed by Mnemosyne LLC.
// License text can be found in the licenses/ folder.
#include <libtransmission/file.h> // tr_sys_dir_get_current()
#include <libtransmission/utils.h> // tr_env_get_string()
#include <fmt/core.h>
#include <fmt/ostream.h>
#include <fstream>
#include <string>
int main(int argc, char** argv)
{
if (argc < 3)
{
return 1;
}
auto const result_path = std::string{ argv[1] };
auto const test_action = std::string{ argv[2] };
auto const tmp_result_path = result_path + ".tmp";
auto out = std::ofstream(tmp_result_path.c_str(), std::ios::out | std::ios::trunc | std::ios::binary);
if (!out)
{
return 1;
}
if (test_action == "--dump-args")
{
for (int i = 3; i < argc; ++i)
{
fmt::print(out, "{:s}\n", argv[i]);
}
}
else if (test_action == "--dump-env")
{
for (int i = 3; i < argc; ++i)
{
fmt::print(out, "{:s}\n", tr_env_get_string(argv[i], "<null>"));
}
}
else if (test_action == "--dump-cwd")
{
auto cwd = tr_sys_dir_get_current(nullptr);
if (std::empty(cwd))
{
cwd = "<null>";
}
fmt::print(out, "{:s}\n", cwd);
}
else
{
out.close();
(void)std::remove(tmp_result_path.c_str());
return 1;
}
out.close();
tr_sys_path_rename(tmp_result_path.c_str(), result_path.c_str());
return 0;
}