CS2-SimpleAdmin/Managers/MuteManager.cs

using Dapper;
using Microsoft.Extensions.Logging;
using MySqlConnector;

namespace CS2_SimpleAdmin;

internal class MuteManager
{
	private readonly Database _database;

	public MuteManager(Database database)
	{
		_database = database;
	}

	public async Task MutePlayer(PlayerInfo player, PlayerInfo issuer, string reason, int time = 0, int type = 0)
	{
		if (player == null || player.SteamId == null) return;

		await using MySqlConnection connection = await _database.GetConnectionAsync();

		DateTime now = DateTime.UtcNow.ToLocalTime();
		DateTime futureTime = now.AddMinutes(time).ToLocalTime();

		string muteType = "GAG";
		if (type == 1)
			muteType = "MUTE";
		else if (type == 2)
			muteType = "SILENCE";

		var sql = "INSERT INTO `sa_mutes` (`player_steamid`, `player_name`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `type`, `server_id`) " +
			"VALUES (@playerSteamid, @playerName, @adminSteamid, @adminName, @banReason, @duration, @ends, @created, @type, @serverid)";

		await connection.ExecuteAsync(sql, new
		{
			playerSteamid = player.SteamId,
			playerName = player.Name,
			adminSteamid = issuer.SteamId == null ? "Console" : issuer.SteamId,
			adminName = issuer.SteamId == null ? "Console" : issuer.Name,
			banReason = reason,
			duration = time,
			ends = futureTime,
			created = now,
			type = muteType,
			serverid = CS2_SimpleAdmin.ServerId
		});
	}

	public async Task AddMuteBySteamid(string playerSteamId, PlayerInfo issuer, string reason, int time = 0, int type = 0)
	{
		if (string.IsNullOrEmpty(playerSteamId)) return;

		await using MySqlConnection connection = await _database.GetConnectionAsync();

		DateTime now = DateTime.UtcNow.ToLocalTime();
		DateTime futureTime = now.AddMinutes(time).ToLocalTime();

		string muteType = "GAG";
		if (type == 1)
			muteType = "MUTE";
		else if (type == 2)
			muteType = "SILENCE";

		var sql = "INSERT INTO `sa_mutes` (`player_steamid`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `type`, `server_id`) " +
			"VALUES (@playerSteamid, @adminSteamid, @adminName, @banReason, @duration, @ends, @created, @type, @serverid)";

		await connection.ExecuteAsync(sql, new
		{
			playerSteamid = playerSteamId,
			adminSteamid = issuer.SteamId == null ? "Console" : issuer.SteamId,
			adminName = issuer.Name == null ? "Console" : issuer.Name,
			banReason = reason,
			duration = time,
			ends = futureTime,
			created = now,
			type = muteType,
			serverid = CS2_SimpleAdmin.ServerId
		});
	}

	public async Task<List<dynamic>> IsPlayerMuted(string steamId)
	{
		if (string.IsNullOrEmpty(steamId))
		{
			return new List<dynamic>();
		}

#if DEBUG
		if (CS2_SimpleAdmin._logger!= null)
			CS2_SimpleAdmin._logger.LogCritical($"IsPlayerMuted for {steamId}");
#endif

		try
		{
			await using MySqlConnection connection = await _database.GetConnectionAsync();
			DateTime currentTime = DateTime.Now.ToLocalTime();
			string sql = "SELECT * FROM sa_mutes WHERE player_steamid = @PlayerSteamID AND status = 'ACTIVE' AND (duration = 0 OR ends > @CurrentTime)";

			var parameters = new { PlayerSteamID = steamId, CurrentTime = currentTime };
			var activeMutes = (await connection.QueryAsync(sql, parameters)).ToList();
			return activeMutes;
		}
		catch (Exception)
		{
			return new List<dynamic>();
		}
	}

	public async Task<int> GetPlayerMutes(string steamId)
	{
		await using MySqlConnection connection = await _database.GetConnectionAsync();

		int muteCount;
		string sql = "SELECT COUNT(*) FROM sa_mutes WHERE player_steamid = @PlayerSteamID";

		muteCount = await connection.ExecuteScalarAsync<int>(sql, new { PlayerSteamID = steamId });

		return muteCount;
	}

	public async Task UnmutePlayer(string playerPattern, int type = 0)
	{
		if (playerPattern == null || playerPattern.Length <= 1)
		{
			return;
		}

		await using MySqlConnection connection = await _database.GetConnectionAsync();

		if (type == 2)
		{
			string _unbanSql = "UPDATE sa_mutes SET status = 'UNMUTED' WHERE (player_steamid = @pattern OR player_name = @pattern) AND status = 'ACTIVE'";
			await connection.ExecuteAsync(_unbanSql, new { pattern = playerPattern });

			return;
		}

		string muteType = "GAG";
		if (type == 1)
		{
			muteType = "MUTE";
		}
		else if (type == 2)
			muteType = "SILENCE";

		string sqlUnban = "UPDATE sa_mutes SET status = 'UNMUTED' WHERE (player_steamid = @pattern OR player_name = @pattern) AND type = @muteType AND status = 'ACTIVE'";
		await connection.ExecuteAsync(sqlUnban, new { pattern = playerPattern, muteType });
	}

	public async Task ExpireOldMutes()
	{
		try
		{
			await using MySqlConnection connection = await _database.GetConnectionAsync();

			string sql = "UPDATE sa_mutes SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND `duration` > 0 AND ends <= @CurrentTime";
			await connection.ExecuteAsync(sql, new { CurrentTime = DateTime.Now.ToLocalTime() });
		}
		catch (Exception)
		{
			if (CS2_SimpleAdmin._logger != null)
				CS2_SimpleAdmin._logger.LogCritical("Unable to remove expired mutes");
		}
	}
}