mirror of
https://github.com/Nereziel/cs2-WeaponPaints.git
synced 2026-02-18 10:43:22 +00:00
375 lines
16 KiB
C#
375 lines
16 KiB
C#
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 CounterStrikeSharp.API.Modules.Menu;
|
|
using CounterStrikeSharp.API.Modules.Utils;
|
|
using Nexd.MySQL;
|
|
using System.Runtime.ExceptionServices;
|
|
using static CounterStrikeSharp.API.Core.Listeners;
|
|
|
|
namespace WeaponPaints;
|
|
public class WeaponPaints : BasePlugin, IPluginConfig<WeaponPaintsConfig>
|
|
{
|
|
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.8";
|
|
|
|
public WeaponPaintsConfig Config { get; set; } = new();
|
|
|
|
MySqlDb? MySql = null;
|
|
private DateTime[] commandCooldown = new DateTime[Server.MaxPlayers];
|
|
private static string PluginPrefix = $" {ChatColors.Green}[WeaponPaints]{ChatColors.White}";
|
|
private Dictionary<ulong, Dictionary<nint, int>> gPlayerWeaponPaints = new();
|
|
private Dictionary<ulong, Dictionary<nint, int>> gPlayerWeaponSeed = new();
|
|
private Dictionary<ulong, Dictionary<nint, float>> gPlayerWeaponWear = new();
|
|
private Dictionary<int, string> g_playersKife = new();
|
|
private static readonly Dictionary<string, string> knifeTypes = new()
|
|
{
|
|
{ "m9", "weapon_knife_m9_bayonet" },
|
|
{ "karambit", "weapon_knife_karambit" },
|
|
{ "bayonet", "weapon_bayonet" },
|
|
{ "bowie", "weapon_knife_survival_bowie" },
|
|
{ "butterfly", "weapon_knife_butterfly" },
|
|
{ "falchion", "weapon_knife_falchion" },
|
|
{ "flip", "weapon_knife_flip" },
|
|
{ "gut", "weapon_knife_gut" },
|
|
{ "tactical", "weapon_knife_tactical" },
|
|
{ "shadow", "weapon_knife_push" },
|
|
{ "navaja", "weapon_knife_gypsy_jackknife" },
|
|
{ "stiletto", "weapon_knife_stiletto" },
|
|
{ "talon", "weapon_knife_widowmaker" },
|
|
{ "ursus", "weapon_knife_ursus" },
|
|
{ "css", "weapon_knife_css" },
|
|
{ "paracord", "weapon_knife_cord" },
|
|
{ "survival", "weapon_knife_canis" },
|
|
{ "nomad", "weapon_knife_outdoor" },
|
|
{ "skeleton", "weapon_knife_skeleton" },
|
|
{ "default", "weapon_knife" }
|
|
};
|
|
private static readonly List<string> weaponList = new()
|
|
{
|
|
"weapon_deagle", "weapon_elite", "weapon_fiveseven", "weapon_glock",
|
|
"weapon_ak47", "weapon_aug", "weapon_awp", "weapon_famas",
|
|
"weapon_g3sg1", "weapon_galilar", "weapon_m249", "weapon_m4a1",
|
|
"weapon_mac10", "weapon_p90", "weapon_mp5sd", "weapon_ump45",
|
|
"weapon_xm1014", "weapon_bizon", "weapon_mag7", "weapon_negev",
|
|
"weapon_sawedoff", "weapon_tec9", "weapon_hkp2000", "weapon_mp7",
|
|
"weapon_mp9", "weapon_nova", "weapon_p250", "weapon_scar20",
|
|
"weapon_sg556", "weapon_ssg08", "weapon_m4a1_silencer", "weapon_usp_silencer",
|
|
"weapon_cz75a", "weapon_revolver", "weapon_bayonet", "weapon_knife"
|
|
};
|
|
public override void Load(bool hotReload)
|
|
{
|
|
base.Load(hotReload);
|
|
SetGlobalExceptionHandler();
|
|
MySql = new MySqlDb(Config.DatabaseHost, Config.DatabaseUser, Config.DatabasePassword, Config.DatabaseName!, Config.DatabasePort);
|
|
RegisterListener<Listeners.OnEntitySpawned>(OnEntitySpawned);
|
|
RegisterListener<Listeners.OnClientPutInServer>(OnClientPutInServer);
|
|
RegisterListener<Listeners.OnClientDisconnect>(OnClientDisconnect);
|
|
RegisterListener<Listeners.OnMapStart>(OnMapStart);
|
|
RegisterEventHandler<EventPlayerSpawn>(OnPlayerSpawn);
|
|
//RegisterEventHandler<EventRoundPrestart>(OnRoundPreStart);
|
|
SetupMenus();
|
|
}
|
|
public void OnConfigParsed(WeaponPaintsConfig config)
|
|
{
|
|
Config = config;
|
|
}
|
|
// TODO: fix for map which change mp_t_default_melee
|
|
/*private HookResult OnRoundPreStart(EventRoundPrestart @event, GameEventInfo info)
|
|
{
|
|
NativeAPI.IssueServerCommand("mp_t_default_melee \"\"");
|
|
NativeAPI.IssueServerCommand("mp_ct_default_melee \"\"");
|
|
return HookResult.Continue;
|
|
}
|
|
*/
|
|
public override void Unload(bool hotReload)
|
|
{
|
|
RemoveGlobalExceptionHandler();
|
|
base.Unload(hotReload);
|
|
}
|
|
private void GlobalExceptionHandler(object? sender, FirstChanceExceptionEventArgs @event)
|
|
{
|
|
Log(@event.Exception.ToString());
|
|
}
|
|
private void SetGlobalExceptionHandler()
|
|
{
|
|
AppDomain.CurrentDomain.FirstChanceException += this.GlobalExceptionHandler;
|
|
}
|
|
private void RemoveGlobalExceptionHandler()
|
|
{
|
|
AppDomain.CurrentDomain.FirstChanceException -= this.GlobalExceptionHandler;
|
|
}
|
|
private void OnMapStart(string mapName)
|
|
{
|
|
// TODO
|
|
// needed for now
|
|
base.AddTimer(2.0f, () => {
|
|
NativeAPI.IssueServerCommand("mp_t_default_melee \"\"");
|
|
NativeAPI.IssueServerCommand("mp_ct_default_melee \"\"");
|
|
});
|
|
}
|
|
|
|
private void OnClientPutInServer(int playerSlot)
|
|
{
|
|
int playerIndex = playerSlot + 1;
|
|
Task.Run(async () =>
|
|
{
|
|
await GetKnifeFromDatabase(playerIndex);
|
|
await GetWeaponPaintsFromDatabase(playerIndex);
|
|
});
|
|
}
|
|
private void OnClientDisconnect(int playerSlot)
|
|
{
|
|
// TODO: Clean up after player
|
|
}
|
|
|
|
private HookResult OnPlayerSpawn(EventPlayerSpawn @event, GameEventInfo info)
|
|
{
|
|
var player = @event.Userid;
|
|
if (!player.IsValid || !player.PlayerPawn.IsValid)
|
|
{
|
|
return HookResult.Continue;
|
|
}
|
|
if (player.IsBot)
|
|
{
|
|
player.GiveNamedItem("weapon_knife");
|
|
return HookResult.Continue;
|
|
}
|
|
|
|
if (!PlayerHasKnife(player))
|
|
{
|
|
if (g_playersKife.ContainsKey((int)player.EntityIndex!.Value.Value))
|
|
{
|
|
player.GiveNamedItem(g_playersKife[(int)player.EntityIndex!.Value.Value]);
|
|
}
|
|
else
|
|
{
|
|
player.GiveNamedItem("weapon_knife");
|
|
}
|
|
}
|
|
|
|
return HookResult.Continue;
|
|
}
|
|
private void OnEntitySpawned(CEntityInstance entity)
|
|
{
|
|
var designerName = entity.DesignerName;
|
|
if (!weaponList.Contains(designerName)) return;
|
|
bool isKnife = false;
|
|
var weapon = new CBasePlayerWeapon(entity.Handle);
|
|
|
|
if (designerName.Contains("knife") || designerName.Contains("bayonet"))
|
|
{
|
|
isKnife = true;
|
|
}
|
|
Server.NextFrame(() =>
|
|
{
|
|
if (!weapon.IsValid) return;
|
|
if (weapon.OwnerEntity.Value == null) return;
|
|
if (!weapon.OwnerEntity.Value.EntityIndex.HasValue) return;
|
|
int weaponOwner = (int)weapon.OwnerEntity.Value.EntityIndex.Value.Value;
|
|
var pawn = new CBasePlayerPawn(NativeAPI.GetEntityFromIndex(weaponOwner));
|
|
if (!pawn.IsValid) return;
|
|
var playerIndex = (int)pawn.Controller.Value.EntityIndex!.Value.Value;
|
|
var player = Utilities.GetPlayerFromIndex(playerIndex);
|
|
if (player == null || !player.IsValid || player.IsBot) return;
|
|
// TODO: Remove knife crashes here, needs another solution
|
|
/*if (isKnife && g_playersKife[(int)player.EntityIndex!.Value.Value] != "weapon_knife" && (weapon.AttributeManager.Item.ItemDefinitionIndex == 42 || weapon.AttributeManager.Item.ItemDefinitionIndex == 59))
|
|
{
|
|
RemoveKnifeFromPlayer(player);
|
|
return;
|
|
}*/
|
|
var steamId = new SteamID(player.SteamID);
|
|
if (!gPlayerWeaponPaints.ContainsKey(steamId.SteamId64)) return;
|
|
if (!gPlayerWeaponPaints[steamId.SteamId64].ContainsKey(weapon.AttributeManager.Item.ItemDefinitionIndex)) return;
|
|
//Log($"Apply on {weapon.DesignerName}({weapon.AttributeManager.Item.ItemDefinitionIndex}) paint {gPlayerWeaponPaints[steamId.SteamId64][weapon.AttributeManager.Item.ItemDefinitionIndex]} seed {gPlayerWeaponSeed[steamId.SteamId64][weapon.AttributeManager.Item.ItemDefinitionIndex]} wear {gPlayerWeaponWear[steamId.SteamId64][weapon.AttributeManager.Item.ItemDefinitionIndex]}");
|
|
weapon.AttributeManager.Item.ItemID = 16384;
|
|
weapon.AttributeManager.Item.ItemIDLow = 16384 & 0xFFFFFFFF;
|
|
weapon.AttributeManager.Item.ItemIDHigh = weapon.AttributeManager.Item.ItemIDLow >> 32;
|
|
weapon.FallbackPaintKit = gPlayerWeaponPaints[steamId.SteamId64][weapon.AttributeManager.Item.ItemDefinitionIndex];
|
|
weapon.FallbackSeed = gPlayerWeaponSeed[steamId.SteamId64][weapon.AttributeManager.Item.ItemDefinitionIndex];
|
|
weapon.FallbackWear = gPlayerWeaponWear[steamId.SteamId64][weapon.AttributeManager.Item.ItemDefinitionIndex];
|
|
if (!isKnife && weapon.CBodyComponent != null && weapon.CBodyComponent.SceneNode != null)
|
|
{
|
|
var skeleton = GetSkeletonInstance(weapon.CBodyComponent.SceneNode);
|
|
skeleton.ModelState.MeshGroupMask = 2;
|
|
}
|
|
});
|
|
}
|
|
public void RemoveKnifeFromPlayer(CCSPlayerController player)
|
|
{
|
|
if (!player.PawnIsAlive) return;
|
|
if (!g_playersKife.ContainsKey((int)player.EntityIndex!.Value.Value)) return;
|
|
var weapons = player.PlayerPawn.Value.WeaponServices!.MyWeapons;
|
|
foreach (var weapon in weapons)
|
|
{
|
|
if (weapon.IsValid && weapon.Value.IsValid)
|
|
{
|
|
//if (weapon.Value.AttributeManager.Item.ItemDefinitionIndex == 42 || weapon.Value.AttributeManager.Item.ItemDefinitionIndex == 59)
|
|
if (weapon.Value.DesignerName.Contains("knife"))
|
|
{
|
|
weapon.Value.Remove();
|
|
player.GiveNamedItem(g_playersKife[(int)player.EntityIndex!.Value.Value]);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
public static bool PlayerHasKnife(CCSPlayerController player)
|
|
{
|
|
if (!player.PawnIsAlive) return false;
|
|
var weapons = player.PlayerPawn.Value.WeaponServices!.MyWeapons;
|
|
foreach (var weapon in weapons)
|
|
{
|
|
if (weapon.IsValid && weapon.Value.IsValid)
|
|
{
|
|
if (weapon.Value.DesignerName.Contains("knife"))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
private void SetupMenus()
|
|
{
|
|
var giveItemMenu = new ChatMenu("Knife Menu");
|
|
var handleGive = (CCSPlayerController player, ChatMenuOption option) =>
|
|
{
|
|
if (knifeTypes.TryGetValue(option.Text, out var knife))
|
|
{
|
|
Task.Run(() => SyncKnifeToDatabase((int)player.EntityIndex!.Value.Value, knife));
|
|
g_playersKife[(int)player.EntityIndex!.Value.Value] = knifeTypes[option.Text];
|
|
player.PrintToChat($"You have chosen {option.Text} as your knife.");
|
|
RemoveKnifeFromPlayer(player);
|
|
}
|
|
};
|
|
foreach (var knife in knifeTypes)
|
|
{
|
|
giveItemMenu.AddMenuOption(knife.Key, handleGive);
|
|
}
|
|
AddCommand("css_knife", "Knife Menu", (player, info) => { if (player == null) return; ChatMenus.OpenMenu(player, giveItemMenu); });
|
|
}
|
|
[ConsoleCommand("css_wp", "refreshskins")]
|
|
public void OnCommandRefresh(CCSPlayerController? player, CommandInfo command)
|
|
{
|
|
if (player == null) return;
|
|
int playerIndex = (int)player.EntityIndex!.Value.Value;
|
|
if (DateTime.UtcNow >= commandCooldown[playerIndex].AddSeconds(Config.CmdRefreshCooldownSeconds))
|
|
{
|
|
commandCooldown[playerIndex] = DateTime.UtcNow;
|
|
Task.Run(async () => await GetWeaponPaintsFromDatabase(playerIndex));
|
|
player.PrintToChat($"{PluginPrefix} Refreshing weapon paints.");
|
|
return;
|
|
}
|
|
player.PrintToChat($"{PluginPrefix} You can't refresh weapon paints right now.");
|
|
}
|
|
[ConsoleCommand("css_ws", "weaponskins")]
|
|
public void OnCommandWS(CCSPlayerController? player, CommandInfo command)
|
|
{
|
|
if (player == null) return;
|
|
player.PrintToChat($"{PluginPrefix} Visit {ChatColors.Purple}{Config.WebSite} {ChatColors.White}where you can change skins.");
|
|
player.PrintToChat($"{PluginPrefix} Type {ChatColors.Purple}!wp {ChatColors.White}in chat to synchronize chosen skins.");
|
|
player.PrintToChat($"{PluginPrefix} Type {ChatColors.Purple}!knife {ChatColors.White}in chat to open knife menu.");
|
|
}
|
|
public static 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 playerIndex)
|
|
{
|
|
try
|
|
{
|
|
CCSPlayerController player = Utilities.GetPlayerFromIndex(playerIndex);
|
|
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 WeaponDefIndex = result.Get<int>(pair.Key, "weapon_defindex");
|
|
int PaintId = result.Get<int>(pair.Key, "weapon_paint_id");
|
|
float Wear = result.Get<float>(pair.Key, "weapon_wear");
|
|
int Seed = result.Get<int>(pair.Key, "weapon_seed");
|
|
|
|
if (!gPlayerWeaponPaints.ContainsKey(steamId.SteamId64))
|
|
{
|
|
gPlayerWeaponPaints[steamId.SteamId64] = new Dictionary<nint, int>();
|
|
}
|
|
if (!gPlayerWeaponWear.ContainsKey(steamId.SteamId64))
|
|
{
|
|
gPlayerWeaponWear[steamId.SteamId64] = new Dictionary<nint, float>();
|
|
}
|
|
if (!gPlayerWeaponSeed.ContainsKey(steamId.SteamId64))
|
|
{
|
|
gPlayerWeaponSeed[steamId.SteamId64] = new Dictionary<nint, int>();
|
|
}
|
|
|
|
gPlayerWeaponPaints[steamId.SteamId64][WeaponDefIndex] = PaintId;
|
|
gPlayerWeaponWear[steamId.SteamId64][WeaponDefIndex] = Wear;
|
|
gPlayerWeaponSeed[steamId.SteamId64][WeaponDefIndex] = Seed;
|
|
});
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log(e.Message);
|
|
return;
|
|
}
|
|
}
|
|
private async Task GetKnifeFromDatabase(int playerIndex)
|
|
{
|
|
try
|
|
{
|
|
CCSPlayerController player = Utilities.GetPlayerFromIndex(playerIndex);
|
|
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_knife").Where(conditions).SelectAsync();
|
|
|
|
string knife = result.Get<string>(0, "knife");
|
|
if (knife != null)
|
|
{
|
|
g_playersKife[playerIndex] = knife;
|
|
}
|
|
//Log($"{player.PlayerName} has this knife -> {g_playersKife[playerIndex]}");
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log(e.Message);
|
|
return;
|
|
}
|
|
}
|
|
private async Task SyncKnifeToDatabase(int playerIndex, string knife)
|
|
{
|
|
try
|
|
{
|
|
CCSPlayerController player = Utilities.GetPlayerFromIndex(playerIndex);
|
|
if (player == null || !player.IsValid) return;
|
|
var steamId = new SteamID(player.SteamID);
|
|
await MySql!.ExecuteNonQueryAsync($"INSERT INTO `wp_player_knife` (`steamid`, `knife`) VALUES('{steamId.SteamId64}', '{knife}') ON DUPLICATE KEY UPDATE `knife` = '{knife}';");
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log(e.Message);
|
|
return;
|
|
}
|
|
}
|
|
private static void Log(string message)
|
|
{
|
|
Console.BackgroundColor = ConsoleColor.DarkGray;
|
|
Console.ForegroundColor = ConsoleColor.Cyan;
|
|
Console.WriteLine(message);
|
|
Console.ResetColor();
|
|
}
|
|
} |