mirror of
https://github.com/Syping/dtranslatebot.git
synced 2025-12-12 22:56:25 +01:00
make code more efficient and added storage setting
This commit is contained in:
Syping
parent
3800da9b00
commit
25d83b243d
10 changed files with 264 additions and 186 deletions
177
src/slashcommands.cpp
Normal file
177
src/slashcommands.cpp
Normal file
|
|
@ -0,0 +1,177 @@
|
|||
/*****************************************************************************
|
||||
* 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 "slashcommands.h"
|
||||
|
||||
void bot::slashcommands::process_translate_command(dpp::cluster *bot, bot::settings::settings *settings, const dpp::slashcommand_t &event)
|
||||
{
|
||||
try {
|
||||
std::variant<dpp::channel,dpp::webhook> v_target;
|
||||
const std::string source = std::get<std::string>(event.get_parameter("source"));
|
||||
const std::string target = std::get<std::string>(event.get_parameter("target"));
|
||||
|
||||
dpp::command_interaction interaction = event.command.get_command_interaction();
|
||||
if (interaction.options[0].name == "channel") {
|
||||
v_target = event.command.get_resolved_channel(
|
||||
std::get<dpp::snowflake>(event.get_parameter("channel")));
|
||||
}
|
||||
else if (interaction.options[0].name == "webhook") {
|
||||
v_target = dpp::webhook(std::get<std::string>(event.get_parameter("webhook")));
|
||||
}
|
||||
|
||||
const std::vector<bot::translate::language> languages = settings->get_translator()->get_languages();
|
||||
|
||||
std::ostringstream language_codes;
|
||||
bool source_valid = false, target_valid = false;
|
||||
for (const bot::translate::language &language : languages) {
|
||||
if (language.code == source)
|
||||
source_valid = true;
|
||||
if (language.code == target)
|
||||
target_valid = true;
|
||||
if (source_valid && target_valid)
|
||||
break;
|
||||
language_codes << " " << language.code;
|
||||
}
|
||||
|
||||
if (source_valid && target_valid) {
|
||||
const std::lock_guard<bot::settings::settings> guard(*settings);
|
||||
if (!settings->get_channel(event.command.guild_id, event.command.channel_id)) {
|
||||
if (dpp::channel *channel = std::get_if<dpp::channel>(&v_target)) {
|
||||
dpp::webhook webhook;
|
||||
webhook.channel_id = channel->id;
|
||||
webhook.guild_id = channel->guild_id;
|
||||
webhook.name = "Translate Bot Webhook <" + std::to_string(event.command.channel_id) + ":" + source + ":" + target + ">";
|
||||
|
||||
bot->create_webhook(webhook, [&bot, &settings, event, source, target](const dpp::confirmation_callback_t &callback) {
|
||||
if (callback.is_error()) {
|
||||
event.reply(dpp::message("Failed to generate webhook!\n" + callback.http_info.body).set_flags(dpp::m_ephemeral));
|
||||
return;
|
||||
}
|
||||
const dpp::webhook webhook = callback.get<dpp::webhook>();
|
||||
|
||||
bot::settings::channel s_channel;
|
||||
s_channel.id = event.command.channel_id;
|
||||
s_channel.source = source;
|
||||
|
||||
bot::settings::target s_target;
|
||||
s_target.target = target;
|
||||
s_target.webhook = webhook;
|
||||
s_channel.targets.push_back(std::move(s_target));
|
||||
|
||||
settings->lock();
|
||||
settings->add_channel(s_channel, event.command.guild_id);
|
||||
settings->add_translatebot_webhook(webhook.id);
|
||||
settings->unlock();
|
||||
|
||||
event.reply(dpp::message("Channel will be now translated!").set_flags(dpp::m_ephemeral));
|
||||
});
|
||||
}
|
||||
else if (dpp::webhook *webhook = std::get_if<dpp::webhook>(&v_target)) {
|
||||
bot::settings::channel s_channel;
|
||||
s_channel.id = event.command.channel_id;
|
||||
s_channel.source = source;
|
||||
|
||||
bot::settings::target s_target;
|
||||
s_target.target = target;
|
||||
s_target.webhook = *webhook;
|
||||
s_channel.targets.push_back(std::move(s_target));
|
||||
|
||||
settings->lock();
|
||||
settings->add_channel(s_channel, event.command.guild_id);
|
||||
settings->add_translatebot_webhook(webhook->id);
|
||||
settings->unlock();
|
||||
|
||||
event.reply(dpp::message("Channel will be now translated!").set_flags(dpp::m_ephemeral));
|
||||
}
|
||||
}
|
||||
else {
|
||||
event.reply(dpp::message("The current channel is already being translated!").set_flags(dpp::m_ephemeral));
|
||||
}
|
||||
}
|
||||
else if (!source_valid && !target_valid) {
|
||||
event.reply(dpp::message("Source and target languages are not valid!\nAvailable languages are:" + language_codes.str()).set_flags(dpp::m_ephemeral));
|
||||
}
|
||||
else if (!source_valid) {
|
||||
event.reply(dpp::message("Source language is not valid!\nAvailable languages are:" + language_codes.str()).set_flags(dpp::m_ephemeral));
|
||||
}
|
||||
else if (!target_valid) {
|
||||
event.reply(dpp::message("Target language is not valid!\nAvailable languages are:" + language_codes.str()).set_flags(dpp::m_ephemeral));
|
||||
}
|
||||
}
|
||||
catch (const std::exception &exception) {
|
||||
std::cerr << "Failed to process command /" << event.command.get_command_name() << ": " << exception.what() << std::endl;
|
||||
event.reply(dpp::message("Failed to process command /" + event.command.get_command_name() + "\n" + exception.what()).set_flags(dpp::m_ephemeral));
|
||||
}
|
||||
}
|
||||
|
||||
void bot::slashcommands::register_commands(dpp::cluster *bot, bot::settings::settings *settings)
|
||||
{
|
||||
settings->lock();
|
||||
const std::vector<bot::translate::language> languages = settings->get_translator()->get_languages();
|
||||
const std::vector<std::string> preferred_languages = settings->get_preferred_languages();
|
||||
settings->unlock();
|
||||
|
||||
std::vector<dpp::slashcommand> commands;
|
||||
|
||||
dpp::slashcommand command_translate("translate", "Translate current channel (ISO 639-1)", bot->me.id);
|
||||
command_translate.set_default_permissions(dpp::p_manage_webhooks);
|
||||
dpp::command_option channel_subcommand(dpp::co_sub_command, "channel", "Translate current channel to a channel (ISO 639-1)");
|
||||
dpp::command_option webhook_subcommand(dpp::co_sub_command, "webhook", "Translate current channel to a webhook (ISO 639-1)");
|
||||
dpp::command_option source_option(dpp::co_string, "source", "Source language", true);
|
||||
source_option.set_max_length(2).set_min_length(2);
|
||||
dpp::command_option target_option(dpp::co_string, "target", "Target language", true);
|
||||
target_option.set_max_length(2).set_min_length(2);
|
||||
dpp::command_option channel_option(dpp::co_channel, "channel", "Target channel", true);
|
||||
channel_option.add_channel_type(dpp::CHANNEL_TEXT);
|
||||
dpp::command_option webhook_option(dpp::co_string, "webhook", "Target webhook", true);
|
||||
channel_subcommand.add_option(source_option);
|
||||
channel_subcommand.add_option(target_option);
|
||||
channel_subcommand.add_option(channel_option);
|
||||
webhook_subcommand.add_option(source_option);
|
||||
webhook_subcommand.add_option(target_option);
|
||||
webhook_subcommand.add_option(webhook_option);
|
||||
command_translate.add_option(channel_subcommand);
|
||||
command_translate.add_option(webhook_subcommand);
|
||||
commands.push_back(std::move(command_translate));
|
||||
|
||||
if (preferred_languages.size() > 1) {
|
||||
dpp::slashcommand command_translate_pref("translate_pref", "Translate current channel (Preferred languages)", bot->me.id);
|
||||
command_translate_pref.set_default_permissions(dpp::p_manage_webhooks);
|
||||
dpp::command_option channel_pref_subcommand(dpp::co_sub_command, "channel", "Translate current channel to a channel (Preferred languages)");
|
||||
dpp::command_option webhook_pref_subcommand(dpp::co_sub_command, "webhook", "Translate current channel to a webhook (Preferred languages)");
|
||||
dpp::command_option source_pref_option(dpp::co_string, "source", "Source language", true);
|
||||
dpp::command_option target_pref_option(dpp::co_string, "target", "Target language", true);
|
||||
for (const bot::translate::language &language : languages) {
|
||||
if (std::find(preferred_languages.begin(), preferred_languages.end(), language.code) != preferred_languages.end()) {
|
||||
source_pref_option.add_choice(dpp::command_option_choice(language.name, language.code));
|
||||
target_pref_option.add_choice(dpp::command_option_choice(language.name, language.code));
|
||||
}
|
||||
}
|
||||
channel_pref_subcommand.add_option(source_pref_option);
|
||||
channel_pref_subcommand.add_option(target_pref_option);
|
||||
channel_pref_subcommand.add_option(channel_option);
|
||||
webhook_pref_subcommand.add_option(source_pref_option);
|
||||
webhook_pref_subcommand.add_option(target_pref_option);
|
||||
webhook_pref_subcommand.add_option(webhook_option);
|
||||
command_translate_pref.add_option(channel_pref_subcommand);
|
||||
command_translate_pref.add_option(webhook_pref_subcommand);
|
||||
commands.push_back(std::move(command_translate_pref));
|
||||
}
|
||||
|
||||
bot->global_bulk_command_create(commands);
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue