MHR-Overlay/reframework/autorun/MHR_Overlay/Misc/player_info.lua

local this = {};

local drawing;
local customization_menu;
local singletons;
local config;
local utils;
local error_handler;
local quest_status;
local time;

local sdk = sdk;
local tostring = tostring;
local pairs = pairs;
local ipairs = ipairs;
local tonumber = tonumber;
local require = require;
local pcall = pcall;
local table = table;
local string = string;
local Vector3f = Vector3f;
local d2d = d2d;
local math = math;
local json = json;
local log = log;
local fs = fs;
local next = next;
local type = type;
local setmetatable = setmetatable;
local getmetatable = getmetatable;
local assert = assert;
local select = select;
local coroutine = coroutine;
local utf8 = utf8;
local re = re;
local imgui = imgui;
local draw = draw;
local Vector2f = Vector2f;
local reframework = reframework;
local os = os;
local ValueType = ValueType;
local package = package;

this.list = {
	attack = 0;
	defense = 0;
	affinity = 0;
	
	stamina = 0;
	max_stamina = 0;
	
	element_type = 0;
	element_attack = 0;
	
	element_type_2 = 0;
	element_attack_2 = 0;
	
	fire_resistance = 0;
	water_resistance = 0;
	thunder_resistance = 0;
	ice_resistance = 0;
	dragon_resistance = 0;
};

local player_manager_type_def = sdk.find_type_definition("snow.player.PlayerManager");
local get_player_method = player_manager_type_def:get_method("getPlayer");
local find_master_player_method = player_manager_type_def:get_method("findMasterPlayer");

local player_base_type_def = find_master_player_method:get_return_type();
local get_player_data_method = player_base_type_def:get_method("get_PlayerData");

local player_data_type_def = get_player_data_method:get_return_type();

local attack_field = player_data_type_def:get_field("_Attack");
local defence_field = player_data_type_def:get_field("_Defence");
local critical_rate_field = player_data_type_def:get_field("_CriticalRate");


local stamina_field = player_data_type_def:get_field("_stamina");
local stamina_max_field = player_data_type_def:get_field("_staminaMax");

local element_type_field = player_data_type_def:get_field("_ElementType");
local element_attack_field = player_data_type_def:get_field("_ElementAttack");

local element_type_2nd_field = player_data_type_def:get_field("_ElementType2nd");
local element_attack_2nd_field = player_data_type_def:get_field("_ElementAttack2nd");

local resistance_element_field = player_data_type_def:get_field("_ResistanceElement");

local system_array_type_def = sdk.find_type_definition("System.Array");
local length_method = system_array_type_def:get_method("get_Length");
local get_value_method = system_array_type_def:get_method("GetValue(System.Int32)");

function this.update()
	if not config.current_config.stats_UI.enabled then
		return;
	end

	if quest_status.flow_state == quest_status.flow_states.NONE then
		return;
	end

	if singletons.player_manager == nil then
		error_handler.report("player_info.update", "Failed to access Data: player_manager");
		return;
	end

	local master_player = find_master_player_method:call(singletons.player_manager); 
	if master_player == nil then
		error_handler.report("player_info.update", "Failed to access Data: master_player");
		return;
	end

	local master_player_data = get_player_data_method:call(master_player);
	if master_player_data == nil then
		error_handler.report("player_info.update", "Failed to access Data: master_player_data");
	end

	this.update_generic("attack", master_player_data, attack_field);
	this.update_generic("affinity", master_player_data, critical_rate_field);
	this.update_generic("defense", master_player_data, defence_field);

	this.update_generic("stamina", master_player_data, stamina_field);
	this.list.stamina = math.floor(this.list.stamina / 30);

	this.update_generic("max_stamina", master_player_data, stamina_max_field);
	this.list.max_stamina = math.floor(this.list.max_stamina / 30);

	this.update_generic("element_type", master_player_data, element_type_field);
	this.update_generic("element_attack", master_player_data, element_attack_field);

	this.update_generic("element_type_2", master_player_data, element_type_2nd_field);
	this.update_generic("element_attack_2", master_player_data, element_attack_2nd_field);
	
	this.update_resistances(master_player_data);
end

function this.update_generic(key, player_data, field)
	local value = field:get_data(player_data);
	if value == nil then
		error_handler.report("player_info.update_generic", string.format("Failed to access Data: %s_value", key));
		return;
	end

	this.list[key] = math.floor(value);
end

function this.update_resistances(player_data)
	local resistance_element_array = resistance_element_field:get_data(player_data);
	if resistance_element_array == nil then
		error_handler.report("player_info.update_resistances", "Failed to access Data: resistance_element_array");
		return;
	end

	-- Fire Resistance
	local fire_resistance_valtype = get_value_method:call(resistance_element_array, 0);
	if fire_resistance_valtype == nil then
		error_handler.report("player_info.update_resistances", "Failed to access Data: fire_resistance_valtype");
		return;
	end

	local fire_resistance = fire_resistance_valtype:get_field("mValue");
	if fire_resistance ~= nil then
		this.list.fire_resistance = math.floor(fire_resistance);
	else
		error_handler.report("player_info.update_resistances", "Failed to access Data: fire_resistance");
		return;
	end

	-- Water Resistance
	local water_resistance_valtype = get_value_method:call(resistance_element_array, 1);
	if water_resistance_valtype == nil then
		error_handler.report("player_info.update_resistances", "Failed to access Data: water_resistance_valtype");
		return;
	end

	local water_resistance = water_resistance_valtype:get_field("mValue");
	if water_resistance ~= nil then
		this.list.water_resistance = math.floor(water_resistance);
	else
		error_handler.report("player_info.update_resistances", "Failed to access Data: water_resistance");
		return;
	end

	-- Thunder Resistance
	local thunder_resistance_valtype = get_value_method:call(resistance_element_array, 2);
	if thunder_resistance_valtype == nil then
		error_handler.report("player_info.update_resistances", "Failed to access Data: thunder_resistance_valtype");
		return;
	end

	local thunder_resistance = thunder_resistance_valtype:get_field("mValue");
	if thunder_resistance ~= nil then
		this.list.thunder_resistance = math.floor(thunder_resistance);
	else
		error_handler.report("player_info.update_resistances", "Failed to access Data: thunder_resistance");
		return;
	end

	-- Ice Resistance
	local ice_resistance_valtype = get_value_method:call(resistance_element_array, 3);
	if ice_resistance_valtype == nil then
		error_handler.report("player_info.update_resistances", "Failed to access Data: ice_resistance_valtype");
		return;
	end

	local ice_resistance = ice_resistance_valtype:get_field("mValue");
	if ice_resistance ~= nil then
		this.list.ice_resistance = math.floor(ice_resistance);
	else
		error_handler.report("player_info.update_resistances", "Failed to access Data: ice_resistance");
		return;
	end

	-- Dragon Resistance
	local dragon_resistance_valtype = get_value_method:call(resistance_element_array, 4);
	if dragon_resistance_valtype == nil then
		error_handler.report("player_info.update_resistances", "Failed to access Data: dragon_resistance_valtype");
		return;
	end

	local dragon_resistance = dragon_resistance_valtype:get_field("mValue");
	if dragon_resistance ~= nil then
		this.list.dragon_resistance = math.floor(dragon_resistance);
	else
		error_handler.report("player_info.update_resistances", "Failed to access Data: dragon_resistance");
		return;
	end
end

function this.init_dependencies()
	singletons = require("MHR_Overlay.Game_Handler.singletons");
	customization_menu = require("MHR_Overlay.UI.customization_menu");
	config = require("MHR_Overlay.Misc.config");
	utils = require("MHR_Overlay.Misc.utils");
	--health_UI_entity = require("MHR_Overlay.UI.UI_Entities.health_UI_entity");
	--stamina_UI_entity = require("MHR_Overlay.UI.UI_Entities.stamina_UI_entity");
	--screen = require("MHR_Overlay.Game_Handler.screen");
	drawing = require("MHR_Overlay.UI.drawing");
	--ailments = require("MHR_Overlay.Monsters.ailments");
	--ailment_UI_entity = require("MHR_Overlay.UI.UI_Entities.ailment_UI_entity");
	error_handler = require("MHR_Overlay.Misc.error_handler");
	quest_status = require("MHR_Overlay.Game_Handler.quest_status");
	time = require("MHR_Overlay.Game_Handler.time");
end

function this.init_module()
end

return this;