Syping/dtranslatebot
1
0
Fork 0
mirror of https://github.com/Syping/dtranslatebot.git synced 2026-04-01 13:20:40 +02:00
Code Issues Projects Releases Packages Wiki Activity

add experimental deepl support

Browse source
This commit is contained in:
Syping 2024-03-23 16:57:54 +01:00
parent 42755811aa
commit 0e369f5a1d
6 changed files with 169 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/core/settings.cpp
View file

@ -21,6 +21,7 @@
#include <iostream>
#include "settings.h"
#include "../database/file/file.h"
#include "../translator/deepl/deepl.h"
#include "../translator/libretranslate/libretranslate.h"
#include "../translator/stub/stub.h"
using namespace bot::settings;
@ -233,7 +234,23 @@ bool process_translator_settings(const dpp::json &json, translator &translator)
std::transform(translator_type.begin(), translator_type.end(), translator_type.begin(), ::tolower);
}
if (translator_type == "libretranslate") {
if (translator_type == "deepl") {
translator.type = TRANSLATOR_DEEPL;
auto json_deepl_hostname = json.find("hostname");
if (json_deepl_hostname != json.end())
translator.hostname = *json_deepl_hostname;
else
translator.hostname = "api-free.deepl.com";
auto json_deepl_apiKey = json.find("apiKey");
if (json_deepl_apiKey == json.end()) {
std::cerr << "[Error] DeepL requires API key for authorization" << std::endl;
return false;
}
translator.apiKey = *json_deepl_apiKey;
}
else if (translator_type == "libretranslate") {
translator.type = TRANSLATOR_LIBRETRANSLATE;
auto json_lt_hostname = json.find("hostname");
@ -459,6 +476,8 @@ std::unique_ptr<bot::translator::translator> settings::get_translator() const
const std::lock_guard<std::recursive_mutex> guard(m_mutex);
switch (m_translator.type) {
case TRANSLATOR_DEEPL:
return std::make_unique<bot::translator::deepl>(m_translator.hostname, m_translator.apiKey);
case TRANSLATOR_LIBRETRANSLATE:
return std::make_unique<bot::translator::libretranslate>(m_translator.hostname, m_translator.port, m_translator.url, m_translator.tls, m_translator.apiKey);
case TRANSLATOR_STUB:

1
src/core/settings_types.h
View file

@ -41,6 +41,7 @@ namespace bot {
std::vector<bot::settings::channel> channel;
};
enum translator_type {
TRANSLATOR_DEEPL,
TRANSLATOR_LIBRETRANSLATE,
TRANSLATOR_STUB
};

4
src/core/slashcommands.cpp
View file

@ -434,9 +434,9 @@ void slashcommands::register_commands(dpp::cluster *bot, bot::settings::settings
std::vector<dpp::slashcommand> commands;
dpp::command_option source_option(dpp::co_string, "source", "Source language (ISO 639-1)", true);
source_option.set_max_length(static_cast<int64_t>(2)).set_min_length(static_cast<int64_t>(2));
source_option.set_max_length(static_cast<int64_t>(5)).set_min_length(static_cast<int64_t>(2));
dpp::command_option target_option(dpp::co_string, "target", "Target language (ISO 639-1)", true);
target_option.set_max_length(static_cast<int64_t>(2)).set_min_length(static_cast<int64_t>(2));
target_option.set_max_length(static_cast<int64_t>(5)).set_min_length(static_cast<int64_t>(2));
dpp::slashcommand command_edit("edit", "Edit current channel settings", bot->me.id);
command_edit.set_default_permissions(dpp::p_manage_webhooks);