🔖 Translate command rework

🎯 Choose the target translation language 🈂️ Support more languages ⚙️ Preparation of translation system
Pexilo · Sep 4, 2022 · baa03b4 · baa03b4
1 parent 02cc5c8
commit baa03b4
Show file tree

Hide file tree

Showing 9 changed files with 154 additions and 51 deletions.
diff --git a/src/commands/context-menu/translate-message.js b/src/commands/context-menu/translate-message.js
@@ -1,14 +1,12 @@
 const { Command } = require("sheweny");
-const { languageFlags } = require("../../languageList");
 
 module.exports = class TranslateMessageContextMenuCommand extends Command {
   constructor(client) {
     super(client, {
       name: "Translate",
       type: "CONTEXT_MENU_MESSAGE",
       description: "🔖 Translate a message.",
-      examples:
-        "Use right click on a message -> `Applications` -> Translate.\n(Translate in server selected language)",
+      examples: "Use right click on a message -> `Applications` -> Translate.",
       usage: "https://i.imgur.com/zjLEvcb.png",
       category: "Context-Menu",
     });
@@ -30,38 +28,42 @@ module.exports = class TranslateMessageContextMenuCommand extends Command {
         ),
       });
 
-    const translated = await this.client.Translate(message.content, lang);
+    // what language the user can translate to
+    const translateTo = {
+      en: "🇺🇸",
+      fr: "🇫🇷",
+      de: "🇩🇪",
+      es: "🇪🇸",
+      it: "🇮🇹",
+      pt: "🇵🇹",
+      ru: "🇷🇺",
+      ja: "🇯🇵",
+    };
+
+    // sort languages with the guild language in first
+    const sortedLanguages = {
+      [lang]: translateTo[lang],
+      ...Object.fromEntries(
+        Object.entries(translateTo).filter(([key]) => key !== lang)
+      ),
+    };
+
+    // create the buttons
+    const languagesButtons = Object.entries(sortedLanguages).map(
+      ([key, value]) => {
+        return {
+          customId: `translate_${key}_${message.id}`,
+          style: key === lang ? "PRIMARY" : "SECONDARY",
+          emoji: value,
+        };
+      }
+    );
+
     await interaction.editReply({
-      embeds: [
-        this.client
-          .Embed()
-          .setAuthor({
-            name: message.author.tag,
-            iconURL: message.author.displayAvatarURL({ dynamic: true }),
-          })
-          .addFields(
-            {
-              name:
-                `${
-                  languageFlags[
-                    translated.translations[0].detected_source_language.toLowerCase()
-                  ]
-                } ` +
-                `${await this.client.FastTranslate("Original", lang)}` +
-                ":",
-              value: `${"```"}${message.content}${"```"}`,
-            },
-            {
-              name:
-                `${languageFlags[lang]} ` +
-                `${await this.client.FastTranslate("Translation", lang)}` +
-                ":",
-              value: `${"```"}${translated.translations[0].text}${"```"}`,
-            }
-          )
-          .setFooter({
-            text: `Powered by DeepL.com`,
-          }),
+      content: "`🔖` Select a language to translate this message",
+      components: [
+        this.client.ButtonRow(languagesButtons.slice(0, 5)),
+        this.client.ButtonRow(languagesButtons.slice(5, 10)),
       ],
     });
   }

diff --git a/src/interactions/buttons/translate-buttons.js b/src/interactions/buttons/translate-buttons.js
@@ -0,0 +1,54 @@
+const { Button } = require("sheweny");
+const { deeplLanguages } = require("../../languages/deeplLanguages");
+
+module.exports = class translateButtons extends Button {
+  constructor(client) {
+    super(client, [/translate_.*/]);
+  }
+
+  async execute(button) {
+    if (!(await this.client.Defer(button))) return;
+    const { channel, customId } = button;
+
+    // Get the language, message from the button id
+    const lang = customId.split("_")[1];
+    const message = await channel.messages.fetch(customId.split("_")[2]);
+
+    // Translate the message
+    const translated = await this.client.Translate(message.content, lang);
+
+    return button.editReply({
+      embeds: [
+        this.client
+          .Embed()
+          .setAuthor({
+            name: message.author.tag,
+            iconURL: message.author.displayAvatarURL({ dynamic: true }),
+          })
+          .addFields(
+            {
+              name:
+                `${
+                  deeplLanguages[
+                    translated.translations[0].detected_source_language.toLowerCase()
+                  ]
+                } ` +
+                `${await this.client.FastTranslate("Original", lang)}` +
+                ":",
+              value: `${"```"}${message.content}${"```"}`,
+            },
+            {
+              name:
+                `${deeplLanguages[lang]} ` +
+                `${await this.client.FastTranslate("Translation", lang)}` +
+                ":",
+              value: `${"```"}${translated.translations[0].text}${"```"}`,
+            }
+          )
+          .setFooter({
+            text: `Powered by DeepL.com`,
+          }),
+      ],
+    });
+  }
+};
diff --git a/src/interactions/select-menus/lang-thirdMsg-select.js b/src/interactions/select-menus/lang-thirdMsg-select.js
@@ -1,5 +1,5 @@
 const { SelectMenu } = require("sheweny");
-const { languageFlags } = require("../../languageList");
+const { supportedLanguages } = require("../../languages/supportedLanguages");
 
 module.exports = class LanguageMenu3MsgSelect extends SelectMenu {
   constructor(client) {
@@ -9,7 +9,7 @@ module.exports = class LanguageMenu3MsgSelect extends SelectMenu {
     if (!(await this.client.Defer(selectMenu))) return;
     const { guild } = selectMenu;
 
-    for (const [key, value] of Object.entries(languageFlags)) {
+    for (const [key, value] of Object.entries(supportedLanguages)) {
       if (selectMenu.values[0] === `${key}_option`) {
         this.client.updateGuild(guild, { language: key });
 

diff --git a/src/interactions/select-menus/setupMenu-secondMsg-select.js b/src/interactions/select-menus/setupMenu-secondMsg-select.js
@@ -1,6 +1,6 @@
 const { SelectMenu } = require("sheweny");
 const { ChannelType } = require("discord.js");
-const { languageFlags } = require("../../languageList");
+const { supportedLanguages } = require("../../languages/supportedLanguages");
 
 module.exports = class SetupMenu2MsgSelect extends SelectMenu {
   constructor(client) {
@@ -16,11 +16,11 @@ module.exports = class SetupMenu2MsgSelect extends SelectMenu {
       case "lang_option":
         const languageRow = this.client.SelectMenuRow(
           "language-select",
-          "Which language do you want to use?"
+          "Select a supported language"
         );
 
         const defaultLanguage = fetchGuild.language;
-        for (const [key, value] of Object.entries(languageFlags)) {
+        for (const [key, value] of Object.entries(supportedLanguages)) {
           languageRow.components[0].addOptions({
             label: key,
             value: `${key}_option`,

diff --git a/src/languageList.js b/src/languageList.js
diff --git a/src/languages/deeplLanguages.js b/src/languages/deeplLanguages.js
@@ -0,0 +1,53 @@
+module.exports = {
+  deeplLanguages: {
+    en: "🇺🇸",
+    //french
+    fr: "🇫🇷",
+    //german
+    de: "🇩🇪",
+    //spanish
+    es: "🇪🇸",
+    //italian
+    it: "🇮🇹",
+    //portuguese
+    pt: "🇵🇹",
+    //russian
+    ru: "🇷🇺",
+    //jap
+    ja: "🇯🇵",
+    //bulgarian
+    bg: "🇧🇬",
+    //chineese
+    zh: "🇨🇳",
+    //czech
+    cs: "🇨🇿",
+    //danish
+    da: "🇩🇰",
+    //dutch
+    nl: "🇳🇱",
+    //finnish
+    fi: "🇫🇮",
+    //estonian
+    et: "🇪🇪",
+    //greek
+    el: "🇬🇷",
+    //hungarian
+    hu: "🇭🇺",
+    //indonesian
+    id: "🇮🇩",
+    //lithuanian
+    lt: "🇱🇹",
+    //polish
+    pl: "🇵🇱",
+    //romanian
+    ro: "🇷🇴",
+    //slovak
+    sk: "🇸🇰",
+    //slovenian
+    sl: "🇸🇮",
+    //swedish
+    sv: "🇸🇪",
+    //turkish
+    tr: "🇹🇷",
+  },
+};
diff --git a/src/languages/en.json b/src/languages/en.json
diff --git a/src/languages/fr.json b/src/languages/fr.json
diff --git a/src/languages/supportedLanguages.js b/src/languages/supportedLanguages.js
@@ -0,0 +1,6 @@
+module.exports = {
+  supportedLanguages: {
+    en: "🇺🇸",
+    fr: "🇫🇷",
+  },
+};