61 lines
2.0 KiB
TypeScript
61 lines
2.0 KiB
TypeScript
export type Listener = (...args: any[]) => any;
|
|
export type TypedEvents = {[k in string | symbol]: Listener} & {'*': (event: string, ...args: any[]) => any};
|
|
|
|
export class TypedEmitter<T extends TypedEvents = TypedEvents> {
|
|
private static listeners: {[key: string]: Listener[]} = {};
|
|
|
|
private listeners: { [key in keyof T]?: Listener[] } = {};
|
|
|
|
static emit(event: any, ...args: any[]) {
|
|
(this.listeners['*'] || []).forEach(l => l(event, ...args));
|
|
(this.listeners[event.toString()] || []).forEach(l => l(...args));
|
|
};
|
|
|
|
static off(event: any, listener: Listener) {
|
|
const e = event.toString();
|
|
this.listeners[e] = (this.listeners[e] || []).filter(l => l === listener);
|
|
}
|
|
|
|
static on(event: any, listener: Listener) {
|
|
const e = event.toString();
|
|
if(!this.listeners[e]) this.listeners[e] = [];
|
|
this.listeners[e]?.push(listener);
|
|
return () => this.off(event, listener);
|
|
}
|
|
|
|
static once(event: any, listener?: Listener): Promise<any> {
|
|
return new Promise(res => {
|
|
const unsubscribe = this.on(event, <any>((...args: any) => {
|
|
res(args.length == 1 ? args[0] : args);
|
|
if(listener) listener(...args);
|
|
unsubscribe();
|
|
}));
|
|
});
|
|
}
|
|
|
|
emit<K extends keyof T>(event: K, ...args: Parameters<T[K]>) {
|
|
(this.listeners['*'] || []).forEach(l => l(event, ...args));
|
|
(this.listeners[event] || []).forEach(l => l(...args));
|
|
};
|
|
|
|
off<K extends keyof T = string>(event: K, listener: T[K]) {
|
|
this.listeners[event] = (this.listeners[event] || []).filter(l => l === listener);
|
|
}
|
|
|
|
on<K extends keyof T = string>(event: K, listener: T[K]) {
|
|
if(!this.listeners[event]) this.listeners[event] = [];
|
|
this.listeners[event]?.push(listener);
|
|
return () => this.off(event, listener);
|
|
}
|
|
|
|
once<K extends keyof T = string>(event: K, listener?: T[K]): Promise<any> {
|
|
return new Promise(res => {
|
|
const unsubscribe = this.on(event, <any>((...args: any) => {
|
|
res(args.length == 1 ? args[0] : args);
|
|
if(listener) listener(...args);
|
|
unsubscribe();
|
|
}));
|
|
});
|
|
}
|
|
}
|