gm api通信

packages/message/client.ts

@@ -1,32 +1,24 @@
-export function sendMessage(action: string, data?: any): Promise<any> {
-  return new Promise((resolve, reject) => {
-    chrome.runtime.sendMessage({ action, data }, (res) => {
-      if (res.code) {
-        console.error(res);
-        reject(res.message);
-      } else {
-        resolve(res.data);
-      }
-    });
-  });
+import { Message, MessageConnect } from "./server";
+
+export function sendMessage(msg: Message, action: string, data?: any): Promise<any> {
+  return msg.sendMessage({ action, data });
 }
 
-export function connect(action: string, data?: any): Promise<chrome.runtime.Port> {
-  return new Promise((resolve) => {
-    const port = chrome.runtime.connect();
-    port.postMessage({ action, data });
-    resolve(port);
-  });
+export function connect(msg: Message, action: string, data?: any): Promise<MessageConnect> {
+  return msg.connect({ action, data });
 }
 
 export class Client {
-  constructor(private prefix: string) {
+  constructor(
+    private msg: Message,
+    private prefix: string
+  ) {
     if (!this.prefix.endsWith("/")) {
       this.prefix += "/";
     }
   }
 
   do(action: string, params?: any): Promise<any> {
-    return sendMessage(this.prefix + action, params);
+    return sendMessage(this.msg, this.prefix + action, params);
   }
 }