add functionality to upload our slash commands to our api

cakedan · cakedan · commit 25b837262a09 · 2022-04-28T07:24:24.000-06:00
diff --git a/src/commands/interactions/context-menu/message/ocr.ts b/src/commands/interactions/context-menu/message/ocr.ts
@@ -13,6 +13,9 @@ export interface OCRCommandArgs extends ContextMenuMessageArgs {
 export const COMMAND_NAME = 'OCR';
 
 export default class OCRCommand extends BaseContextMenuMessageCommand {
+  metadata = {
+    id: Formatter.Commands.ToolsOCR.COMMAND_ID,
+  };
   name = COMMAND_NAME;
 
   onBeforeRun(context: Interaction.InteractionContext, args: OCRCommandArgs) {
diff --git a/src/commands/interactions/context-menu/message/qr.scan.ts b/src/commands/interactions/context-menu/message/qr.scan.ts
@@ -13,6 +13,9 @@ export interface QRScanCommandArgs extends ContextMenuMessageArgs {
 export const COMMAND_NAME = 'QR Scan';
 
 export default class QRScanCommand extends BaseContextMenuMessageCommand {
+  metadata = {
+    id: Formatter.Commands.ToolsQrScan.COMMAND_ID,
+  };
   name = COMMAND_NAME;
 
   onBeforeRun(context: Interaction.InteractionContext, args: QRScanCommandArgs) {
diff --git a/src/commands/interactions/context-menu/message/translate.ts b/src/commands/interactions/context-menu/message/translate.ts
@@ -14,6 +14,9 @@ export interface TranslateCommandArgs extends ContextMenuMessageArgs {
 export const COMMAND_NAME = 'Translate (w/ OCR)';
 
 export default class TranslateCommand extends BaseContextMenuMessageCommand {
+  metadata = {
+    id: Formatter.Commands.ToolsTranslate.COMMAND_ID,
+  };
   name = COMMAND_NAME;
 
   onBeforeRun(context: Interaction.InteractionContext, args: TranslateCommandArgs) {
diff --git a/src/commands/interactions/context-menu/user/allowlist.add.ts b/src/commands/interactions/context-menu/user/allowlist.add.ts
@@ -10,6 +10,9 @@ export const COMMAND_NAME = 'Allowlist Add';
 
 export default class AllowlistAddCommand extends BaseContextMenuUserCommand {
   disableDm = true;
+  metadata = {
+    id: Formatter.Commands.ModerationAllowlistAddUsers.COMMAND_ID,
+  };
   name = COMMAND_NAME;
   permissions = [Permissions.ADMINISTRATOR];
 
diff --git a/src/commands/interactions/context-menu/user/allowlist.remove.ts b/src/commands/interactions/context-menu/user/allowlist.remove.ts
@@ -10,6 +10,9 @@ export const COMMAND_NAME = 'Allowlist Remove';
 
 export default class AllowlistRemoveCommand extends BaseContextMenuUserCommand {
   disableDm = true;
+  metadata = {
+    id: Formatter.Commands.ModerationAllowlistRemoveUsers.COMMAND_ID,
+  };
   name = COMMAND_NAME;
   permissions = [Permissions.ADMINISTRATOR];
 
diff --git a/src/commands/interactions/context-menu/user/blocklist.add.ts b/src/commands/interactions/context-menu/user/blocklist.add.ts
@@ -10,6 +10,9 @@ export const COMMAND_NAME = 'Blocklist Add';
 
 export default class BlocklistAddCommand extends BaseContextMenuUserCommand {
   disableDm = true;
+  metadata = {
+    id: Formatter.Commands.ModerationBlocklistAddUsers.COMMAND_ID,
+  };
   name = COMMAND_NAME;
   permissions = [Permissions.ADMINISTRATOR];
 
diff --git a/src/commands/interactions/context-menu/user/blocklist.remove.ts b/src/commands/interactions/context-menu/user/blocklist.remove.ts
@@ -10,6 +10,9 @@ export const COMMAND_NAME = 'Blocklist Remove';
 
 export default class BlocklistRemoveCommand extends BaseContextMenuUserCommand {
   disableDm = true;
+  metadata = {
+    id: Formatter.Commands.ModerationBlocklistRemoveUsers.COMMAND_ID,
+  };
   name = COMMAND_NAME;
   permissions = [Permissions.ADMINISTRATOR];
 
diff --git a/src/commands/prefixed/moderation/allowlist.add.users.ts b/src/commands/prefixed/moderation/allowlist.add.users.ts
@@ -2,7 +2,7 @@ import { Command, CommandClient, Structures } from 'detritus-client';
 import { Permissions } from 'detritus-client/lib/constants';
 
 import { CommandCategories, EmbedColors, GuildAllowlistTypes } from '../../../constants';
-import { Parameters } from '../../../utils';
+import { Formatter, Parameters } from '../../../utils';
 
 import { BaseCommand } from '../basecommand';
 
@@ -29,12 +29,13 @@ export default class AllowlistAddUsersCommand extends BaseCommand {
       disableDm: true,
       label: 'users',
       metadata: {
+        category: CommandCategories.MODERATION,
         description: 'Add users to the allowlist.',
         examples: [
           `${COMMAND_NAME} cake`,
           `${COMMAND_NAME} <@300505364032389122> <@61189081970774016>`,
         ],
-        category: CommandCategories.MODERATION,
+        id: Formatter.Commands.ModerationAllowlistAddUsers.COMMAND_ID,
         usage: '...<user:id|mention|name>',
       },
       permissionsClient: [Permissions.EMBED_LINKS],
diff --git a/src/commands/prefixed/moderation/allowlist.remove.users.ts b/src/commands/prefixed/moderation/allowlist.remove.users.ts
@@ -2,7 +2,7 @@ import { Command, CommandClient, Structures } from 'detritus-client';
 import { Permissions } from 'detritus-client/lib/constants';
 
 import { CommandCategories, EmbedColors, GuildAllowlistTypes } from '../../../constants';
-import { Parameters } from '../../../utils';
+import { Formatter, Parameters } from '../../../utils';
 
 import { BaseCommand } from '../basecommand';
 
@@ -33,12 +33,13 @@ export default class AllowlistRemoveUsersCommand extends BaseCommand {
       disableDm: true,
       label: 'users',
       metadata: {
+        category: CommandCategories.MODERATION,
         description: 'Remove users from the allowlist.',
         examples: [
           `${COMMAND_NAME} cake`,
           `${COMMAND_NAME} <@300505364032389122> <@61189081970774016>`,
         ],
-        category: CommandCategories.MODERATION,
+        id: Formatter.Commands.ModerationAllowlistRemoveUsers.COMMAND_ID,
         usage: '...<user:id|mention|name>',
       },
       permissionsClient: [Permissions.EMBED_LINKS],
diff --git a/src/commands/prefixed/moderation/blocklist.add.users.ts b/src/commands/prefixed/moderation/blocklist.add.users.ts
@@ -2,7 +2,7 @@ import { Command, CommandClient, Structures } from 'detritus-client';
 import { Permissions } from 'detritus-client/lib/constants';
 
 import { CommandCategories, EmbedColors, GuildBlocklistTypes } from '../../../constants';
-import { Parameters } from '../../../utils';
+import { Formatter, Parameters } from '../../../utils';
 
 import { BaseCommand } from '../basecommand';
 
@@ -29,12 +29,13 @@ export default class BlocklistAddUsersCommand extends BaseCommand {
       disableDm: true,
       label: 'users',
       metadata: {
+        category: CommandCategories.MODERATION,
         description: 'Add users to the blocklist.',
         examples: [
           `${COMMAND_NAME} cake`,
           `${COMMAND_NAME} <@300505364032389122> <@61189081970774016>`,
         ],
-        category: CommandCategories.MODERATION,
+        id: Formatter.Commands.ModerationBlocklistAddUsers.COMMAND_ID,
         usage: '...<user:id|mention|name>',
       },
       permissionsClient: [Permissions.EMBED_LINKS],
diff --git a/src/commands/prefixed/moderation/blocklist.remove.users.ts b/src/commands/prefixed/moderation/blocklist.remove.users.ts
@@ -2,7 +2,7 @@ import { Command, CommandClient, Structures } from 'detritus-client';
 import { Permissions } from 'detritus-client/lib/constants';
 
 import { CommandCategories, EmbedColors, GuildBlocklistTypes } from '../../../constants';
-import { Parameters } from '../../../utils';
+import { Formatter, Parameters } from '../../../utils';
 
 import { BaseCommand } from '../basecommand';
 
@@ -33,12 +33,13 @@ export default class BlocklistRemoveUsersCommand extends BaseCommand {
       disableDm: true,
       label: 'users',
       metadata: {
+        category: CommandCategories.MODERATION,
         description: 'Remove users from the blocklist.',
         examples: [
           `${COMMAND_NAME} cake`,
           `${COMMAND_NAME} <@300505364032389122> <@61189081970774016>`,
         ],
-        category: CommandCategories.MODERATION,
+        id: Formatter.Commands.ModerationBlocklistRemoveUsers.COMMAND_ID,
         usage: '...<user:id|mention|name>',
       },
       permissionsClient: [Permissions.EMBED_LINKS],
diff --git a/src/commands/prefixed/owner/upload-commands.ts b/src/commands/prefixed/owner/upload-commands.ts
@@ -1,4 +1,5 @@
-import { Command, CommandClient } from 'detritus-client';
+import { Command, CommandClient, Interaction } from 'detritus-client';
+import { ApplicationCommandTypes } from 'detritus-client/lib/constants';
 
 import { uploadCommands } from '../../../api';
 import { CommandCategories, CommandTypes } from '../../../constants';
@@ -30,9 +31,12 @@ export default class UploadCommandsCommand extends BaseCommand {
   }
 
   async run(context: Command.Context) {
-    const commands = context.commandClient.commands.filter((command) => {
-      return !!command.metadata && !!command.metadata.category;
-    }).map((command) => {
+    const commands: Array<any> = [];
+    for (let command of context.commandClient.commands) {
+      if (!command.metadata || !command.metadata.category) {
+        continue;
+      }
+
       const { args } = command.argParser;
       const metadata = command.metadata as CommandMetadata;
 
@@ -49,7 +53,7 @@ export default class UploadCommandsCommand extends BaseCommand {
           }
         }
       }
-      return {
+      commands.push({
         aliases: names,
         args: args.map((arg) => {
           const argMetadata = arg.metadata || {};
@@ -78,8 +82,62 @@ export default class UploadCommandsCommand extends BaseCommand {
         }),
         type: CommandTypes.PREFIXED,
         usage: metadata.usage || '',
-      };
-    });
+      });
+    }
+    if (context.interactionCommandClient) {
+      const invokers: Array<[Interaction.InteractionCommand, Interaction.InteractionCommand | Interaction.InteractionCommandOption]> = [];
+      for (let command of context.interactionCommandClient.commands) {
+        if (command._options && command.isGroup) {
+          for (let [x, option] of command._options) {
+            if (option._options && option.isSubCommandGroup) {
+              for (let [y, opt] of option._options) {
+                if (opt.isSubCommand) {
+                  invokers.push([command, opt]);
+                }
+              }
+            } else if (option.isSubCommand) {
+              invokers.push([command, option]);
+            }
+          }
+        } else {
+          invokers.push([command, command]);
+        }
+      }
+      for (let [command, invoker] of invokers) {
+        if (!invoker.metadata) {
+          continue;
+        }
+
+        let commandType: number;
+        switch (command.type) {
+          case ApplicationCommandTypes.CHAT_INPUT: commandType = CommandTypes.APPLICATION_SLASH; break;
+          case ApplicationCommandTypes.MESSAGE: commandType = CommandTypes.APPLICATION_MENU_MESSAGE; break;
+          case ApplicationCommandTypes.USER: commandType = CommandTypes.APPLICATION_MENU_USER; break;
+          default: continue;
+        }
+
+        // add in args
+        commands.push({
+          args: [],
+          category: '',
+          enabled: true,
+          description: invoker.description,
+          dmable: !invoker.disableDm,
+          id: invoker.metadata.id || invoker.fullName.split(' ').join('.'),
+          name: invoker.fullName,
+          ratelimits: invoker.ratelimits?.map((ratelimit) => {
+            return {
+              duration: ratelimit.duration,
+              key: ratelimit.key,
+              limit: ratelimit.limit,
+              type: String(ratelimit.type),
+            };
+          }),
+          type: commandType,
+          usage: '',
+        });
+      }
+    }
     await uploadCommands(context, {commands});
 
     return editOrReply(context, 'ok');
diff --git a/src/utils/formatter/commands/index.ts b/src/utils/formatter/commands/index.ts
@@ -48,6 +48,11 @@ import * as ImageWall from './image.wall';
 
 import * as InfoUser from './info.user';
 
+import * as ModerationAllowlistAddUsers from './moderation.allowlist.add.users';
+import * as ModerationAllowlistRemoveUsers from './moderation.allowlist.remove.users';
+import * as ModerationBlocklistAddUsers from './moderation.blocklist.add.users';
+import * as ModerationBlocklistRemoveUsers from './moderation.blocklist.remove.users';
+
 import * as PrefixesAdd from './prefixes.add';
 import * as PrefixesClear from './prefixes.clear';
 import * as PrefixesList from './prefixes.list';
@@ -143,6 +148,10 @@ export {
   ImageToolsRotate,
   ImageWall,
   InfoUser,
+  ModerationAllowlistAddUsers,
+  ModerationAllowlistRemoveUsers,
+  ModerationBlocklistAddUsers,
+  ModerationBlocklistRemoveUsers,
   PrefixesAdd,
   PrefixesClear,
   PrefixesList,
diff --git a/src/utils/formatter/commands/moderation.allowlist.add.users.ts b/src/utils/formatter/commands/moderation.allowlist.add.users.ts
@@ -0,0 +1 @@
+export const COMMAND_ID = 'moderation.allowlist.add.users';
diff --git a/src/utils/formatter/commands/moderation.allowlist.remove.users.ts b/src/utils/formatter/commands/moderation.allowlist.remove.users.ts
@@ -0,0 +1 @@
+export const COMMAND_ID = 'moderation.allowlist.remove.users';
diff --git a/src/utils/formatter/commands/moderation.blocklist.add.users.ts b/src/utils/formatter/commands/moderation.blocklist.add.users.ts
@@ -0,0 +1 @@
+export const COMMAND_ID = 'moderation.blocklist.add.users';
diff --git a/src/utils/formatter/commands/moderation.blocklist.remove.users.ts b/src/utils/formatter/commands/moderation.blocklist.remove.users.ts
@@ -0,0 +1 @@
+export const COMMAND_ID = 'moderation.blocklist.remove.users';
