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

chroe: remove cache file and print statements

Browse source
This commit is contained in:
Arthur Khachaturov 2024-11-26 15:11:29 +03:00
parent 94f58c1857
commit 495009e270
No known key found for this signature in database
GPG key ID: CAC2B7EB6DF45D55
3 changed files with 21 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

4
docker-compose.yml
View file

@ -2,8 +2,10 @@ services:
puppies:
build: puppy
ports:
- "2390:80"
- "127.0.0.1:2000-2030:80"
stop_signal: SIGINT
volumes:
- '/dev/shm/mgx-temp:/tmpfs'
deploy:
mode: replicated
replicas: 30

1
puppy/Dockerfile
View file

@ -24,7 +24,6 @@ RUN xvfb-run -a winetricks -q mfc42 && \
wine regedit $WINEPREFIX/drive_c/registry.reg && \
apt remove -y --autoremove --purge xvfb winetricks && \
apt-get clean && rm -rf /var/lib/apt/lists/* && \
ln -sf /cache $WINEPREFIX/drive_c/cache && \
mkdir /tmpfs && ln -sf /tmpfs $WINEPREFIX/drive_c/tmpfs && \
echo "Sleeping for 15 seconds" && sleep 15

51
puppy/app/main.cpp
View file

@ -4,6 +4,7 @@
#include <fstream>
#include <functional>
#include <mutex>
#include <csignal>
#include <string>
#include <windows.h>
#include <wingdi.h>
@ -11,14 +12,6 @@
#include "PromtCtlDocument.hpp"
#include "PromtFTManager.hpp"
// TODO: remove
template<typename T>
void print(T arg) {
std::cout << arg << std::endl;
}
static const char CACHE_FOLDER[] = "C:\\cache\\";
static const char TMP_FOLDER[] = "C:\\tmpfs\\";
static inline std::string random_filename(int len = 65) {
static const char ASCII_PRINTABLE[] = {'1', '2', '3', '4', '5', '6', '7', '8', '9', '0', 'A', 'B', 'C', 'D', 'E', 'F',
@ -31,9 +24,13 @@ static inline std::string random_filename(int len = 65) {
std::string random_string;
for (std::size_t i = 0; i < len; ++i)
random_string += ASCII_PRINTABLE[distribution(generator)];
return TMP_FOLDER + random_string;
return random_string;
}
static const char TMP_FOLDER[] = "C:\\tmpfs\\";
static const auto TMP_IN = TMP_FOLDER + random_filename();
static const auto TMP_OUT = TMP_FOLDER + random_filename();
// https://www.linux.org.ru/forum/development/3968525?cid=3971135
static inline size_t cp1251_to_utf8(const char *in, char *out = nullptr) {
static const int table[128] = {
@ -104,7 +101,6 @@ class WebServer {
private:
void TranslateHandler(const httplib::Request &req, httplib::Response &res) {
std::lock_guard _(m_global_lock);
print("got request");
auto target_dir_it = req.headers.find("x-translation-direction");
if (target_dir_it != req.headers.end()) {
@ -144,46 +140,35 @@ class WebServer {
auto translator = m_ftman.Translator(ft, direction);
auto hash = std::to_string(std::hash<std::string>{}(req.body));
auto translated_file = CACHE_FOLDER + hash;
print(CACHE_FOLDER);
print(hash);
print(translated_file);
if (std::filesystem::exists(translated_file))
std::remove(translated_file.c_str());
if (!std::filesystem::exists(translated_file)) {
print("not found");
auto src = random_filename();
auto dest = random_filename();
{
std::ofstream ofs(src);
ofs << req.body;
}
translator.Translate(src, dest);
print(dest);
convert_file(dest);
print("file converted");
std::filesystem::copy(dest, translated_file);
std::remove(src.c_str());
std::remove(dest.c_str());
{
std::ofstream ofs(TMP_IN);
ofs << req.body;
}
res.set_file_content(translated_file, content_type);
translator.Translate(TMP_IN, TMP_OUT);
convert_file(TMP_OUT);
res.set_file_content(TMP_OUT, content_type);
}
public:
WebServer() {
if (!std::filesystem::exists(TMP_FOLDER)) std::filesystem::create_directory(TMP_FOLDER);
if (!std::filesystem::exists(CACHE_FOLDER)) std::filesystem::create_directory(CACHE_FOLDER);
m_svr.Post("/translate", [this](const httplib::Request &req, httplib::Response &res) { TranslateHandler(req, res); });
}
~WebServer() {
stop();
}
void listen() {
m_svr.listen("0.0.0.0", 80);
}
void stop() {
m_svr.stop();
std::remove(TMP_IN.c_str());
std::remove(TMP_OUT.c_str());
}
};