diff options
Diffstat (limited to 'src/headless')
| -rw-r--r-- | src/headless/main.cpp | 146 |
1 files changed, 100 insertions, 46 deletions
diff --git a/src/headless/main.cpp b/src/headless/main.cpp index 3935c2d..6cb6661 100644 --- a/src/headless/main.cpp +++ b/src/headless/main.cpp @@ -1,3 +1,5 @@ +#include <cstring> + #include <openvic-simulation/GameManager.hpp> #include <openvic-simulation/dataloader/Dataloader.hpp> #include <openvic-simulation/utility/Logger.hpp> @@ -5,36 +7,20 @@ using namespace OpenVic; -static char const* get_program_name(char const* name) { - static constexpr char const* missing_name = "<program>"; - if (name == nullptr) return missing_name; - char const* last_separator = name; - while (*name != '\0') { - if (*name == '/' || *name == '\\') { - last_separator = name + 1; - } - ++name; - } - if (*last_separator == '\0') return missing_name; - return last_separator; +static void print_help(std::ostream& stream, char const* program_name) { + stream + << "Usage: " << program_name << " [-h] [-t] [-b <path>] [path]+\n" + << " -h : Print this help message and exit the program.\n" + << " -t : Run tests after loading defines.\n" + << " -b : Use the following path as the base directory (instead of searching for one).\n" + << " -s : Use the following path as a hint to search for a base directory.\n" + << "Any following paths are read as mod directories, with priority starting at one above the base directory.\n" + << "(Paths with spaces need to be enclosed in \"quotes\").\n"; } -static bool headless_load(Dataloader::path_vector_t const& roots) { +static bool headless_load(GameManager& game_manager, Dataloader const& dataloader) { bool ret = true; - Logger::set_info_func([](std::string&& str) { std::cout << str; }); - Logger::set_warning_func([](std::string&& str) { std::cerr << str; }); - Logger::set_error_func([](std::string&& str) { std::cerr << str; }); - - GameManager game_manager { []() { - Logger::info("State updated"); - } }; - Dataloader dataloader; - - if (!dataloader.set_roots(roots)) { - Logger::error("Failed to set dataloader roots!"); - ret = false; - } if (!dataloader.load_defines(game_manager)) { Logger::error("Failed to load defines!"); ret = false; @@ -52,37 +38,105 @@ static bool headless_load(Dataloader::path_vector_t const& roots) { ret = false; } + return ret; +} + +static bool run_headless(Dataloader::path_vector_t const& roots, bool run_tests) { + bool ret = true; + + Dataloader dataloader; + if (!dataloader.set_roots(roots)) { + Logger::error("Failed to set dataloader roots!"); + ret = false; + } + + GameManager game_manager { []() { + Logger::info("State updated"); + } }; + + ret &= headless_load(game_manager, dataloader); - Testing testing = Testing(&game_manager); - std::cout << std::endl << "Testing Loaded" << std::endl << std::endl; - testing.execute_all_scripts(); - testing.report_results(); - std::cout << "Testing Executed" << std::endl << std::endl; + if (run_tests) { + Testing testing = Testing(&game_manager); + std::cout << std::endl << "Testing Loaded" << std::endl << std::endl; + testing.execute_all_scripts(); + testing.report_results(); + std::cout << "Testing Executed" << std::endl << std::endl; + } return ret; } +/* + $ program [-h] [-t] [-b] [path]+ +*/ + int main(int argc, char const* argv[]) { - Dataloader::path_vector_t roots; - if (argc < 2) { - // TODO - non-Windows default paths - static const fs::path default_path = "C:/Program Files (x86)/Steam/steamapps/common/Victoria 2"; - - std::cout - << "Usage: " << get_program_name(argc > 0 ? argv[0] : nullptr) << " <base defines dir> [[mod defines dir]+]\n" - << "Requires defines path(s) as arguments, starting with the base defines and continuing with mods.\n" - << "(Paths with spaces need to be enclosed in \"quotes\").\n" - << "Defaulting to " << default_path << std::endl; - roots.push_back(default_path); - } else { - for (int i = 1; i < argc; ++i) { - roots.push_back(argv[i]); + Logger::set_logger_funcs(); + + char const* program_name = Logger::get_filename(argc > 0 ? argv[0] : nullptr, "<program>"); + fs::path root; + bool run_tests = false; + int argn = 0; + + /* Reads the next argument and converts it to a path via path_transform. If reading or converting fails, an error + * message and the help text are displayed, along with returning false to signify the program should exit. + */ + const auto _read = [&root, &argn, argc, argv, program_name](std::string_view command, std::string_view path_use, auto path_transform) -> bool { + if (root.empty()) { + if (++argn < argc) { + char const* path = argv[argn]; + root = path_transform(path); + if (!root.empty()) { + return true; + } else { + std::cerr << "Empty path after giving \"" << path << "\" to " << path_use << " command line argument \"" << command << "\"." << std::endl; + } + } else { + std::cerr << "Missing path after " << path_use << " command line argument \"" << command << "\"." << std::endl; + } + } else { + std::cerr << "Duplicate " << path_use << " command line argument \"-b\"." << std::endl; } + print_help(std::cerr, program_name); + return false; + }; + + while (++argn < argc) { + char const* arg = argv[argn]; + if (strcmp(arg, "-h") == 0) { + print_help(std::cout, program_name); + return 0; + } else if (strcmp(arg, "-t") == 0) { + run_tests = true; + } else if (strcmp(arg, "-b") == 0) { + if (!_read("-b", "base directory", std::identity{})) { + return -1; + } + } else if (strcmp(arg, "-s") == 0) { + if (!_read("-s", "search hint", Dataloader::search_for_game_path)) { + return -1; + } + } else { + break; + } + } + if (root.empty()) { + root = Dataloader::search_for_game_path(); + if (root.empty()) { + std::cerr << "Search for base directory path failed!" << std::endl; + print_help(std::cerr, program_name); + return -1; + } + } + Dataloader::path_vector_t roots = { root }; + while (argn < argc) { + roots.emplace_back(root / argv[argn++]); } std::cout << "!!! HEADLESS SIMULATION START !!!" << std::endl; - const bool ret = headless_load(roots); + const bool ret = run_headless(roots, run_tests); std::cout << "!!! HEADLESS SIMULATION END !!!" << std::endl; |