@nocobase/server 1.5.0-beta.30 → 1.5.0-beta.31

package/lib/gateway/index.js

@@ -355,51 +355,6 @@ const _Gateway = class _Gateway extends import_events.EventEmitter {
     }
     this.server = import_http.default.createServer(this.getCallback());
     this.wsServer = new import_ws_server.WSServer();
-    this.wsServer.on("message", async ({ client, message }) => {
-      const app = await import_app_supervisor.AppSupervisor.getInstance().getApp(client.app);
-      if (!app) {
-        return;
-      }
-      const parsedMessage = JSON.parse(message.toString());
-      if (!parsedMessage.type) {
-        return;
-      }
-      if (!app.listenerCount(`ws:setTag`)) {
-        app.on("ws:setTag", ({ clientId, tagKey, tagValue }) => {
-          this.wsServer.setClientTag(clientId, tagKey, tagValue);
-        });
-        app.on("ws:removeTag", ({ clientId, tagKey }) => {
-          this.wsServer.removeClientTag(clientId, tagKey);
-        });
-        app.on("ws:sendToTag", ({ tagKey, tagValue, message: message2 }) => {
-          this.wsServer.sendToConnectionsByTags(
-            [
-              { tagName: tagKey, tagValue },
-              { tagName: "app", tagValue: app.name }
-            ],
-            message2
-          );
-        });
-        app.on("ws:sendToTags", ({ tags, message: message2 }) => {
-          this.wsServer.sendToConnectionsByTags(tags, message2);
-        });
-        app.on("ws:authorized", ({ clientId, userId }) => {
-          this.wsServer.sendToConnectionsByTags(
-            [
-              { tagName: "userId", tagValue: userId },
-              { tagName: "app", tagValue: app.name }
-            ],
-            { type: "authorized" }
-          );
-        });
-      }
-      const eventName = `ws:message:${parsedMessage.type}`;
-      app.emit(eventName, {
-        clientId: client.id,
-        tags: [...client.tags],
-        payload: parsedMessage.payload
-      });
-    });
     this.server.on("upgrade", (request, socket, head) => {
       const { pathname } = (0, import_url.parse)(request.url);
       if (pathname === import_node_process.default.env.WS_PATH) {

package/lib/gateway/ws-server.d.ts

@@ -28,6 +28,7 @@ export declare class WSServer extends EventEmitter {
     webSocketClients: Map<string, WebSocketClient>;
     logger: Logger;
     constructor();
+    bindAppWSEvents(app: any): void;
     addNewConnection(ws: WebSocketWithId, request: IncomingMessage): WebSocketClient;
     setClientTag(clientId: string, tagKey: string, tagValue: string): void;
     removeClientTag(clientId: string, tagKey: string): void;
@@ -44,6 +45,7 @@ export declare class WSServer extends EventEmitter {
         tagName: string;
         tagValue: string;
     }>, sendMessage: object): void;
+    sendToClient(clientId: string, sendMessage: object): void;
     loopThroughConnections(callback: (client: WebSocketClient) => void): void;
     close(): void;
 }

package/lib/gateway/ws-server.js

@@ -133,6 +133,57 @@ const _WSServer = class _WSServer extends import_events.default {
         }
       });
     });
+    import_app_supervisor.AppSupervisor.getInstance().on("afterAppAdded", (app) => {
+      this.bindAppWSEvents(app);
+    });
+    this.on("message", async ({ client, message }) => {
+      const app = await import_app_supervisor.AppSupervisor.getInstance().getApp(client.app);
+      if (!app) {
+        return;
+      }
+      const parsedMessage = JSON.parse(message.toString());
+      if (!parsedMessage.type) {
+        return;
+      }
+      const eventName = `ws:message:${parsedMessage.type}`;
+      app.emit(eventName, {
+        clientId: client.id,
+        tags: [...client.tags],
+        payload: parsedMessage.payload
+      });
+    });
+  }
+  bindAppWSEvents(app) {
+    if (app.listenerCount("ws:setTag") > 0) {
+      return;
+    }
+    app.on("ws:setTag", ({ clientId, tagKey, tagValue }) => {
+      this.setClientTag(clientId, tagKey, tagValue);
+    });
+    app.on("ws:removeTag", ({ clientId, tagKey }) => {
+      this.removeClientTag(clientId, tagKey);
+    });
+    app.on("ws:sendToTag", ({ tagKey, tagValue, message }) => {
+      this.sendToConnectionsByTags(
+        [
+          { tagName: tagKey, tagValue },
+          { tagName: "app", tagValue: app.name }
+        ],
+        message
+      );
+    });
+    app.on("ws:sendToClient", ({ clientId, message }) => {
+      this.sendToClient(clientId, message);
+    });
+    app.on("ws:sendToCurrentApp", ({ message }) => {
+      this.sendToConnectionsByTag("app", app.name, message);
+    });
+    app.on("ws:sendToTags", ({ tags, message }) => {
+      this.sendToConnectionsByTags(tags, message);
+    });
+    app.on("ws:authorized", ({ clientId, userId }) => {
+      this.sendToClient(clientId, { type: "authorized" });
+    });
   }
   addNewConnection(ws, request) {
     const id = (0, import_nanoid.nanoid)();
@@ -149,6 +200,9 @@ const _WSServer = class _WSServer extends import_events.default {
   }
   setClientTag(clientId, tagKey, tagValue) {
     const client = this.webSocketClients.get(clientId);
+    if (!client) {
+      return;
+    }
     client.tags.add(`${tagKey}#${tagValue}`);
     console.log(`client tags: ${Array.from(client.tags)}`);
   }
@@ -197,6 +251,12 @@ const _WSServer = class _WSServer extends import_events.default {
       }
     });
   }
+  sendToClient(clientId, sendMessage) {
+    const client = this.webSocketClients.get(clientId);
+    if (client) {
+      this.sendMessageToConnection(client, sendMessage);
+    }
+  }
   loopThroughConnections(callback) {
     this.webSocketClients.forEach((client) => {
       callback(client);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@nocobase/server",
-    "version": "1.5.0-beta.30",
+    "version": "1.5.0-beta.31",
     "main": "lib/index.js",
     "types": "./lib/index.d.ts",
     "license": "AGPL-3.0",
@@ -10,19 +10,19 @@
         "@koa/cors": "^3.1.0",
         "@koa/multer": "^3.0.2",
         "@koa/router": "^9.4.0",
-        "@nocobase/acl": "1.5.0-beta.30",
-        "@nocobase/actions": "1.5.0-beta.30",
-        "@nocobase/auth": "1.5.0-beta.30",
-        "@nocobase/cache": "1.5.0-beta.30",
-        "@nocobase/data-source-manager": "1.5.0-beta.30",
-        "@nocobase/database": "1.5.0-beta.30",
-        "@nocobase/evaluators": "1.5.0-beta.30",
-        "@nocobase/lock-manager": "1.5.0-beta.30",
-        "@nocobase/logger": "1.5.0-beta.30",
-        "@nocobase/resourcer": "1.5.0-beta.30",
-        "@nocobase/sdk": "1.5.0-beta.30",
-        "@nocobase/telemetry": "1.5.0-beta.30",
-        "@nocobase/utils": "1.5.0-beta.30",
+        "@nocobase/acl": "1.5.0-beta.31",
+        "@nocobase/actions": "1.5.0-beta.31",
+        "@nocobase/auth": "1.5.0-beta.31",
+        "@nocobase/cache": "1.5.0-beta.31",
+        "@nocobase/data-source-manager": "1.5.0-beta.31",
+        "@nocobase/database": "1.5.0-beta.31",
+        "@nocobase/evaluators": "1.5.0-beta.31",
+        "@nocobase/lock-manager": "1.5.0-beta.31",
+        "@nocobase/logger": "1.5.0-beta.31",
+        "@nocobase/resourcer": "1.5.0-beta.31",
+        "@nocobase/sdk": "1.5.0-beta.31",
+        "@nocobase/telemetry": "1.5.0-beta.31",
+        "@nocobase/utils": "1.5.0-beta.31",
         "@types/decompress": "4.2.7",
         "@types/ini": "^1.3.31",
         "@types/koa-send": "^4.1.3",
@@ -56,5 +56,5 @@
         "@types/serve-handler": "^6.1.1",
         "@types/ws": "^8.5.5"
     },
-    "gitHead": "40c48b2ecf3e2315a76dcb729dbda825b6a1331c"
+    "gitHead": "75fe2500e19dfbb2ae1a29a27f136f3b794e633a"
 }