harmony/src/models/extensions.ts
2020-11-11 14:34:49 +05:30

112 lines
No EOL
3.2 KiB
TypeScript

import { Collection } from "../utils/collection.ts";
import { Command } from "./command.ts";
import { CommandClient } from "./commandClient.ts";
export type ExtensionEventCallback = (ext: Extension, ...args: any[]) => any
export class ExtensionCommands {
extension: Extension
constructor(ext: Extension) {
this.extension = ext
}
get list(): Collection<string, Command> {
return this.extension.client.commands.list.filter(c => c.extension?.name === this.extension.name)
}
get(cmd: string): Command | undefined {
const find = this.extension.client.commands.find(cmd)
// linter sucks
if (find === undefined) return undefined
else if (find.extension === undefined) return undefined
else if (find.extension.name !== this.extension.name) return undefined
else return find
}
add(Cmd: Command | typeof Command): boolean {
const cmd = Cmd instanceof Command ? Cmd : new Cmd()
cmd.extension = this.extension
return this.extension.client.commands.add(cmd)
}
delete(cmd: Command | string): boolean {
const find = this.extension.client.commands.find(typeof cmd === 'string' ? cmd : cmd.name)
if (find === undefined) return false
if (find.extension !== undefined && find.extension.name !== this.extension.name) return false
else return this.extension.client.commands.delete(find)
}
deleteAll(): void {
for (const [cmd] of this.list) {
this.delete(cmd)
}
}
}
export class Extension {
client: CommandClient
name: string = ''
description?: string
commands: ExtensionCommands = new ExtensionCommands(this)
events: { [name: string]: (...args: any[]) => {} } = {}
constructor(client: CommandClient) {
this.client = client
}
listen(event: string, cb: ExtensionEventCallback): boolean {
if (this.events[event] !== undefined) return false
else {
const fn = (...args: any[]): any => {
// eslint-disable-next-line standard/no-callback-literal
cb(this, ...args)
};
this.client.on(event, fn)
this.events[event] = fn
return true
}
}
load(): any {}
unload(): any {}
}
export class ExtensionsManager {
client: CommandClient
list: Collection<string, Extension> = new Collection()
constructor(client: CommandClient) {
this.client = client
}
get(ext: string): Extension | undefined {
return this.list.get(ext)
}
exists(ext: string): boolean {
return this.get(ext) !== undefined
}
load(ext: Extension | typeof Extension): void {
// eslint-disable-next-line new-cap
if (!(ext instanceof Extension)) ext = new ext(this.client)
if (this.exists(ext.name)) throw new Error(`Extension with name '${ext.name}' already exists`)
this.list.set(ext.name, ext)
ext.load()
}
unload(ext: Extension | string): boolean {
const name = typeof ext === 'string' ? ext : ext.name
const extension = this.get(name)
if (extension === undefined) return false
extension.commands.deleteAll()
for (const [k, v] of Object.entries(extension.events)) {
this.client.removeListener(k, v)
// eslint-disable-next-line @typescript-eslint/no-dynamic-delete
delete extension.events[k]
}
extension.unload()
return this.list.delete(name)
}
}