#pragma once

#include "gta_util.hpp"

#include "pointers.hpp"

#include "util/scripts.hpp"

#include <script/scrProgram.hpp>

namespace big

{

class script_function

{

rage::joaat_t m_script;

const memory::pattern m_pattern;

uint32_t m_ip;

std::string m_name;

public:

script_function(const std::string& name, const rage::joaat_t script, const std::string& pattern);

uint32_t get_ip(rage::scrProgram* program);

template<typename Arg>

void push_arg(uint64_t* stack, uint32_t& stack_pointer, Arg&& value)

{

*reinterpret_cast<std::remove_cv_t<std::remove_reference_t<Arg>>*>(reinterpret_cast<uint64_t*>(stack) + (stack_pointer++)) = std::forward<Arg>(value);

}

template<typename Ret, typename... Args>

Ret call(Args... args)

{

auto thread = gta_util::find_script_thread(m_script);

auto program = gta_util::find_script_program(m_script);

if (!thread || !program)

{

LOG(FATAL) << m_name << " failed to find its associated script running.";

return Ret();

}

auto ip = get_ip(program);

if (!ip)

{

return Ret();

}

auto tls_ctx = rage::tlsContext::get();

auto stack = (uint64_t*)thread->m_stack;

auto og_thread = tls_ctx->m_script_thread;

tls_ctx->m_script_thread = thread;

tls_ctx->m_is_script_thread_active = true;

rage::scrThreadContext ctx = thread->m_context;

auto top_stack = ctx.m_stack_pointer; // This will be the top item in the stack after the args and return address are cleaned off

(push_arg(stack, ctx.m_stack_pointer, std::forward<Args>(args)), ...);

stack[ctx.m_stack_pointer++] = 0;

ctx.m_instruction_pointer = ip;

ctx.m_state = rage::eThreadState::running;

g_pointers->m_script_vm(stack, g_pointers->m_script_globals, program, &ctx);

tls_ctx->m_script_thread = og_thread;

tls_ctx->m_is_script_thread_active = og_thread != nullptr;

if constexpr (!std::is_same_v<Ret, void>)

{

return *reinterpret_cast<Ret*>(stack + top_stack);

}

}

};

namespace scr_functions

{

inline script_function set_as_ceo("SAC", RAGE_JOAAT("freemode"), "2D 04 1D 00 00 5D");

inline script_function set_freemode_session_active("SFSA", RAGE_JOAAT("freemode"), "2D 00 02 00 00 75 5D ? ? ? 50");

inline script_function give_owned_freemode_weapons("GOFM", RAGE_JOAAT("freemode"), "2D 00 02 00 00 62 ? ? ? 72 57 ? ? 76");

inline script_function save_to_datafile("STD", RAGE_JOAAT("fm_race_creator"), "2D 01 03 00 00 71 2C");

inline script_function load_from_datafile("LFD", RAGE_JOAAT("fm_race_creator"), "2D 03 0C 00 00 71 2C");

inline script_function reset_session_data("RSD", RAGE_JOAAT("main_persistent"), "2D 02 7D 00 00");

inline script_function add_clan_logo_to_vehicle("ACLTV", RAGE_JOAAT("main_persistent"), "2D 02 04 00 00 5D ? ? ? 61");

inline script_function vehicle_cannot_accept_clan_logo("CVACL", RAGE_JOAAT("main_persistent"), "2D 01 03 00 00 2C 01 00 A1 06 ? 04");

inline script_function get_component_name_string("GCNS", RAGE_JOAAT("mp_weapons"), "2D 02 43 00 00 38 01");

inline script_function get_component_desc_string("GCDS", RAGE_JOAAT("mp_weapons"), "2D 02 43 00 00 38 00");

inline script_function get_weapon_desc_string("GWDS", RAGE_JOAAT("mp_weapons"), "2D 02 A0 00 00");

}

}