优化消息

packages/message/server.ts

@@ -0,0 +1,45 @@
+export type ApiFunction = (params: any, con: chrome.runtime.Port | chrome.runtime.MessageSender) => any;
+
+export class Server {
+  apiFunctionMap: Map<string, ApiFunction> = new Map();
+
+  constructor() {
+    chrome.runtime.onConnect.addListener((port) => {
+      port.onMessage.addListener((msg: { action: string }) => {
+        this.connectHandle(msg.action, msg, port);
+      });
+    });
+
+    chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
+      this.messageHandle(msg.action, msg, sender, sendResponse);
+    });
+  }
+
+  on(name: string, func: ApiFunction) {
+    this.apiFunctionMap.set(name, func);
+  }
+
+  private connectHandle(msg: string, params: any, con: chrome.runtime.Port) {
+    const func = this.apiFunctionMap.get(msg);
+    if (func) {
+      func(params, con);
+    }
+  }
+
+  private messageHandle(
+    msg: string,
+    params: any,
+    sender: chrome.runtime.MessageSender,
+    sendResponse: (response: any) => void
+  ) {
+    const func = this.apiFunctionMap.get(msg);
+    if (func) {
+      try {
+        const ret = func(params, sender);
+        sendResponse({ code: 0, data: ret });
+      } catch (e: any) {
+        sendResponse({ code: -1, message: e.message });
+      }
+    }
+  }
+}