ms/packages/plugin/src/command.ts

import { command, plugin, server } from '@ccms/api'
import { provideSingleton, Autowired } from '@ccms/container'
import { interfaces } from './interfaces'
import { getPluginCommandMetadata, getPluginTabCompleterMetadata } from './utils'

@provideSingleton(PluginCommandManager)
export class PluginCommandManager {
    @Autowired()
    private CommandManager: command.Command
    @Autowired()
    private ServerChecker: server.ServerChecker

    constructor() {
        process.on('plugin.before.enable', this.registryCommand.bind(this))
        process.on('plugin.after.disable', this.unregistryCommand.bind(this))
    }

    private registryCommand(pluginInstance: plugin.Plugin) {
        let cmds = getPluginCommandMetadata(pluginInstance)
        let tabs = getPluginTabCompleterMetadata(pluginInstance)
        for (const cmd of cmds) {
            if (!this.ServerChecker.check(cmd.servers)) {
                console.debug(`[${pluginInstance.description.name}] ${cmd.target.constructor.name} incompatible command ${cmd.name} server(${cmd.servers}) ignore.`)
                continue
            }
            for (let command of [cmd.name, ...cmd.alias]) {
                let [cmdExecutor, cmdCompleter] = this.generateAutoMainCommand(pluginInstance, cmd, tabs.get(command))
                this.CommandManager.on(pluginInstance, command, {
                    cmd: cmdExecutor.bind(pluginInstance),
                    tab: cmdCompleter?.bind(pluginInstance)
                })
            }
        }
    }

    private unregistryCommand(pluginInstance: plugin.Plugin) {
        let cmds = getPluginCommandMetadata(pluginInstance)
        for (const cmd of cmds) {
            if (!this.ServerChecker.check(cmd.servers)) {
                console.debug(`[${pluginInstance.description.name}] ${cmd.target.constructor.name} incompatible command ${cmd.name} server(${cmd.servers}) ignore.`)
                continue
            }
            for (let command of [cmd.name, ...cmd.alias]) {
                this.CommandManager.off(pluginInstance, command)
            }
        }
    }

    private generateAutoMainCommand(pluginInstance: plugin.Plugin, cmd: interfaces.CommandMetadata, tab: interfaces.CommandMetadata) {
        let cmdExecutor = pluginInstance[cmd.executor]
        let cmdCompleter = tab ? pluginInstance[tab.executor] : undefined
        let cmdSubCache = Object.keys(pluginInstance.constructor.prototype).filter(s => s.startsWith('cmd')).map(s => s.substring(3))
        if (cmd.autoMain) {
            cmdExecutor = (sender: any, command: string, args: string[]) => {
                let subcommand = args[0]
                let cmdKey = 'cmd' + subcommand
                let subcommandexec = pluginInstance[cmdKey]
                if (!subcommandexec) {
                    subcommandexec = pluginInstance['cmdmain']
                } else {
                    args.shift()
                }
                if (!subcommandexec) {
                    subcommand && pluginInstance.logger.sender(sender, '§4未知的子命令: §c' + subcommand)
                    pluginInstance.logger.sender(
                        sender,
                        pluginInstance['cmdhelp'] ?
                            `§6请执行 §b/${command} §ahelp §6查看帮助!` :
                            `§b版本: §a ${pluginInstance.description.version}`
                    )
                    return
                }
                let permission: string
                if (typeof cmd.permission == "string") {
                    permission = cmd.permission as string
                } else if (cmd.permission) {
                    permission = `${pluginInstance.description.name.toLocaleLowerCase()}.${command}.${subcommand}`
                }
                if (sender.hasPermission && !sender.hasPermission(permission)) {
                    return pluginInstance.logger.sender(sender, `§c你需要 ${permission} 权限 才可执行此命令.`)
                }
                return subcommandexec.apply(pluginInstance, [sender, ...args])
            }
            let originCompleter = cmdCompleter
            cmdCompleter = (sender: any, command: string, args: string[]) => {
                let permission: string
                if (typeof cmd.permission == "string") {
                    permission = cmd.permission as string
                } else if (cmd.permission) {
                    permission = `${pluginInstance.description.name.toLocaleLowerCase()}.${command}`
                }
                if (sender.hasPermission && !sender.hasPermission(permission)) { return [] }
                return (args.length == 1 ? cmdSubCache : []).concat(originCompleter?.apply(pluginInstance, [sender, command, args]) || [])
            }
        }
        if (!cmdCompleter) { console.warn(`[${pluginInstance.description.name}] command ${cmd.name} is not registry tabCompleter`) }
        return [cmdExecutor, cmdCompleter]
    }
}