SimpleAdmin/CS2-SimpleAdmin/Database/MysqlDatabaseProvider.cs

using System.Data.Common;
using MySqlConnector;

namespace CS2_SimpleAdmin.Database;

public class MySqlDatabaseProvider(string connectionString) : IDatabaseProvider
{
    public async Task<DbConnection> CreateConnectionAsync()
    {
        var connection = new MySqlConnection(connectionString);
        await connection.OpenAsync();
        
        await using var cmd = connection.CreateCommand();
        
        cmd.CommandText = "SET NAMES 'utf8mb4' COLLATE 'utf8mb4_general_ci';";
        await cmd.ExecuteNonQueryAsync();
        
        // cmd.CommandText = "SET time_zone = '+00:00';";
        await cmd.ExecuteNonQueryAsync();

        return connection;
    }

    public async Task<(bool Success, string? Exception)> CheckConnectionAsync()
    {
        try
        {
            await using var conn = await CreateConnectionAsync();
            return (true, null);
        }
        catch (Exception ex)
        {
            return (false, ex.Message);
        }
    }
    
    public Task DatabaseMigrationAsync()
    {
        var migration = new Migration(CS2_SimpleAdmin.Instance.ModuleDirectory + "/Database/Migrations/Mysql");
        return migration.ExecuteMigrationsAsync();
    }

    public string GetBanSelectQuery(bool multiServer)
    {
        return multiServer ? """
                             SELECT 
                             id AS Id,
                             player_name AS PlayerName,
                             player_steamid AS PlayerSteamId,
                             player_ip AS PlayerIp,
                             status AS Status
                             FROM sa_bans
                             """ : """
                                   SELECT
                                   id AS Id,
                                       player_name AS PlayerName,
                                       player_steamid AS PlayerSteamId,
                                       player_ip AS PlayerIp,
                                       status AS Status 
                                   FROM sa_bans
                                   WHERE server_id = @serverId
                                   """;
    }    
    
    public static string GetBanUpdatedSelectQuery(bool multiServer)
    {
        return multiServer ? """
                             SELECT id AS Id,
                             player_name AS PlayerName,
                             player_steamid AS PlayerSteamId,
                             player_ip AS PlayerIp,
                             status AS Status 
                             FROM `sa_bans` WHERE updated_at > @lastUpdate OR created > @lastUpdate ORDER BY updated_at DESC
                             """ : """
                                   SELECT id AS Id,
                                   player_name AS PlayerName,
                                   player_steamid AS PlayerSteamId,
                                   player_ip AS PlayerIp,
                                   status AS Status 
                                   FROM `sa_bans` WHERE (updated_at > @lastUpdate OR created > @lastUpdate) AND server_id = @serverId ORDER BY updated_at DESC
                                   """;
    }

    public string GetIpHistoryQuery()
    {
        return "SELECT steamid, name, address, used_at FROM sa_players_ips ORDER BY used_at DESC";
    }

    public string GetUpsertPlayerIpQuery()
    {
        return """
            INSERT INTO `sa_players_ips` (steamid, name, address, used_at)
            VALUES (@SteamID, @playerName, @IPAddress, CURRENT_TIMESTAMP)
            ON DUPLICATE KEY UPDATE
                used_at = CURRENT_TIMESTAMP,
                name = @playerName;
            """;
    }

    public string GetBanUpdateQuery(bool multiServer)
    {
        return multiServer ? """
                             UPDATE sa_bans
                             SET 
                                 player_ip = COALESCE(player_ip, @PlayerIP),
                                 player_name = COALESCE(player_name, @PlayerName)
                             WHERE 
                                 (player_steamid = @PlayerSteamID OR player_ip = @PlayerIP)
                                 AND status = 'ACTIVE'
                                 AND (duration = 0 OR ends > @CurrentTime)
                             """ : """
                                   UPDATE sa_bans
                                   SET 
                                       player_ip = COALESCE(player_ip, @PlayerIP),
                                       player_name = COALESCE(player_name, @PlayerName)
                                   WHERE 
                                       (player_steamid = @PlayerSteamID OR player_ip = @PlayerIP)
                                       AND status = 'ACTIVE'
                                       AND (duration = 0 OR ends > @CurrentTime) AND server_id = @ServerId
                                   """;
    }

    public string GetAdminsQuery()
    {
        return """
               SELECT sa_admins.player_steamid, sa_admins.player_name, sa_admins_flags.flag, sa_admins.immunity, sa_admins.ends
               FROM sa_admins_flags
               JOIN sa_admins ON sa_admins_flags.admin_id = sa_admins.id
               WHERE (sa_admins.ends IS NULL OR sa_admins.ends > @CurrentTime)
               AND (sa_admins.server_id IS NULL OR sa_admins.server_id = @serverid)
               ORDER BY sa_admins.player_steamid
               """;
    }
    
    public string GetDeleteAdminQuery(bool globalDelete) =>
        globalDelete
            ? "DELETE FROM sa_admins WHERE player_steamid = @PlayerSteamID"
            : "DELETE FROM sa_admins WHERE player_steamid = @PlayerSteamID AND server_id = @ServerId";
    
    public string GetAddAdminQuery() =>
        "INSERT INTO sa_admins (player_steamid, player_name, immunity, ends, created, server_id) " +
        "VALUES (@playerSteamId, @playerName, @immunity, @ends, @created, @serverid); SELECT LAST_INSERT_ID();";

    public string GetGroupsQuery()
    {
        return """
               SELECT g.group_id, sg.name AS group_name, sg.immunity, f.flag
               FROM sa_groups_flags f
               JOIN sa_groups_servers g ON f.group_id = g.group_id
               JOIN sa_groups sg ON sg.id = g.group_id
               WHERE (g.server_id = @serverid OR server_id IS NULL)
               """;
    }

    public string GetAddAdminFlagsQuery() =>
        "INSERT INTO sa_admins_flags (admin_id, flag) VALUES (@adminId, @flag);";

    public string GetUpdateAdminGroupQuery() =>
        "UPDATE sa_admins SET group_id = @groupId WHERE id = @adminId;";

    public string GetAddGroupQuery() =>
        "INSERT INTO sa_groups (name, immunity) VALUES (@groupName, @immunity); SELECT LAST_INSERT_ID();";
    
    public string GetGroupIdByNameQuery() =>
        """
        SELECT sgs.group_id
                  FROM sa_groups_servers sgs
                  JOIN sa_groups sg ON sgs.group_id = sg.id
                  WHERE sg.name = @groupName
                  ORDER BY (sgs.server_id = @serverId) DESC, sgs.server_id ASC
                  LIMIT 1;
        """;
    public string GetAddGroupFlagsQuery() =>
        "INSERT INTO sa_groups_flags (group_id, flag) VALUES (@groupId, @flag);";

    public string GetAddGroupServerQuery() =>
        "INSERT INTO sa_groups_servers (group_id, server_id) VALUES (@groupId, @server_id);";

    public string GetDeleteGroupQuery() =>
        "DELETE FROM sa_groups WHERE name = @groupName;";

    public string GetDeleteOldAdminsQuery() =>
        "DELETE FROM sa_admins WHERE ends IS NOT NULL AND ends <= @CurrentTime;";
    
    public string GetAddBanQuery()
    {
        return """
                   INSERT INTO `sa_bans` 
                       (`player_steamid`, `player_name`, `player_ip`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `server_id`) 
                   VALUES 
                       (@playerSteamid, @playerName, @playerIp, @adminSteamid, @adminName, @banReason, @duration, @ends, @created, @serverid);
                   SELECT LAST_INSERT_ID();
               """;
    }
    
    public string GetAddBanBySteamIdQuery()
    {
        return """
                   INSERT INTO `sa_bans` 
                       (`player_steamid`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `server_id`) 
                   VALUES 
                       (@playerSteamid, @adminSteamid, @adminName, @banReason, @duration, @ends, @created, @serverid);
                   SELECT LAST_INSERT_ID();
               """;
    }

    public string GetAddBanByIpQuery()
    {
        return """
                   INSERT INTO `sa_bans` 
                       (`player_ip`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `server_id`) 
                   VALUES 
                       (@playerIp, @adminSteamid, @adminName, @banReason, @duration, @ends, @created, @serverid);
               """;
    }
    
    public string GetUnbanRetrieveBansQuery(bool multiServer)
    {
        return multiServer
            ? "SELECT id, player_steamid FROM sa_bans WHERE (player_steamid = @pattern OR player_name = @pattern OR player_ip = @pattern) AND status = 'ACTIVE'"
            : "SELECT id, player_steamid FROM sa_bans WHERE (player_steamid = @pattern OR player_name = @pattern OR player_ip = @pattern) AND status = 'ACTIVE' AND server_id = @serverid";
    }

    public string GetUnbanAdminIdQuery()
    {
        return "SELECT id FROM sa_admins WHERE player_steamid = @adminSteamId";
    }

    public string GetInsertUnbanQuery(bool includeReason)
    {
        return includeReason
            ? "INSERT INTO sa_unbans (ban_id, admin_id, reason) VALUES (@banId, @adminId, @reason); SELECT LAST_INSERT_ID();"
            : "INSERT INTO sa_unbans (ban_id, admin_id) VALUES (@banId, @adminId); SELECT LAST_INSERT_ID();";
    }

    public string GetUpdateBanStatusQuery()
    {
        return "UPDATE sa_bans SET status = 'UNBANNED', unban_id = @unbanId WHERE id = @banId";
    }

    public string GetRenamesQuery() =>
        "SELECT player_steamid, name FROM sa_renames";

    public string GetUpsertRenameQuery() =>
        "INSERT INTO sa_renames (player_steamid, name) VALUES (@steamId, @name) ON DUPLICATE KEY UPDATE name = @name";

    public string GetDeleteRenameQuery() =>
        "DELETE FROM sa_renames WHERE player_steamid = @steamId";

    public string GetExpireBansQuery(bool multiServer)
    {
        return multiServer
            ? "UPDATE sa_bans SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND duration > 0 AND ends <= @currentTime"
            : "UPDATE sa_bans SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND duration > 0 AND ends <= @currentTime AND server_id = @serverid";
    }

    public string GetExpireIpBansQuery(bool multiServer)
    {
        return multiServer
            ? "UPDATE sa_bans SET player_ip = NULL WHERE status = 'ACTIVE' AND ends <= @ipBansTime"
            : "UPDATE sa_bans SET player_ip = NULL WHERE status = 'ACTIVE' AND ends <= @ipBansTime AND server_id = @serverid";
    }

    public string GetExpireOldPlayerIpsQuery()
    {
        return "DELETE FROM sa_players_ips WHERE used_at <= @ipBansTime";
    }
    
    public string GetAddMuteQuery(bool includePlayerName) =>
        includePlayerName
            ? """
              INSERT INTO `sa_mutes`
              (`player_steamid`, `player_name`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `type`, `server_id`) 
              VALUES (@playerSteamid, @playerName, @adminSteamid, @adminName, @muteReason, @duration, @ends, @created, @type, @serverid);
              SELECT LAST_INSERT_ID();
              """
            : """
              INSERT INTO `sa_mutes`
              (`player_steamid`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `type`, `server_id`) 
              VALUES (@playerSteamid, @adminSteamid, @adminName, @muteReason, @duration, @ends, @created, @type, @serverid);
              SELECT LAST_INSERT_ID();
              """;

    public string GetIsMutedQuery(bool multiServer, int timeMode) =>
        multiServer
            ? (timeMode == 1
                ? "SELECT * FROM sa_mutes WHERE player_steamid = @PlayerSteamID AND status = 'ACTIVE' AND (duration = 0 OR ends > @CurrentTime)"
                : "SELECT * FROM sa_mutes WHERE player_steamid = @PlayerSteamID AND status = 'ACTIVE' AND (duration = 0 OR duration > COALESCE(passed, 0))")
            : (timeMode == 1
                ? "SELECT * FROM sa_mutes WHERE player_steamid = @PlayerSteamID AND status = 'ACTIVE' AND (duration = 0 OR ends > @CurrentTime) AND server_id = @serverid"
                : "SELECT * FROM sa_mutes WHERE player_steamid = @PlayerSteamID AND status = 'ACTIVE' AND (duration = 0 OR duration > COALESCE(passed, 0)) AND server_id = @serverid");

    public string GetMuteStatsQuery(bool multiServer) =>
        multiServer
            ? """
              SELECT
                  COUNT(CASE WHEN type = 'MUTE' THEN 1 END) AS TotalMutes,
                  COUNT(CASE WHEN type = 'GAG' THEN 1 END) AS TotalGags,
                  COUNT(CASE WHEN type = 'SILENCE' THEN 1 END) AS TotalSilences
              FROM sa_mutes
              WHERE player_steamid = @PlayerSteamID;
              """
            : """
              SELECT
                  COUNT(CASE WHEN type = 'MUTE' THEN 1 END) AS TotalMutes,
                  COUNT(CASE WHEN type = 'GAG' THEN 1 END) AS TotalGags,
                  COUNT(CASE WHEN type = 'SILENCE' THEN 1 END) AS TotalSilences
              FROM sa_mutes
              WHERE player_steamid = @PlayerSteamID AND server_id = @ServerId;
              """;

    public string GetUpdateMutePassedQuery(bool multiServer) =>
        multiServer
            ? "UPDATE `sa_mutes` SET passed = COALESCE(passed, 0) + 1 WHERE (player_steamid = @PlayerSteamID) AND duration > 0 AND status = 'ACTIVE'"
            : "UPDATE `sa_mutes` SET passed = COALESCE(passed, 0) + 1 WHERE (player_steamid = @PlayerSteamID) AND duration > 0 AND status = 'ACTIVE' AND server_id = @serverid";

    public string GetCheckExpiredMutesQuery(bool multiServer) =>
        multiServer
            ? "SELECT * FROM `sa_mutes` WHERE player_steamid = @PlayerSteamID AND passed >= duration AND duration > 0 AND status = 'ACTIVE'"
            : "SELECT * FROM `sa_mutes` WHERE player_steamid = @PlayerSteamID AND passed >= duration AND duration > 0 AND status = 'ACTIVE' AND server_id = @serverid";

    public string GetRetrieveMutesQuery(bool multiServer) =>
        multiServer
            ? "SELECT id FROM sa_mutes WHERE (player_steamid = @pattern OR player_name = @pattern) AND type = @muteType AND status = 'ACTIVE'"
            : "SELECT id FROM sa_mutes WHERE (player_steamid = @pattern OR player_name = @pattern) AND type = @muteType AND status = 'ACTIVE' AND server_id = @serverid";

    public string GetUnmuteAdminIdQuery() =>
        "SELECT id FROM sa_admins WHERE player_steamid = @adminSteamId";

    public string GetInsertUnmuteQuery(bool includeReason) =>
        includeReason
            ? "INSERT INTO sa_unmutes (mute_id, admin_id, reason) VALUES (@muteId, @adminId, @reason); SELECT LAST_INSERT_ID();"
            : "INSERT INTO sa_unmutes (mute_id, admin_id) VALUES (@muteId, @adminId); SELECT LAST_INSERT_ID();";

    public string GetUpdateMuteStatusQuery() =>
        "UPDATE sa_mutes SET status = 'UNMUTED', unmute_id = @unmuteId WHERE id = @muteId";

    public string GetExpireMutesQuery(bool multiServer, int timeMode) =>
        multiServer
            ? (timeMode == 1
                ? "UPDATE sa_mutes SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND `duration` > 0 AND ends <= @CurrentTime"
                : "UPDATE sa_mutes SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND `duration` > 0 AND `passed` >= `duration`")
            : (timeMode == 1
                ? "UPDATE sa_mutes SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND `duration` > 0 AND ends <= @CurrentTime AND server_id = @serverid"
                : "UPDATE sa_mutes SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND `duration` > 0 AND `passed` >= `duration` AND server_id = @serverid");
    
    
    public string GetAddWarnQuery(bool includePlayerName) =>
        includePlayerName
            ? """
              INSERT INTO `sa_warns` 
              (`player_steamid`, `player_name`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `server_id`) 
              VALUES 
              (@playerSteamid, @playerName, @adminSteamid, @adminName, @warnReason, @duration, @ends, @created, @serverid);
              SELECT LAST_INSERT_ID();
              """
            : """
              INSERT INTO `sa_warns` 
              (`player_steamid`, `admin_steamid`, `admin_name`, `reason`, `duration`, `ends`, `created`, `server_id`) 
              VALUES 
              (@playerSteamid, @adminSteamid, @adminName, @warnReason, @duration, @ends, @created, @serverid);
              SELECT LAST_INSERT_ID();
              """;

    public string GetPlayerWarnsQuery(bool multiServer, bool active) =>
        multiServer
            ? active
                ? "SELECT * FROM sa_warns WHERE player_steamid = @PlayerSteamID AND status = 'ACTIVE' ORDER BY id DESC"
                : "SELECT * FROM sa_warns WHERE player_steamid = @PlayerSteamID ORDER BY id DESC"
            : active
                ? "SELECT * FROM sa_warns WHERE player_steamid = @PlayerSteamID AND server_id = @serverid AND status = 'ACTIVE' ORDER BY id DESC"
                : "SELECT * FROM sa_warns WHERE player_steamid = @PlayerSteamID AND server_id = @serverid ORDER BY id DESC";

    public string GetPlayerWarnsCountQuery(bool multiServer, bool active) =>
        multiServer
            ? active
                ? "SELECT COUNT(*) FROM sa_warns WHERE player_steamid = @PlayerSteamID AND status = 'ACTIVE'"
                : "SELECT COUNT(*) FROM sa_warns WHERE player_steamid = @PlayerSteamID"
            : active
                ? "SELECT COUNT(*) FROM sa_warns WHERE player_steamid = @PlayerSteamID AND server_id = @serverid AND status = 'ACTIVE'"
                : "SELECT COUNT(*) FROM sa_warns WHERE player_steamid = @PlayerSteamID AND server_id = @serverid";

    public string GetUnwarnByIdQuery(bool multiServer) =>
        multiServer
            ? "UPDATE sa_warns SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND player_steamid = @steamid AND id = @warnId"
            : "UPDATE sa_warns SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND player_steamid = @steamid AND id = @warnId AND server_id = @serverid";

    public string GetUnwarnLastQuery(bool multiServer) =>
        multiServer
            ? """
              UPDATE sa_warns 
              JOIN (
                  SELECT MAX(id) AS max_id
                  FROM sa_warns
                  WHERE player_steamid = @steamid AND status = 'ACTIVE'
              ) AS subquery ON sa_warns.id = subquery.max_id
              SET sa_warns.status = 'EXPIRED'
              WHERE sa_warns.status = 'ACTIVE' AND sa_warns.player_steamid = @steamid;
              """
            : """
              UPDATE sa_warns 
              JOIN (
                  SELECT MAX(id) AS max_id
                  FROM sa_warns
                  WHERE player_steamid = @steamid AND status = 'ACTIVE' AND server_id = @serverid
              ) AS subquery ON sa_warns.id = subquery.max_id
              SET sa_warns.status = 'EXPIRED'
              WHERE sa_warns.status = 'ACTIVE' AND sa_warns.player_steamid = @steamid AND sa_warns.server_id = @serverid;
              """;

    public string GetExpireWarnsQuery(bool multiServer) =>
        multiServer
            ? "UPDATE sa_warns SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND `duration` > 0 AND ends <= @CurrentTime"
            : "UPDATE sa_warns SET status = 'EXPIRED' WHERE status = 'ACTIVE' AND `duration` > 0 AND ends <= @CurrentTime AND server_id = @serverid";

    public string GetPenaltyHistoryQuery(bool multiServer) =>
        $"""
        SELECT b.id, 'BAN' AS type, b.player_name, b.admin_name, b.reason, b.duration, b.created, b.ends, b.status,
               ub.reason AS lift_reason, ub.date AS lift_date, ua.player_name AS lift_admin
        FROM sa_bans b
        LEFT JOIN sa_unbans ub ON ub.id = b.unban_id
        LEFT JOIN sa_admins ua ON ua.id = ub.admin_id
        WHERE b.player_steamid = @PlayerSteamID {(multiServer ? "" : "AND b.server_id = @serverid")}
        UNION ALL
        SELECT m.id, m.type, m.player_name, m.admin_name, m.reason, m.duration, m.created, m.ends, m.status,
               um.reason AS lift_reason, um.date AS lift_date, ua.player_name AS lift_admin
        FROM sa_mutes m
        LEFT JOIN sa_unmutes um ON um.id = m.unmute_id
        LEFT JOIN sa_admins ua ON ua.id = um.admin_id
        WHERE m.player_steamid = @PlayerSteamID {(multiServer ? "" : "AND m.server_id = @serverid")}
        UNION ALL
        SELECT w.id, 'WARN' AS type, w.player_name, w.admin_name, w.reason, w.duration, w.created, w.ends, w.status,
               NULL AS lift_reason, NULL AS lift_date, NULL AS lift_admin
        FROM sa_warns w
        WHERE w.player_steamid = @PlayerSteamID {(multiServer ? "" : "AND w.server_id = @serverid")}
        ORDER BY created DESC
        """;

    public ValueTask DisposeAsync() => ValueTask.CompletedTask;
}