feat: add the possibility to specify if context menu is message or user in the type

Author: davidabou

src/define.ts

@@ -70,20 +70,26 @@ export const defineCommand = <
 }
 
 export const defineContextMenu: DefineContextMenu &
-  DefineContextMenuWithOptions = (
-  ...args: [ContextMenuOptions | ContextMenuCallback, ContextMenuCallback?]
+  DefineContextMenuWithOptions = <Type extends 'message' | 'user'>(
+  ...args: [
+    ContextMenuOptions | ContextMenuCallback<Type>,
+    ContextMenuCallback<Type>?
+  ]
 ): HarmonixContextMenu => {
   let options: ContextMenuOptions = {}
 
   if (args.length === 1) {
-    const [callback] = args as [ContextMenuCallback]
+    const [callback] = args as [ContextMenuCallback<Type>]
 
     return {
       options,
       callback
     }
   } else {
-    const [opts, callback] = args as [ContextMenuOptions, ContextMenuCallback]
+    const [opts, callback] = args as [
+      ContextMenuOptions,
+      ContextMenuCallback<Type>
+    ]
 
     options = opts
     return {

src/types/contextMenus.ts

@@ -1,14 +1,13 @@
 import type {
-  CacheType,
   MessageContextMenuCommandInteraction,
   PermissionsString,
   UserContextMenuCommandInteraction
 } from 'discord.js'
 
-export type ContextMenuCallback = (
-  interaction:
-    | MessageContextMenuCommandInteraction<CacheType>
-    | UserContextMenuCommandInteraction<CacheType>
+export type ContextMenuCallback<Type extends 'message' | 'user'> = (
+  interaction: Type extends 'message'
+    ? MessageContextMenuCommandInteraction
+    : UserContextMenuCommandInteraction
 ) => void
 
 export interface ContextMenuOptions {
@@ -18,17 +17,17 @@ export interface ContextMenuOptions {
   preconditions?: string[]
 }
 
-export type DefineContextMenu = (
-  callback: ContextMenuCallback
+export type DefineContextMenu = <Type extends 'message' | 'user'>(
+  callback: ContextMenuCallback<Type>
 ) => HarmonixContextMenu
-export type DefineContextMenuWithOptions = (
+export type DefineContextMenuWithOptions = <Type extends 'message' | 'user'>(
   options: ContextMenuOptions,
-  callback: ContextMenuCallback
+  callback: ContextMenuCallback<Type>
 ) => HarmonixContextMenu
 
 export type HarmonixContextMenuInput = string | HarmonixContextMenu
 
 export interface HarmonixContextMenu {
   options: ContextMenuOptions
-  callback: ContextMenuCallback
+  callback: ContextMenuCallback<'message' | 'user'>
 }