RankMgr: Implemented the basic API functions.
This commit is contained in:
parent
670e94bfeb
commit
65b81b4ab7
@ -5,15 +5,13 @@
|
|||||||
|
|
||||||
#include "Globals.h"
|
#include "Globals.h"
|
||||||
#include "RankManager.h"
|
#include "RankManager.h"
|
||||||
#include "Protocol/MojangAPI.h"
|
#include "SQLiteCpp/Transaction.h"
|
||||||
#include "inifile/iniFile.h"
|
|
||||||
#include <iostream>
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
//*
|
/*
|
||||||
// This code is for internal testing while developing the cRankManager class
|
// This code is for internal testing while developing the cRankManager class
|
||||||
static class cRankMgrTest
|
static class cRankMgrTest
|
||||||
{
|
{
|
||||||
@ -21,44 +19,72 @@ public:
|
|||||||
cRankMgrTest(void) :
|
cRankMgrTest(void) :
|
||||||
m_Mgr()
|
m_Mgr()
|
||||||
{
|
{
|
||||||
// Initialize the cMojangAPI so that it can convert playernames to UUIDs:
|
// Initialize logging:
|
||||||
cIniFile Ini;
|
new cMCLogger();
|
||||||
Ini.ReadFile("settings.ini");
|
AString UUID = "b1caf24202a841a78055a079c460eee7"; // UUID for "xoft"
|
||||||
m_API.Start(Ini);
|
LOG("Testing UUID %s", UUID.c_str());
|
||||||
|
|
||||||
// Test the cRankManager class:
|
// Test the initial state of the ranks:
|
||||||
ReportPlayer("xoft");
|
LOG("Initial test:");
|
||||||
|
ReportPlayer(UUID);
|
||||||
|
|
||||||
|
// Add a rank, a few groups and permissions and set the player to use them:
|
||||||
|
LOG("Adding data...");
|
||||||
|
m_Mgr.AddRank("TestRank", "[test]", "[/test]", "7");
|
||||||
|
m_Mgr.AddGroup("TestGroup1");
|
||||||
|
m_Mgr.AddGroup("TestGroup2");
|
||||||
|
m_Mgr.AddGroupToRank("TestGroup1", "TestRank");
|
||||||
|
m_Mgr.AddGroupToRank("TestGroup2", "TestRank");
|
||||||
|
m_Mgr.AddPermissionToGroup("testpermission1.1", "TestGroup1");
|
||||||
|
m_Mgr.AddPermissionToGroup("testpermission1.2", "TestGroup1");
|
||||||
|
m_Mgr.AddPermissionToGroup("testpermission2.1", "TestGroup2");
|
||||||
|
m_Mgr.SetPlayerRank(UUID, "TestRank");
|
||||||
|
|
||||||
|
// Test the added data:
|
||||||
|
LOG("Testing the added data:");
|
||||||
|
LOG("IsGroupInRank(TestGroup1, TestRank) = %s", m_Mgr.IsGroupInRank("TestGroup1", "TestRank") ? "true" : "false");
|
||||||
|
LOG("IsGroupInRank(TestGroup3, TestRank) = %s", m_Mgr.IsGroupInRank("TestGroup3", "TestRank") ? "true" : "false");
|
||||||
|
LOG("IsPermissionInGroup(testpermission1.2, TestGroup1) = %s", m_Mgr.IsPermissionInGroup("testpermission1.2", "TestGroup1") ? "true" : "false");
|
||||||
|
LOG("IsPermissionInGroup(testpermission1.2, TestGroup2) = %s", m_Mgr.IsPermissionInGroup("testpermission1.2", "TestGroup2") ? "true" : "false");
|
||||||
|
LOG("IsPermissionInGroup(testpermission1.3, TestGroup2) = %s", m_Mgr.IsPermissionInGroup("testpermission1.3", "TestGroup2") ? "true" : "false");
|
||||||
|
LOG("IsPlayerRankSet(%s) = %s", UUID.c_str(), m_Mgr.IsPlayerRankSet(UUID) ? "true" : "false");
|
||||||
|
LOG("IsPlayerRankSet(%s1) = %s", UUID.c_str(), m_Mgr.IsPlayerRankSet(UUID + "1") ? "true" : "false");
|
||||||
|
LOG("GroupExists(TestGroup1) = %s", m_Mgr.GroupExists("TestGroup1") ? "true" : "false");
|
||||||
|
LOG("GroupExists(TestGroup3) = %s", m_Mgr.GroupExists("TestGroup3") ? "true" : "false");
|
||||||
|
LOG("RankExists(TestRank) = %s", m_Mgr.RankExists("TestRank") ? "true" : "false");
|
||||||
|
LOG("RankExists(NonexistentRank) = %s", m_Mgr.RankExists("NonexistentRank") ? "true" : "false");
|
||||||
|
|
||||||
|
// Test the assignments above:
|
||||||
|
LOG("After-assignment test:");
|
||||||
|
ReportPlayer(UUID);
|
||||||
|
|
||||||
|
LOG("Done.");
|
||||||
}
|
}
|
||||||
|
|
||||||
void ReportPlayer(const AString & a_PlayerName)
|
void ReportPlayer(const AString & a_PlayerUUID)
|
||||||
{
|
{
|
||||||
// Get the player's UUID and rank:
|
// Get the player's UUID and rank:
|
||||||
AString UUID = m_API.GetUUIDFromPlayerName(a_PlayerName);
|
LOG(" Rank: '%s'", m_Mgr.GetPlayerRankName(a_PlayerUUID).c_str());
|
||||||
std::cout << "Player " << a_PlayerName << " has UUID '" << UUID <<"'." << std::endl;
|
|
||||||
std::cout << " Rank: '" << m_Mgr.GetPlayerRankName(UUID) << "'." << std::endl;
|
|
||||||
|
|
||||||
// List all the permission groups for the player:
|
// List all the permission groups for the player:
|
||||||
AStringVector Groups = m_Mgr.GetPlayerGroups(UUID);
|
AStringVector Groups = m_Mgr.GetPlayerGroups(a_PlayerUUID);
|
||||||
std::cout << " Groups(" << Groups.size() << "):" << std::endl;
|
LOG(" Groups(%u):", (unsigned)Groups.size());
|
||||||
for (AStringVector::const_iterator itr = Groups.begin(), end = Groups.end(); itr != end; ++itr)
|
for (AStringVector::const_iterator itr = Groups.begin(), end = Groups.end(); itr != end; ++itr)
|
||||||
{
|
{
|
||||||
std::cout << " '" << *itr << "'." << std::endl;
|
LOG(" '%s'" , itr->c_str());
|
||||||
} // for itr - Groups[]
|
} // for itr - Groups[]
|
||||||
|
|
||||||
// List all the permissions for the player:
|
// List all the permissions for the player:
|
||||||
AStringVector Permissions = m_Mgr.GetPlayerPermissions(UUID);
|
AStringVector Permissions = m_Mgr.GetPlayerPermissions(a_PlayerUUID);
|
||||||
std::cout << " Permissions(" << Permissions.size() << "):" << std::endl;
|
LOG(" Permissions(%u):", (unsigned)Permissions.size());
|
||||||
for (AStringVector::const_iterator itr = Permissions.begin(), end = Permissions.end(); itr != end; ++itr)
|
for (AStringVector::const_iterator itr = Permissions.begin(), end = Permissions.end(); itr != end; ++itr)
|
||||||
{
|
{
|
||||||
std::cout << " '" << *itr << "'." << std::endl;
|
LOG(" '%s'", itr->c_str());
|
||||||
} // for itr - Groups[]
|
} // for itr - Groups[]
|
||||||
|
|
||||||
std::cout << "Done." << std::endl;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
cRankManager m_Mgr;
|
cRankManager m_Mgr;
|
||||||
cMojangAPI m_API;
|
|
||||||
} g_RankMgrTest;
|
} g_RankMgrTest;
|
||||||
//*/
|
//*/
|
||||||
|
|
||||||
@ -70,7 +96,7 @@ cRankManager::cRankManager(void) :
|
|||||||
m_DB("Ranks.sqlite", SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE)
|
m_DB("Ranks.sqlite", SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE)
|
||||||
{
|
{
|
||||||
// Create the DB tables, if they don't exist:
|
// Create the DB tables, if they don't exist:
|
||||||
m_DB.exec("CREATE TABLE IF NOT EXISTS Rank (RankID INTEGER PRIMARY KEY, Name, MsgPrefix, MsgPostfix, MsgNameColorCode)");
|
m_DB.exec("CREATE TABLE IF NOT EXISTS Rank (RankID INTEGER PRIMARY KEY, Name, MsgPrefix, MsgSuffix, MsgNameColorCode)");
|
||||||
m_DB.exec("CREATE TABLE IF NOT EXISTS PlayerRank (PlayerUUID, PlayerName, RankID INTEGER)");
|
m_DB.exec("CREATE TABLE IF NOT EXISTS PlayerRank (PlayerUUID, PlayerName, RankID INTEGER)");
|
||||||
m_DB.exec("CREATE TABLE IF NOT EXISTS PermGroup (PermGroupID INTEGER PRIMARY KEY, Name)");
|
m_DB.exec("CREATE TABLE IF NOT EXISTS PermGroup (PermGroupID INTEGER PRIMARY KEY, Name)");
|
||||||
m_DB.exec("CREATE TABLE IF NOT EXISTS RankPermGroup (RankID INTEGER, PermGroupID INTEGER)");
|
m_DB.exec("CREATE TABLE IF NOT EXISTS RankPermGroup (RankID INTEGER, PermGroupID INTEGER)");
|
||||||
@ -85,15 +111,23 @@ cRankManager::cRankManager(void) :
|
|||||||
|
|
||||||
AString cRankManager::GetPlayerRankName(const AString & a_PlayerUUID)
|
AString cRankManager::GetPlayerRankName(const AString & a_PlayerUUID)
|
||||||
{
|
{
|
||||||
SQLite::Statement stmt(m_DB, "SELECT Rank.Name FROM Rank LEFT JOIN PlayerRank ON Rank.RankID = PlayerRank.RankID WHERE PlayerRank.PlayerUUID = ?");
|
try
|
||||||
stmt.bind(1, a_PlayerUUID);
|
|
||||||
stmt.executeStep();
|
|
||||||
if (stmt.isDone())
|
|
||||||
{
|
{
|
||||||
// No data returned from the DB
|
SQLite::Statement stmt(m_DB, "SELECT Rank.Name FROM Rank LEFT JOIN PlayerRank ON Rank.RankID = PlayerRank.RankID WHERE PlayerRank.PlayerUUID = ?");
|
||||||
return AString();
|
stmt.bind(1, a_PlayerUUID);
|
||||||
|
stmt.executeStep();
|
||||||
|
if (stmt.isDone())
|
||||||
|
{
|
||||||
|
// No data returned from the DB
|
||||||
|
return AString();
|
||||||
|
}
|
||||||
|
return stmt.getColumn(0).getText();
|
||||||
}
|
}
|
||||||
return stmt.getColumn(0).getText();
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Cannot get player rank name: %s", __FUNCTION__, ex.what());
|
||||||
|
}
|
||||||
|
return AString();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -102,22 +136,29 @@ AString cRankManager::GetPlayerRankName(const AString & a_PlayerUUID)
|
|||||||
|
|
||||||
AStringVector cRankManager::GetPlayerGroups(const AString & a_PlayerUUID)
|
AStringVector cRankManager::GetPlayerGroups(const AString & a_PlayerUUID)
|
||||||
{
|
{
|
||||||
// Prepare the DB statement:
|
|
||||||
SQLite::Statement stmt(m_DB,
|
|
||||||
"SELECT Group.Name FROM Group "
|
|
||||||
"LEFT JOIN RankGroups "
|
|
||||||
"ON Group.GroupID = RankGroups.GroupID "
|
|
||||||
"LEFT JOIN PlayerRank "
|
|
||||||
"ON PlayerRank.RankID = RankGroups.RankID "
|
|
||||||
"WHERE PlayerRank.PlayerUUID = ?"
|
|
||||||
);
|
|
||||||
stmt.bind(1, a_PlayerUUID);
|
|
||||||
|
|
||||||
// Execute and get results:
|
|
||||||
AStringVector res;
|
AStringVector res;
|
||||||
while (stmt.executeStep())
|
try
|
||||||
{
|
{
|
||||||
res.push_back(stmt.getColumn(0).getText());
|
// Prepare the DB statement:
|
||||||
|
SQLite::Statement stmt(m_DB,
|
||||||
|
"SELECT PermGroup.Name FROM PermGroup "
|
||||||
|
"LEFT JOIN RankPermGroup "
|
||||||
|
"ON PermGroup.PermGroupID = RankPermGroup.PermGroupID "
|
||||||
|
"LEFT JOIN PlayerRank "
|
||||||
|
"ON PlayerRank.RankID = RankPermGroup.RankID "
|
||||||
|
"WHERE PlayerRank.PlayerUUID = ?"
|
||||||
|
);
|
||||||
|
stmt.bind(1, a_PlayerUUID);
|
||||||
|
|
||||||
|
// Execute and get results:
|
||||||
|
while (stmt.executeStep())
|
||||||
|
{
|
||||||
|
res.push_back(stmt.getColumn(0).getText());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Cannot get player groups: %s", __FUNCTION__, ex.what());
|
||||||
}
|
}
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
@ -128,22 +169,29 @@ AStringVector cRankManager::GetPlayerGroups(const AString & a_PlayerUUID)
|
|||||||
|
|
||||||
AStringVector cRankManager::GetPlayerPermissions(const AString & a_PlayerUUID)
|
AStringVector cRankManager::GetPlayerPermissions(const AString & a_PlayerUUID)
|
||||||
{
|
{
|
||||||
// Prepare the DB statement:
|
|
||||||
SQLite::Statement stmt(m_DB,
|
|
||||||
"SELECT PermissionItem.Permission FROM PermissionItem "
|
|
||||||
"LEFT JOIN RankGroups "
|
|
||||||
"ON PermissionItem.GroupID = RankGroups.GroupID "
|
|
||||||
"LEFT JOIN PlayerRank "
|
|
||||||
"ON PlayerRank.RankID = RankGroups.RankID "
|
|
||||||
"WHERE PlayerRank.PlayerUUID = ?"
|
|
||||||
);
|
|
||||||
stmt.bind(1, a_PlayerUUID);
|
|
||||||
|
|
||||||
// Execute and get results:
|
|
||||||
AStringVector res;
|
AStringVector res;
|
||||||
while (stmt.executeStep())
|
try
|
||||||
{
|
{
|
||||||
res.push_back(stmt.getColumn(0).getText());
|
// Prepare the DB statement:
|
||||||
|
SQLite::Statement stmt(m_DB,
|
||||||
|
"SELECT PermissionItem.Permission FROM PermissionItem "
|
||||||
|
"LEFT JOIN RankPermGroup "
|
||||||
|
"ON PermissionItem.PermGroupID = RankPermGroup.PermGroupID "
|
||||||
|
"LEFT JOIN PlayerRank "
|
||||||
|
"ON PlayerRank.RankID = RankPermGroup.RankID "
|
||||||
|
"WHERE PlayerRank.PlayerUUID = ?"
|
||||||
|
);
|
||||||
|
stmt.bind(1, a_PlayerUUID);
|
||||||
|
|
||||||
|
// Execute and get results:
|
||||||
|
while (stmt.executeStep())
|
||||||
|
{
|
||||||
|
res.push_back(stmt.getColumn(0).getText());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Cannot get player permissions: %s", __FUNCTION__, ex.what());
|
||||||
}
|
}
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
@ -151,3 +199,575 @@ AStringVector cRankManager::GetPlayerPermissions(const AString & a_PlayerUUID)
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
AStringVector cRankManager::GetRankGroups(const AString & a_RankName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
AStringVector res;
|
||||||
|
res.push_back(Printf("%s: DummyGroup", __FUNCTION__));
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
AStringVector cRankManager::GetGroupPermissions(const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
AStringVector res;
|
||||||
|
res.push_back(Printf("%s: DummyPermission", __FUNCTION__));
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
AStringVector cRankManager::GetRankPermissions(const AString & a_RankName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
AStringVector res;
|
||||||
|
res.push_back(Printf("%s: DummyPermission", __FUNCTION__));
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
AStringVector cRankManager::GetAllRanks(void)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
AStringVector res;
|
||||||
|
res.push_back(Printf("%s: DummyRank", __FUNCTION__));
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
AStringVector cRankManager::GetAllGroups(void)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
AStringVector res;
|
||||||
|
res.push_back(Printf("%s: DummyGroup", __FUNCTION__));
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
AStringVector cRankManager::GetAllPermissions(void)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
AStringVector res;
|
||||||
|
res.push_back(Printf("%s: DummyPermission", __FUNCTION__));
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::GetPlayerMsgVisuals(
|
||||||
|
const AString & a_PlayerUUID,
|
||||||
|
AString & a_MsgPrefix,
|
||||||
|
AString & a_MsgSuffix,
|
||||||
|
AString & a_MsgNameColorCode
|
||||||
|
)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
a_MsgPrefix = Printf("%s: DummyPrefix", __FUNCTION__);
|
||||||
|
a_MsgSuffix = Printf("%s: DummySuffix", __FUNCTION__);
|
||||||
|
a_MsgNameColorCode = Printf("%s: DummyMsgNameColorCode", __FUNCTION__);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::AddRank(
|
||||||
|
const AString & a_RankName,
|
||||||
|
const AString & a_MsgPrefix,
|
||||||
|
const AString & a_MsgSuffix,
|
||||||
|
const AString & a_MsgNameColorCode
|
||||||
|
)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
// Check if such a rank name is already used:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT COUNT(*) FROM Rank WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_RankName);
|
||||||
|
if (stmt.executeStep())
|
||||||
|
{
|
||||||
|
if (stmt.getColumn(0).getInt() > 0)
|
||||||
|
{
|
||||||
|
// Rank already exists, do nothing:
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Insert a new rank:
|
||||||
|
SQLite::Statement stmt(m_DB, "INSERT INTO Rank (Name, MsgPrefix, MsgSuffix, MsgNameColorCode) VALUES (?, ?, ?, ?)");
|
||||||
|
stmt.bind(1, a_RankName);
|
||||||
|
stmt.bind(2, a_MsgPrefix);
|
||||||
|
stmt.bind(3, a_MsgSuffix);
|
||||||
|
stmt.bind(4, a_MsgNameColorCode);
|
||||||
|
if (stmt.exec() <= 0)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add a new rank \"%s\".", __FUNCTION__, a_RankName.c_str());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add a new rank \"%s\": %s", __FUNCTION__, a_RankName.c_str(), ex.what());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::AddGroup(const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
// Check if such a rank name is already used:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT COUNT(*) FROM PermGroup WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_GroupName);
|
||||||
|
if (stmt.executeStep())
|
||||||
|
{
|
||||||
|
if (stmt.getColumn(0).getInt() > 0)
|
||||||
|
{
|
||||||
|
// Group already exists, do nothing:
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Insert a new rank:
|
||||||
|
SQLite::Statement stmt(m_DB, "INSERT INTO PermGroup (Name) VALUES (?)");
|
||||||
|
stmt.bind(1, a_GroupName);
|
||||||
|
if (stmt.exec() <= 0)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add a new group \"%s\".", __FUNCTION__, a_GroupName.c_str());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add a new group \"%s\": %s", __FUNCTION__, a_GroupName.c_str(), ex.what());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::AddGroupToRank(const AString & a_GroupName, const AString & a_RankName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SQLite::Transaction trans(m_DB);
|
||||||
|
int GroupID, RankID;
|
||||||
|
|
||||||
|
// Get the group's ID:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT PermGroupID FROM PermGroup WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_GroupName);
|
||||||
|
if (!stmt.executeStep())
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: No such group (%s), aborting.", __FUNCTION__, a_GroupName.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
GroupID = stmt.getColumn(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the rank's ID:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT RankID FROM Rank WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_RankName);
|
||||||
|
if (!stmt.executeStep())
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: No such rank (%s), aborting.", __FUNCTION__, a_RankName.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
RankID = stmt.getColumn(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if the group is already there:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT COUNT(*) FROM RankPermGroup WHERE RankID = ? AND PermGroupID = ?");
|
||||||
|
stmt.bind(1, RankID);
|
||||||
|
stmt.bind(2, GroupID);
|
||||||
|
if (!stmt.executeStep())
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to check binding between rank %s and group %s, aborting.", __FUNCTION__, a_RankName.c_str(), a_GroupName.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (stmt.getColumn(0).getInt() > 0)
|
||||||
|
{
|
||||||
|
LOGD("%s: Group %s already present in rank %s, skipping and returning success.",
|
||||||
|
__FUNCTION__, a_GroupName.c_str(), a_RankName.c_str()
|
||||||
|
);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add the group:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "INSERT INTO RankPermGroup (RankID, PermGroupID) VALUES (?, ?)");
|
||||||
|
stmt.bind(1, RankID);
|
||||||
|
stmt.bind(2, GroupID);
|
||||||
|
if (stmt.exec() <= 0)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add group %s to rank %s, aborting.", __FUNCTION__, a_GroupName.c_str(), a_RankName.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Adding succeeded:
|
||||||
|
trans.commit();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add group %s to rank %s: %s", __FUNCTION__, a_GroupName.c_str(), a_RankName.c_str(), ex.what());
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::AddPermissionToGroup(const AString & a_Permission, const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
// Wrapp the entire operation into a transaction:
|
||||||
|
SQLite::Transaction trans(m_DB);
|
||||||
|
|
||||||
|
// Get the group's ID:
|
||||||
|
int GroupID;
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT PermGroupID FROM PermGroup WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_GroupName);
|
||||||
|
if (!stmt.executeStep())
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: No such group (%s), aborting.", __FUNCTION__, a_GroupName.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
GroupID = stmt.getColumn(0).getInt();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if the permission is already present:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT COUNT(*) FROM PermissionItem WHERE PermGroupID = ? AND Permission = ?");
|
||||||
|
stmt.bind(1, GroupID);
|
||||||
|
stmt.bind(2, a_Permission);
|
||||||
|
if (!stmt.executeStep())
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to check binding between permission %s and group %s, aborting.", __FUNCTION__, a_Permission.c_str(), a_GroupName.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (stmt.getColumn(0).getInt() > 0)
|
||||||
|
{
|
||||||
|
LOGD("%s: Permission %s is already present in group %s, skipping and returning success.",
|
||||||
|
__FUNCTION__, a_Permission.c_str(), a_GroupName.c_str()
|
||||||
|
);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add the permission:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "INSERT INTO PermissionItem (Permission, PermGroupID) VALUES (?, ?)");
|
||||||
|
stmt.bind(1, a_Permission);
|
||||||
|
stmt.bind(2, GroupID);
|
||||||
|
if (stmt.exec() <= 0)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add permission %s to group %s, aborting.", __FUNCTION__, a_Permission.c_str(), a_GroupName.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Adding succeeded:
|
||||||
|
trans.commit();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to add permission %s to group %s: %s",
|
||||||
|
__FUNCTION__, a_Permission.c_str(), a_GroupName.c_str(), ex.what()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::RemoveRank(const AString & a_RankName, const AString & a_ReplacementRankName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::RemoveGroup(const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::RemoveGroupFromRank(const AString & a_RankName, const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::RemovePermissionFromGroup(const AString & a_Permission, const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::RenameRank(const AString & a_OldName, const AString & a_NewName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::RenameGroup(const AString & a_OldName, const AString & a_NewName)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::SetPlayerRank(const AString & a_PlayerUUID, const AString & a_RankName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
// Wrap the entire operation into a transaction:
|
||||||
|
SQLite::Transaction trans(m_DB);
|
||||||
|
|
||||||
|
// Get the rank ID:
|
||||||
|
int RankID;
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT RankID FROM Rank WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_RankName);
|
||||||
|
if (!stmt.executeStep())
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: There is no rank %s, aborting.", __FUNCTION__, a_RankName.c_str());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
RankID = stmt.getColumn(0).getInt();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update the player's rank, if already in DB:
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "UPDATE PlayerRank SET RankID = ? WHERE PlayerUUID = ?");
|
||||||
|
stmt.bind(1, RankID);
|
||||||
|
stmt.bind(2, a_PlayerUUID);
|
||||||
|
if (stmt.exec() > 0)
|
||||||
|
{
|
||||||
|
// Successfully updated the player's rank
|
||||||
|
trans.commit();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// The player is not yet in the DB, add them:
|
||||||
|
SQLite::Statement stmt(m_DB, "INSERT INTO PlayerRank (RankID, PlayerUUID) VALUES (?, ?)");
|
||||||
|
stmt.bind(1, RankID);
|
||||||
|
stmt.bind(2, a_PlayerUUID);
|
||||||
|
if (stmt.exec() > 0)
|
||||||
|
{
|
||||||
|
// Successfully added the player
|
||||||
|
trans.commit();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
LOGWARNING("%s: Failed to set player UUID %s to rank %s.",
|
||||||
|
__FUNCTION__, a_PlayerUUID.c_str(), a_RankName.c_str()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to set player UUID %s to rank %s: %s",
|
||||||
|
__FUNCTION__, a_PlayerUUID.c_str(), a_RankName.c_str(), ex.what()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void cRankManager::SetRankVisuals(
|
||||||
|
const AString & a_RankName,
|
||||||
|
const AString & a_MsgPrefix,
|
||||||
|
const AString & a_MsgSuffix,
|
||||||
|
const AString & a_MsgNameColorCode
|
||||||
|
)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Not implemented yet", __FUNCTION__);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::RankExists(const AString & a_RankName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT * FROM Rank WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_RankName);
|
||||||
|
if (stmt.executeStep())
|
||||||
|
{
|
||||||
|
// The rank was found
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to query DB for rank %s: %s", __FUNCTION__, a_RankName.c_str(), ex.what());
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::GroupExists(const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT * FROM PermGroup WHERE Name = ?");
|
||||||
|
stmt.bind(1, a_GroupName);
|
||||||
|
if (stmt.executeStep())
|
||||||
|
{
|
||||||
|
// The group was found
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to query DB for group %s: %s", __FUNCTION__, a_GroupName.c_str(), ex.what());
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::IsPlayerRankSet(const AString & a_PlayerUUID)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB, "SELECT * FROM PlayerRank WHERE PlayerUUID = ?");
|
||||||
|
stmt.bind(1, a_PlayerUUID);
|
||||||
|
if (stmt.executeStep())
|
||||||
|
{
|
||||||
|
// The player UUID was found, they have a rank
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to query DB for player UUID %s: %s", __FUNCTION__, a_PlayerUUID.c_str(), ex.what());
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::IsGroupInRank(const AString & a_GroupName, const AString & a_RankName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB,
|
||||||
|
"SELECT * FROM Rank "
|
||||||
|
"LEFT JOIN RankPermGroup ON Rank.RankID = RankPermGroup.RankID "
|
||||||
|
"LEFT JOIN PermGroup ON PermGroup.PermGroupID = RankPermGroup.PermGroupID "
|
||||||
|
"WHERE Rank.Name = ? AND PermGroup.Name = ?"
|
||||||
|
);
|
||||||
|
stmt.bind(1, a_RankName);
|
||||||
|
stmt.bind(2, a_GroupName);
|
||||||
|
if (stmt.executeStep())
|
||||||
|
{
|
||||||
|
// The group is in the rank
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to query DB: %s", __FUNCTION__, ex.what());
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool cRankManager::IsPermissionInGroup(const AString & a_Permission, const AString & a_GroupName)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
SQLite::Statement stmt(m_DB,
|
||||||
|
"SELECT * FROM PermissionItem "
|
||||||
|
"LEFT JOIN PermGroup ON PermGroup.PermGroupID = PermissionItem.PermGroupID "
|
||||||
|
"WHERE PermissionItem.Permission = ? AND PermGroup.Name = ?"
|
||||||
|
);
|
||||||
|
stmt.bind(1, a_Permission);
|
||||||
|
stmt.bind(2, a_GroupName);
|
||||||
|
if (stmt.executeStep())
|
||||||
|
{
|
||||||
|
// The permission is in the group
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (const SQLite::Exception & ex)
|
||||||
|
{
|
||||||
|
LOGWARNING("%s: Failed to query DB: %s", __FUNCTION__, ex.what());
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -68,7 +68,7 @@ public:
|
|||||||
/** Adds the specified permission group to the specified rank.
|
/** Adds the specified permission group to the specified rank.
|
||||||
Fails if the rank or group names are not found.
|
Fails if the rank or group names are not found.
|
||||||
Returns true if successful, false on error. */
|
Returns true if successful, false on error. */
|
||||||
bool AddGroupToRank(const AString & a_RankName, const AString & a_GroupName);
|
bool AddGroupToRank(const AString & a_GroupName, const AString & a_RankName);
|
||||||
|
|
||||||
/** Adds the specified permission to the specified permission group.
|
/** Adds the specified permission to the specified permission group.
|
||||||
Fails if the permission group name is not found.
|
Fails if the permission group name is not found.
|
||||||
@ -113,6 +113,22 @@ public:
|
|||||||
const AString & a_MsgSuffix,
|
const AString & a_MsgSuffix,
|
||||||
const AString & a_MsgNameColorCode
|
const AString & a_MsgNameColorCode
|
||||||
);
|
);
|
||||||
|
|
||||||
|
/** Returns true iff the specified rank exists in the DB. */
|
||||||
|
bool RankExists(const AString & a_RankName);
|
||||||
|
|
||||||
|
/** Returns true iff the specified group exists in the DB. */
|
||||||
|
bool GroupExists(const AString & a_GroupName);
|
||||||
|
|
||||||
|
/** Returns true iff the specified player has a rank assigned to them in the DB. */
|
||||||
|
bool IsPlayerRankSet(const AString & a_PlayerUUID);
|
||||||
|
|
||||||
|
/** Returns true iff the specified rank contains the specified group. */
|
||||||
|
bool IsGroupInRank(const AString & a_GroupName, const AString & a_RankName);
|
||||||
|
|
||||||
|
/** Returns true iff the specified group contains the specified permission. */
|
||||||
|
bool IsPermissionInGroup(const AString & a_Permission, const AString & a_GroupName);
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
|
|
||||||
SQLite::Database m_DB;
|
SQLite::Database m_DB;
|
||||||
|
Loading…
Reference in New Issue
Block a user