From d62a72a7787e95649b8592a7eda1be4243fc86fb Mon Sep 17 00:00:00 2001
From: GreenComfyTea <klunics97@gmail.com>
Date: Sat, 29 Jun 2024 10:38:29 +0300
Subject: [PATCH] [Monster UI] Fix an issue when run away monsters UI was not
 disappearing

---
 .../MHR_Overlay/Monsters/monster_hook.lua     | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/reframework/autorun/MHR_Overlay/Monsters/monster_hook.lua b/reframework/autorun/MHR_Overlay/Monsters/monster_hook.lua
index 6513ebb..638d30d 100644
--- a/reframework/autorun/MHR_Overlay/Monsters/monster_hook.lua
+++ b/reframework/autorun/MHR_Overlay/Monsters/monster_hook.lua
@@ -44,6 +44,7 @@ local enemy_character_base_type_def = sdk.find_type_definition("snow.enemy.Enemy
 local enemy_character_base_update_method = enemy_character_base_type_def:get_method("update");
 
 local is_boss_enemy_method = enemy_character_base_type_def:get_method("get_isBossEnemy");
+local on_destroy_method = enemy_character_base_type_def:get_method("onDestroy");
 
 local enemy_damage_check_type_def = sdk.find_type_definition("snow.enemy.EnemyDamageCheck");
 local damage_check_update_param_update_method = enemy_damage_check_type_def:get_method("updateParam");
@@ -261,6 +262,20 @@ function this.update_rage_timer(anger_param, enemy)
 	large_monster.update_rage_timer(enemy, monster, anger_param);
 end
 
+function this.on_destroy(enemy)
+	local is_large = is_boss_enemy_method:call(enemy);
+	if is_large == nil then
+		error_handler.report("monster_hook.on_destroy", "Failed to Access Data: is_large");
+		return;
+	end
+
+	if is_large then
+		large_monster.list[enemy] = nil;
+	else
+		small_monster.list[enemy] = nil;
+	end
+end
+
 function this.init_dependencies()
 	small_monster = require("MHR_Overlay.Monsters.small_monster");
 	large_monster = require("MHR_Overlay.Monsters.large_monster");
@@ -296,6 +311,12 @@ function this.init_module()
 	end, function(retval)
 		return retval;
 	end);
+
+	sdk.hook(on_destroy_method, function(args)
+		pcall(this.on_destroy, sdk.to_managed_object(args[2]));
+	end, function(retval)
+		return retval;
+	end);
 end
 
 return this;