
2 changed files with 295 additions and 0 deletions
@ -0,0 +1,253 @@ |
|||
#include "listmanager_dummy.hh" |
|||
#include "pEpSQLite.hh" |
|||
#include <exception> |
|||
|
|||
using namespace std; |
|||
namespace pEp { |
|||
bool ListManagerDummy::log_enabled = false; |
|||
|
|||
// public
|
|||
ListManagerDummy::ListManagerDummy(const string& db_path) : db(pEpSQLite(db_path)) |
|||
{ |
|||
pEpLogClass("called"); |
|||
} |
|||
|
|||
// private
|
|||
void ListManagerDummy::ensure_db_initialized() |
|||
{ |
|||
if (!db.is_open()) { |
|||
is_db_initialized = false; |
|||
try { |
|||
db.create_or_open_db(); |
|||
} catch (...) { |
|||
db.close_db(); |
|||
runtime_error e("ensure_db_initialized() - error opening db"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
if (!is_db_initialized) { |
|||
try { |
|||
db_config(); |
|||
create_tables(); |
|||
} catch (...) { |
|||
db.close_db(); |
|||
runtime_error e("ensure_db_initialized() - db init failed"); |
|||
throw_with_nested(e); |
|||
} |
|||
is_db_initialized = true; |
|||
} |
|||
} |
|||
|
|||
// private
|
|||
void ListManagerDummy::db_config() |
|||
{ |
|||
try { |
|||
string sql; |
|||
sql = "PRAGMA foreign_keys=ON"; |
|||
db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e("db_config() - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
|
|||
// private
|
|||
void ListManagerDummy::create_tables() |
|||
{ |
|||
try { |
|||
string sql; |
|||
|
|||
sql = "CREATE TABLE IF NOT EXISTS lists(" |
|||
"address TEXT NOT NULL," |
|||
"moderator_address TEXT NOT NULL," |
|||
"PRIMARY KEY(address));"; |
|||
db.execute(sql); |
|||
|
|||
sql = "CREATE TABLE IF NOT EXISTS member_of(" |
|||
"address TEXT NOT NULL," |
|||
"list_address TEXT NOT NULL," |
|||
"PRIMARY KEY (address, list_address)," |
|||
"FOREIGN KEY(list_address) REFERENCES lists(address) ON DELETE CASCADE);"; |
|||
db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e("create_tables() - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
|
|||
// public
|
|||
void ListManagerDummy::close_db() |
|||
{ |
|||
pEpLogClass("called"); |
|||
db.close_db(); |
|||
} |
|||
|
|||
// public
|
|||
void ListManagerDummy::delete_db() |
|||
{ |
|||
pEpLogClass("called"); |
|||
try { |
|||
db.delete_db(); |
|||
} catch (...) { |
|||
runtime_error e("delete_db() - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
|
|||
// public
|
|||
void ListManagerDummy::list_add(const std::string& addr_list, const std::string& addr_mgr) |
|||
{ |
|||
pEpLogClass("list_add(addr_list: \"" + addr_list + "\", addr_mgr: \"" + addr_mgr + "\")"); |
|||
ensure_db_initialized(); |
|||
try { |
|||
string sql = "INSERT INTO lists(address, moderator_address) VALUES ('" + addr_list + |
|||
"','" + addr_mgr + "');"; |
|||
|
|||
db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e( |
|||
"list_add(addr_list: \"" + addr_list + "\"\taddr_mgr: \"" + addr_mgr + |
|||
"\") - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
|
|||
// public
|
|||
void ListManagerDummy::list_delete(const std::string& addr_list) |
|||
{ |
|||
pEpLogClass("list_delete(addr_list: \"" + addr_list + "\")"); |
|||
ensure_db_initialized(); |
|||
try { |
|||
string sql; |
|||
sql = "DELETE FROM lists WHERE lists.address = '" + addr_list + "';"; |
|||
db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e("list_delete(addr_list: \"" + addr_list + "\") - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
|
|||
// public
|
|||
void ListManagerDummy::member_add(const std::string& addr_list, const std::string& addr_member) |
|||
{ |
|||
pEpLogClass( |
|||
"member_add(addr_list: \"" + addr_list + "\", addr_member: \"" + addr_member + "\")"); |
|||
ensure_db_initialized(); |
|||
try { |
|||
string sql = "INSERT INTO member_of(address, list_address) VALUES ('" + addr_member + |
|||
"', '" + addr_list + "');"; |
|||
db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e( |
|||
"member_add(addr_list: \"" + addr_list + "\", addr_member: \"" + addr_member + |
|||
"\") - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
|
|||
// public
|
|||
void ListManagerDummy::member_remove(const std::string& addr_list, const std::string& addr_member) |
|||
{ |
|||
pEpLogClass( |
|||
"member_remove(addr_list: \"" + addr_list + "\", addr_member: '\"" + addr_member + "\")"); |
|||
ensure_db_initialized(); |
|||
try { |
|||
string sql; |
|||
sql = "DELETE FROM member_of WHERE (member_of.address = '" + addr_member + |
|||
"') AND (member_of.list_address = '" + addr_list + "');"; |
|||
db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e( |
|||
"member_remove(" + addr_list + ", " + addr_member + ") - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
} |
|||
|
|||
// public
|
|||
std::vector<std::string> ListManagerDummy::lists() |
|||
{ |
|||
pEpLogClass("called"); |
|||
ensure_db_initialized(); |
|||
vector<string> ret; |
|||
ResultSet rs; |
|||
|
|||
try { |
|||
string sql; |
|||
sql = "SELECT address FROM lists"; |
|||
rs = db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e("lists() - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
|
|||
for (const RSRecord& rec : rs) { |
|||
ret.push_back(rec.at("address")); |
|||
} |
|||
|
|||
return ret; |
|||
} |
|||
|
|||
// public
|
|||
std::string ListManagerDummy::moderator(std::string list_address) |
|||
{ |
|||
pEpLogClass("called"); |
|||
ensure_db_initialized(); |
|||
string ret; |
|||
ResultSet rs; |
|||
|
|||
try { |
|||
string sql; |
|||
sql = "SELECT moderator_address FROM lists " |
|||
"WHERE lists.address = '" + |
|||
list_address + "';"; |
|||
rs = db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e("lists() - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
|
|||
if (!rs.empty()) { |
|||
for (const RSRecord& rec : rs) { |
|||
ret = rec.at("moderator_address"); |
|||
} |
|||
} |
|||
|
|||
return ret; |
|||
} |
|||
|
|||
// public
|
|||
std::vector<std::string> ListManagerDummy::members(std::string list_address) |
|||
{ |
|||
pEpLogClass("called"); |
|||
ensure_db_initialized(); |
|||
vector<string> ret; |
|||
ResultSet rs; |
|||
|
|||
try { |
|||
string sql; |
|||
sql = "SELECT address FROM member_of " |
|||
"WHERE list_address = '" + |
|||
list_address + "'"; |
|||
rs = db.execute(sql); |
|||
} catch (...) { |
|||
runtime_error e("lists() - failed with exception"); |
|||
throw_with_nested(e); |
|||
} |
|||
|
|||
if (!rs.empty()) { |
|||
for (const RSRecord& rec : rs) { |
|||
ret.push_back(rec.at("address")); |
|||
} |
|||
} |
|||
|
|||
return ret; |
|||
} |
|||
|
|||
// public
|
|||
ListManagerDummy::~ListManagerDummy() |
|||
{ |
|||
pEpLogClass("called"); |
|||
db.close_db(); |
|||
} |
|||
|
|||
} // namespace pEp
|
@ -0,0 +1,42 @@ |
|||
// This file is under GNU General Public License 3.0
|
|||
// see LICENSE.txt
|
|||
|
|||
#ifndef LIBPEPADAPTER_LISTMANAGER_DUMMY_HH |
|||
#define LIBPEPADAPTER_LISTMANAGER_DUMMY_HH |
|||
|
|||
#include "pEpSQLite.hh" |
|||
#include "pEpLog.hh" |
|||
#include <vector> |
|||
#include <string> |
|||
|
|||
namespace pEp { |
|||
class ListManagerDummy { |
|||
public: |
|||
ListManagerDummy(const std::string& db_path); |
|||
// Update
|
|||
void list_add(const std::string& addr_list, const std::string& addr_mgr); |
|||
void list_delete(const std::string& addr_list); |
|||
void member_add(const std::string& addr_list, const std::string& addr_member); |
|||
void member_remove(const std::string& addr_list, const std::string& addr_member); |
|||
// Query
|
|||
std::vector<std::string> lists(); |
|||
std::string moderator(std::string list_address); |
|||
std::vector<std::string> members(std::string list_address); |
|||
// db
|
|||
void close_db(); |
|||
void delete_db(); |
|||
// Logging
|
|||
static bool log_enabled; |
|||
Adapter::pEpLog::pEpLogger logger{"ListManagerDummy", log_enabled}; |
|||
~ListManagerDummy(); |
|||
private: |
|||
pEpSQLite db; |
|||
bool is_db_initialized {false}; |
|||
void ensure_db_initialized(); |
|||
void db_config(); |
|||
void create_tables(); |
|||
Adapter::pEpLog::pEpLogger& m4gic_logger_n4ame = logger; |
|||
}; |
|||
} // namespace pEp
|
|||
|
|||
#endif // LIBPEPADAPTER_LISTMANAGER_DUMMY_HH
|
Loading…
Reference in new issue