fix & refactor: pass actions through constructor

src/onebot/network/active-http.ts

@@ -1,37 +1,22 @@
 import { IOB11NetworkAdapter, OB11EmitEventContent } from '@/onebot/network/index';
-import BaseAction from '@/onebot/action/BaseAction';
 import { createHmac } from 'crypto';
 import { LogWrapper } from '@/common/utils/log';
 import { QuickAction, QuickActionEvent } from '../types';
 import { NapCatCore } from '@/core';
 import { handleQuickOperation } from '../helper/quick';
-import { NapCatOneBot11Adapter } from '@/onebot';
 
 export class OB11ActiveHttpAdapter implements IOB11NetworkAdapter {
-    url: string;
-    heartbeatInterval: number;
-    secret: string | undefined;
-    coreContext: NapCatCore;
-    obContext: NapCatOneBot11Adapter;
     logger: LogWrapper;
     isOpen: boolean = false;
 
-    constructor(url: string, heartbeatInterval: number, secret: string | undefined, coreContext: NapCatCore, onebotContext: NapCatOneBot11Adapter) {
-        this.heartbeatInterval = heartbeatInterval;
-        this.url = url;
-        this.secret = secret;
-        this.coreContext = coreContext;
-        this.obContext = onebotContext;
+    constructor(
+        public url: string,
+        public secret: string | undefined,
+        public coreContext: NapCatCore
+    ) {
         this.logger = coreContext.context.logger;
     }
 
-    registerActionMap(actionMap: Map<string, BaseAction<any, any>>) {
-    }
-
-    registerAction<T extends BaseAction<P, R>, P, R>(action: T) {
-        // Passive http adapter does not need to register actions
-    }
-
     onEvent<T extends OB11EmitEventContent>(event: T) {
         if (!this.isOpen) {
             return;

src/onebot/network/active-websocket.ts

@@ -1,44 +1,29 @@
 import { IOB11NetworkAdapter, OB11EmitEventContent } from '@/onebot/network/index';
 import { WebSocket } from 'ws';
-import BaseAction from '@/onebot/action/BaseAction';
 import { OB11HeartbeatEvent } from '../event/meta/OB11HeartbeatEvent';
 import { NapCatCore } from '@/core';
 import { ActionName } from '@/onebot/action/types';
 import { OB11Response } from '@/onebot/action/OB11Response';
 import { LogWrapper } from '@/common/utils/log';
-import { NapCatOneBot11Adapter } from '@/onebot';
+import { ActionMap } from '@/onebot/action';
 
 export class OB11ActiveWebSocketAdapter implements IOB11NetworkAdapter {
-    url: string;
-    reconnectIntervalInMillis: number;
     isClosed: boolean = false;
-    heartbeatInterval: number;
-    obContext: NapCatOneBot11Adapter;
-    coreContext: NapCatCore;
     logger: LogWrapper;
     private connection: WebSocket | null = null;
-    private actionMap: Map<string, BaseAction<any, any>> = new Map();
     private heartbeatRef: NodeJS.Timeout | null = null;
-    private readonly token: string;
-    
-    constructor(url: string, reconnectIntervalInMillis: number, heartbeatInterval: number, token:string, coreContext: NapCatCore, onebotContext: NapCatOneBot11Adapter) {
-        this.url = url;
-        this.token = token;
-        this.heartbeatInterval = heartbeatInterval;
-        this.reconnectIntervalInMillis = reconnectIntervalInMillis;
-        this.coreContext = coreContext;
-        this.obContext = onebotContext;
+
+    constructor(
+        public url: string,
+        public reconnectIntervalInMillis: number,
+        public heartbeatIntervalInMillis: number,
+        private token: string,
+        public coreContext: NapCatCore,
+        public actions: ActionMap,
+    ) {
         this.logger = coreContext.context.logger;
     }
 
-    registerActionMap(actionMap: Map<string, BaseAction<any, any>>) {
-        this.actionMap = actionMap;
-    }
-
-    registerAction<T extends BaseAction<P, R>, P, R>(action: T) {
-        this.actionMap.set(action.actionName, action);
-    }
-
     onEvent<T extends OB11EmitEventContent>(event: T) {
         if (this.connection) {
             this.connection.send(JSON.stringify(event));
@@ -51,9 +36,9 @@ export class OB11ActiveWebSocketAdapter implements IOB11NetworkAdapter {
         }
         this.heartbeatRef = setInterval(() => {
             if (this.connection && this.connection.readyState === WebSocket.OPEN) {
-                this.connection.send(JSON.stringify(new OB11HeartbeatEvent(this.coreContext, this.heartbeatInterval, this.coreContext.selfInfo.online, true)));
+                this.connection.send(JSON.stringify(new OB11HeartbeatEvent(this.coreContext, this.heartbeatIntervalInMillis, this.coreContext.selfInfo.online, true)));
             }
-        }, this.heartbeatInterval);
+        }, this.heartbeatIntervalInMillis);
         await this.tryConnect();
     }
 
@@ -61,7 +46,6 @@ export class OB11ActiveWebSocketAdapter implements IOB11NetworkAdapter {
         if (this.isClosed) {
             this.logger.logDebug('Cannot close a closed WebSocket connection');
             return;
-            //throw new Error('Cannot close a closed WebSocket connection');
         }
         this.isClosed = true;
         if (this.connection) {
@@ -130,7 +114,7 @@ export class OB11ActiveWebSocketAdapter implements IOB11NetworkAdapter {
                 this.checkStateAndReply<any>(OB11Response.error('json解析失败,请检查数据格式', 1400, echo));
             }
             receiveData.params = (receiveData?.params) ? receiveData.params : {};//兼容类型验证
-            const retdata = await this.actionMap.get(receiveData.action)
+            const retdata = await this.actions.get(receiveData.action)
                 ?.websocketHandle(receiveData.params, echo || '');
             const packet = Object.assign({}, retdata);
             this.checkStateAndReply<any>(packet);

src/onebot/network/index.ts

@@ -1,13 +1,12 @@
 import BaseAction from '@/onebot/action/BaseAction';
 import { OB11BaseEvent } from '@/onebot/event/OB11BaseEvent';
 import { OB11Message } from '@/onebot';
+import { ActionMap } from '@/onebot/action';
 
 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;
+    actions?: ActionMap;
 
     onEvent<T extends OB11EmitEventContent>(event: T): void;
 
@@ -19,27 +18,22 @@ export interface IOB11NetworkAdapter {
 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) {
         //console.log('adapters', this.adapters.length);
         return Promise.all(this.adapters.map(adapter => adapter.onEvent(event)));
     }
 
-    async registerAdapter(adapter: IOB11NetworkAdapter) {
-        //console.log('Registering adapter:', adapter);
+    registerAdapter(adapter: IOB11NetworkAdapter) {
         this.adapters.push(adapter);
-        //console.log('Current adapters:', this.adapters.length);
+    }
+
+    async registerAdapterAndOpen(adapter: IOB11NetworkAdapter) {
+        this.registerAdapter(adapter);
+        await adapter.open();
     }
 
     async closeSomeAdapters(adaptersToClose: IOB11NetworkAdapter[]) {
@@ -55,10 +49,8 @@ export class OB11NetworkManager {
     }
 
     async closeAllAdapters() {
-        //console.log('Closing all adapters');
         await Promise.all(this.adapters.map(adapter => adapter.close()));
         this.adapters = [];
-        //console.log('All adapters closed. Current adapters:', this.adapters.length);
     }
 }
 

src/onebot/network/passive-http.ts

@@ -1,38 +1,25 @@
 import { IOB11NetworkAdapter, OB11EmitEventContent } from './index';
-import BaseAction from '@/onebot/action/BaseAction';
 import express, { Express, Request, Response } from 'express';
 import http from 'http';
 import { NapCatCore } from '@/core';
 import { OB11Response } from '../action/OB11Response';
-import { NapCatOneBot11Adapter } from '@/onebot';
+import { ActionMap } from '@/onebot/action';
 
 export class OB11PassiveHttpAdapter implements IOB11NetworkAdapter {
-    token: string;
-    coreContext: NapCatCore;
-    obContext: NapCatOneBot11Adapter;
     private app: Express | undefined;
     private server: http.Server | undefined;
     private isOpen: boolean = false;
-    private actionMap: Map<string, BaseAction<any, any>> = new Map();
-    private port: number;
 
-    constructor(port: number, token: string, coreContext: NapCatCore, onebotContext: NapCatOneBot11Adapter) {
-        this.port = port;
-        this.token = token;
-        this.coreContext = coreContext;
-        this.obContext = onebotContext;
+    constructor(
+        public port: number,
+        public token: string,
+        public coreContext: NapCatCore,
+        public actions: ActionMap,
+    ) {
     }
 
-    registerAction<T extends BaseAction<P, R>, P, R>(action: T) {
-        this.actionMap.set(action.actionName, action);
-    }
-
-    registerActionMap(actionMap: Map<string, BaseAction<any, any>>) {
-        this.actionMap = actionMap;
-    }
-
-    onEvent<T extends OB11EmitEventContent>(event: T) {
-        // 事件处理逻辑可以在这里实现
+    onEvent() {
+        // http server is passive, no need to emit event
     }
 
     open() {
@@ -97,7 +84,7 @@ export class OB11PassiveHttpAdapter implements IOB11NetworkAdapter {
         }
 
         const actionName = req.path.split('/')[1];
-        const action = this.actionMap.get(actionName);
+        const action = this.actions.get(actionName);
         if (action) {
             try {
                 const result = await action.handle(payload);

src/onebot/network/passive-websocket.ts

@@ -10,6 +10,7 @@ import { NapCatOneBot11Adapter } from '..';
 import { LogWrapper } from '@/common/utils/log';
 import { OB11HeartbeatEvent } from '../event/meta/OB11HeartbeatEvent';
 import { IncomingMessage } from 'http';
+import { ActionMap } from '@/onebot/action';
 
 export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
     wsServer: WebSocketServer;
@@ -18,15 +19,19 @@ export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
     isOpen: boolean = false;
     hasBeenClosed: boolean = false;
     heartbeatInterval: number = 0;
-    obContext: NapCatOneBot11Adapter;
     coreContext: NapCatCore;
     logger: LogWrapper;
-    private actionMap: Map<string, BaseAction<any, any>> = new Map();
     private heartbeatIntervalId: NodeJS.Timeout | null = null;
 
-    constructor(ip: string, port: number, heartbeatInterval: number, token: string, coreContext: NapCatCore, onebotContext: NapCatOneBot11Adapter) {
+    constructor(
+        ip: string,
+        port: number,
+        heartbeatInterval: number,
+        token: string,
+        coreContext: NapCatCore,
+        public actions: ActionMap
+    ) {
         this.coreContext = coreContext;
-        this.obContext = onebotContext;
         this.logger = coreContext.context.logger;
 
         this.heartbeatInterval = heartbeatInterval;
@@ -58,14 +63,6 @@ export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
         }).on('error', (err) => this.logger.log('[OneBot] [WebSocket Server] Server Error:', err.message));
     }
 
-    registerActionMap(actionMap: Map<string, BaseAction<any, any>>) {
-        this.actionMap = actionMap;
-    }
-
-    registerAction<T extends BaseAction<P, R>, P, R>(action: T) {
-        this.actionMap.set(action.actionName, action);
-    }
-
     onEvent<T extends OB11EmitEventContent>(event: T) {
         this.wsClientsMutex.runExclusive(async () => {
             this.wsClients.forEach((wsClient) => {
@@ -79,7 +76,11 @@ export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
             this.logger.logError('[OneBot] [WebSocket Server] Cannot open a opened WebSocket server');
             return;
         }
-        let addressInfo = this.wsServer.address();
+        if (this.hasBeenClosed) {
+            this.logger.logError('[OneBot] [WebSocket Server] Cannot open a WebSocket server that has been closed');
+            return;
+        }
+        const addressInfo = this.wsServer.address();
         this.logger.log('[OneBot] [WebSocket Server] Server Started', typeof (addressInfo) === 'string' ? addressInfo : addressInfo?.address + ':' + addressInfo?.port);
         
         this.isOpen = true;
@@ -137,7 +138,7 @@ export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
                 this.checkStateAndReply<any>(OB11Response.error('json解析失败,请检查数据格式', 1400, echo), wsClient);
             }
             receiveData.params = (receiveData?.params) ? receiveData.params : {};//兼容类型验证
-            const retdata = await this.actionMap.get(receiveData.action)?.websocketHandle(receiveData.params, echo || '');
+            const retdata = await this.actions.get(receiveData.action)?.websocketHandle(receiveData.params, echo || '');
             const packet = Object.assign({}, retdata);
             this.checkStateAndReply<any>(packet, wsClient);
         } catch (e) {