mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-12-23 05:53:34 +00:00
a60865cd19
* simplify leaderboard code, fully abstract database * update exception catching * update exception catching and sql references, remove ugc from gamemessages fix deleting model remove unrelated changes Update GameMessages.cpp * remove ugc from gamemessages * Update GameMessages.cpp * Update Leaderboard.cpp * bug fixes * fix racing leaderboard * remove extra stuff * update * add sqlite * use a default for optimizations * update sqlite * Fix limits on update and delete * fix bugs * use definition to switch between databases * add switch for different backends * fix include guard and includes * always build both * add mysql if block * Update Database.cpp * add new options and add check to prevent overriding mysql * correct config names * Update README.md * Update README.md * merge to 1 sql file for sqlite database * move to sqlite folder * add back mysql migrations * Update README.md * add migration to correct the folder name or mysql * yes aron * updates * Update CMakeLists.txt * dont use paths at all, add where check to only update if folder name still exist check also doesnt check for slashes and assumes one will be there since it will be. * default dont auto create account for releases we can change this flag * default 0 * add times played query * fix leaderboard not incrementing on a not better score * add env vars with defaults for docker * use an "enum" * default to mariadb * Update .env.example
271 lines
12 KiB
C++
271 lines
12 KiB
C++
#ifndef SQLITEDATABASE_H
|
|
#define SQLITEDATABASE_H
|
|
|
|
#include "CppSQLite3.h"
|
|
|
|
#include "GameDatabase.h"
|
|
|
|
using PreppedStmtRef = CppSQLite3Statement&;
|
|
|
|
// Purposefully no definition for this to provide linker errors in the case someone tries to
|
|
// bind a parameter to a type that isn't defined.
|
|
template<typename ParamType>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const ParamType param);
|
|
|
|
// This is a function to set each parameter in a prepared statement.
|
|
// This is accomplished with a combination of parameter packing and Fold Expressions.
|
|
// The constexpr if statement is used to prevent the compiler from trying to call SetParam with 0 arguments.
|
|
template<typename... Args>
|
|
void SetParams(PreppedStmtRef stmt, Args&&... args) {
|
|
if constexpr (sizeof...(args) != 0) {
|
|
int i = 1;
|
|
(SetParam(stmt, i++, args), ...);
|
|
}
|
|
}
|
|
|
|
class SQLiteDatabase : public GameDatabase {
|
|
public:
|
|
void Connect() override;
|
|
void Destroy(std::string source = "") override;
|
|
|
|
void Commit() override;
|
|
bool GetAutoCommit() override;
|
|
void SetAutoCommit(bool value) override;
|
|
void ExecuteCustomQuery(const std::string_view query) override;
|
|
|
|
// Overloaded queries
|
|
std::optional<IServers::MasterInfo> GetMasterInfo() override;
|
|
|
|
std::vector<std::string> GetApprovedCharacterNames() override;
|
|
|
|
std::vector<FriendData> GetFriendsList(uint32_t charID) override;
|
|
|
|
std::optional<IFriends::BestFriendStatus> GetBestFriendStatus(const uint32_t playerCharacterId, const uint32_t friendCharacterId) override;
|
|
void SetBestFriendStatus(const uint32_t playerAccountId, const uint32_t friendAccountId, const uint32_t bestFriendStatus) override;
|
|
void AddFriend(const uint32_t playerAccountId, const uint32_t friendAccountId) override;
|
|
void RemoveFriend(const uint32_t playerAccountId, const uint32_t friendAccountId) override;
|
|
void UpdateActivityLog(const uint32_t characterId, const eActivityType activityType, const LWOMAPID mapId) override;
|
|
void DeleteUgcModelData(const LWOOBJID& modelId) override;
|
|
void UpdateUgcModelData(const LWOOBJID& modelId, std::istringstream& lxfml) override;
|
|
std::vector<IUgc::Model> GetAllUgcModels() override;
|
|
void CreateMigrationHistoryTable() override;
|
|
bool IsMigrationRun(const std::string_view str) override;
|
|
void InsertMigration(const std::string_view str) override;
|
|
std::optional<ICharInfo::Info> GetCharacterInfo(const uint32_t charId) override;
|
|
std::optional<ICharInfo::Info> GetCharacterInfo(const std::string_view charId) override;
|
|
std::string GetCharacterXml(const uint32_t accountId) override;
|
|
void UpdateCharacterXml(const uint32_t characterId, const std::string_view lxfml) override;
|
|
std::optional<IAccounts::Info> GetAccountInfo(const std::string_view username) override;
|
|
void InsertNewCharacter(const ICharInfo::Info info) override;
|
|
void InsertCharacterXml(const uint32_t accountId, const std::string_view lxfml) override;
|
|
std::vector<uint32_t> GetAccountCharacterIds(uint32_t accountId) override;
|
|
void DeleteCharacter(const uint32_t characterId) override;
|
|
void SetCharacterName(const uint32_t characterId, const std::string_view name) override;
|
|
void SetPendingCharacterName(const uint32_t characterId, const std::string_view name) override;
|
|
void UpdateLastLoggedInCharacter(const uint32_t characterId) override;
|
|
void SetPetNameModerationStatus(const LWOOBJID& petId, const IPetNames::Info& info) override;
|
|
std::optional<IPetNames::Info> GetPetNameInfo(const LWOOBJID& petId) override;
|
|
std::optional<IProperty::Info> GetPropertyInfo(const LWOMAPID mapId, const LWOCLONEID cloneId) override;
|
|
void UpdatePropertyModerationInfo(const IProperty::Info& info) override;
|
|
void UpdatePropertyDetails(const IProperty::Info& info) override;
|
|
void InsertNewProperty(const IProperty::Info& info, const uint32_t templateId, const LWOZONEID& zoneId) override;
|
|
std::vector<IPropertyContents::Model> GetPropertyModels(const LWOOBJID& propertyId) override;
|
|
void RemoveUnreferencedUgcModels() override;
|
|
void InsertNewPropertyModel(const LWOOBJID& propertyId, const IPropertyContents::Model& model, const std::string_view name) override;
|
|
void UpdateModel(const LWOOBJID& propertyId, const NiPoint3& position, const NiQuaternion& rotation, const std::array<std::pair<int32_t, std::string>, 5>& behaviors) override;
|
|
void RemoveModel(const LWOOBJID& modelId) override;
|
|
void UpdatePerformanceCost(const LWOZONEID& zoneId, const float performanceCost) override;
|
|
void InsertNewBugReport(const IBugReports::Info& info) override;
|
|
void InsertCheatDetection(const IPlayerCheatDetections::Info& info) override;
|
|
void InsertNewMail(const IMail::MailInfo& mail) override;
|
|
void InsertNewUgcModel(
|
|
std::istringstream& sd0Data,
|
|
const uint32_t blueprintId,
|
|
const uint32_t accountId,
|
|
const uint32_t characterId) override;
|
|
std::vector<IMail::MailInfo> GetMailForPlayer(const uint32_t characterId, const uint32_t numberOfMail) override;
|
|
std::optional<IMail::MailInfo> GetMail(const uint64_t mailId) override;
|
|
uint32_t GetUnreadMailCount(const uint32_t characterId) override;
|
|
void MarkMailRead(const uint64_t mailId) override;
|
|
void DeleteMail(const uint64_t mailId) override;
|
|
void ClaimMailItem(const uint64_t mailId) override;
|
|
void InsertSlashCommandUsage(const uint32_t characterId, const std::string_view command) override;
|
|
void UpdateAccountUnmuteTime(const uint32_t accountId, const uint64_t timeToUnmute) override;
|
|
void UpdateAccountBan(const uint32_t accountId, const bool banned) override;
|
|
void UpdateAccountPassword(const uint32_t accountId, const std::string_view bcryptpassword) override;
|
|
void InsertNewAccount(const std::string_view username, const std::string_view bcryptpassword) override;
|
|
void SetMasterIp(const std::string_view ip, const uint32_t port) override;
|
|
std::optional<uint32_t> GetCurrentPersistentId() override;
|
|
void InsertDefaultPersistentId() override;
|
|
void UpdatePersistentId(const uint32_t id) override;
|
|
std::optional<uint32_t> GetDonationTotal(const uint32_t activityId) override;
|
|
std::optional<bool> IsPlaykeyActive(const int32_t playkeyId) override;
|
|
std::vector<IUgc::Model> GetUgcModels(const LWOOBJID& propertyId) override;
|
|
void AddIgnore(const uint32_t playerId, const uint32_t ignoredPlayerId) override;
|
|
void RemoveIgnore(const uint32_t playerId, const uint32_t ignoredPlayerId) override;
|
|
std::vector<IIgnoreList::Info> GetIgnoreList(const uint32_t playerId) override;
|
|
void InsertRewardCode(const uint32_t account_id, const uint32_t reward_code) override;
|
|
std::vector<uint32_t> GetRewardCodesByAccountID(const uint32_t account_id) override;
|
|
void AddBehavior(const IBehaviors::Info& info) override;
|
|
std::string GetBehavior(const int32_t behaviorId) override;
|
|
void RemoveBehavior(const int32_t characterId) override;
|
|
void UpdateAccountGmLevel(const uint32_t accountId, const eGameMasterLevel gmLevel) override;
|
|
std::optional<IProperty::PropertyEntranceResult> GetProperties(const IProperty::PropertyLookup& params) override;
|
|
std::vector<ILeaderboard::Entry> GetDescendingLeaderboard(const uint32_t activityId) override;
|
|
std::vector<ILeaderboard::Entry> GetAscendingLeaderboard(const uint32_t activityId) override;
|
|
std::vector<ILeaderboard::Entry> GetNsLeaderboard(const uint32_t activityId) override;
|
|
std::vector<ILeaderboard::Entry> GetAgsLeaderboard(const uint32_t activityId) override;
|
|
void SaveScore(const uint32_t playerId, const uint32_t gameId, const Score& score) override;
|
|
void UpdateScore(const uint32_t playerId, const uint32_t gameId, const Score& score) override;
|
|
std::optional<ILeaderboard::Score> GetPlayerScore(const uint32_t playerId, const uint32_t gameId) override;
|
|
void IncrementNumWins(const uint32_t playerId, const uint32_t gameId) override;
|
|
void IncrementTimesPlayed(const uint32_t playerId, const uint32_t gameId) override;
|
|
void InsertUgcBuild(const std::string& modules, const LWOOBJID bigId, const std::optional<uint32_t> characterId) override;
|
|
void DeleteUgcBuild(const LWOOBJID bigId) override;
|
|
uint32_t GetAccountCount() override;
|
|
private:
|
|
CppSQLite3Statement CreatePreppedStmt(const std::string& query);
|
|
|
|
// Generic query functions that can be used for any query.
|
|
// Return type may be different depending on the query, so it is up to the caller to check the return type.
|
|
// The first argument is the query string, and the rest are the parameters to bind to the query.
|
|
// The return type is a unique_ptr to the result set, which is deleted automatically when it goes out of scope
|
|
template<typename... Args>
|
|
inline std::pair<CppSQLite3Statement, CppSQLite3Query> ExecuteSelect(const std::string& query, Args&&... args) {
|
|
std::pair<CppSQLite3Statement, CppSQLite3Query> toReturn;
|
|
toReturn.first = CreatePreppedStmt(query);
|
|
SetParams(toReturn.first, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(toReturn.second = toReturn.first.execQuery());
|
|
return toReturn;
|
|
}
|
|
|
|
template<typename... Args>
|
|
inline void ExecuteDelete(const std::string& query, Args&&... args) {
|
|
auto preppedStmt = CreatePreppedStmt(query);
|
|
SetParams(preppedStmt, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(preppedStmt.execDML());
|
|
}
|
|
|
|
template<typename... Args>
|
|
inline int32_t ExecuteUpdate(const std::string& query, Args&&... args) {
|
|
auto preppedStmt = CreatePreppedStmt(query);
|
|
SetParams(preppedStmt, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(return preppedStmt.execDML());
|
|
}
|
|
|
|
template<typename... Args>
|
|
inline int ExecuteInsert(const std::string& query, Args&&... args) {
|
|
auto preppedStmt = CreatePreppedStmt(query);
|
|
SetParams(preppedStmt, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(return preppedStmt.execDML());
|
|
}
|
|
};
|
|
|
|
// Below are each of the definitions of SetParam for each supported type.
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::string_view param) {
|
|
LOG("%s", param.data());
|
|
stmt.bind(index, param.data());
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const char* param) {
|
|
LOG("%s", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::string param) {
|
|
LOG("%s", param.c_str());
|
|
stmt.bind(index, param.c_str());
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int8_t param) {
|
|
LOG("%u", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint8_t param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int16_t param) {
|
|
LOG("%u", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint16_t param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint32_t param) {
|
|
LOG("%u", param);
|
|
stmt.bind(index, static_cast<int32_t>(param));
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int32_t param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int64_t param) {
|
|
LOG("%llu", param);
|
|
stmt.bind(index, static_cast<sqlite_int64>(param));
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint64_t param) {
|
|
LOG("%llu", param);
|
|
stmt.bind(index, static_cast<sqlite_int64>(param));
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const float param) {
|
|
LOG("%f", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const double param) {
|
|
LOG("%f", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const bool param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::istream* param) {
|
|
LOG("Blob");
|
|
// This is the one time you will ever see me use const_cast.
|
|
std::stringstream stream;
|
|
stream << param->rdbuf();
|
|
stmt.bind(index, reinterpret_cast<const unsigned char*>(stream.str().c_str()), stream.str().size());
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::optional<uint32_t> param) {
|
|
if (param) {
|
|
LOG("%d", param.value());
|
|
stmt.bind(index, static_cast<int>(param.value()));
|
|
} else {
|
|
LOG("Null");
|
|
stmt.bindNull(index);
|
|
}
|
|
}
|
|
|
|
#endif //!SQLITEDATABASE_H
|