feat: add `websocket.ts`

.claude/settings.local.json

@@ -11,7 +11,11 @@
       "mcp__chrome-devtools__evaluate_script",
       "mcp__chrome-devtools__get_network_request",
       "Bash(cat:*)",
-      "mcp__chrome-devtools__wait_for"
+      "mcp__chrome-devtools__wait_for",
+      "Bash(find:*)",
+      "Bash(xargs -I {} sh -c 'echo \"\"\"\"=== {} ===\"\"\"\" && grep -E \"\"\"\"<module>|<artifactId>\"\"\"\" {} 2>/dev/null | head -20')",
+      "mcp__chrome-devtools__list_console_messages",
+      "Bash(dir:*)"
     ],
     "deny": [],
     "ask": []

package.json

@@ -46,6 +46,10 @@
       "types": "./dist/plugs/http/index.d.ts",
       "default": "./dist/plugs/http/index.js"
     },
+    "./ws": {
+      "types": "./dist/plugs/websocket.d.ts",
+      "default": "./dist/plugs/websocket.js"
+    },
     "./i18n": {
       "types": "./dist/plugs/i18n/index.d.ts",
       "default": "./dist/plugs/i18n/index.js"

plugs/index.ts

@@ -4,6 +4,7 @@ export * as Store from "./store";
 export * as Http from "./http";
 export * as Lang from "./i18n";
 export * as Api from "./api";
+export * as WebSocket from "./websocket";
 export * from "./constant";
 export * from "./util/asyncUtil";
 export * from "./util/objectUtil";

plugs/websocket.ts

@@ -0,0 +1,52 @@
+let websocket: WebSocket;
+const msgHandlersMap = {};
+let lastUrl;
+let closeFlag = false;
+
+export const closeWebSocket = () => {
+  closeFlag = false;
+  websocket?.close();
+};
+
+export const sendSocketMsg = (msg) => {
+  websocket?.send(JSON.stringify(msg));
+};
+
+export const openWebSocket = (url) => {
+  if (websocket && websocket.readyState === WebSocket.OPEN) {
+    if (lastUrl == url) {
+      return;
+    }
+    closeWebSocket();
+  }
+  websocket = new WebSocket(url);
+  lastUrl = url;
+  websocket.onopen = () => {
+    console.log("websocket已连接");
+    closeFlag = true;
+  };
+  websocket.onmessage = (msg) => {
+    const event = JSON.parse(msg.data);
+    const handlers = msgHandlersMap[event.type];
+    if (!handlers) {
+      return;
+    }
+    Object.values(handlers).forEach((handler: any) => {
+      handler(event.data);
+    });
+  };
+  websocket.onclose = () => {
+    console.log("websocket已断开");
+    if (closeFlag) {
+      setTimeout(() => {
+        openWebSocket(url);
+      }, 2000);
+    }
+  };
+};
+
+export const registerHandler = (type, key, handler) => {
+  const handlers = msgHandlersMap[type] || {};
+  handlers[key] = handler;
+  msgHandlersMap[type] = handlers;
+};

tsconfig.json

@@ -42,6 +42,9 @@
       "noob-mengyxu/http": [
         "./plugs/http/index.ts"
       ],
+      "noob-mengyxu/ws": [
+        "./plugs/websockets.ts"
+      ],
       "noob-mengyxu/i18n": [
         "./plugs/i18n/index.ts"
       ],

vite.config.ts

@@ -50,6 +50,7 @@ export default defineConfig(({ command, mode }) => {
               "plugs/element/index": resolve(__dirname, "plugs/element/index.ts"),
               "plugs/store/index": resolve(__dirname, "plugs/store/index.ts"),
               "plugs/util/index": resolve(__dirname, "plugs/util/index.ts"),
+              "plugs/websocket": resolve(__dirname, "plugs/websocket.ts"),
             },
             formats: ["es"],
           },