
1 changed files with 0 additions and 323 deletions
@ -1,323 +0,0 @@ |
|||||
#include "../src/sqlite3.h" |
|
||||
#include "../src/pEpLog.hh" |
|
||||
#include <iostream> |
|
||||
#include <cstdio> |
|
||||
#include <exception> |
|
||||
#include <stdexcept> |
|
||||
#include <vector> |
|
||||
#include <map> |
|
||||
|
|
||||
//TODO: add const
|
|
||||
using namespace std; |
|
||||
using namespace pEp; |
|
||||
|
|
||||
using ResultSet = vector<map<string, string>>; |
|
||||
using RSRecord = map<string, string>; |
|
||||
|
|
||||
namespace pEp { |
|
||||
namespace SQLite3 { |
|
||||
::sqlite3 *db; |
|
||||
string db_path; |
|
||||
|
|
||||
ResultSet resultset; |
|
||||
|
|
||||
void create_or_open_db() |
|
||||
{ |
|
||||
pEpLog("called"); |
|
||||
int rc{::sqlite3_open(db_path.c_str(), &db)}; |
|
||||
|
|
||||
if (rc) { |
|
||||
runtime_error e{string("Can't open database (" + db_path + "):" + ::sqlite3_errmsg(db))}; |
|
||||
throw (e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
void close_db() |
|
||||
{ |
|
||||
::sqlite3_close(db); |
|
||||
} |
|
||||
|
|
||||
void delete_db() |
|
||||
{ |
|
||||
pEpLog("called"); |
|
||||
remove(db_path.c_str()); |
|
||||
if (errno) { |
|
||||
cerr << "could not delete db (" + db_path + "): " << strerror(errno) << endl; |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
ResultSet execute(const string& stmt) |
|
||||
{ |
|
||||
// pEpLog("execute(\"" + stmt + "\")");
|
|
||||
resultset.clear(); |
|
||||
char *zErrMsg = nullptr; |
|
||||
int rc = ::sqlite3_exec(db, stmt.c_str(), [](void *NotUsed, int argc, char **argv, char **azColName) -> int { |
|
||||
RSRecord record; |
|
||||
for (int col = 0; col < argc; col++) { |
|
||||
const string key = string{azColName[col]}; |
|
||||
// TODO: NULL is not correct, could be a valid value
|
|
||||
const string val = string{argv[col] ? argv[col] : "NULL"}; |
|
||||
record.insert({key, val}); |
|
||||
} |
|
||||
resultset.push_back(record); |
|
||||
return 0; |
|
||||
}, 0, &zErrMsg); |
|
||||
if (rc != SQLITE_OK) { |
|
||||
runtime_error e{string("execute: " + string(::sqlite3_errmsg(db)) + ":" + string(zErrMsg))}; |
|
||||
::sqlite3_free(zErrMsg); |
|
||||
throw (e); |
|
||||
} |
|
||||
return resultset; |
|
||||
} |
|
||||
|
|
||||
string resultset_to_string(ResultSet rs) |
|
||||
{ |
|
||||
stringstream ss; |
|
||||
int i = 0; |
|
||||
for (const RSRecord& rec : rs) { |
|
||||
for (const auto& item : rec) { |
|
||||
ss << "RESULTSET[" << i << "][" << item.first << "] = " << item.second << "\"" << endl; |
|
||||
} |
|
||||
i++; |
|
||||
} |
|
||||
return ss.str(); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
|
|
||||
void print_exception(const exception& e, int level = 0) |
|
||||
{ |
|
||||
cerr << string(level, ' ') << "exception: " << e.what() << endl; |
|
||||
try { |
|
||||
rethrow_if_nested(e); |
|
||||
} catch (const exception& e) { |
|
||||
print_exception(e, level + 1); |
|
||||
} catch (...) { |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
void db_config() |
|
||||
{ |
|
||||
pEpLog("called"); |
|
||||
try { |
|
||||
string sql; |
|
||||
sql = "PRAGMA foreign_keys=ON"; |
|
||||
SQLite3::execute(sql); |
|
||||
} catch (...) { |
|
||||
runtime_error e("db_config() - failed with exception"); |
|
||||
throw_with_nested(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
void create_tables() |
|
||||
{ |
|
||||
pEpLog("called"); |
|
||||
try { |
|
||||
string sql; |
|
||||
|
|
||||
sql = "CREATE TABLE IF NOT EXISTS lists(" |
|
||||
"address TEXT NOT NULL," |
|
||||
"moderator_address TEXT NOT NULL," |
|
||||
"PRIMARY KEY(address));"; |
|
||||
SQLite3::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);"; |
|
||||
SQLite3::execute(sql); |
|
||||
} catch (...) { |
|
||||
runtime_error e("create_tables() - failed with exception"); |
|
||||
throw_with_nested(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
void list_add(const string& addr_list, const string& addr_mgr) |
|
||||
{ |
|
||||
pEpLog("list_add(addr_list: \"" + addr_list + "\", addr_mgr: \"" + addr_mgr + "\")"); |
|
||||
try { |
|
||||
string sql = "INSERT INTO lists(address, moderator_address) VALUES ('" + addr_list + "','" + |
|
||||
addr_mgr + "');"; |
|
||||
SQLite3::execute(sql); |
|
||||
} catch (...) { |
|
||||
runtime_error e( |
|
||||
"list_add(addr_list: \"" + addr_list + "\"\taddr_mgr: \"" + addr_mgr + |
|
||||
"\") - failed with exception"); |
|
||||
throw_with_nested(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
void list_delete(const string& addr_list) |
|
||||
{ |
|
||||
pEpLog("list_delete(addr_list: \"" + addr_list + "\")"); |
|
||||
try { |
|
||||
string sql; |
|
||||
sql = "DELETE FROM lists WHERE lists.address = '" + addr_list + "';"; |
|
||||
SQLite3::execute(sql); |
|
||||
} catch (...) { |
|
||||
runtime_error e("list_delete(addr_list: \"" + addr_list + "\") - failed with exception"); |
|
||||
throw_with_nested(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
void member_add(const string& addr_list, const string& addr_member) |
|
||||
{ |
|
||||
pEpLog("member_add(addr_list: \"" + addr_list + "\", addr_member: \"" + addr_member + "\")"); |
|
||||
try { |
|
||||
string sql = "INSERT INTO member_of(address, list_address) VALUES ('" + addr_member + |
|
||||
"', '" + addr_list + "');"; |
|
||||
SQLite3::execute(sql); |
|
||||
} catch (...) { |
|
||||
runtime_error e( |
|
||||
"member_add(addr_list: \"" + addr_list + "\", addr_member: \"" + addr_member + |
|
||||
"\") - failed with exception"); |
|
||||
throw_with_nested(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
void member_remove(const string& addr_list, const string& addr_member) |
|
||||
{ |
|
||||
pEpLog("member_remove(addr_list: \"" + addr_list + "\", addr_member: '\"" + addr_member + "\")"); |
|
||||
try { |
|
||||
string sql; |
|
||||
sql = "DELETE FROM member_of WHERE (member_of.address = '" + addr_member + |
|
||||
"') AND (member_of.list_address = '" + addr_list + "');"; |
|
||||
SQLite3::execute(sql); |
|
||||
} catch (...) { |
|
||||
runtime_error e( |
|
||||
"member_remove(" + addr_list + ", " + addr_member + ") - failed with exception"); |
|
||||
throw_with_nested(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
vector<string> lists() |
|
||||
{ |
|
||||
pEpLog("called"); |
|
||||
vector<string> ret; |
|
||||
ResultSet rs; |
|
||||
|
|
||||
try { |
|
||||
string sql; |
|
||||
sql = "SELECT address FROM lists"; |
|
||||
rs = SQLite3::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; |
|
||||
} |
|
||||
|
|
||||
// Exceptions:
|
|
||||
// * ListNotFound
|
|
||||
string moderator(string list_address) |
|
||||
{ |
|
||||
pEpLog("called"); |
|
||||
string ret; |
|
||||
ResultSet rs; |
|
||||
|
|
||||
try { |
|
||||
string sql; |
|
||||
sql = "SELECT moderator_address FROM lists " |
|
||||
"WHERE lists.address = '" + list_address + "';"; |
|
||||
rs = SQLite3::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; |
|
||||
} |
|
||||
|
|
||||
// Exceptions:
|
|
||||
// * ListNotFound
|
|
||||
vector<string> members(string list_address) |
|
||||
{ |
|
||||
pEpLog("called"); |
|
||||
vector<string> ret; |
|
||||
ResultSet rs; |
|
||||
|
|
||||
try { |
|
||||
string sql; |
|
||||
sql = "SELECT address FROM member_of " |
|
||||
"WHERE list_address = '" + list_address + "'"; |
|
||||
rs = SQLite3::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; |
|
||||
} |
|
||||
|
|
||||
template<typename T> |
|
||||
string vector_to_string(vector<T> v) |
|
||||
{ |
|
||||
stringstream ss; |
|
||||
for (const T& elem : v) { |
|
||||
ss << elem << endl; |
|
||||
} |
|
||||
return ss.str(); |
|
||||
} |
|
||||
|
|
||||
int main(int argc, char *argv[]) |
|
||||
{ |
|
||||
pEp::Adapter::pEpLog::set_enabled(true); |
|
||||
SQLite3::db_path = "test.db"; |
|
||||
string dummy_in; |
|
||||
|
|
||||
|
|
||||
try { |
|
||||
SQLite3::db_path = "test.db"; |
|
||||
SQLite3::delete_db(); |
|
||||
SQLite3::create_or_open_db(); |
|
||||
db_config(); |
|
||||
create_tables(); |
|
||||
|
|
||||
list_add("grp1@peptest.org", "alice@peptest.org"); |
|
||||
cout << vector_to_string(lists()); |
|
||||
list_add("grp2@peptest.org", "alice@peptest.org"); |
|
||||
|
|
||||
cout << vector_to_string(lists()); |
|
||||
|
|
||||
try { |
|
||||
list_add("grp1@peptest.org", "bob@peptest.org"); |
|
||||
assert(false); |
|
||||
} catch (const exception& e) { |
|
||||
print_exception(e); |
|
||||
} |
|
||||
|
|
||||
member_add("grp1@peptest.org", "bob@peptest.org"); |
|
||||
member_add("grp1@peptest.org", "carol@peptest.org"); |
|
||||
member_add("grp1@peptest.org", "joe@peptest.org"); |
|
||||
|
|
||||
cout << vector_to_string(members("grp1@peptest.org")) << endl; |
|
||||
|
|
||||
cout << moderator("grp1@peptest.org") << endl; |
|
||||
member_remove("grp1@peptest.org", "joe@peptest.org"); |
|
||||
|
|
||||
list_delete("grp1@peptest.org"); |
|
||||
cout << vector_to_string(lists()); |
|
||||
} catch (const exception& e) { |
|
||||
print_exception(e); |
|
||||
} |
|
||||
SQLite3::close_db(); |
|
||||
} |
|
Loading…
Reference in new issue