PokemonScriptTester/src/main.cpp

86 lines
3.8 KiB
C++

#include <PkmnLib/ScriptResolving/AngelScript/AngelScriptResolver.hpp>
#include <chrono>
#include <filesystem>
#include <iostream>
#include "../extern/args.hpp"
#include "BuildData/BuildLibrary.hpp"
#include "Globals.hpp"
#include "Tester/AngelScript/BattleFunctions.hpp"
#include "Tester/AngelScript/MiscMockFunctions.hpp"
#include "Tester/AngelScript/PokemonBuilder.hpp"
#include "Tester/AngelScript/TestFunctions.hpp"
#include "Tester/TestRunner.hpp"
#include "git.h"
int main(int argc, char** argv) {
std::cout << "PokemonScript Tester version: " << GIT_DESCRIBE << std::endl;
args::ArgumentParser parser("PkmnLib Script Tester.", "");
args::HelpFlag help(parser, "help", "Display this help menu", {'h', "help"});
std::string workingDirectory;
args::HelpFlag helpFlag(parser, "help", "Display this help menu", {'h', "help"});
args::ValueFlag<std::string> workingDirFlag(parser, "Working Directory", "Which work directory to use.",
{"workdir"});
args::Flag logTimeFlag(parser, "time-log", "Whether to show time logging.", {'t', "time-log"});
args::Flag forceColorFlag(parser, "force-color", "Whether to force color text output.", {'c', "force-color"});
args::Flag waitForDebuggerFlag(parser, "wait-for-debugger", "Whether to wait until a debugger is connected or not.",
{'w', "wait-for-debugger"});
try {
parser.ParseCLI(argc, argv);
} catch (args::Help&) {
std::cout << parser;
return 0;
} catch (args::ParseError& e) {
std::cerr << e.what() << std::endl;
std::cerr << parser;
return 1;
} catch (args::ValidationError& e) {
std::cerr << e.what() << std::endl;
std::cerr << parser;
return 1;
}
if (workingDirFlag) {
workingDirectory = args::get(workingDirFlag);
}
if (!workingDirectory.empty()) {
chdir((const char*)std::filesystem::path(workingDirectory).c_str());
}
if (forceColorFlag.Get()) {
termcolor::colorize(std::cout);
}
std::chrono::steady_clock::time_point begin = std::chrono::steady_clock::now();
std::function<void(PkmnLib::Battling::ScriptResolver*)> initialize =
[](PkmnLib::Battling::ScriptResolver* resolver) {
auto* scriptResolver = dynamic_cast<AngelScriptResolver*>(resolver);
scriptResolver->DefineWord("TESTS");
TestFunctions::Register(scriptResolver);
BattleFunctions::Register(scriptResolver);
MiscMockFunctions::Register(scriptResolver);
PokemonBuilderRegistration::Register(scriptResolver);
};
Globals::Library = BuildLibrary::Build("", initialize);
std::chrono::steady_clock::time_point end_build = std::chrono::steady_clock::now();
if (!Globals::Library.HasValue()) {
return 1;
}
if (logTimeFlag.Get()) {
std::cout << "Finished building library in: "
<< std::chrono::duration_cast<std::chrono::milliseconds>(end_build - begin).count() << "ms"
<< std::endl;
BuildLibrary::LogBuildTimes();
std::cout << std::endl;
}
auto* scriptResolver = dynamic_cast<AngelScriptResolver*>(Globals::Library.GetValue()->GetScriptResolver().get());
auto testRunner = TestRunner(scriptResolver);
std::chrono::steady_clock::time_point beginTests = std::chrono::steady_clock::now();
auto v = testRunner.RunAll(waitForDebuggerFlag.Get());
std::chrono::steady_clock::time_point end = std::chrono::steady_clock::now();
std::cout << std::endl
<< "Total Run time: " << std::chrono::duration_cast<std::chrono::milliseconds>(end - begin).count()
<< "ms (of which " << std::chrono::duration_cast<std::chrono::milliseconds>(end - beginTests).count()
<< " ms was test runs)" << std::endl;
return v;
}