add direct message support

src/core/http_request.cpp

@@ -17,6 +17,7 @@
 *****************************************************************************/
 
 #include "http_request.h"
+using namespace std::string_literals;
 
 http_request::http_request() {
     instance = curl_easy_init();
@@ -55,7 +56,7 @@ const http_response http_request::post(const std::string &url, const std::string
     curl_easy_setopt(instance, CURLOPT_URL, url.c_str());
     curl_slist *header_slist = nullptr;
     if (!content_type.empty()) {
-        curl_slist *new_header_slist = curl_slist_append(header_slist, std::string("Content-Type: " + content_type).c_str());
+        curl_slist *new_header_slist = curl_slist_append(header_slist, std::string("Content-Type: "s + content_type).c_str());
         if (!new_header_slist) {
             curl_slist_free_all(header_slist);
             throw std::bad_alloc();
@@ -87,7 +88,7 @@ const http_response http_request::post(const std::string &url, const std::string
 }
 
 std::string http_request::legacy_url(const std::string &hostname, uint16_t port, const std::string &url, bool tls) {
-    return (tls ? "https://" : "http://") + hostname + ":" + std::to_string(port) + (url.empty() ? "/" : (url.front() != '/' ? "/" + url : url));
+    return (tls ? "https://"s : "http://"s) + hostname + ":"s + std::to_string(port) + (url.empty() ? "/"s : (url.front() != '/' ? "/"s + url : url));
 }
 
 http_request::~http_request() {

src/core/main.cpp

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2023-2024 Syping
+* Copyright (C) 2023-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -77,7 +77,8 @@ int main(int argc, char* argv[]) {
     bot::message_queue message_queue;
     std::thread message_queue_loop(&bot::message_queue::run, &message_queue, &settings, &submit_queue);
 
-    bot.on_message_create(std::bind(&bot::message_queue::process_message_event, &message_queue, &bot, &settings, std::placeholders::_1));
+    bot.on_message_context_menu(std::bind(&bot::slashcommands::process_message_menu_event, &message_queue, &bot, &settings, std::placeholders::_1));
+    bot.on_message_create(std::bind(&bot::message_queue::process_guild_message_event, &message_queue, &bot, &settings, std::placeholders::_1));
     bot.on_slashcommand(std::bind(&bot::slashcommands::process_command_event, &bot, &settings, std::placeholders::_1));
     bot.on_ready([&bot, &settings]([[maybe_unused]] const dpp::ready_t &event) {
         if (dpp::run_once<struct register_bot_commands>()) {

src/core/message_queue.cpp

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2023-2024 Syping
+* Copyright (C) 2023-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -21,6 +21,7 @@
 #include "settings.h"
 using bot::message_queue;
 using namespace std::chrono_literals;
+using namespace std::string_literals;
 
 void message_queue::add(const message &message)
 {
@@ -34,7 +35,30 @@ void message_queue::add(message &&message)
     m_queue.push(message);
 }
 
-void message_queue::process_message_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_create_t &event)
+void message_queue::process_direct_message_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_context_menu_t &event)
+{
+    try {
+        // We check for conditions we want to skip translation for
+        if (event.ctx_message.author.id == bot->me.id || event.ctx_message.content.empty()) {
+            event.reply(dpp::message("Invalid message").set_flags(dpp::m_ephemeral));
+            return;
+        }
+
+        event.thinking(false);
+
+        bot::direct_message direct_message;
+        direct_message.event = event;
+        direct_message.message = event.ctx_message.content;
+
+        add(std::move(direct_message));
+    }
+    catch (const std::exception &exception) {
+        std::cerr << "[Exception] " << exception.what() << std::endl;
+        event.reply(dpp::message("Exception while processing command:\n"s + exception.what()).set_flags(dpp::m_ephemeral));
+    }
+}
+
+void message_queue::process_guild_message_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_create_t &event)
 {
     // We check for conditions we want to skip translation for
     if (event.msg.author.id == bot->me.id || event.msg.content.empty() || event.msg.has_thread())
@@ -50,7 +74,7 @@ void message_queue::process_message_event(dpp::cluster *bot, bot::settings::sett
 
     const std::lock_guard<bot::settings::settings> guard(*settings);
     if (const bot::settings::channel *channel = settings->get_channel(event.msg.guild_id, event.msg.channel_id)) {
-        bot::message message;
+        bot::guild_message message;
         message.id = event.msg.id;
 
         message.author = event.msg.member.get_nickname();
@@ -81,13 +105,21 @@ void message_queue::run(bot::settings::settings *settings, submit_queue *submit_
 
             auto translator = settings->get_translator();
 
-            for (auto target = message.targets.begin(); target != message.targets.end(); target++) {
-                translated_message tr_message;
-                tr_message.author = message.author;
-                tr_message.avatar = message.avatar;
-                tr_message.message = translator->translate(message.message, message.source, target->target);
-                tr_message.webhook = target->webhook;
-                submit_queue->add(std::move(tr_message));
+            if (const auto *direct_message = std::get_if<bot::direct_message>(&message)) {
+                translated_direct_message translated_message;
+                translated_message.event = direct_message->event;
+                translated_message.message = translator->translate(direct_message->message, {}, "en");
+                submit_queue->add(std::move(translated_message));
+            }
+            else if (const auto *guild_message = std::get_if<bot::guild_message>(&message)) {
+                for (auto target = guild_message->targets.begin(); target != guild_message->targets.end(); target++) {
+                    translated_guild_message translated_message;
+                    translated_message.author = guild_message->author;
+                    translated_message.avatar = guild_message->avatar;
+                    translated_message.message = translator->translate(guild_message->message, guild_message->source, target->target);
+                    translated_message.webhook = target->webhook;
+                    submit_queue->add(std::move(translated_message));
+                }
             }
 
             std::this_thread::yield();

src/core/message_queue.h

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2023-2024 Syping
+* Copyright (C) 2023-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -22,12 +22,18 @@
 #include <mutex>
 #include <string>
 #include <queue>
+#include <variant>
 #include <vector>
 #include "settings.h"
 #include "submit_queue.h"
 
 namespace bot {
-    struct message {
+    struct direct_message {
+        dpp::message_context_menu_t event;
+        std::string message;
+    };
+
+    struct guild_message {
         uint64_t id;
         std::string author;
         std::string avatar;
@@ -36,11 +42,14 @@ namespace bot {
         std::vector<bot::settings::target> targets;
     };
 
+    typedef std::variant<direct_message, guild_message> message;
+
     class message_queue {
     public:
         void add(const message &message);
         void add(message &&message);
-        void process_message_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_create_t &event);
+        void process_direct_message_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_context_menu_t &event);
+        void process_guild_message_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_create_t &event);
         void run(bot::settings::settings *settings, submit_queue *submit_queue);
         void terminate();
 

src/core/slashcommands.cpp

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2024 Syping
+* Copyright (C) 2024-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -27,10 +27,16 @@ void slashcommands::process_command_event(dpp::cluster *bot, bot::settings::sett
         slashcommands::process_edit_command(bot, settings, event);
     else if (event.command.get_command_name() == "list")
         slashcommands::process_list_command(bot, settings, event);
-    else if (event.command.get_command_name() == "translate" || event.command.get_command_name() == "translate_pref")
+    else if (event.command.get_command_name() == "translate" || event.command.get_command_name() == "translate pref")
         slashcommands::process_translate_command(bot, settings, event);
 }
 
+void slashcommands::process_message_menu_event(bot::message_queue *message_queue, dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_context_menu_t &event)
+{
+    if (event.command.get_command_name() == "translate message")
+        message_queue->process_direct_message_event(bot, settings, event);
+}
+
 void slashcommands::process_edit_command(dpp::cluster *bot, bot::settings::settings *settings, const dpp::slashcommand_t &event)
 {
     try {
@@ -475,7 +481,7 @@ void slashcommands::register_commands(dpp::cluster *bot, bot::settings::settings
     commands.push_back(command_translate);
 
     if (preferred_languages.size() > 1) {
-        dpp::slashcommand command_translate_pref("translate_pref", "Translate current channel (Preferred languages)", bot->me.id);
+        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)");
@@ -498,5 +504,9 @@ void slashcommands::register_commands(dpp::cluster *bot, bot::settings::settings
         commands.push_back(command_translate_pref);
     }
 
+    dpp::slashcommand command_translate_message("translate message", dpp::ctxm_message, bot->me.id);
+    command_translate_message.set_dm_permission(true);
+    commands.push_back(command_translate_message);
+
     bot->global_bulk_command_create(commands);
 }

src/core/slashcommands.h

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2024 Syping
+* Copyright (C) 2024-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -20,6 +20,7 @@
 #define SLASHCOMMANDS_H
 
 #include <dpp/cluster.h>
+#include "message_queue.h"
 #include "settings.h"
 
 namespace bot {
@@ -27,6 +28,7 @@ namespace bot {
     public:
         slashcommands() = delete;
         static void process_command_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::slashcommand_t &event);
+        static void process_message_menu_event(bot::message_queue *message_queue, dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_context_menu_t &event);
         static void register_commands(dpp::cluster *bot, bot::settings::settings *settings);
 
     private:

src/core/submit_queue.cpp

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2024 Syping
+* Copyright (C) 2024-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -44,7 +44,12 @@ void submit_queue::run(dpp::cluster *bot)
             m_queue.pop();
             m_mutex.unlock();
 
-            webhook_push::run(message, bot);
+            if (const auto *direct_message = std::get_if<bot::translated_direct_message>(&message)) {
+                direct_message->event.edit_original_response(dpp::message(direct_message->message));
+            }
+            else if (const auto *guild_message = std::get_if<bot::translated_guild_message>(&message)) {
+                webhook_push::run(*guild_message, bot);
+            }
 
             std::this_thread::yield();
         }

src/core/submit_queue.h

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2024 Syping
+* Copyright (C) 2024-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -25,13 +25,20 @@
 #include <queue>
 
 namespace bot {
-    struct translated_message {
+    struct translated_direct_message {
+        dpp::message_context_menu_t event;
+        std::string message;
+    };
+
+    struct translated_guild_message {
         std::string author;
         std::string avatar;
         std::string message;
         dpp::webhook webhook;
     };
 
+    typedef std::variant<translated_direct_message, translated_guild_message> translated_message;
+
     class submit_queue {
     public:
         void add(const translated_message &message);

src/core/webhook_push.cpp

@@ -1,6 +1,6 @@
 /*****************************************************************************
 * dtranslatebot Discord Translate Bot
-* Copyright (C) 2024 Syping
+* Copyright (C) 2024-2026 Syping
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
@@ -22,7 +22,7 @@
 using namespace std::string_literals;
 using namespace std::string_view_literals;
 
-void bot::webhook_push::run(const bot::translated_message &message, dpp::cluster *bot)
+void bot::webhook_push::run(const bot::translated_guild_message &message, dpp::cluster *bot)
 {
     dpp::json json_body = {
         {"username"s, message.author}

src/core/webhook_push.h

@@ -27,7 +27,7 @@ namespace bot {
     class webhook_push {
     public:
         webhook_push() = delete;
-        static void run(const bot::translated_message &message, dpp::cluster *bot);
+        static void run(const bot::translated_guild_message &message, dpp::cluster *bot);
 
     private:
         static void push_request(dpp::snowflake webhook_id, const std::string &webhook_token, const std::string &json, dpp::cluster *bot);