fix(Cooldown): Avoid a nil error in ChargesInfo()

herotc · Jul 24, 2024 · 5cf8fc8 · 5cf8fc8
1 parent 9361f72
commit 5cf8fc8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/HeroLib/Class/Spell/Cooldown.lua b/HeroLib/Class/Spell/Cooldown.lua
@@ -40,6 +40,10 @@ end
 function Spell:ChargesInfo()
   local SpellID = self:ID()
   local SpellInfo = GetSpellCharges(SpellID)
+  -- Non-charged spells now return nil, so let's return default values to avoid a nil error.
+  if not SpellInfo then
+    return 1, 1, 0, 0
+  end
 
   return SpellInfo.currentCharges, SpellInfo.maxCharges, SpellInfo.cooldownStartTime, SpellInfo.cooldownDuration
 end