Helper method to restore all attack uses on a Creature.

Signed-off-by: Deukhoofd <Deukhoofd@gmail.com>

CInterface/Battling/Creature.cpp

@@ -57,6 +57,7 @@ export uint8_t CreatureLib_Creature_Damage(Creature* p, uint32_t damage, DamageS
 export uint8_t CreatureLib_Creature_Heal(Creature* p, uint32_t health, bool canRevive) {
     Try(p->Heal(health, canRevive);)
 }
+export uint8_t CreatureLib_Creature_RestoreAllAttackUses(Creature* p) { Try(p->RestoreAllAttackUses();) }
 export uint8_t CreatureLib_Creature_GetActiveTalent(const Creature* p, const char*& out) {
     Try(out = p->GetActiveTalent().c_str();)
 }

src/Battling/Models/Creature.cpp

@@ -171,6 +171,14 @@ void Battling::Creature::Heal(uint32_t amount, bool canRevive) {
     _currentHealth = newHealth;
 }
 
+void Battling::Creature::RestoreAllAttackUses() noexcept {
+    for (auto& a : _attacks) {
+        if (a != nullptr) {
+            a->RestoreAllUses();
+        }
+    }
+}
+
 void Battling::Creature::OverrideActiveTalent(const ArbUt::StringView& talent) {
     _hasOverridenTalent = true;
     if (_activeTalent != nullptr) {
@@ -262,4 +270,4 @@ void Battling::Creature::AddVolatileScript(const ArbUt::StringView& name) {
 void Battling::Creature::AddVolatileScript(Script* script) { _volatile.Add(script); }
 void Battling::Creature::RemoveVolatileScript(const ArbUt::BasicStringView& name) { _volatile.Remove(name); }
 void Battling::Creature::RemoveVolatileScript(Battling::Script* script) { _volatile.Remove(script->GetName()); }
-bool Battling::Creature::HasVolatileScript(const ArbUt::BasicStringView& name) const { return _volatile.Has(name); }
+bool Battling::Creature::HasVolatileScript(const ArbUt::BasicStringView& name) const { return _volatile.Has(name); }

src/Battling/Models/Creature.hpp

@@ -120,6 +120,7 @@ namespace CreatureLib::Battling {
         void ChangeLevelBy(int8_t amount);
         void Damage(uint32_t damage, DamageSource source);
         void Heal(uint32_t amount, bool canRevive = false);
+        void RestoreAllAttackUses () noexcept;
         void OverrideActiveTalent(const ArbUt::StringView& talent);
         void AddExperience(uint32_t amount);