wzray/promty
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: split response logic into functions

Browse source
This commit is contained in:
Arthur Khachaturov 2024-11-26 22:32:50 +03:00
parent 8aa7b759cf
commit 87a758ecd8
No known key found for this signature in database
GPG key ID: CAC2B7EB6DF45D55
1 changed files with 44 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

73
puppy/app/main.cpp
View file

@ -1,4 +1,4 @@
#include "httplib.hpp" #include "httplib.hpp"
#include <filesystem> #include <filesystem>
#include <fstream> #include <fstream>
@ -60,11 +60,27 @@ class WebServer {
std::mutex m_global_lock; std::mutex m_global_lock;
private: private:
void TranslateHandler(const httplib::Request &req, httplib::Response &res) { int TranslateHtml(const std::string &text, httplib::Response &res) {
std::lock_guard _(m_global_lock); const auto ft = PromtFTManager::FileType::kHTML;
auto direction = m_doc.direction();
auto translator = m_ftman.Translator(ft, direction);
{
std::ofstream ofs(TMP_IN);
ofs << text;
}
translator.Translate(TMP_IN, TMP_OUT);
res.set_content(read_file(TMP_OUT), "text/html");
auto target_dir_it = req.headers.find("x-translation-direction"); }
if (target_dir_it != req.headers.end()) {
void TranslateText(const std::string &text, httplib::Response &res) {
res.set_content("Plain text translation not implemented", "text/plain");
res.status = 400;
}
bool SetDirection(const httplib::Headers &headers, httplib::Response &res) {
const auto target_dir_it = headers.find("x-translation-direction");
if (target_dir_it != headers.end()) {
auto target_dir = target_dir_it->second; auto target_dir = target_dir_it->second;
if (target_dir == "en-ru") if (target_dir == "en-ru")
m_doc.direction(PromtCtlDocument::Direction::kEngRus); m_doc.direction(PromtCtlDocument::Direction::kEngRus);
@ -74,40 +90,39 @@ class WebServer {
res.set_content("X-Translation-Direction must be one of: [\"en-ru\", \"ru-en\"]", res.set_content("X-Translation-Direction must be one of: [\"en-ru\", \"ru-en\"]",
"text/plain"); "text/plain");
res.status = 400; res.status = 400;
return; return false;
} }
} else } else
m_doc.direction(PromtCtlDocument::Direction::kEngRus); m_doc.direction(PromtCtlDocument::Direction::kEngRus);
auto content_type_it = req.headers.find("content-type"); return true;
if (content_type_it == req.headers.end()) { }
res.set_content("Missing Content-Type header", "text/plain");
void TranslateHandler(const httplib::Request &req, httplib::Response &res) {
const std::lock_guard lock(m_global_lock);
if (!SetDirection(req.headers, res))
return;
const auto &body = req.body;
if (body.empty()) {
res.set_content("Request body must not be empty", "text/plain");
res.status = 400; res.status = 400;
return; return;
} }
const auto content_type_it = req.headers.find("content-type");
if (content_type_it == req.headers.end()) {
TranslateText(body, res);
return;
}
auto content_type = content_type_it->second; auto content_type = content_type_it->second;
PromtFTManager::FileType ft;
if (content_type == "text/html") if (content_type == "text/html")
ft = PromtFTManager::FileType::kHTML; TranslateHtml(body, res);
else { else
res.set_content("Can't translate this content type", "text/plain"); TranslateText(body, res);
res.status = 400;
return;
}
auto direction = m_doc.direction();
auto translator = m_ftman.Translator(ft, direction);
auto hash = std::to_string(std::hash<std::string>{}(req.body));
{
std::ofstream ofs(TMP_IN);
ofs << req.body;
}
translator.Translate(TMP_IN, TMP_OUT);
res.set_content(read_file(TMP_OUT), content_type);
} }
public: public: