88 lines
4.5 KiB
C++
88 lines
4.5 KiB
C++
|
#include <sql/SqlQuery.h>
|
||
|
|
#include <log/LogUtils.h>
|
||
|
|
#include "LetterManager.h"
|
||
|
|
#include "src/TSServer.h"
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
using namespace std::chrono;
|
||
|
|
using namespace ts;
|
||
|
|
using namespace ts::letter;
|
||
|
|
using namespace ts::server;
|
||
|
|
|
||
|
|
//`serverId` INT NOT NULL, `sender` TEXT, `receiver` TEXT, `created` INT, `subject` TEXT, `message` TEXT
|
||
|
|
|
||
|
|
LetterManager::LetterManager(server::TSServer* server) : server(server) {}
|
||
|
|
LetterManager::~LetterManager() {}
|
||
|
|
|
||
|
|
std::vector<std::shared_ptr<LetterHeader>> LetterManager::avariableLetters(ClientUid cluid) {
|
||
|
|
vector<shared_ptr<LetterHeader>> result;
|
||
|
|
|
||
|
|
auto res = sql::command(this->server->getSql(), "SELECT `letterId`, `sender`, `created`, `subject`, `read` FROM `letters` WHERE `serverId` = :sid AND `receiver` = :uid", variable{":sid", this->server->getServerId()}, variable{":uid", cluid})
|
||
|
|
.query([cluid](vector<shared_ptr<LetterHeader>>* list, int length, char** values, char** columns){
|
||
|
|
shared_ptr<LetterHeader> letter = make_shared<LetterHeader>();
|
||
|
|
|
||
|
|
for(int index = 0; index < length; index++)
|
||
|
|
if(strcmp(columns[index], "sender") == 0)
|
||
|
|
letter->sender = values[index];
|
||
|
|
else if(strcmp(columns[index], "created") == 0)
|
||
|
|
letter->created = system_clock::now() + milliseconds(stoull(values[index]));
|
||
|
|
else if(strcmp(columns[index], "letterId") == 0)
|
||
|
|
letter->id = static_cast<LetterId>(stoull(values[index]));
|
||
|
|
else if(strcmp(columns[index], "subject") == 0)
|
||
|
|
letter->subject = values[index];
|
||
|
|
else if(strcmp(columns[index], "read") == 0)
|
||
|
|
letter->read = strcmp(values[index], "1") == 0;
|
||
|
|
letter->receiver = cluid;
|
||
|
|
list->push_back(letter);
|
||
|
|
return 0;
|
||
|
|
}, &result);
|
||
|
|
(LOG_SQL_CMD)(res);
|
||
|
|
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
void LetterManager::deleteLetter(LetterId letter) {
|
||
|
|
sql::command(this->server->getSql(), "DELETE FROM `letters` WHERE `letterId` = :id", variable{":id", letter}).executeLater().waitAndGetLater(LOG_SQL_CMD, {1, "future failed"});
|
||
|
|
}
|
||
|
|
|
||
|
|
void LetterManager::updateReadFlag(LetterId letter, bool flag) {
|
||
|
|
sql::command(this->server->getSql(), "UPDATE `letters` SET `read` = :flag WHERE `letterId` = :id", variable{":id", letter}, variable{":flag", flag}).executeLater().waitAndGetLater(LOG_SQL_CMD, {1, "future failed"});
|
||
|
|
}
|
||
|
|
|
||
|
|
shared_ptr<Letter> LetterManager::getFullLetter(LetterId letter) {
|
||
|
|
shared_ptr<Letter> res = std::make_shared<Letter>();
|
||
|
|
|
||
|
|
sql::command(this->server->getSql(), "SELECT `letterId`, `sender`, `receiver`, `created`, `subject`, `read`,`message` FROM `letters` WHERE `letterId` = :id", variable{":id", letter}).query([](Letter* letter, int length, char** values, char** columns){
|
||
|
|
for(int index = 0; index < length; index++)
|
||
|
|
if(strcmp(columns[index], "sender") == 0)
|
||
|
|
letter->sender = values[index];
|
||
|
|
else if(strcmp(columns[index], "created") == 0)
|
||
|
|
letter->created = system_clock::now() + milliseconds(stoull(values[index]));
|
||
|
|
else if(strcmp(columns[index], "letterId") == 0)
|
||
|
|
letter->id = static_cast<LetterId>(stoull(values[index]));
|
||
|
|
else if(strcmp(columns[index], "subject") == 0)
|
||
|
|
letter->subject = values[index];
|
||
|
|
else if(strcmp(columns[index], "read") == 0)
|
||
|
|
letter->read = strcmp(values[index], "1") == 0;
|
||
|
|
else if(strcmp(columns[index], "receiver") == 0)
|
||
|
|
letter->receiver = values[index];
|
||
|
|
else if(strcmp(columns[index], "message") == 0)
|
||
|
|
letter->message = values[index];
|
||
|
|
return 0;
|
||
|
|
}, res.get());
|
||
|
|
|
||
|
|
return res;
|
||
|
|
}
|
||
|
|
|
||
|
|
void LetterManager::createLetter(ClientUid sender, ClientUid reciver, std::string subject, std::string message) {
|
||
|
|
sql::command(this->server->getSql(), "INSERT INTO `letters` (`serverId`, `sender`, `receiver`, `created`, `subject`, `message`, `read`) VALUES (:sid, :sender, :receiver, :created, :subject, :message, :read)",
|
||
|
|
variable{":sid", this->server->getServerId()},
|
||
|
|
variable{":sender", sender},
|
||
|
|
variable{":receiver", reciver},
|
||
|
|
variable{":created", duration_cast<milliseconds>(system_clock::now().time_since_epoch()).count()},
|
||
|
|
variable{":subject", subject},
|
||
|
|
variable{":message", message},
|
||
|
|
variable{":read", false})
|
||
|
|
.executeLater().waitAndGetLater(LOG_SQL_CMD, {1, "future failed"});
|
||
|
|
}
|