feat(scripting/player/weapons): SetActiveWeapon

Author: skuzzis

docgen/data/data.json

@@ -2980,6 +2980,21 @@
 						"params": {},
 						"additional": {}
 					},
+					"setactiveweapon": {
+						"title": "SetActiveWeapon",
+						"template": "function-syntax",
+						"language": "both",
+						"description": "Sets the active weapon.",
+						"variable": {
+							"cpp": "player->weapons->SetActiveWeapon",
+							"lua": "player:weapons():SetActiveWeapon"
+						},
+						"return": "void",
+						"params": {
+							"slot": "WeaponSlot"
+						},
+						"additional": {}
+					},
 					"getweapon": {
 						"title": "GetWeapon",
 						"template": "function-syntax",

plugin_files/scripting/includes/swiftly/player/weapons.h

@@ -15,6 +15,7 @@ typedef void (*Player_DropWeapons)(uint32_t);
 typedef void (*Player_RemoveWeapons)(uint32_t);
 typedef void (*Player_GiveWeapon)(uint32_t, const char *);
 typedef uint32_t (*Player_GetWeaponID)(uint32_t, uint32_t);
+typedef void (*Player_SetActiveWeapon)(uint32_t, uint32_t);
 
 class Weapons
 {
@@ -68,6 +69,15 @@ class Weapons
     {
         return (new Weapon(this->m_playerSlot, weaponID));
     }
+
+    void SetActiveWeapon(WeaponSlot slot)
+    {
+        void *player_SetActiveWeapon = FetchFunctionPtr(nullptr, "scripting_Player_Weapons_SetActiveWeapon");
+        if (player_SetActiveWeapon)
+            reinterpret_cast<Player_SetActiveWeapon>(player_SetActiveWeapon)(this->m_playerSlot, slot);
+        else
+            NOT_SUPPORTED("scripting_Player_Weapons_SetActiveWeapon");
+    }
 };
 
 #endif

src/components/Plugins/inc/Scripting.h

@@ -187,6 +187,7 @@ SMM_API int scripting_Player_GetLatency(uint32 playerId);
 
 SMM_API void scripting_Player_ExecuteCommand(uint32 playerId, const char *cmd);
 
+SMM_API void scripting_Player_Weapons_SetActiveWeapon(uint32 playerId, uint32 slot);
 SMM_API void scripting_Player_Weapons_Drop(uint32 playerId);
 SMM_API void scripting_Player_Weapons_Remove(uint32 playerId);
 SMM_API void scripting_Player_Weapons_Give(uint32 playerId, const char *name);

src/components/Plugins/src/language/lua/player.cpp

@@ -280,6 +280,8 @@ void SetupLuaPlayer(luacpp::LuaState *state, Plugin *plugin)
                    { scripting_Player_Weapons_Remove(base->playerSlot); })
         .DefMember("GiveWeapons", [](LuaPlayerArgsClass *base, const char *name) -> void
                    { scripting_Player_Weapons_Give(base->playerSlot, name); })
+        .DefMember("SetActiveWeapon", [](LuaPlayerArgsClass *base, uint32 slot) -> void
+                   { scripting_Player_Weapons_SetActiveWeapon(base->playerSlot, slot); })
         .DefMember("GetWeaponFromSlot", [weaponClass](LuaPlayerArgsClass *base, uint32 slot) -> luacpp::LuaObject
                    { return weaponClass.CreateInstance(base->playerSlot, scripting_Player_Weapons_GetWeaponID(base->playerSlot, slot)); })
         .DefMember("GetWeapon", [weaponClass](LuaPlayerArgsClass *base, uint32 weaponID) -> luacpp::LuaObject

src/components/Plugins/src/scripting/Player.cpp

@@ -756,6 +756,31 @@ SMM_API uint32_t scripting_Player_Weapon_GetSilencerType(uint32 playerId, uint32
     return vData->m_eSilencerType();
 }
 
+SMM_API void scripting_Player_Weapons_SetActiveWeapon(uint32 playerId, uint32 slot)
+{
+    Player *player = g_playerManager->GetPlayer(playerId);
+    if (!player)
+        return;
+
+    CBasePlayerWeapon *weapon = player->GetPlayerWeaponFromSlot((gear_slot_t)slot);
+    if (!weapon)
+        return;
+
+    CCSPlayerPawn *pawn = player->GetPlayerPawn();
+    if (!pawn)
+        return;
+
+    CPlayer_WeaponServices *weaponServices = pawn->m_pWeaponServices();
+    if (!weaponServices)
+        return;
+
+    CCSPlayer_ItemServices *itemServices = pawn->m_pItemServices();
+    if (!itemServices)
+        return;
+
+    weaponServices->m_hActiveWeapon = weapon;
+}
+
 SMM_API void scripting_Player_Weapon_SetReserveAmmo(uint32 playerId, uint32 slot, int ammo)
 {
     Player *player = g_playerManager->GetPlayer(playerId);