mirror of
https://github.com/Syping/dtranslatebot.git
synced 2024-10-07 00:26:58 +02:00
add experimental deepl support
This commit is contained in:
parent
42755811aa
commit
0e369f5a1d
6 changed files with 169 additions and 3 deletions
|
@ -33,6 +33,7 @@ set(DTRANSLATEBOT_HEADERS
|
|||
src/core/translator.h
|
||||
src/core/webhook_push.h
|
||||
src/database/file/file.h
|
||||
src/translator/deepl/deepl.h
|
||||
src/translator/libretranslate/libretranslate.h
|
||||
src/translator/stub/stub.h
|
||||
)
|
||||
|
@ -46,6 +47,7 @@ set(DTRANSLATEBOT_SOURCES
|
|||
src/core/translator.cpp
|
||||
src/core/webhook_push.cpp
|
||||
src/database/file/file.cpp
|
||||
src/translator/deepl/deepl.cpp
|
||||
src/translator/libretranslate/libretranslate.cpp
|
||||
src/translator/stub/stub.cpp
|
||||
)
|
||||
|
|
|
@ -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:
|
||||
|
|
|
@ -41,6 +41,7 @@ namespace bot {
|
|||
std::vector<bot::settings::channel> channel;
|
||||
};
|
||||
enum translator_type {
|
||||
TRANSLATOR_DEEPL,
|
||||
TRANSLATOR_LIBRETRANSLATE,
|
||||
TRANSLATOR_STUB
|
||||
};
|
||||
|
|
|
@ -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);
|
||||
|
|
104
src/translator/deepl/deepl.cpp
Normal file
104
src/translator/deepl/deepl.cpp
Normal file
|
@ -0,0 +1,104 @@
|
|||
/*****************************************************************************
|
||||
* dtranslatebot Discord Translate Bot
|
||||
* Copyright (C) 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 <dpp/json.h>
|
||||
#include <dpp/httpsclient.h>
|
||||
#include "deepl.h"
|
||||
using namespace bot::translator;
|
||||
using namespace std::string_literals;
|
||||
|
||||
deepl::deepl(const std::string &hostname, const std::string apiKey) :
|
||||
m_hostname(hostname), m_apiKey(apiKey)
|
||||
{
|
||||
}
|
||||
|
||||
deepl::~deepl()
|
||||
{
|
||||
}
|
||||
|
||||
const std::vector<language> deepl::get_languages()
|
||||
{
|
||||
std::vector<language> languages;
|
||||
|
||||
try {
|
||||
dpp::https_client http_request(m_hostname, 443, "/v2/languages?type=target", "GET", {}, { {"Authorization"s, "DeepL-Auth-Key " + m_apiKey} }, false);
|
||||
if (http_request.get_status() == 200) {
|
||||
const dpp::json response = dpp::json::parse(http_request.get_content());
|
||||
if (response.is_array()) {
|
||||
for (const auto &json_language : response) {
|
||||
if (json_language.is_object()) {
|
||||
language language;
|
||||
|
||||
auto json_lang_code = json_language.find("language");
|
||||
if (json_lang_code != json_language.end())
|
||||
language.code = *json_lang_code;
|
||||
|
||||
std::transform(language.code.begin(), language.code.end(), language.code.begin(), ::tolower);
|
||||
|
||||
auto json_lang_name = json_language.find("name");
|
||||
if (json_lang_name != json_language.end())
|
||||
language.name = *json_lang_name;
|
||||
|
||||
if (!language.code.empty() && !language.name.empty())
|
||||
languages.push_back(std::move(language));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (const std::exception &exception) {
|
||||
std::cerr << "[Exception] " << exception.what() << std::endl;
|
||||
}
|
||||
|
||||
return languages;
|
||||
}
|
||||
|
||||
const std::string deepl::translate(const std::string &text, const std::string &source, const std::string &target)
|
||||
{
|
||||
const dpp::http_headers http_headers = {
|
||||
{"Authorization"s, "DeepL-Auth-Key "s + m_apiKey},
|
||||
{"Content-Type"s, "application/json"s}
|
||||
};
|
||||
|
||||
dpp::json json_body = {
|
||||
{"text"s, { text } },
|
||||
{"target_lang"s, target},
|
||||
};
|
||||
|
||||
try {
|
||||
dpp::https_client http_request(m_hostname, 443, "/v2/translate", "POST", json_body.dump(), http_headers, false);
|
||||
if (http_request.get_status() == 200) {
|
||||
const dpp::json response = dpp::json::parse(http_request.get_content());
|
||||
if (response.is_object()) {
|
||||
auto translations = response.find("translations");
|
||||
if (translations != response.end() && translations->is_array()) {
|
||||
for (auto translation = translations->begin(); translation != translations->end(); translation++) {
|
||||
auto tr_text = translation->find("text");
|
||||
if (tr_text != translation->end())
|
||||
return *tr_text;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (const std::exception &exception) {
|
||||
std::cerr << "[Exception] " << exception.what() << std::endl;
|
||||
}
|
||||
|
||||
return text;
|
||||
}
|
40
src/translator/deepl/deepl.h
Normal file
40
src/translator/deepl/deepl.h
Normal file
|
@ -0,0 +1,40 @@
|
|||
/*****************************************************************************
|
||||
* dtranslatebot Discord Translate Bot
|
||||
* Copyright (C) 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.
|
||||
*****************************************************************************/
|
||||
|
||||
#ifndef TRANSLATOR_DEEPL_H
|
||||
#define TRANSLATOR_DEEPL_H
|
||||
|
||||
#include "../../core/translator.h"
|
||||
|
||||
namespace bot {
|
||||
namespace translator {
|
||||
class deepl : public translator {
|
||||
public:
|
||||
explicit deepl(const std::string &hostname, const std::string apiKey = {});
|
||||
~deepl() override;
|
||||
const std::vector<language> get_languages() override;
|
||||
const std::string translate(const std::string &text, const std::string &source, const std::string &target) override;
|
||||
|
||||
private:
|
||||
std::string m_apiKey;
|
||||
std::string m_hostname;
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
#endif // TRANSLATOR_DEEPL_H
|
Loading…
Reference in a new issue