Kayne Ruse
910e51f637
I'm mostly just testing the waters at this stage, and i've left some debug code in. There's no way to unload, delete, etc. the accounts, but that comes later.
121 lines
4.2 KiB
C++
121 lines
4.2 KiB
C++
/* Copyright: (c) Kayne Ruse 2014
|
|
*
|
|
* This software is provided 'as-is', without any express or implied
|
|
* warranty. In no event will the authors be held liable for any damages
|
|
* arising from the use of this software.
|
|
*
|
|
* Permission is granted to anyone to use this software for any purpose,
|
|
* including commercial applications, and to alter it and redistribute it
|
|
* freely, subject to the following restrictions:
|
|
*
|
|
* 1. The origin of this software must not be misrepresented; you must not
|
|
* claim that you wrote the original software. If you use this software
|
|
* in a product, an acknowledgment in the product documentation would be
|
|
* appreciated but is not required.
|
|
*
|
|
* 2. Altered source versions must be plainly marked as such, and must not be
|
|
* misrepresented as being the original software.
|
|
*
|
|
* 3. This notice may not be removed or altered from any source
|
|
* distribution.
|
|
*/
|
|
#include "server_application.hpp"
|
|
|
|
#include "sqlite3/sqlite3.h"
|
|
|
|
#include <stdexcept>
|
|
#include <iostream>
|
|
|
|
//-------------------------
|
|
//Define the queries
|
|
//-------------------------
|
|
|
|
static const char* CREATE_USER_ACCOUNT = "INSERT INTO UserAccounts (username) VALUES (?);";
|
|
static const char* LOAD_USER_ACCOUNT = "SELECT uid, username, blacklisted, whitelisted FROM UserAccounts WHERE username = ?;";
|
|
|
|
//-------------------------
|
|
//Define the methods
|
|
//-------------------------
|
|
|
|
int ServerApplication::CreateUserAccount(std::string username, int clientIndex) {
|
|
std::cout << "Calling CreateUserAccount(" << username << ")\n";
|
|
//create this user account, failing if it exists, leave this account in memory
|
|
sqlite3_stmt* statement = nullptr;
|
|
|
|
//prep
|
|
if (sqlite3_prepare_v2(database, CREATE_USER_ACCOUNT, -1, &statement, nullptr) != SQLITE_OK) {
|
|
throw( std::runtime_error(std::string() + "Failed to prepare an SQL statement: " + sqlite3_errmsg(database)) );
|
|
}
|
|
|
|
//parameter
|
|
if (sqlite3_bind_text(statement, 1, username.c_str(), username.size() + 1, SQLITE_STATIC) != SQLITE_OK) {
|
|
throw( std::runtime_error(std::string() + "Failed to replace a prepared statement's parameter: " + sqlite3_errmsg(database)) );
|
|
}
|
|
|
|
//execute
|
|
if (sqlite3_step(statement) != SQLITE_DONE) {
|
|
//if this returns something, than this account exists
|
|
sqlite3_finalize(statement);
|
|
return -1;
|
|
}
|
|
|
|
sqlite3_finalize(statement);
|
|
|
|
//load this account into memory
|
|
return LoadUserAccount(username, clientIndex);
|
|
}
|
|
|
|
int ServerApplication::LoadUserAccount(std::string username, int clientIndex) {
|
|
std::cout << "Calling LoadUserAccount(" << username << ")\n";
|
|
//load this user account, creating it if it doesn't exist
|
|
sqlite3_stmt* statement = nullptr;
|
|
|
|
//prep
|
|
if (sqlite3_prepare_v2(database, LOAD_USER_ACCOUNT, -1, &statement, nullptr) != SQLITE_OK) {
|
|
throw( std::runtime_error(std::string() + "Failed to prepare an SQL statement: " + sqlite3_errmsg(database)) );
|
|
}
|
|
|
|
//parameter
|
|
if (sqlite3_bind_text(statement, 1, username.c_str(), username.size() + 1, SQLITE_STATIC) != SQLITE_OK) {
|
|
throw( std::runtime_error(std::string() + "Failed to replace a prepared statement's parameter: " + sqlite3_errmsg(database)) );
|
|
}
|
|
|
|
//execute
|
|
int ret = sqlite3_step(statement);
|
|
|
|
//process the result
|
|
if (ret == SQLITE_ROW) {
|
|
std::cout << "ret = ROW\n";
|
|
//extract the data into memory
|
|
int uid = sqlite3_column_int(statement, 0);
|
|
accountMap[uid].username = reinterpret_cast<const char*>(sqlite3_column_text(statement, 1));
|
|
accountMap[uid].blackListed = sqlite3_column_int(statement, 2);
|
|
accountMap[uid].whiteListed = sqlite3_column_int(statement, 3);
|
|
accountMap[uid].clientIndex = clientIndex;
|
|
return uid;
|
|
}
|
|
|
|
if (ret == SQLITE_DONE) {
|
|
std::cout << "ret = DONE\n";
|
|
//create the non-existant account instead
|
|
return CreateUserAccount(username, clientIndex);
|
|
}
|
|
|
|
throw(std::runtime_error(std::string() + "Unknown SQL error in LoadUserAccount: " + sqlite3_errmsg(database) ));
|
|
}
|
|
|
|
void ServerApplication::SaveUserAccount(std::string username) {
|
|
//save this user account, replacing it if it exists
|
|
//TODO
|
|
}
|
|
|
|
void ServerApplication::UnloadUserAccount(std::string username) {
|
|
//save this user account, and then unload it
|
|
//TODO
|
|
}
|
|
|
|
void ServerApplication::DeleteUserAccount(std::string username) {
|
|
//delete a user account from the database, and remove it from memory
|
|
//TODO
|
|
}
|