#pragma once
#include "stats.hpp"

#include "natives.hpp"

namespace lua::stats
{
	// Lua API: Table
	// Name: stats
	// Table for manipulating GTA stats.
	// For stats that get prefixed by either `MP0` or `MP1`, you can use `MPX` instead and the menu will resolve to the correct number automatically.

	// Lua API: Function
	// Table: stats
	// Name: get_character_index
	// Returns: integer: The current multiplayer character index (0 or 1).
	static int get_character_index()
	{
		return self::char_index;
	}

	static Hash stat_text_to_hash(std::string& text)
	{
		if (text[0] == '$')
		{
			text = text.substr(1);
		}

		std::transform(text.begin(), text.end(), text.begin(), ::toupper);

		if (text.substr(0, 3) == "MPX")
			text[2] = get_character_index() + '0';

		return rage::joaat(text);
	}

	// Lua API: Function
	// Table: stats
	// Name: get_bool
	// Param: stat_hash: integer: the stat hash.
	// Returns: boolean: The value of the given stat.
	static bool get_bool_hash(Hash stat_hash)
	{
		int out_value;
		STATS::STAT_GET_BOOL(stat_hash, &out_value, -1);

		return out_value;
	}

	// Lua API: Function
	// Table: stats
	// Name: get_bool
	// Param: stat_name: string: the stat name.
	// Returns: boolean: The value of the given stat.
	static bool get_bool_name(std::string stat_name)
	{
		return get_bool_hash(stat_text_to_hash(stat_name));
	}

	// Lua API: Function
	// Table: stats
	// Name: get_bool_masked
	// Param: stat_hash: integer: the stat hash.
	// Param: bit_index: integer: bit index.
	// Returns: boolean: The value of the given stat.
	static bool get_bool_masked_hash(Hash stat_hash, int bit_index)
	{
		int out_value;
		constexpr int bit_count = 8;
		STATS::STAT_GET_MASKED_INT(stat_hash, &out_value, bit_index, bit_count, -1);

		return out_value;
	}

	// Lua API: Function
	// Table: stats
	// Name: get_bool_masked
	// Param: stat_name: string: the stat name.
	// Param: bit_index: integer: bit index.
	// Returns: boolean: The value of the given stat.
	static bool get_bool_masked_name(std::string stat_name, int bit_index)
	{
		return get_bool_masked_hash(stat_text_to_hash(stat_name), bit_index);
	}

	// Lua API: Function
	// Table: stats
	// Name: get_float
	// Param: stat_hash: integer: the stat hash.
	// Returns: float: The value of the given stat.
	static float get_float_hash(Hash stat_hash)
	{
		float out_value;
		STATS::STAT_GET_FLOAT(stat_hash, &out_value, -1);

		return out_value;
	}

	// Lua API: Function
	// Table: stats
	// Name: get_float
	// Param: stat_name: string: the stat name.
	// Returns: float: The value of the given stat.
	static float get_float_name(std::string stat_name)
	{
		return get_float_hash(stat_text_to_hash(stat_name));
	}

	// Lua API: Function
	// Table: stats
	// Name: get_int
	// Param: stat_hash: integer: the stat hash.
	// Returns: integer: The value of the given stat.
	static int get_int_hash(Hash stat_hash)
	{
		int out_value;
		STATS::STAT_GET_INT(stat_hash, &out_value, -1);

		return out_value;
	}

	// Lua API: Function
	// Table: stats
	// Name: get_int
	// Param: stat_name: string: the stat name.
	// Returns: integer: The value of the given stat.
	static int get_int_name(std::string stat_name)
	{
		return get_int_hash(stat_text_to_hash(stat_name));
	}

	// Lua API: Function
	// Table: stats
	// Name: get_masked_int
	// Param: stat_hash: integer: the stat hash.
	// Param: bit_start: integer: bit start.
	// Param: bit_size: integer: bit size.
	// Returns: integer: The value of the given stat.
	static int get_masked_int_hash(Hash stat_hash, int bit_index, int bit_size)
	{
		int out_value;
		STATS::STAT_GET_MASKED_INT(stat_hash, &out_value, bit_index, bit_size, -1);

		return out_value;
	}

	// Lua API: Function
	// Table: stats
	// Name: get_masked_int
	// Param: stat_name: string: the stat name.
	// Param: bit_index: integer: bit index.
	// Param: bit_size: integer: bit size.
	// Returns: integer: The value of the given stat.
	static int get_masked_int_name(std::string stat_name, int bit_index, int bit_size)
	{
		return get_masked_int_hash(stat_text_to_hash(stat_name), bit_index, bit_size);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_bool
	// Param: stat_hash: integer: the stat hash.
	// Param: new_value: boolean: the new value for the stat.
	// Returns: boolean: True if succeeded.
	static bool set_bool_hash(Hash stat_hash, bool new_value)
	{
		return STATS::STAT_SET_BOOL(stat_hash, new_value, true);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_bool
	// Param: stat_name: string: the stat name.
	// Param: new_value: boolean: the new value for the stat.
	// Returns: boolean: True if succeeded.
	static bool set_bool_name(std::string stat_name, bool new_value)
	{
		return set_bool_hash(stat_text_to_hash(stat_name), new_value);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_bool_masked
	// Param: stat_hash: integer: the stat hash.
	// Param: new_value: boolean: the new value for the stat.
	// Param: bit_index: integer: bit_index.
	// Returns: boolean: True if succeeded.
	static bool set_bool_masked_hash(Hash stat_hash, bool new_value, int bit_index)
	{
		constexpr int bit_count = 8;
		return STATS::STAT_SET_MASKED_INT(stat_hash, new_value, bit_index, bit_count, true);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_bool_masked
	// Param: stat_name: string: the stat name.
	// Param: new_value: boolean: the new value for the stat.
	// Param: bit_index: integer: bit_index.
	// Returns: boolean: True if succeeded.
	static bool set_bool_masked_name(std::string stat_name, bool new_value, int bit_index)
	{
		return set_bool_masked_hash(stat_text_to_hash(stat_name), new_value, bit_index);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_float
	// Param: stat_hash: integer: the stat hash.
	// Param: new_value: float: the new value for the stat.
	// Returns: boolean: True if succeeded.
	static bool set_float_hash(Hash stat_hash, float new_value)
	{
		return STATS::STAT_SET_FLOAT(stat_hash, new_value, true);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_float
	// Param: stat_name: string: the stat name.
	// Param: new_value: float: the new value for the stat.
	// Returns: boolean: True if succeeded.
	static bool set_float_name(std::string stat_name, float new_value)
	{
		return set_float_hash(stat_text_to_hash(stat_name), new_value);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_int
	// Param: stat_hash: integer: the stat hash.
	// Param: new_value: integer: the new value for the stat.
	// Returns: boolean: True if succeeded.
	static bool set_int_hash(Hash stat_hash, int new_value)
	{
		return STATS::STAT_SET_INT(stat_hash, new_value, true);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_int
	// Param: stat_name: string: the stat name.
	// Param: new_value: integer: the new value for the stat.
	// Returns: boolean: True if succeeded.
	static bool set_int_name(std::string stat_name, int new_value)
	{
		return set_int_hash(stat_text_to_hash(stat_name), new_value);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_masked_int
	// Param: stat_hash: integer: the stat hash.
	// Param: new_value: integer: the new value for the stat.
	// Param: bit_start: integer: bit_start.
	// Param: bit_size: integer: bit_size.
	// Returns: boolean: True if succeeded.
	static bool set_masked_int_hash(Hash stat_hash, int new_value, int bit_start, int bit_size)
	{
		return STATS::STAT_SET_MASKED_INT(stat_hash, new_value, bit_start, bit_size, true);
	}

	// Lua API: Function
	// Table: stats
	// Name: set_masked_int
	// Param: stat_name: string: the stat name.
	// Param: new_value: integer: the new value for the stat.
	// Param: bit_start: integer: bit_start.
	// Param: bit_size: integer: bit_size.
	// Returns: boolean: True if succeeded.
	static bool set_masked_int_name(std::string stat_name, int new_value, int bit_start, int bit_size)
	{
		return set_masked_int_hash(stat_text_to_hash(stat_name), new_value, bit_start, bit_size);
	}

	void bind(sol::state& state)
	{
		auto ns = state["stats"].get_or_create<sol::table>();

		ns["get_bool"]        = sol::overload(get_bool_hash, get_bool_name);
		ns["get_bool_masked"] = sol::overload(get_bool_masked_hash, get_bool_masked_name);
		ns["get_float"]       = sol::overload(get_float_hash, get_float_name);
		ns["get_int"]         = sol::overload(get_int_hash, get_int_name);
		ns["get_masked_int"]  = sol::overload(get_masked_int_hash, get_masked_int_name);

		ns["set_bool"]        = sol::overload(set_bool_hash, set_bool_name);
		ns["set_bool_masked"] = sol::overload(set_bool_masked_hash, set_bool_masked_name);
		ns["set_float"]       = sol::overload(set_float_hash, set_float_name);
		ns["set_int"]         = sol::overload(set_int_hash, set_int_name);
		ns["set_masked_int"]  = sol::overload(set_masked_int_hash, set_masked_int_name);
	}
}