#include <PkmnLib/ScriptResolving/AngelScript/AngelScriptResolver.hpp>
#include <filesystem>
#include <iostream>
#include "../extern/args.hpp"
#include "BuildData/BuildLibrary.hpp"
#include "Tester/TestFunctions.hpp"
#include "Tester/TestRunner.hpp"

int main(int argc, char** argv) {
    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"});
    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(std::filesystem::path(workingDirectory).c_str());
    }

    std::function<void(PkmnLib::Battling::ScriptResolver*)> initialize =
        [](PkmnLib::Battling::ScriptResolver* resolver) {
            auto* scriptResolver = dynamic_cast<AngelScriptResolver*>(resolver);
            scriptResolver->DefineWord("TEST");
            TestFunctions::Register(scriptResolver);
        };

    auto battleLib = ArbUt::OptionalScopedPtr(BuildLibrary::Build("", initialize));
    if (!battleLib.HasValue()) {
        return 1;
    }
    auto* scriptResolver = dynamic_cast<AngelScriptResolver*>(battleLib.GetValue()->GetScriptResolver().get());
    auto testRunner = TestRunner(scriptResolver);
    auto engine = scriptResolver->GetBuilder().GetEngine();
    testRunner.RunAll(engine);

    return 0;
}