Merge pull request #61 from DjDeveloperr/slash

Add Sub Commands and Sub Command Group handlers and decorators

Author: Helloyunho

src/gateway/handlers/index.ts

@@ -330,7 +330,7 @@ export interface ClientEvents extends EventTypes {
    */
   webhooksUpdate: (guild: Guild, channel: GuildTextChannel) => void
   /**
-   * A Slash Command was triggered
+   * An Interaction was created
    */
   interactionCreate: (interaction: Interaction) => void
 }

src/models/client.ts

@@ -15,6 +15,7 @@ import { Extension } from './extensions.ts'
 import { SlashClient } from './slashClient.ts'
 import { Interaction } from '../structures/slash.ts'
 import { SlashModule } from './slashModule.ts'
+import type { ShardManager } from './shard.ts'
 
 /** OS related properties sent with Gateway Identify */
 export interface ClientProperties {
@@ -93,6 +94,8 @@ export class Client extends EventEmitter {
   _decoratedSlash?: Array<{
     name: string
     guild?: string
+    parent?: string
+    group?: string
     handler: (interaction: Interaction) => any
   }>
 
@@ -110,6 +113,11 @@ export class Client extends EventEmitter {
     ...args: Parameters<ClientEvents[K]>
   ): boolean => this._untypedEmit(event, ...args)
 
+  /** Shard on which this Client is */
+  shard: number = 0
+  /** Shard Manager of this Client if Sharded */
+  shardManager?: ShardManager
+
   constructor(options: ClientOptions = {}) {
     super()
     this.token = options.token
@@ -231,6 +239,47 @@ export function slash(name?: string, guild?: string) {
   }
 }
 
+export function subslash(parent: string, name?: string, guild?: string) {
+  return function (client: Client | SlashModule, prop: string) {
+    if (client._decoratedSlash === undefined) client._decoratedSlash = []
+    const item = (client as { [name: string]: any })[prop]
+    if (typeof item !== 'function') {
+      item.parent = parent
+      client._decoratedSlash.push(item)
+    } else
+      client._decoratedSlash.push({
+        parent,
+        name: name ?? prop,
+        guild,
+        handler: item
+      })
+  }
+}
+
+export function groupslash(
+  parent: string,
+  group: string,
+  name?: string,
+  guild?: string
+) {
+  return function (client: Client | SlashModule, prop: string) {
+    if (client._decoratedSlash === undefined) client._decoratedSlash = []
+    const item = (client as { [name: string]: any })[prop]
+    if (typeof item !== 'function') {
+      item.parent = parent
+      item.group = group
+      client._decoratedSlash.push(item)
+    } else
+      client._decoratedSlash.push({
+        group,
+        parent,
+        name: name ?? prop,
+        guild,
+        handler: item
+      })
+  }
+}
+
 export function slashModule() {
   return function (client: Client, prop: string) {
     if (client._decoratedSlashModules === undefined)

src/models/command.ts

@@ -503,12 +503,13 @@ export const parseCommand = (
   client: CommandClient,
   msg: Message,
   prefix: string
-): ParsedCommand => {
+): ParsedCommand | undefined => {
   let content = msg.content.slice(prefix.length)
   if (client.spacesAfterPrefix === true) content = content.trim()
   const args = parse(content)._.map((e) => e.toString())
 
-  const name = args.shift() as string
+  const name = args.shift()
+  if (name === undefined) return
   const argString = content.slice(name.length).trim()
 
   return {

src/models/commandClient.ts

@@ -187,6 +187,7 @@ export class CommandClient extends Client implements CommandClientOptions {
     prefix = usedPrefix
 
     const parsed = parseCommand(this, msg, prefix)
+    if (parsed === undefined) return
     const command = this.commands.fetch(parsed)
 
     if (command === undefined) return

src/models/shard.ts

@@ -1 +1,69 @@
-// TODO: write code
+import { Collection } from '../utils/collection.ts'
+import { Client, ClientOptions } from './client.ts'
+import EventEmitter from 'https://deno.land/[email protected]/node/events.ts'
+import { RESTManager } from './rest.ts'
+// import { GATEWAY_BOT } from '../types/endpoint.ts'
+// import { GatewayBotPayload } from '../types/gatewayBot.ts'
+
+// TODO(DjDeveloperr)
+// I'm kinda confused; will continue on this later once
+// Deno namespace in Web Woker is stable!
+export interface ShardManagerOptions {
+  client: Client | typeof Client
+  token?: string
+  intents?: number[]
+  options?: ClientOptions
+  shards: number
+}
+
+export interface ShardManagerInitOptions {
+  file: string
+  token?: string
+  intents?: number[]
+  options?: ClientOptions
+  shards?: number
+}
+
+export class ShardManager extends EventEmitter {
+  workers: Collection<string, Worker> = new Collection()
+  token: string
+  intents: number[]
+  shardCount: number
+  private readonly __client: Client
+
+  get rest(): RESTManager {
+    return this.__client.rest
+  }
+
+  constructor(options: ShardManagerOptions) {
+    super()
+    this.__client =
+      options.client instanceof Client
+        ? options.client
+        : // eslint-disable-next-line new-cap
+          new options.client(options.options)
+
+    if (this.__client.token === undefined || options.token === undefined)
+      throw new Error('Token should be provided when constructing ShardManager')
+    if (this.__client.intents === undefined || options.intents === undefined)
+      throw new Error(
+        'Intents should be provided when constructing ShardManager'
+      )
+
+    this.token = this.__client.token ?? options.token
+    this.intents = this.__client.intents ?? options.intents
+    this.shardCount = options.shards
+  }
+
+  // static async init(): Promise<ShardManager> {}
+
+  // async start(): Promise<ShardManager> {
+  //   const info = ((await this.rest.get(
+  //     GATEWAY_BOT()
+  //   )) as unknown) as GatewayBotPayload
+
+  //   const totalShards = this.__shardCount ?? info.shards
+
+  //   return this
+  // }
+}

src/models/slashClient.ts

@@ -34,7 +34,7 @@ export class SlashCommand {
     this.applicationID = data.application_id
     this.name = data.name
     this.description = data.description
-    this.options = data.options
+    this.options = data.options ?? []
   }
 
   async delete(): Promise<void> {
@@ -158,6 +158,8 @@ export type SlashCommandHandlerCallback = (interaction: Interaction) => any
 export interface SlashCommandHandler {
   name: string
   guild?: string
+  parent?: string
+  group?: string
   handler: SlashCommandHandlerCallback
 }
 
@@ -182,38 +184,45 @@ export class SlashClient {
     }
 
     this.client.on('interactionCreate', (interaction) =>
-      this.process(interaction)
+      this._process(interaction)
     )
   }
 
   /** Adds a new Slash Command Handler */
-  handle(
-    name: string,
-    handler: SlashCommandHandlerCallback,
-    guild?: string
-  ): SlashClient {
-    this.handlers.push({
-      name,
-      guild,
-      handler
-    })
+  handle(handler: SlashCommandHandler): SlashClient {
+    this.handlers.push(handler)
     return this
   }
 
+  private _getCommand(i: Interaction): SlashCommandHandler | undefined {
+    return this.handlers.find((e) => {
+      const hasGroupOrParent = e.group !== undefined || e.parent !== undefined
+      const groupMatched =
+        e.group !== undefined && e.parent !== undefined
+          ? i.options
+              .find((o) => o.name === e.group)
+              ?.options?.find((o) => o.name === e.name) !== undefined
+          : true
+      const subMatched =
+        e.group === undefined && e.parent !== undefined
+          ? i.options.find((o) => o.name === e.name) !== undefined
+          : true
+      const nameMatched1 = e.name === i.name
+      const parentMatched = hasGroupOrParent ? e.parent === i.name : true
+      const nameMatched = hasGroupOrParent ? parentMatched : nameMatched1
+
+      const matched = groupMatched && subMatched && nameMatched
+      return matched
+    })
+  }
+
   /** Process an incoming Slash Command (interaction) */
-  private process(interaction: Interaction): void {
+  private _process(interaction: Interaction): void {
     if (!this.enabled) return
 
     if (interaction.type !== InteractionType.APPLICATION_COMMAND) return
 
-    let cmd
-
-    if (interaction.guild !== undefined)
-      cmd =
-        this.handlers.find(
-          (e) => e.guild !== undefined && e.name === interaction.name
-        ) ?? this.handlers.find((e) => e.name === interaction.name)
-    else cmd = this.handlers.find((e) => e.name === interaction.name)
+    const cmd = this._getCommand(interaction)
 
     if (cmd === undefined) return
 

src/structures/slash.ts

@@ -3,6 +3,7 @@ import { MessageOption } from '../types/channel.ts'
 import { INTERACTION_CALLBACK, WEBHOOK_MESSAGE } from '../types/endpoint.ts'
 import {
   InteractionData,
+  InteractionOption,
   InteractionPayload,
   InteractionResponsePayload,
   InteractionResponseType
@@ -76,10 +77,15 @@ export class Interaction {
     return this.data.name
   }
 
+  get options(): InteractionOption[] {
+    return this.data.options ?? []
+  }
+
   option<T = any>(name: string): T {
-    return this.data.options.find((e) => e.name === name)?.value
+    return this.options.find((e) => e.name === name)?.value
   }
 
+  /** Respond to an Interaction */
   async respond(data: InteractionResponse): Promise<Interaction> {
     const payload: InteractionResponsePayload = {
       type: data.type ?? InteractionResponseType.CHANNEL_MESSAGE_WITH_SOURCE,
@@ -105,6 +111,7 @@ export class Interaction {
     return this
   }
 
+  /** Edit the original Interaction response */
   async editResponse(data: {
     content?: string
     embeds?: Embed[]
@@ -121,6 +128,7 @@ export class Interaction {
     return this
   }
 
+  /** Delete the original Interaction Response */
   async deleteResponse(): Promise<Interaction> {
     const url = WEBHOOK_MESSAGE(
       this.client.user?.id as string,
@@ -135,6 +143,7 @@ export class Interaction {
     return `https://discord.com/api/v8/webhooks/${this.client.user?.id}/${this.token}`
   }
 
+  /** Send a followup message */
   async send(
     text?: string | AllWebhookMessageOptions,
     option?: AllWebhookMessageOptions
@@ -195,6 +204,7 @@ export class Interaction {
     return res
   }
 
+  /** Edit a Followup message */
   async editMessage(
     msg: Message | string,
     data: {