fix: use enum

Garlic-Team · Feb 13, 2022 · ff09917 · ff09917
1 parent 9060f66
commit ff09917
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/src/handlers/MessageCommandHandler.ts b/src/handlers/MessageCommandHandler.ts
@@ -48,11 +48,8 @@ const checkValidation = async(arg: MessageArgumentTypes, content: string | Messa
 		} else {
 			channel.send(text);
 			const message = await channel.awaitMessages({ filter: (m) => m.author.id === user.id && m.channelId === channel.id, time: 60000, max: 1 });
-
-			/**
-			 * TODO: Use ArgumentType.ATTACHMENT | Need wait for https://github.com/Garlic-Team/gcommands/pull/314 to be merged (:
-			 * @ts-expect-error */
-			if (argument.type == 11) {
+
+			if (argument.type == ArgumentType.ATTACHMENT) {
 				const attachments = [...message.values()]?.[0]?.attachments;
 				content = attachments ? [...attachments.values()][0] : null;
 			}

diff --git a/src/lib/structures/arguments/base.ts b/src/lib/structures/arguments/base.ts
@@ -59,8 +59,7 @@ export class MessageArgumentTypeBase {
 			const { UserType } = await import('./User');
 			return new UserType(guild);
 		}
-		// @ts-expect-error TODO: Use ArgumentType.ATTACHMENT | Need wait for https://github.com/Garlic-Team/gcommands/pull/314 to be merged (:
-		case 11: {
+		case ArgumentType.ATTACHMENT: {
 			const { AttachmentType } = await import('./Attachment');
 			return new AttachmentType();
 		}