Update PlayerMethods.h

azerothcore · Feb 7, 2025 · df1b97e · df1b97e
1 parent d869057
commit df1b97e
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/LuaEngine/methods/PlayerMethods.h b/src/LuaEngine/methods/PlayerMethods.h
@@ -2281,12 +2281,22 @@ namespace LuaPlayer
      * Sends a trainer window to the [Player] from the [Creature] specified
      *
      * @param [Creature] sender
+     * @param uint32 entry : trainer entry
      */
+
     int SendTrainerList(lua_State* L, Player* player)
     {
-        Creature* obj = Eluna::CHECKOBJ<Creature>(L, 2);
+        uint32 entry = Eluna::CHECKVAL<uint32>(L, 3, 0);
+
+        if (!entry) {
+            Creature* obj = Eluna::CHECKOBJ<Creature>(L, 2);
+            player->GetSession()->SendTrainerList(obj->GET_GUID());
+        }
+        else {
+            WorldObject* obj = Eluna::CHECKOBJ<WorldObject>(L, 2);
+            player->GetSession()->SendTrainerList(obj->GET_GUID(), entry);
+        }
 
-        player->GetSession()->SendTrainerList(obj->GET_GUID());
         return 0;
     }