GarpBot/cs2-WeaponPaints
1
0
Fork 0
mirror of https://github.com/Nereziel/cs2-WeaponPaints.git synced 2026-03-11 16:59:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update WeaponPaints.cs

Browse Source 
add async mysql query
add command !wp with 2min cooldown (for now hardcoded)
This commit is contained in:
Nereziel
2023-11-08 13:45:49 +01:00
parent 15ea69254d
commit 5a14b3f7c7
1 changed files with 112 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

199
WeaponPaints.cs
View File

@@ -1,100 +1,125 @@
using CounterStrikeSharp.API;
using CounterStrikeSharp.API.Core;
using CounterStrikeSharp.API.Core.Attributes.Registration;
using CounterStrikeSharp.API.Modules.Commands;
using CounterStrikeSharp.API.Modules.Entities;
using CounterStrikeSharp.API.Modules.Memory;
using Nexd.MySQL;
namespace WeaponPaints;
public class WeaponPaints : BasePlugin
{
public override string ModuleName => "WeaponPaints";
public override string ModuleDescription => "Connector for web-based player chosen wepaon paints.";
public override string ModuleAuthor => "Nereziel";
public override string ModuleVersion => "0.4";
MySqlDb? MySql = null;
private Dictionary<ulong, Dictionary<nint, int>> g_playersSkins = new Dictionary<ulong, Dictionary<nint, int>>();
public override string ModuleName => "WeaponPaints";
public override string ModuleDescription => "Connector for web-based player chosen wepaon paints.";
public override string ModuleAuthor => "Nereziel";
public override string ModuleVersion => "0.5";
MySqlDb? MySql = null;
public DateTime[] commandCooldown = new DateTime[Server.MaxPlayers];
private Dictionary<ulong, Dictionary<nint, int>> g_playersSkins = new Dictionary<ulong, Dictionary<nint, int>>();
public override void Load(bool hotReload)
{
new Cfg().CheckConfig(ModuleDirectory);
MySql = new MySqlDb(Cfg.config.DatabaseHost!, Cfg.config.DatabaseUser!, Cfg.config.DatabasePassword!, Cfg.config.DatabaseName!, (int)Cfg.config.DatabasePort);
RegisterListener<Listeners.OnEntitySpawned>(OnEntitySpawned);
RegisterListener<Listeners.OnClientPutInServer>(OnClientConnect);
}
private void OnEntitySpawned(CEntityInstance entity)
{
var designerName = entity.DesignerName;
if (!designerName.Contains("weapon_")) return;
if (designerName.Contains("knife")) return;
if (designerName.Contains("bayonet")) return;
var weapon = new CBasePlayerWeapon(entity.Handle);
Server.NextFrame(() =>
public override void Load(bool hotReload)
{
if (!weapon.IsValid) return;
var pawn = new CBasePlayerPawn(NativeAPI.GetEntityFromIndex((int)weapon.OwnerEntity.Value.EntityIndex!.Value.Value));
if (!pawn.IsValid) return;
var playerIndex = (int)pawn.Controller.Value.EntityIndex!.Value.Value;
CCSPlayerController player = Utilities.GetPlayerFromIndex(playerIndex);
if (player == null || !player.IsValid) return;
var steamId = new SteamID(player.SteamID);
if (g_playersSkins.TryGetValue(steamId.SteamId64, out var weaponIDs))
{
if (weaponIDs.TryGetValue(weapon.AttributeManager.Item.ItemDefinitionIndex, out var weaponPaint))
{
weapon.AttributeManager.Item.ItemIDLow = unchecked((uint)-1);
weapon.AttributeManager.Item.ItemIDHigh = unchecked((uint)-1);
weapon.FallbackPaintKit = weaponPaint;
weapon.FallbackSeed = 0;
weapon.FallbackWear = 0.0001f;
}
}
});
}
private static void Log(string message)
{
Console.BackgroundColor = ConsoleColor.DarkGray;
Console.ForegroundColor = ConsoleColor.DarkMagenta;
Console.WriteLine(message);
Console.ResetColor();
}
private void OnClientConnect(int playerSlot)
{
try
{
CCSPlayerController player = Utilities.GetPlayerFromSlot(playerSlot);
if (player == null || !player.IsValid) return;
var steamId = new SteamID(player.SteamID);
MySqlQueryCondition conditions = new MySqlQueryCondition()
.Add("steamid", "=", steamId.SteamId64.ToString());
MySqlQueryResult result = MySql!.Table("wp_player_skins").Where(conditions).Select();
result.ToList().ForEach(pair =>
{
int weponId = result.Get<int>(pair.Key, "weapon_defindex");
int weponPaint = result.Get<int>(pair.Key, "weapon_paint_id");
if (!g_playersSkins.ContainsKey(steamId.SteamId64))
{
g_playersSkins[steamId.SteamId64] = new Dictionary<nint, int>();
}
g_playersSkins[steamId.SteamId64][weponId] = weponPaint;
}
);
new Cfg().CheckConfig(ModuleDirectory);
MySql = new MySqlDb(Cfg.config.DatabaseHost!, Cfg.config.DatabaseUser!, Cfg.config.DatabasePassword!, Cfg.config.DatabaseName!, (int)Cfg.config.DatabasePort);
RegisterListener<Listeners.OnEntitySpawned>(OnEntitySpawned);
RegisterListener<Listeners.OnClientConnect>(OnClientConnect);
RegisterListener<Listeners.OnClientDisconnect>(OnClientDisconnect);
}
catch (Exception)
private void OnClientConnect(int playerSlot, string name, string ipAddress)
{
return;
int slot = playerSlot;
Server.NextFrame(() =>
{
Task.Run(() => GetWeaponPaintsFromDatabase(slot));
});
}
}
}
private void OnClientDisconnect(int playerSlot)
{
// Clean up after player
}
private void OnEntitySpawned(CEntityInstance entity)
{
var designerName = entity.DesignerName;
if (!designerName.Contains("weapon_")) return;
if (designerName.Contains("knife")) return;
if (designerName.Contains("bayonet")) return;
var weapon = new CBasePlayerWeapon(entity.Handle);
Server.NextFrame(() =>
{
if (!weapon.IsValid) return;
var pawn = new CBasePlayerPawn(NativeAPI.GetEntityFromIndex((int)weapon.OwnerEntity.Value.EntityIndex!.Value.Value));
if (!pawn.IsValid) return;
var playerIndex = (int)pawn.Controller.Value.EntityIndex!.Value.Value;
CCSPlayerController player = Utilities.GetPlayerFromIndex(playerIndex);
if (player == null || !player.IsValid) return;
var steamId = new SteamID(player.SteamID);
if (g_playersSkins.TryGetValue(steamId.SteamId64, out var weaponIDs))
{
if (weaponIDs.TryGetValue(weapon.AttributeManager.Item.ItemDefinitionIndex, out var weaponPaint))
{
weapon.AttributeManager.Item.ItemIDLow = unchecked((uint)-1);
weapon.AttributeManager.Item.ItemIDHigh = unchecked((uint)-1);
weapon.FallbackPaintKit = weaponPaint;
weapon.FallbackSeed = 0;
weapon.FallbackWear = 0.0001f;
if (weapon.AttributeManager.Item.AccountID > 0 && weapon.CBodyComponent != null && weapon.CBodyComponent.SceneNode != null)
{
var skeleton = GetSkeletonInstance(weapon.CBodyComponent.SceneNode);
skeleton.ModelState.MeshGroupMask = 2;
}
}
}
});
}
[ConsoleCommand("css_wp", "refreshskins")]
public void OnCommandRefresh(CCSPlayerController? player, CommandInfo command)
{
if (player == null) return;
int playerSlot = (int)player.EntityIndex!.Value.Value - 1;
if (DateTime.UtcNow >= commandCooldown[playerSlot].AddMinutes(2))
{
commandCooldown[playerSlot] = DateTime.UtcNow;
Task.Run(async () => await GetWeaponPaintsFromDatabase(playerSlot));
player.PrintToChat("Refreshed weapon paints.");
return;
}
player.PrintToChat("You can't refresh weapon paints right now.");
}
public CSkeletonInstance GetSkeletonInstance(CGameSceneNode node)
{
Func<nint, nint> GetSkeletonInstance = VirtualFunction.Create<nint, nint>(node.Handle, 8);
return new CSkeletonInstance(GetSkeletonInstance(node.Handle));
}
private async Task GetWeaponPaintsFromDatabase(int playerSlot)
{
try
{
CCSPlayerController player = Utilities.GetPlayerFromSlot(playerSlot);
if (player == null || !player.IsValid) return;
var steamId = new SteamID(player.SteamID);
MySqlQueryCondition conditions = new MySqlQueryCondition()
.Add("steamid", "=", steamId.SteamId64.ToString());
MySqlQueryResult result = await MySql!.Table("wp_player_skins").Where(conditions).SelectAsync();
result.ToList().ForEach(pair =>
{
int weaponId = result.Get<int>(pair.Key, "weapon_defindex");
int weaponPaint = result.Get<int>(pair.Key, "weapon_paint_id");
if (!g_playersSkins.ContainsKey(steamId.SteamId64))
{
g_playersSkins[steamId.SteamId64] = new Dictionary<nint, int>();
}
g_playersSkins[steamId.SteamId64][weaponId] = weaponPaint;
}
);
}
catch (Exception)
{
return;
}
}
}