feat(Translator): add context menu item (needs testing)

Socketlike · Aug 21, 2024 · c0c724f · c0c724f
1 parent 5bccffd
commit c0c724f
Show file tree

Hide file tree

Showing 2 changed files with 2,262 additions and 2,606 deletions.
diff --git a/plugins/Translator/src/index.tsx b/plugins/Translator/src/index.tsx
@@ -4,6 +4,8 @@ import { APIMessage } from 'discord-api-types/v9';
 
 import { Injector, webpack } from 'replugged';
 import { fluxDispatcher, i18n, messages, toast } from 'replugged/common';
+import { ContextMenu } from 'replugged/components';
+import { ContextMenuTypes } from 'replugged/types';
 
 import { TranslateButton, TranslateIcon, TranslateOffIcon, TranslatedTag } from './components';
 import { config, events, logger } from './util';
@@ -86,6 +88,18 @@ export const start = async (): Promise<void> => {
         : void translateMessage(message),
   }));
 
+  injector.utils.addMenuItem(ContextMenuTypes.Message, ({ message }: { message: APIMessage }) => (
+    <ContextMenu.MenuItem
+      id='translate-message'
+      label={originalCache.has(message.id) ? 'Untranslate' : 'Translate'}
+      action={() =>
+        originalCache.has(message.id)
+          ? untranslateMessage(message.id)
+          : void translateMessage(message)
+      }
+    />
+  ));
+
   injector.instead(messages, 'sendMessage', async (args, orig) => {
     if (config.get('sendTranslateEnabled'))
       args[1].content = (await _translate(args[1].content, true)).text;