From 901b49f7b37e0e580a28da2a8d3fce9db9a90faa Mon Sep 17 00:00:00 2001 From: Parnic Date: Mon, 3 Jun 2019 22:46:24 -0500 Subject: [PATCH] Attempt to fix pet health/power not updating when entering instance --- modules/PetHealth.lua | 6 ++++-- modules/PetMana.lua | 6 ++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/modules/PetHealth.lua b/modules/PetHealth.lua index 472ea72..d944bc2 100644 --- a/modules/PetHealth.lua +++ b/modules/PetHealth.lua @@ -54,7 +54,7 @@ function PetHealth.prototype:Enable(core) self:RegisterEvent("UNIT_ENTERED_VEHICLE", "EnteringVehicle") self:RegisterEvent("UNIT_EXITED_VEHICLE", "ExitingVehicle") - self:RegisterEvent("PLAYER_ENTERING_WORLD", "CheckVehicle") + self:RegisterEvent("PLAYER_ENTERING_WORLD", "EnteringWorld") self.frame:SetAttribute("unit", self.unit) RegisterUnitWatch(self.frame) @@ -220,7 +220,9 @@ function PetHealth.prototype:ExitingVehicle(event, unit) end end -function PetHealth.prototype:CheckVehicle() +function PetHealth.prototype:EnteringWorld() + self:Update(self.unit) + if UnitHasVehicleUI("player") then self:EnteringVehicle(nil, "player", true) else diff --git a/modules/PetMana.lua b/modules/PetMana.lua index 87a55da..a8f4a07 100644 --- a/modules/PetMana.lua +++ b/modules/PetMana.lua @@ -88,7 +88,7 @@ function PetMana.prototype:Enable(core) self:RegisterEvent("UNIT_ENTERED_VEHICLE", "EnteringVehicle") self:RegisterEvent("UNIT_EXITED_VEHICLE", "ExitingVehicle") - self:RegisterEvent("PLAYER_ENTERING_WORLD", "CheckVehicle") + self:RegisterEvent("PLAYER_ENTERING_WORLD", "EnteringWorld") self:CheckPet() self:ManaType(nil, self.unit) @@ -239,7 +239,9 @@ function PetMana.prototype:ExitingVehicle(event, unit) end end -function PetMana.prototype:CheckVehicle() +function PetMana.prototype:EnteringWorld() + self:Update(self.unit) + if UnitHasVehicleUI("player") then self:EnteringVehicle(nil, "player", true) else