@pikokr/command.ts 5.0.0-dev.a0bc517 → 5.0.0-dev.d04bd18

package/src/applicationCommand/ApplicationCommandOption.ts

@@ -1,6 +1,13 @@
+/*
+* File: ApplicationCommandOption.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 import { APIApplicationCommandOption } from 'discord.js'
-import { createArgumentDecorator } from '../core'
-import { ComponentArgumentDecorator } from '../core/components/ComponentArgumentDecorator'
+import { createArgumentDecorator, ComponentArgumentDecorator } from '../core'
 
 type Options = APIApplicationCommandOption
 

package/src/applicationCommand/index.ts

@@ -1,2 +1,10 @@
+/*
+* File: index.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 export * from './ApplicationCommand'
 export { option } from './ApplicationCommandOption'

package/src/core/components/BaseComponent.ts

@@ -1,22 +1,28 @@
+/*
+ * File: BaseComponent.ts
+ *
+ * Copyright (c) 2022-2022 pikokr
+ *
+ * Licensed under MIT License. Please see more defails in LICENSE file.
+ */
+
 import { Collection } from 'discord.js'
 import _ from 'lodash'
+import type { ComponentHookStore } from '../hooks'
 import { ComponentArgument } from './ComponentArgument'
 
-export class BaseComponent<Options = unknown, RequiredOptions = unknown> {
-  options: Options & RequiredOptions
+export class BaseComponent<Options = unknown, OptionsArg = Options> {
+  options: Options
 
   method: Function
 
+  hooks: ComponentHookStore = new Collection()
+
   argTypes: Collection<number, ComponentArgument> = new Collection()
 
-  constructor(options: Partial<Options> & RequiredOptions, method: Function, argTypes: unknown[]) {
-    if (typeof options === 'object') {
-      this.options = _.merge(this.defaultOptions(), options)
-    } else if (typeof options === 'string') {
-      this.options = options as this['options']
-    } else {
-      this.options = null as unknown as this['options']
-    }
+  constructor(options: OptionsArg, method: Function, argTypes: unknown[]) {
+    this.options = this.convertOptions(options)
+
     this.method = method
     for (let i = 0; i < argTypes.length; i++) {
       const element = argTypes[i]
@@ -24,11 +30,27 @@ export class BaseComponent<Options = unknown, RequiredOptions = unknown> {
     }
   }
 
-  defaultOptions(): Options & Partial<RequiredOptions> {
-    return {} as unknown as ReturnType<this['defaultOptions']>
+  convertOptions(options: OptionsArg): Options {
+    return options as unknown as Options
   }
 
-  execute(target: object, args: unknown[]) {
-    return this.method.apply(target, args)
+  async executeHook(target: object, name: string, args: unknown[]) {
+    const hook = this.hooks.get(name)
+
+    if (!hook) return
+
+    const { CommandClient } = await import('../structures/CommandClient')
+
+    for (const fn of hook) {
+      await fn.call(null, CommandClient.getFromModule(target), ...args)
+    }
+  }
+
+  async execute(target: object, args: unknown[], beforeCallArgs: unknown[] = args) {
+    await this.executeHook(target, 'beforeCall', beforeCallArgs)
+    const result = await this.method.call(target, ...args)
+    await this.executeHook(target, 'afterCall', [result])
+
+    return result
   }
 }

package/src/core/components/ComponentArgument.ts

@@ -1,3 +1,11 @@
+/*
+* File: ComponentArgument.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 import { ComponentArgumentDecorator } from './ComponentArgumentDecorator'
 
 export class ComponentArgument {

package/src/core/components/ComponentArgumentDecorator.ts

@@ -1,3 +1,11 @@
+/*
+* File: ComponentArgumentDecorator.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 import _ from 'lodash'
 
 export class ComponentArgumentDecorator<Options = unknown> {

package/src/core/components/decoratorCreator.ts

@@ -1,4 +1,14 @@
+/*
+* File: decoratorCreator.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 import { Collection } from 'discord.js'
+import { ComponentHookStore } from '../hooks'
+import { getComponentHookStore } from '../hooks/componentHook'
 import { ComponentStoreSymbol } from '../symbols'
 import { BaseComponent } from './BaseComponent'
 import { ComponentArgumentDecorator } from './ComponentArgumentDecorator'
@@ -24,10 +34,14 @@ export const getComponent = (target: object, key: string | symbol) => {
   return store.get(key)
 }
 
-export const createComponentDecorator = <Options, RequiredOptions>(type: typeof BaseComponent<Options, RequiredOptions>) => {
-  return (options: Partial<Options> & RequiredOptions): MethodDecorator => {
+export const createComponentDecorator = <Options, OptionArgs>(type: typeof BaseComponent<Options, OptionArgs>) => {
+  return (options: OptionArgs): MethodDecorator => {
     return (target, key) => {
-      var component: BaseComponent<Options> = new type(options, Reflect.get(target, key), Reflect.getMetadata('design:paramtypes', target, key))
+      var component: BaseComponent<Options, OptionArgs> = new type(options, Reflect.get(target, key), Reflect.getMetadata('design:paramtypes', target, key))
+
+      const componentHookStore: ComponentHookStore = getComponentHookStore(target, key)
+
+      component.hooks = componentHookStore
 
       const store = getComponentStore(target)
 

package/src/core/components/index.ts

@@ -1,3 +1,13 @@
-import 'reflect-metadata'
-export * from './BaseComponent'
-export * from './decoratorCreator'
+/*
+ * File: index.ts
+ *
+ * Copyright (c) 2022-2022 pikokr
+ *
+ * Licensed under MIT License. Please see more defails in LICENSE file.
+ */
+
+import 'reflect-metadata'
+export * from './decoratorCreator'
+export * from './ComponentArgument'
+export * from './ComponentArgumentDecorator'
+export * from './BaseComponent'

package/src/core/converter/index.ts

@@ -0,0 +1,16 @@
+/*
+* File: index.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+import { BaseComponent } from '../components/BaseComponent'
+import { createComponentDecorator } from '../components/decoratorCreator'
+
+type Options = { component: typeof BaseComponent<unknown>; type: Function; parameterless: boolean }
+
+export class ConverterComponent extends BaseComponent<Options, Options & { parameterless?: boolean }> {}
+
+export const argConverter = createComponentDecorator(ConverterComponent)

package/src/core/extensions/CTSExtension.ts

@@ -0,0 +1,17 @@
+/*
+* File: CTSExtension.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+import chalk from 'chalk'
+import { Extension } from './Extension'
+
+export class CTSExtension extends Extension {
+  protected get logger() {
+    if (!this._logger) this._logger = this.commandClient.ctsLogger.getChildLogger({ prefix: [chalk.green(`[${this.constructor.name}]`)], displayFunctionName: false })
+    return this._logger
+  }
+}

package/src/core/extensions/Extension.ts

@@ -0,0 +1,62 @@
+/*
+* File: Extension.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+import chalk from 'chalk'
+import { Collection } from 'discord.js'
+import { Logger } from 'tslog'
+import { BaseComponent } from '../components'
+import { ComponentArgument } from '../components/ComponentArgument'
+import { ConverterComponent } from '../converter'
+import { CommandClient } from '../structures'
+
+export class Extension {
+  protected get commandClient() {
+    return CommandClient.getFromModule(this)
+  }
+
+  protected get client() {
+    return this.commandClient.discord
+  }
+
+  protected _logger?: Logger
+
+  protected get logger() {
+    if (!this._logger) this._logger = this.commandClient.logger.getChildLogger({ prefix: [chalk.green(`[${this.constructor.name}]`)], displayFunctionName: false })
+    return this._logger
+  }
+
+  protected async convertArguments(
+    component: typeof BaseComponent<unknown>,
+    argList: unknown[],
+    args: Collection<number, ComponentArgument>,
+    getConverterArgs: (arg: ComponentArgument, index: number) => unknown[] | Promise<unknown[]>,
+  ) {
+    const items = new Collection<unknown, { ext: object; component: ConverterComponent }>()
+
+    for (const extension of this.commandClient.registry.extensions) {
+      for (const converter of this.commandClient.registry.getComponentsWithType(extension, ConverterComponent)) {
+        if (converter.options.component != component) continue
+
+        items.set(converter.options.type, { component: converter, ext: extension })
+      }
+    }
+
+    for (const [index, arg] of args) {
+      const converter = items.get(arg.type)
+
+      if (!converter) {
+        argList[index] = undefined
+        continue
+      }
+
+      const converterArgs = await getConverterArgs(arg, index)
+
+      argList[index] = await converter.component.execute(converter.ext, converterArgs)
+    }
+  }
+}

package/src/core/extensions/index.ts

@@ -0,0 +1,9 @@
+/*
+* File: index.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+export * from './Extension'

package/src/core/hooks/componentHook.ts

@@ -0,0 +1,40 @@
+/*
+ * File: componentHook.ts
+ *
+ * Copyright (c) 2022-2022 pikokr
+ *
+ * Licensed under MIT License. Please see more defails in LICENSE file.
+ */
+
+import { Collection } from 'discord.js'
+import { ComponentHookSymbol } from '../symbols'
+
+export type ComponentHookFn = (...args: any[]) => void | Promise<void>
+
+export type ComponentHookStore = Collection<string, ComponentHookFn[]>
+
+export const getComponentHookStore = (target: object, property: string | symbol): ComponentHookStore => {
+  let data = Reflect.getMetadata(ComponentHookSymbol, target, property) as ComponentHookStore
+
+  if (!data) {
+    data = new Collection()
+    Reflect.defineMetadata(ComponentHookSymbol, data, target, property)
+  }
+
+  return data
+}
+
+export const createComponentHook = (name: string, fn: ComponentHookFn): MethodDecorator => {
+  return (target, key) => {
+    const store = getComponentHookStore(target, key)
+
+    let hooks = store.get(name)
+
+    if (!hooks) {
+      hooks = []
+      store.set(name, hooks)
+    }
+
+    hooks.push(fn)
+  }
+}

package/src/core/hooks/index.ts

@@ -1 +1,11 @@
-export * from './moduleHook'
+/*
+ * File: index.ts
+ *
+ * Copyright (c) 2022-2022 pikokr
+ *
+ * Licensed under MIT License. Please see more defails in LICENSE file.
+ */
+
+export * from './moduleHook'
+export { createComponentHook } from './componentHook'
+export type { ComponentHookStore } from './componentHook'

package/src/core/hooks/moduleHook.ts

@@ -1,7 +1,15 @@
+/*
+* File: moduleHook.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 import { Collection } from 'discord.js'
 import { ModuleHookStoreSymbol } from '../symbols'
 
-type ModuleHookStore = Collection<string | symbol, Function[]>
+type ModuleHookStore = Collection<string, Function[]>
 
 export const getModuleHookStore = (target: object) => {
   let result: ModuleHookStore | null = Reflect.getMetadata(ModuleHookStoreSymbol, target)
@@ -19,11 +27,11 @@ export const moduleHook = (name: string): MethodDecorator => {
   return (target, key) => {
     const store = getModuleHookStore(target)
 
-    let v = store.get(key)
+    let v = store.get(name)
 
     if (!v) {
       v = []
-      store.set(key, v)
+      store.set(name, v)
     }
 
     v.push(Reflect.get(target, key))

package/src/core/index.ts

@@ -1,3 +1,15 @@
+/*
+* File: index.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 export * from './components'
-export * from './structures'
 export * from './hooks'
+export * from './converter'
+export * from './utils'
+export * from './listener'
+export * from './structures'
+export * from './extensions'

package/src/core/listener/index.ts

@@ -1,9 +1,29 @@
-import { BaseComponent, createComponentDecorator } from '../components'
+/*
+* File: index.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+import { BaseComponent } from '../components/BaseComponent'
+import { createComponentDecorator } from '../components/decoratorCreator'
 
-export class ListenerComponent extends BaseComponent<{ emitter: string }, { event: string }> {
+export class ListenerComponent extends BaseComponent<{ emitter: string; event: string }, { emitter?: string; event: string }> {
   defaultOptions() {
     return { emitter: 'discord' }
   }
+
+  constructor(options: ListenerComponent['options'], method: Function, argTypes: unknown[]) {
+    super(
+      {
+        event: options.event,
+        emitter: options.emitter ?? 'discord',
+      },
+      method,
+      argTypes,
+    )
+  }
 }
 
 export const listener = createComponentDecorator(ListenerComponent)

package/src/core/structures/CommandClient.ts

@@ -1,14 +1,71 @@
-import { Client } from 'discord.js'
+/*
+* File: CommandClient.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+import chalk from 'chalk'
+import { Client, Snowflake, Team, User } from 'discord.js'
 import EventEmitter from 'events'
+import { Logger } from 'tslog'
+import { ApplicationCommandExtension, ApplicationCommandExtensionConfig } from '../../applicationCommand/ApplicationCommandExtension'
+import { CommandClientSymbol } from '../symbols'
 import { Registry } from './Registry'
-
 export class CommandClient extends EventEmitter {
-  registry = new Registry()
+  ctsLogger: Logger
+  registry: Registry
+
+  owners: Set<Snowflake> = new Set()
 
-  constructor(public discord: Client) {
+  constructor(public discord: Client, public logger: Logger = new Logger({ dateTimeTimezone: Intl.DateTimeFormat().resolvedOptions().timeZone })) {
     super()
 
+    this.ctsLogger = logger.getChildLogger({ prefix: [chalk.blue('[command.ts]')], displayFilePath: 'hidden', displayFunctionName: false })
+
+    this.registry = new Registry(this.ctsLogger, this)
+
     this.registry.registerEventEmitter('cts', this)
     this.registry.registerEventEmitter('discord', this.discord)
   }
+
+  async fetchOwners() {
+    if (!this.discord.application) throw new Error('The client is not logged in.')
+
+    this.ctsLogger.info('Fetching owners...')
+
+    await this.discord.application.fetch()
+
+    const owner = this.discord.application.owner
+
+    if (!owner) throw new Error('Cannot find application owner')
+
+    const owners: string[] = []
+
+    if (owner instanceof User) {
+      this.owners.add(owner.id)
+      owners.push(owner.tag)
+    } else if (owner instanceof Team) {
+      for (const [id, member] of owner.members) {
+        this.owners.add(id)
+        owners.push(member.user.tag)
+      }
+    }
+
+    this.ctsLogger.info(`Fetched ${chalk.green(owners.length)} owners(${owners.map((x) => chalk.blue(x)).join(', ')})`)
+  }
+
+  async enableApplicationCommandsExtension(config: ApplicationCommandExtensionConfig) {
+    await this.registry.registerModule(new ApplicationCommandExtension(config))
+    this.ctsLogger.info('Application command extension enabled.')
+  }
+
+  getApplicationCommandsExtension() {
+    return this.registry.extensions.find((x) => x.constructor === ApplicationCommandExtension) as ApplicationCommandExtension | undefined
+  }
+
+  static getFromModule(ext: object): CommandClient {
+    return Reflect.getMetadata(CommandClientSymbol, ext)
+  }
 }

package/src/core/structures/Registry.ts

@@ -1,17 +1,37 @@
+/*
+* File: Registry.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+import chalk from 'chalk'
 import { Collection } from 'discord.js'
 import EventEmitter from 'events'
 import _ from 'lodash'
-import { BaseComponent, getComponentStore } from '../components'
+import { Logger } from 'tslog'
+import { getComponentStore } from '../components'
+import type { BaseComponent } from '../components'
 import { getModuleHookStore } from '../hooks'
 import { ListenerComponent } from '../listener'
-import { ListenersSymbol } from '../symbols'
+import { CommandClientSymbol } from '../symbols'
+import { CommandClient } from './CommandClient'
 
 export class Registry {
   extensions: object[] = []
 
   emitters: Collection<string, EventEmitter> = new Collection()
 
-  getComponentsWithTypeGlobal<T extends typeof BaseComponent<Config, RequiredConfig>, Config, RequiredConfig>(type: T): InstanceType<T>[] {
+  logger: Logger
+
+  constructor(logger: Logger, public client: CommandClient) {
+    this.logger = logger.getChildLogger({
+      prefix: [chalk.green('[Registry]')],
+    })
+  }
+
+  getComponentsWithTypeGlobal<T extends typeof BaseComponent<Config>, Config>(type: T): InstanceType<T>[] {
     const result: InstanceType<T>[] = []
 
     for (const ext of this.extensions) {
@@ -21,7 +41,7 @@ export class Registry {
     return result
   }
 
-  getComponentsWithType<T extends typeof BaseComponent<Config, RequiredConfig>, Config, RequiredConfig>(ext: object, type: T): InstanceType<T>[] {
+  getComponentsWithType<T extends typeof BaseComponent<Config>, Config>(ext: object, type: T): InstanceType<T>[] {
     const componentStore = getComponentStore(ext)
 
     return Array.from(componentStore.filter((x) => (x.constructor as unknown) === type).values() as Iterable<InstanceType<T>>)
@@ -57,15 +77,19 @@ export class Registry {
   }
 
   async registerModule(ext: object) {
+    Reflect.defineMetadata(CommandClientSymbol, this.client, ext)
+
     this.registerEventListeners(ext)
     await this.runModuleHook(ext, 'load')
     this.extensions.push(ext)
+    this.logger.info(`Module registered: ${chalk.green(ext.constructor.name)}`)
   }
 
   async unregisterModule(ext: object) {
     this.unregisterEventListeners(ext)
     await this.runModuleHook(ext, 'unload')
     _.remove(this.extensions, (x) => x === ext)
+    this.logger.info(`Module unregistered: ${chalk.green(ext.constructor.name)}`)
   }
 
   runModuleHook(ext: object, hookName: string, ...args: unknown[]) {
@@ -75,7 +99,7 @@ export class Registry {
 
     if (functions) {
       for (const fn of functions) {
-        fn.apply(ext, args)
+        fn.call(ext, ...args)
       }
     }
   }

package/src/core/structures/index.ts

@@ -1,2 +1,10 @@
+/*
+* File: index.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
 export * from './Registry'
 export * from './CommandClient'

package/src/core/symbols.ts

@@ -1,4 +1,13 @@
-export const ComponentStoreSymbol = Symbol()
-export const ComponentArgStoreSymbol = Symbol()
-export const ModuleHookStoreSymbol = Symbol()
-export const ListenersSymbol = Symbol()
+/*
+ * File: symbols.ts
+ *
+ * Copyright (c) 2022-2022 pikokr
+ *
+ * Licensed under MIT License. Please see more defails in LICENSE file.
+ */
+
+export const ComponentStoreSymbol = Symbol()
+export const ComponentArgStoreSymbol = Symbol()
+export const ModuleHookStoreSymbol = Symbol()
+export const CommandClientSymbol = Symbol()
+export const ComponentHookSymbol = Symbol()

package/src/core/utils/checks.ts

@@ -0,0 +1,27 @@
+/*
+* File: checks.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+import { BaseInteraction, Interaction, Message } from 'discord.js'
+import { createComponentHook } from '../hooks'
+import { ComponentHookFn } from '../hooks/componentHook'
+import { CommandClient } from '../structures'
+import { OwnerOnlyError } from './errors'
+
+export const createCheckDecorator = (fn: ComponentHookFn) => createComponentHook('beforeCall', fn)
+
+export const ownerOnly = createCheckDecorator(async (client: CommandClient, i: Interaction | Message) => {
+  let isOwner = false
+
+  if (i instanceof BaseInteraction) {
+    isOwner = client.owners.has(i.user.id)
+  } else if (i instanceof Message) {
+    isOwner = client.owners.has(i.author.id)
+  }
+
+  if (!isOwner) throw new OwnerOnlyError()
+})

package/src/core/utils/errors.ts

@@ -0,0 +1,9 @@
+/*
+* File: errors.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+export class OwnerOnlyError {}

package/src/core/utils/index.ts

@@ -0,0 +1,10 @@
+/*
+* File: index.ts
+* 
+* Copyright (c) 2022-2022 pikokr
+* 
+* Licensed under MIT License. Please see more defails in LICENSE file.
+*/
+
+export * from './checks'
+export * from './errors'