NapCatQQ/src/onebot/network/index.ts

import BaseAction from '@/onebot/action/BaseAction';
import { OB11BaseEvent } from '@/onebot/event/OB11BaseEvent';
import { OB11Message } from '@/onebot';

export type OB11EmitEventContent = OB11BaseEvent | OB11Message;

export interface IOB11NetworkAdapter {
    registerAction<T extends BaseAction<P, R>, P, R>(action: T): void;

    registerActionMap(actionMap: Map<string, BaseAction<any, any>>): void;

    onEvent<T extends OB11EmitEventContent>(event: T): void;

    open(): void | Promise<void>;

    close(): void | Promise<void>;
}

export class OB11NetworkManager {
    adapters: IOB11NetworkAdapter[] = [];
    async getAllAdapters() {
        return this.adapters;
    }
    async openAllAdapters() {
        return Promise.all(this.adapters.map(adapter => adapter.open()));
    }
    async registerAllActions(actions: Map<string,BaseAction<any, any>>) {
        return Promise.all(this.adapters.map(adapter => adapter.registerActionMap(actions)));
    }
    async emitEvent(event: OB11EmitEventContent) {
        // Mlikiowa V2.0.0 Refactor Todo
        return Promise.all(this.adapters.map(adapter => adapter.onEvent(event)));
    }

    async registerAdapter(adapter: IOB11NetworkAdapter) {
        return this.adapters.push(adapter);
    }

    async closeSomeAdapters(adapters: IOB11NetworkAdapter[]) {
        this.adapters = this.adapters.filter(adapter => !adapters.includes(adapter));
        await Promise.all(adapters.map(adapter => adapter.close()));
    }

    async closeAllAdapters() {
        this.adapters = [];
        await Promise.all(this.adapters.map(adapter => adapter.close()));
    }
}

export * from './active-http';
export * from './active-websocket';
export * from './passive-http';
export * from './passive-websocket';