2024-01-02 03:45:06 +01:00
|
|
|
/*****************************************************************************
|
|
|
|
* dtranslatebot Discord Translate Bot
|
|
|
|
* Copyright (C) 2023-2024 Syping
|
|
|
|
*
|
|
|
|
* Redistribution and use in source and binary forms, with or without modification,
|
|
|
|
* are permitted provided that the following conditions are met:
|
|
|
|
*
|
|
|
|
* 1. Redistributions of source code must retain the above copyright notice,
|
|
|
|
* this list of conditions and the following disclaimer.
|
|
|
|
*
|
|
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
|
|
* and/or other materials provided with the distribution.
|
|
|
|
*
|
|
|
|
* This software is provided as-is, no warranties are given to you, we are not
|
|
|
|
* responsible for anything with use of the software, you are self responsible.
|
|
|
|
*****************************************************************************/
|
|
|
|
|
|
|
|
#include <iostream>
|
|
|
|
#include <thread>
|
2024-01-11 21:49:01 +01:00
|
|
|
#include "message_queue.h"
|
2024-01-02 03:45:06 +01:00
|
|
|
#include "settings.h"
|
|
|
|
using namespace std::chrono_literals;
|
|
|
|
|
2024-01-11 21:49:01 +01:00
|
|
|
void bot::message_queue::add(const bot::message &message)
|
2024-01-02 03:45:06 +01:00
|
|
|
{
|
|
|
|
m_mutex.lock();
|
|
|
|
m_queue.push_back(message);
|
|
|
|
m_mutex.unlock();
|
|
|
|
}
|
|
|
|
|
2024-01-11 21:49:01 +01:00
|
|
|
void bot::message_queue::run(dpp::cluster *bot, bot::settings::settings *settings)
|
2024-01-02 03:45:06 +01:00
|
|
|
{
|
|
|
|
while (true) {
|
|
|
|
m_mutex.lock();
|
|
|
|
if (!m_queue.empty()) {
|
|
|
|
const bot::message message = m_queue.front();
|
|
|
|
m_queue.erase(m_queue.begin());
|
|
|
|
m_mutex.unlock();
|
|
|
|
|
|
|
|
settings->lock();
|
|
|
|
bot::settings::translate *translate = settings->get_translate();
|
|
|
|
|
|
|
|
dpp::json body = {
|
|
|
|
{"q", message.message},
|
|
|
|
{"source", message.source},
|
|
|
|
{"target", message.target},
|
|
|
|
{"format", "text"},
|
|
|
|
};
|
|
|
|
|
|
|
|
if (!translate->apiKey.empty())
|
|
|
|
body.emplace("apiKey", translate->apiKey);
|
|
|
|
|
|
|
|
const std::string tr_hostname = translate->hostname;
|
|
|
|
const uint16_t tr_port = translate->port;
|
|
|
|
const std::string tr_url = translate->url;
|
|
|
|
const bool tr_tls = translate->tls;
|
|
|
|
settings->unlock();
|
|
|
|
|
|
|
|
dpp::http_headers http_headers;
|
|
|
|
http_headers.emplace("Content-Type", "application/json");
|
|
|
|
|
|
|
|
std::string tr_message = message.message;
|
|
|
|
dpp::https_client http_request(tr_hostname, tr_port, tr_url, "POST", body.dump(), http_headers, !tr_tls);
|
|
|
|
if (http_request.get_status() == 200) {
|
|
|
|
dpp::json response = dpp::json::parse(http_request.get_content());
|
|
|
|
if (response.is_object()) {
|
|
|
|
auto tr_text = response.find("translatedText");
|
|
|
|
if (tr_text != response.end())
|
|
|
|
tr_message = tr_text.value();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
dpp::webhook webhook(message.webhook);
|
|
|
|
webhook.name = message.author;
|
|
|
|
|
|
|
|
try {
|
|
|
|
bot->execute_webhook_sync(webhook, dpp::message(tr_message));
|
|
|
|
}
|
|
|
|
catch (dpp::rest_exception exception) {
|
|
|
|
std::cerr << "REST Error: " << exception.what() << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::this_thread::yield();
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
m_mutex.unlock();
|
|
|
|
std::this_thread::sleep_for(100ms);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|