mirror of
https://github.com/Nereziel/cs2-WeaponPaints.git
synced 2026-02-18 18:49:21 +00:00
650 lines
24 KiB
C#
650 lines
24 KiB
C#
using Dapper;
|
||
using System.Collections.Concurrent;
|
||
using CounterStrikeSharp.API.Modules.Utils;
|
||
using System.Globalization;
|
||
using System.Data;
|
||
|
||
namespace WeaponPaints;
|
||
|
||
internal class WeaponSynchronization
|
||
{
|
||
private readonly WeaponPaintsConfig _config;
|
||
private readonly Database _database;
|
||
|
||
internal WeaponSynchronization(Database database, WeaponPaintsConfig config)
|
||
{
|
||
_database = database;
|
||
_config = config;
|
||
}
|
||
|
||
internal async Task GetPlayerData(PlayerInfo? player)
|
||
{
|
||
try
|
||
{
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
|
||
if (_config.Additional.KnifeEnabled)
|
||
GetKnifeFromDatabase(player, connection.GetConnection());
|
||
if (_config.Additional.GloveEnabled)
|
||
GetGloveFromDatabase(player, connection.GetConnection());
|
||
if (_config.Additional.AgentEnabled)
|
||
GetAgentFromDatabase(player, connection.GetConnection());
|
||
if (_config.Additional.MusicEnabled)
|
||
GetMusicFromDatabase(player, connection.GetConnection());
|
||
if (_config.Additional.SkinEnabled)
|
||
GetWeaponPaintsFromDatabase(player, connection.GetConnection());
|
||
if (_config.Additional.PinsEnabled)
|
||
GetPinsFromDatabase(player, connection.GetConnection());
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
// Log the exception or handle it appropriately
|
||
Console.WriteLine($"An error occurred: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
private void GetKnifeFromDatabase(PlayerInfo? player, IDbConnection connection)
|
||
{
|
||
try
|
||
{
|
||
if (!_config.Additional.KnifeEnabled || string.IsNullOrEmpty(player?.SteamId))
|
||
return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "SELECT knife, weapon_team FROM wp_player_knife WHERE steamid = @steamid ORDER BY weapon_team ASC"
|
||
: "SELECT `knife`, `weapon_team` FROM `wp_player_knife` WHERE `steamid` = @steamid ORDER BY `weapon_team` ASC";
|
||
var rows = connection.Query<dynamic>(query, new { steamid = player.SteamId }); // Retrieve all records for the player
|
||
|
||
foreach (var row in rows)
|
||
{
|
||
// Check if knife is null or empty
|
||
if (string.IsNullOrEmpty(row.knife)) continue;
|
||
|
||
// Determine the weapon team based on the query result
|
||
CsTeam weaponTeam = Convert.ToInt32(row.weapon_team) switch
|
||
{
|
||
2 => CsTeam.Terrorist,
|
||
3 => CsTeam.CounterTerrorist,
|
||
_ => CsTeam.None,
|
||
};
|
||
|
||
// Get or create entries for the player's slot
|
||
var playerKnives = WeaponPaints.GPlayersKnife.GetOrAdd(player.Slot, _ => new ConcurrentDictionary<CsTeam, string>());
|
||
|
||
if (weaponTeam == CsTeam.None)
|
||
{
|
||
// Assign knife to both teams if weaponTeam is None
|
||
playerKnives[CsTeam.Terrorist] = row.knife;
|
||
playerKnives[CsTeam.CounterTerrorist] = row.knife;
|
||
}
|
||
else
|
||
{
|
||
// Assign knife to the specific team
|
||
playerKnives[weaponTeam] = row.knife;
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Utility.Log($"An error occurred in GetKnifeFromDatabase: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
private void GetGloveFromDatabase(PlayerInfo? player, IDbConnection connection)
|
||
{
|
||
try
|
||
{
|
||
if (!_config.Additional.GloveEnabled || string.IsNullOrEmpty(player?.SteamId))
|
||
return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "SELECT weapon_defindex, weapon_team FROM wp_player_gloves WHERE steamid = @steamid ORDER BY weapon_team ASC"
|
||
: "SELECT `weapon_defindex`, `weapon_team` FROM `wp_player_gloves` WHERE `steamid` = @steamid ORDER BY `weapon_team` ASC";
|
||
var rows = connection.Query<dynamic>(query, new { steamid = player.SteamId }); // Retrieve all records for the player
|
||
|
||
foreach (var row in rows)
|
||
{
|
||
// Check if weapon_defindex is null
|
||
if (row.weapon_defindex == null) continue;
|
||
// Determine the weapon team based on the query result
|
||
var playerGloves = WeaponPaints.GPlayersGlove.GetOrAdd(player.Slot, _ => new ConcurrentDictionary<CsTeam, ushort>());
|
||
CsTeam weaponTeam = Convert.ToInt32(row.weapon_team) switch
|
||
{
|
||
2 => CsTeam.Terrorist,
|
||
3 => CsTeam.CounterTerrorist,
|
||
_ => CsTeam.None,
|
||
};
|
||
|
||
// Get or create entries for the player’s slot
|
||
|
||
if (weaponTeam == CsTeam.None)
|
||
{
|
||
// Assign glove ID to both teams if weaponTeam is None
|
||
playerGloves[CsTeam.Terrorist] = Convert.ToUInt16(row.weapon_defindex);
|
||
playerGloves[CsTeam.CounterTerrorist] = Convert.ToUInt16(row.weapon_defindex);
|
||
}
|
||
else
|
||
{
|
||
// Assign glove ID to the specific team
|
||
playerGloves[weaponTeam] = Convert.ToUInt16(row.weapon_defindex);
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Utility.Log($"An error occurred in GetGlovesFromDatabase: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
private void GetAgentFromDatabase(PlayerInfo? player, IDbConnection connection)
|
||
{
|
||
try
|
||
{
|
||
if (!_config.Additional.AgentEnabled || string.IsNullOrEmpty(player?.SteamId))
|
||
return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "SELECT agent_ct, agent_t FROM wp_player_agents WHERE steamid = @steamid"
|
||
: "SELECT `agent_ct`, `agent_t` FROM `wp_player_agents` WHERE `steamid` = @steamid";
|
||
var agentData = connection.QueryFirstOrDefault<(string, string)>(query, new { steamid = player.SteamId });
|
||
|
||
if (agentData == default) return;
|
||
var agentCT = agentData.Item1;
|
||
var agentT = agentData.Item2;
|
||
|
||
if (!string.IsNullOrEmpty(agentCT) || !string.IsNullOrEmpty(agentT))
|
||
{
|
||
WeaponPaints.GPlayersAgent[player.Slot] = (
|
||
agentCT,
|
||
agentT
|
||
);
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Utility.Log($"An error occurred in GetAgentFromDatabase: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
private void GetWeaponPaintsFromDatabase(PlayerInfo? player, IDbConnection connection)
|
||
{
|
||
try
|
||
{
|
||
if (!_config.Additional.SkinEnabled || player == null || string.IsNullOrEmpty(player.SteamId))
|
||
return;
|
||
|
||
var playerWeapons = WeaponPaints.GPlayerWeaponsInfo.GetOrAdd(player.Slot,
|
||
_ => new ConcurrentDictionary<CsTeam, ConcurrentDictionary<int, WeaponInfo>>());
|
||
|
||
// var weaponInfos = new ConcurrentDictionary<int, WeaponInfo>();
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "SELECT * FROM wp_player_skins WHERE steamid = @steamid ORDER BY weapon_team ASC"
|
||
: "SELECT * FROM `wp_player_skins` WHERE `steamid` = @steamid ORDER BY `weapon_team` ASC";
|
||
var playerSkins = connection.Query<dynamic>(query, new { steamid = player.SteamId });
|
||
|
||
foreach (var row in playerSkins)
|
||
{
|
||
int weaponDefIndex = Convert.ToInt32(row.weapon_defindex ?? 0);
|
||
int weaponPaintId = Convert.ToInt32(row.weapon_paint_id ?? 0);
|
||
float weaponWear = Convert.ToSingle(row.weapon_wear ?? 0f);
|
||
int weaponSeed = Convert.ToInt32(row.weapon_seed ?? 0);
|
||
string weaponNameTag = row.weapon_nametag?.ToString() ?? "";
|
||
bool weaponStatTrak = Convert.ToBoolean(row.weapon_stattrak ?? false);
|
||
int weaponStatTrakCount = Convert.ToInt32(row.weapon_stattrak_count ?? 0);
|
||
|
||
CsTeam weaponTeam = Convert.ToInt32(row.weapon_team) switch
|
||
{
|
||
2 => CsTeam.Terrorist,
|
||
3 => CsTeam.CounterTerrorist,
|
||
_ => CsTeam.None,
|
||
};
|
||
|
||
string[]? keyChainParts = row.weapon_keychain?.ToString().Split(';');
|
||
|
||
KeyChainInfo keyChainInfo = new KeyChainInfo();
|
||
|
||
if (keyChainParts!.Length == 5 &&
|
||
uint.TryParse(keyChainParts[0], out uint keyChainId) &&
|
||
float.TryParse(keyChainParts[1], NumberStyles.Float, CultureInfo.InvariantCulture, out float keyChainOffsetX) &&
|
||
float.TryParse(keyChainParts[2], NumberStyles.Float, CultureInfo.InvariantCulture, out float keyChainOffsetY) &&
|
||
float.TryParse(keyChainParts[3], NumberStyles.Float, CultureInfo.InvariantCulture, out float keyChainOffsetZ) &&
|
||
uint.TryParse(keyChainParts[4], out uint keyChainSeed))
|
||
{
|
||
// Successfully parsed the values
|
||
keyChainInfo.Id = keyChainId;
|
||
keyChainInfo.OffsetX = keyChainOffsetX;
|
||
keyChainInfo.OffsetY = keyChainOffsetY;
|
||
keyChainInfo.OffsetZ = keyChainOffsetZ;
|
||
keyChainInfo.Seed = keyChainSeed;
|
||
}
|
||
else
|
||
{
|
||
// Failed to parse the values, default to 0
|
||
keyChainInfo.Id = 0;
|
||
keyChainInfo.OffsetX = 0f;
|
||
keyChainInfo.OffsetY = 0f;
|
||
keyChainInfo.OffsetZ = 0f;
|
||
keyChainInfo.Seed = 0;
|
||
}
|
||
|
||
// Create the WeaponInfo object
|
||
WeaponInfo weaponInfo = new WeaponInfo
|
||
{
|
||
Paint = weaponPaintId,
|
||
Seed = weaponSeed,
|
||
Wear = weaponWear,
|
||
Nametag = weaponNameTag,
|
||
KeyChain = keyChainInfo,
|
||
StatTrak = weaponStatTrak,
|
||
StatTrakCount = weaponStatTrakCount,
|
||
};
|
||
|
||
// Retrieve and parse sticker data (up to 5 slots)
|
||
for (int i = 0; i <= 4; i++)
|
||
{
|
||
// Access the sticker data dynamically using reflection
|
||
string stickerColumn = $"weapon_sticker_{i}";
|
||
var stickerData = ((IDictionary<string, object>)row!)[stickerColumn]; // Safely cast row to a dictionary
|
||
|
||
if (string.IsNullOrEmpty(stickerData.ToString())) continue;
|
||
|
||
var parts = stickerData.ToString()!.Split(';');
|
||
|
||
//"id;schema;x;y;wear;scale;rotation"
|
||
if (parts.Length != 7 ||
|
||
!uint.TryParse(parts[0], out uint stickerId) ||
|
||
!uint.TryParse(parts[1], out uint stickerSchema) ||
|
||
!float.TryParse(parts[2], NumberStyles.Float, CultureInfo.InvariantCulture, out float stickerOffsetX) ||
|
||
!float.TryParse(parts[3], NumberStyles.Float, CultureInfo.InvariantCulture, out float stickerOffsetY) ||
|
||
!float.TryParse(parts[4], NumberStyles.Float, CultureInfo.InvariantCulture, out float stickerWear) ||
|
||
!float.TryParse(parts[5], NumberStyles.Float, CultureInfo.InvariantCulture, out float stickerScale) ||
|
||
!float.TryParse(parts[6], NumberStyles.Float, CultureInfo.InvariantCulture, out float stickerRotation)) continue;
|
||
|
||
StickerInfo stickerInfo = new StickerInfo
|
||
{
|
||
Id = stickerId,
|
||
Schema = stickerSchema,
|
||
OffsetX = stickerOffsetX,
|
||
OffsetY = stickerOffsetY,
|
||
Wear = stickerWear,
|
||
Scale = stickerScale,
|
||
Rotation = stickerRotation
|
||
};
|
||
|
||
weaponInfo.Stickers.Add(stickerInfo);
|
||
}
|
||
|
||
if (weaponTeam == CsTeam.None)
|
||
{
|
||
// Get or create entries for both teams
|
||
var terroristWeapons = playerWeapons.GetOrAdd(CsTeam.Terrorist, _ => new ConcurrentDictionary<int, WeaponInfo>());
|
||
var counterTerroristWeapons = playerWeapons.GetOrAdd(CsTeam.CounterTerrorist, _ => new ConcurrentDictionary<int, WeaponInfo>());
|
||
|
||
// Add weaponInfo to both team weapon dictionaries
|
||
terroristWeapons[weaponDefIndex] = weaponInfo;
|
||
counterTerroristWeapons[weaponDefIndex] = weaponInfo;
|
||
}
|
||
else
|
||
{
|
||
// Add to the specific team
|
||
var teamWeapons = playerWeapons.GetOrAdd(weaponTeam, _ => new ConcurrentDictionary<int, WeaponInfo>());
|
||
teamWeapons[weaponDefIndex] = weaponInfo;
|
||
}
|
||
|
||
// weaponInfos[weaponDefIndex] = weaponInfo;
|
||
}
|
||
|
||
// WeaponPaints.GPlayerWeaponsInfo[player.Slot][weaponTeam] = weaponInfos;
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Utility.Log($"An error occurred in GetWeaponPaintsFromDatabase: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
private void GetMusicFromDatabase(PlayerInfo? player, IDbConnection connection)
|
||
{
|
||
try
|
||
{
|
||
if (!_config.Additional.MusicEnabled || string.IsNullOrEmpty(player?.SteamId))
|
||
return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "SELECT music_id, weapon_team FROM wp_player_music WHERE steamid = @steamid ORDER BY weapon_team ASC"
|
||
: "SELECT `music_id`, `weapon_team` FROM `wp_player_music` WHERE `steamid` = @steamid ORDER BY `weapon_team` ASC";
|
||
var rows = connection.Query<dynamic>(query, new { steamid = player.SteamId }); // Retrieve all records for the player
|
||
|
||
foreach (var row in rows)
|
||
{
|
||
// Check if music_id is null
|
||
if (row.music_id == null) continue;
|
||
|
||
// Determine the weapon team based on the query result
|
||
CsTeam weaponTeam = Convert.ToInt32(row.weapon_team) switch
|
||
{
|
||
2 => CsTeam.Terrorist,
|
||
3 => CsTeam.CounterTerrorist,
|
||
_ => CsTeam.None,
|
||
};
|
||
|
||
// Get or create entries for the player's slot
|
||
var playerMusic = WeaponPaints.GPlayersMusic.GetOrAdd(player.Slot, _ => new ConcurrentDictionary<CsTeam, ushort>());
|
||
|
||
if (weaponTeam == CsTeam.None)
|
||
{
|
||
// Assign music ID to both teams if weaponTeam is None
|
||
playerMusic[CsTeam.Terrorist] = Convert.ToUInt16(row.music_id);
|
||
playerMusic[CsTeam.CounterTerrorist] = Convert.ToUInt16(row.music_id);
|
||
}
|
||
else
|
||
{
|
||
// Assign music ID to the specific team
|
||
playerMusic[weaponTeam] = Convert.ToUInt16(row.music_id);
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Utility.Log($"An error occurred in GetMusicFromDatabase: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
private void GetPinsFromDatabase(PlayerInfo? player, IDbConnection connection)
|
||
{
|
||
try
|
||
{
|
||
if (string.IsNullOrEmpty(player?.SteamId))
|
||
return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "SELECT id, weapon_team FROM wp_player_pins WHERE steamid = @steamid ORDER BY weapon_team ASC"
|
||
: "SELECT `id`, `weapon_team` FROM `wp_player_pins` WHERE `steamid` = @steamid ORDER BY `weapon_team` ASC";
|
||
var rows = connection.Query<dynamic>(query, new { steamid = player.SteamId }); // Retrieve all records for the player
|
||
|
||
foreach (var row in rows)
|
||
{
|
||
// Check if id is null
|
||
if (row.id == null) continue;
|
||
|
||
// Determine the weapon team based on the query result
|
||
CsTeam weaponTeam = Convert.ToInt32(row.weapon_team) switch
|
||
{
|
||
2 => CsTeam.Terrorist,
|
||
3 => CsTeam.CounterTerrorist,
|
||
_ => CsTeam.None,
|
||
};
|
||
|
||
// Get or create entries for the player's slot
|
||
var playerPins = WeaponPaints.GPlayersPin.GetOrAdd(player.Slot, _ => new ConcurrentDictionary<CsTeam, ushort>());
|
||
|
||
if (weaponTeam == CsTeam.None)
|
||
{
|
||
// Assign pin ID to both teams if weaponTeam is None
|
||
playerPins[CsTeam.Terrorist] = Convert.ToUInt16(row.id);
|
||
playerPins[CsTeam.CounterTerrorist] = Convert.ToUInt16(row.id);
|
||
}
|
||
else
|
||
{
|
||
// Assign pin ID to the specific team
|
||
playerPins[weaponTeam] = Convert.ToUInt16(row.id);
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Utility.Log($"An error occurred in GetPinsFromDatabase: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
internal async Task SyncKnifeToDatabase(PlayerInfo player, string knife, CsTeam[] teams)
|
||
{
|
||
if (!_config.Additional.KnifeEnabled || string.IsNullOrEmpty(player.SteamId) || string.IsNullOrEmpty(knife) || teams.Length == 0) return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "INSERT OR REPLACE INTO wp_player_knife (steamid, weapon_team, knife) VALUES(@steamid, @team, @newKnife)"
|
||
: "INSERT INTO `wp_player_knife` (`steamid`, `weapon_team`, `knife`) VALUES(@steamid, @team, @newKnife) ON DUPLICATE KEY UPDATE `knife` = @newKnife";
|
||
|
||
try
|
||
{
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
|
||
// Loop through each team and insert/update accordingly
|
||
foreach (var team in teams)
|
||
{
|
||
await connection.GetConnection().ExecuteAsync(query, new { steamid = player.SteamId, team, newKnife = knife });
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utility.Log($"Error syncing knife to database: {e.Message}");
|
||
}
|
||
}
|
||
|
||
internal async Task SyncGloveToDatabase(PlayerInfo player, ushort gloveDefIndex, CsTeam[] teams)
|
||
{
|
||
// Check if the necessary conditions are met
|
||
if (!_config.Additional.GloveEnabled || string.IsNullOrEmpty(player.SteamId) || teams.Length == 0)
|
||
return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "INSERT OR REPLACE INTO wp_player_gloves (steamid, weapon_team, weapon_defindex) VALUES(@steamid, @team, @gloveDefIndex)"
|
||
: @"INSERT INTO `wp_player_gloves` (`steamid`, `weapon_team`, `weapon_defindex`)
|
||
VALUES(@steamid, @team, @gloveDefIndex)
|
||
ON DUPLICATE KEY UPDATE `weapon_defindex` = @gloveDefIndex";
|
||
|
||
try
|
||
{
|
||
// Get a database connection
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
|
||
// Loop through each team and insert/update accordingly
|
||
foreach (var team in teams)
|
||
{
|
||
// Execute the SQL command for each team
|
||
await connection.GetConnection().ExecuteAsync(query, new {
|
||
steamid = player.SteamId,
|
||
team = (int)team, // Cast the CsTeam enum to int for insertion
|
||
gloveDefIndex
|
||
});
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
// Log any exceptions that occur
|
||
Utility.Log($"Error syncing glove to database: {e.Message}");
|
||
}
|
||
}
|
||
|
||
internal async Task SyncAgentToDatabase(PlayerInfo player)
|
||
{
|
||
if (!_config.Additional.AgentEnabled || string.IsNullOrEmpty(player.SteamId)) return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "INSERT OR REPLACE INTO wp_player_agents (steamid, agent_ct, agent_t) VALUES(@steamid, @agent_ct, @agent_t)"
|
||
: "INSERT INTO `wp_player_agents` (`steamid`, `agent_ct`, `agent_t`) VALUES(@steamid, @agent_ct, @agent_t) ON DUPLICATE KEY UPDATE `agent_ct` = @agent_ct, `agent_t` = @agent_t";
|
||
try
|
||
{
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
|
||
await connection.GetConnection().ExecuteAsync(query, new { steamid = player.SteamId, agent_ct = WeaponPaints.GPlayersAgent[player.Slot].CT, agent_t = WeaponPaints.GPlayersAgent[player.Slot].T });
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utility.Log($"Error syncing agents to database: {e.Message}");
|
||
}
|
||
}
|
||
|
||
internal async Task SyncWeaponPaintsToDatabase(PlayerInfo player)
|
||
{
|
||
if (string.IsNullOrEmpty(player.SteamId) || !WeaponPaints.GPlayerWeaponsInfo.TryGetValue(player.Slot, out var teamWeaponInfos))
|
||
return;
|
||
|
||
try
|
||
{
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
|
||
// Loop through each team (Terrorist and CounterTerrorist)
|
||
foreach (var (teamId, weaponsInfo) in teamWeaponInfos)
|
||
{
|
||
foreach (var (weaponDefIndex, weaponInfo) in weaponsInfo)
|
||
{
|
||
var paintId = weaponInfo.Paint;
|
||
var wear = weaponInfo.Wear;
|
||
var seed = weaponInfo.Seed;
|
||
|
||
// Prepare the queries to check and update/insert weapon skin data
|
||
string queryCheckExistence = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "SELECT COUNT(*) FROM wp_player_skins WHERE steamid = @steamid AND weapon_defindex = @weaponDefIndex AND weapon_team = @weaponTeam"
|
||
: "SELECT COUNT(*) FROM `wp_player_skins` WHERE `steamid` = @steamid AND `weapon_defindex` = @weaponDefIndex AND `weapon_team` = @weaponTeam";
|
||
|
||
var existingRecordCount = await connection.GetConnection().ExecuteScalarAsync<int>(
|
||
queryCheckExistence,
|
||
new { steamid = player.SteamId, weaponDefIndex, weaponTeam = teamId }
|
||
);
|
||
|
||
string query;
|
||
object parameters;
|
||
|
||
if (existingRecordCount > 0)
|
||
{
|
||
// Update existing record
|
||
query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "UPDATE wp_player_skins SET weapon_paint_id = @paintId, weapon_wear = @wear, weapon_seed = @seed WHERE steamid = @steamid AND weapon_defindex = @weaponDefIndex AND weapon_team = @weaponTeam"
|
||
: "UPDATE `wp_player_skins` SET `weapon_paint_id` = @paintId, `weapon_wear` = @wear, `weapon_seed` = @seed WHERE `steamid` = @steamid AND `weapon_defindex` = @weaponDefIndex AND `weapon_team` = @weaponTeam";
|
||
parameters = new { steamid = player.SteamId, weaponDefIndex, weaponTeam = (int)teamId, paintId, wear, seed };
|
||
}
|
||
else
|
||
{
|
||
// Insert new record
|
||
query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "INSERT INTO wp_player_skins (steamid, weapon_defindex, weapon_team, weapon_paint_id, weapon_wear, weapon_seed) VALUES (@steamid, @weaponDefIndex, @weaponTeam, @paintId, @wear, @seed)"
|
||
: "INSERT INTO `wp_player_skins` (`steamid`, `weapon_defindex`, `weapon_team`, `weapon_paint_id`, `weapon_wear`, `weapon_seed`) VALUES (@steamid, @weaponDefIndex, @weaponTeam, @paintId, @wear, @seed)";
|
||
parameters = new { steamid = player.SteamId, weaponDefIndex, weaponTeam = (int)teamId, paintId, wear, seed };
|
||
}
|
||
|
||
await connection.GetConnection().ExecuteAsync(query, parameters);
|
||
}
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utility.Log($"Error syncing weapon paints to database: {e.Message}");
|
||
}
|
||
}
|
||
|
||
internal async Task SyncMusicToDatabase(PlayerInfo player, ushort music, CsTeam[] teams)
|
||
{
|
||
if (!_config.Additional.MusicEnabled || string.IsNullOrEmpty(player.SteamId)) return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "INSERT OR REPLACE INTO wp_player_music (steamid, weapon_team, music_id) VALUES(@steamid, @team, @newMusic)"
|
||
: "INSERT INTO `wp_player_music` (`steamid`, `weapon_team`, `music_id`) VALUES(@steamid, @team, @newMusic) ON DUPLICATE KEY UPDATE `music_id` = @newMusic";
|
||
|
||
try
|
||
{
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
|
||
// Loop through each team and insert/update accordingly
|
||
foreach (var team in teams)
|
||
{
|
||
await connection.GetConnection().ExecuteAsync(query, new { steamid = player.SteamId, team, newMusic = music });
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utility.Log($"Error syncing music kit to database: {e.Message}");
|
||
}
|
||
}
|
||
|
||
internal async Task SyncPinToDatabase(PlayerInfo player, ushort pin, CsTeam[] teams)
|
||
{
|
||
if (!_config.Additional.PinsEnabled || string.IsNullOrEmpty(player.SteamId)) return;
|
||
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "INSERT OR REPLACE INTO wp_player_pins (steamid, weapon_team, id) VALUES(@steamid, @team, @newPin)"
|
||
: "INSERT INTO `wp_player_pins` (`steamid`, `weapon_team`, `id`) VALUES(@steamid, @team, @newPin) ON DUPLICATE KEY UPDATE `id` = @newPin";
|
||
|
||
try
|
||
{
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
|
||
// Loop through each team and insert/update accordingly
|
||
foreach (var team in teams)
|
||
{
|
||
await connection.GetConnection().ExecuteAsync(query, new { steamid = player.SteamId, team, newPin = pin });
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utility.Log($"Error syncing pin to database: {e.Message}");
|
||
}
|
||
}
|
||
|
||
internal async Task SyncStatTrakToDatabase(PlayerInfo player)
|
||
{
|
||
if (WeaponPaints.WeaponSync == null || WeaponPaints.GPlayerWeaponsInfo.IsEmpty) return;
|
||
if (string.IsNullOrEmpty(player.SteamId))
|
||
return;
|
||
|
||
try
|
||
{
|
||
await using var connection = await _database.GetConnectionAsync();
|
||
using var transaction = await connection.BeginTransactionAsync();
|
||
|
||
// Check if player's slot exists in GPlayerWeaponsInfo
|
||
if (!WeaponPaints.GPlayerWeaponsInfo.TryGetValue(player.Slot, out var teamWeaponsInfo))
|
||
return;
|
||
|
||
// Iterate through each team in the player's weapon info
|
||
foreach (var teamInfo in teamWeaponsInfo)
|
||
{
|
||
// Retrieve weaponInfos for the current team
|
||
var weaponInfos = teamInfo.Value;
|
||
|
||
// Get StatTrak weapons for the current team
|
||
var statTrakWeapons = weaponInfos
|
||
.ToDictionary(
|
||
w => w.Key,
|
||
w => (w.Value.StatTrak, w.Value.StatTrakCount) // Store both StatTrak and StatTrakCount in a tuple
|
||
);
|
||
|
||
// Check if there are StatTrak weapons to sync
|
||
if (statTrakWeapons.Count == 0) continue;
|
||
|
||
// Get the current team ID
|
||
int weaponTeam = (int)teamInfo.Key;
|
||
|
||
// Sync StatTrak values for the current team
|
||
foreach (var (defindex, (statTrak, statTrakCount)) in statTrakWeapons)
|
||
{
|
||
string query = Utility.Config?.DatabaseType?.ToLower() == "sqlite"
|
||
? "UPDATE wp_player_skins SET weapon_stattrak = @StatTrak, weapon_stattrak_count = @StatTrakCount WHERE steamid = @steamid AND weapon_defindex = @weaponDefIndex AND weapon_team = @weaponTeam"
|
||
: @"UPDATE `wp_player_skins`
|
||
SET `weapon_stattrak` = @StatTrak,
|
||
`weapon_stattrak_count` = @StatTrakCount
|
||
WHERE `steamid` = @steamid
|
||
AND `weapon_defindex` = @weaponDefIndex
|
||
AND `weapon_team` = @weaponTeam";
|
||
|
||
var parameters = new
|
||
{
|
||
steamid = player.SteamId,
|
||
weaponDefIndex = defindex,
|
||
StatTrak = statTrak,
|
||
StatTrakCount = statTrakCount,
|
||
weaponTeam
|
||
};
|
||
|
||
await connection.GetConnection().ExecuteAsync(query, parameters, transaction);
|
||
}
|
||
}
|
||
|
||
await connection.CommitTransactionAsync(transaction);
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utility.Log($"Error syncing stattrak to database: {e.Message}");
|
||
}
|
||
}
|
||
} |