feat: add onSetArmor #95

Author: ShrBox

src/legacy/api/EventAPI.cpp

@@ -432,18 +432,9 @@ void EnableEventListener(int eventId) {
         lse::events::PlayerOpenContainerScreenEvent();
         break;
 
-        // case EVENT_TYPES::onSetArmor:
-        //   Event::PlayerSetArmorEvent::subscribe([](const PlayerSetArmorEvent &ev)
-        //   {
-        //     IF_LISTENED(EVENT_TYPES::onSetArmor) {
-        //       CallEvent(EVENT_TYPES::onSetArmor,
-        //       PlayerClass::newPlayer(ev.mPlayer),
-        //                 Number::newNumber(ev.mSlot),
-        //                 ItemClass::newItem(ev.mArmorItem));
-        //     }
-        //     IF_LISTENED_END(EVENT_TYPES::onSetArmor);
-        //   });
-        //   break;
+    case EVENT_TYPES::onSetArmor:
+        lse::events::PlayerSetArmorEvent();
+        break;
 
     case EVENT_TYPES::onEat:
         bus.emplaceListener<PlayerUseItemEvent>([](PlayerUseItemEvent& ev) {

src/legacy/events/EventHooks.cpp

@@ -935,14 +935,35 @@ LL_TYPE_INSTANCE_HOOK(
 //     return origin(instance, entity, pos, face, clickPos);
 // }
 
-LL_TYPE_INSTANCE_HOOK(PlayerConsumeTotemHook, HookPriority::Normal, Player, &Player::consumeTotem, bool) {
+LL_TYPE_INSTANCE_HOOK(PlayerConsumeTotemHook, HookPriority::Normal, Player, "?consumeTotem@Player@@UEAA_NXZ", bool) {
     IF_LISTENED(EVENT_TYPES::onConsumeTotem) {
         CallEventRtnValue(EVENT_TYPES::onConsumeTotem, false, PlayerClass::newPlayer(this));
     }
     IF_LISTENED_END(EVENT_TYPES::onConsumeTotem);
     return origin();
 }
 
+LL_TYPE_INSTANCE_HOOK(
+    PlayerSetArmorHook,
+    HookPriority::Normal,
+    ServerPlayer,
+    "?setArmor@ServerPlayer@@UEAAXW4ArmorSlot@@AEBVItemStack@@@Z",
+    void,
+    ArmorSlot        armorSlot,
+    ItemStack const& item
+) {
+    IF_LISTENED(EVENT_TYPES::onSetArmor) {
+        CallEventVoid(
+            EVENT_TYPES::onSetArmor,
+            PlayerClass::newPlayer(this),
+            Number::newNumber((int)armorSlot),
+            ItemClass::newItem(&const_cast<ItemStack&>(item), false)
+        );
+    }
+    IF_LISTENED_END(EVENT_TYPES::onSetArmor);
+    origin(std::move(armorSlot), item);
+}
+
 void PlayerStartDestroyBlock() { PlayerStartDestroyHook::hook(); }
 void PlayerDropItem() { PlayerDropItemHook::hook(); }
 void PlayerOpenContainerEvent() { PlayerOpenContainerHook::hook(); }
@@ -993,6 +1014,7 @@ void PlayerUseBucketTakeEvent() {
     PlayerUseBucketTakeHook2::hook();
 }
 void PlayerConsumeTotemEvent() { PlayerConsumeTotemHook::hook(); }
+void PlayerSetArmorEvent() { PlayerSetArmorHook::hook(); }
 
 // NOLINTEND(cppcoreguidelines-pro-type-reinterpret-cast)
 // NOLINTEND(cppcoreguidelines-avoid-non-const-global-variables)

src/legacy/events/EventHooks.h

@@ -35,5 +35,6 @@ void ScoreChangedEvent();
 void PlayerUseBucketPlaceEvent();
 void PlayerUseBucketTakeEvent();
 void PlayerConsumeTotemEvent();
+void PlayerSetArmorEvent();
 
 } // namespace lse::events