mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2026-01-31 16:09:55 +00:00
85 lines
3.3 KiB
C++
85 lines
3.3 KiB
C++
#include "MySQLDatabase.h"
|
|
|
|
#include "eGameMasterLevel.h"
|
|
|
|
std::optional<IAccounts::Info> MySQLDatabase::GetAccountInfo(const std::string_view username) {
|
|
auto result = ExecuteSelect("SELECT id, password, banned, locked, play_key_id, gm_level, mute_expire FROM accounts WHERE name = ? LIMIT 1;", username);
|
|
|
|
if (!result->next()) {
|
|
return std::nullopt;
|
|
}
|
|
|
|
IAccounts::Info toReturn;
|
|
toReturn.id = result->getUInt("id");
|
|
toReturn.maxGmLevel = static_cast<eGameMasterLevel>(result->getInt("gm_level"));
|
|
toReturn.bcryptPassword = result->getString("password").c_str();
|
|
toReturn.banned = result->getBoolean("banned");
|
|
toReturn.locked = result->getBoolean("locked");
|
|
toReturn.playKeyId = result->getUInt("play_key_id");
|
|
toReturn.muteExpire = result->getUInt64("mute_expire");
|
|
|
|
return toReturn;
|
|
}
|
|
|
|
void MySQLDatabase::UpdateAccountUnmuteTime(const uint32_t accountId, const uint64_t timeToUnmute) {
|
|
ExecuteUpdate("UPDATE accounts SET mute_expire = ? WHERE id = ?;", timeToUnmute, accountId);
|
|
}
|
|
|
|
void MySQLDatabase::UpdateAccountBan(const uint32_t accountId, const bool banned) {
|
|
ExecuteUpdate("UPDATE accounts SET banned = ? WHERE id = ?;", banned, accountId);
|
|
}
|
|
|
|
void MySQLDatabase::UpdateAccountPassword(const uint32_t accountId, const std::string_view bcryptpassword) {
|
|
ExecuteUpdate("UPDATE accounts SET password = ? WHERE id = ?;", bcryptpassword, accountId);
|
|
}
|
|
|
|
void MySQLDatabase::InsertNewAccount(const std::string_view username, const std::string_view bcryptpassword) {
|
|
ExecuteInsert("INSERT INTO accounts (name, password, gm_level) VALUES (?, ?, ?);", username, bcryptpassword, static_cast<int32_t>(eGameMasterLevel::OPERATOR));
|
|
}
|
|
|
|
void MySQLDatabase::UpdateAccountGmLevel(const uint32_t accountId, const eGameMasterLevel gmLevel) {
|
|
ExecuteUpdate("UPDATE accounts SET gm_level = ? WHERE id = ?;", static_cast<int32_t>(gmLevel), accountId);
|
|
}
|
|
|
|
uint32_t MySQLDatabase::GetAccountCount() {
|
|
auto res = ExecuteSelect("SELECT COUNT(*) as count FROM accounts;");
|
|
return res->next() ? res->getUInt("count") : 0;
|
|
}
|
|
|
|
void MySQLDatabase::RecordFailedAttempt(const uint32_t accountId) {
|
|
ExecuteUpdate("UPDATE accounts SET failed_attempts = failed_attempts + 1 WHERE id = ?;", accountId);
|
|
}
|
|
|
|
void MySQLDatabase::ClearFailedAttempts(const uint32_t accountId) {
|
|
ExecuteUpdate("UPDATE accounts SET failed_attempts = 0, lockout_time = NULL, last_login = NOW() WHERE id = ?;", accountId);
|
|
}
|
|
|
|
void MySQLDatabase::SetLockout(const uint32_t accountId, const int64_t lockoutUntil) {
|
|
ExecuteUpdate("UPDATE accounts SET lockout_time = FROM_UNIXTIME(?) WHERE id = ?;", lockoutUntil, accountId);
|
|
}
|
|
|
|
bool MySQLDatabase::IsLockedOut(const uint32_t accountId) {
|
|
auto result = ExecuteSelect("SELECT lockout_time FROM accounts WHERE id = ?;", accountId);
|
|
if (!result->next()) {
|
|
return false;
|
|
}
|
|
|
|
// If lockout_time is set and in the future, account is locked
|
|
const char* lockoutTime = result->getString("lockout_time").c_str();
|
|
if (lockoutTime == nullptr || strlen(lockoutTime) == 0) {
|
|
return false;
|
|
}
|
|
|
|
// Simplified check - if lockout_time exists and is not null, it's locked
|
|
return true;
|
|
}
|
|
|
|
uint8_t MySQLDatabase::GetFailedAttempts(const uint32_t accountId) {
|
|
auto result = ExecuteSelect("SELECT failed_attempts FROM accounts WHERE id = ?;", accountId);
|
|
if (!result->next()) {
|
|
return 0;
|
|
}
|
|
|
|
return result->getUInt("failed_attempts");
|
|
}
|