#ifndef POKEMONSCRIPTTESTER_TESTFUNCTIONS_HPP
#define POKEMONSCRIPTTESTER_TESTFUNCTIONS_HPP

#include <angelscript.h>
#include "AssertionFailed.hpp"
#include "TestEnvironment.hpp"

class TestFunctions {
private:
    struct AssertionData {
        std::string FileName;
        size_t Line;
    };

    static AssertionData GetAssertionData() {
        auto* ctx = asGetActiveContext();
        TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData());
        env->AssertionCount += 1;
        auto filename = ctx->GetFunction()->GetScriptSectionName();
        auto line = ctx->GetLineNumber();
        return AssertionData{
            .FileName = filename,
            .Line = (size_t)line,
        };
    }

    static bool Assert(bool value) {
        if (!value) {
            auto data = GetAssertionData();
            throw AssertionFailed(data.FileName, data.Line);
        }
        return true;
    }

#define GetEqualityMessage(expected, actual)                                                                           \
    std::string eqMsg;                                                                                                 \
    {                                                                                                                  \
        std::ostringstream ss;                                                                                         \
        ss << "Expected: '" << std::to_string(expected) << "', but was: '" << std::to_string(actual) << "'";           \
        eqMsg = ss.str();                                                                                              \
    }

    static bool AssertEquals(i32 expected, i32 actual) {
        if (expected != actual) {
            auto data = GetAssertionData();
            GetEqualityMessage(expected, actual);
            throw AssertionFailed(data.FileName, data.Line, eqMsg);
        }
        return true;
    }

public:
    static void Register(AngelScriptResolver* scriptResolver) {
        auto engine = scriptResolver->GetBuilder().GetEngine();
        Ensure(engine->RegisterGlobalFunction("bool Assert(bool expression)", asFUNCTION(Assert), asCALL_CDECL) >= 0);
        Ensure(engine->RegisterGlobalFunction("bool AssertEquals(int expected, int actual)", asFUNCTION(AssertEquals),
                                              asCALL_CDECL) >= 0);
    }
};

#endif // POKEMONSCRIPTTESTER_TESTFUNCTIONS_HPP