@talex-touch/utils 1.0.14 → 1.0.15

package/plugin/channel.ts

@@ -1,193 +1,203 @@
-// const { ipcRenderer, IpcMainEvent } = require("electron");
-import { ipcRenderer, type IpcRendererEvent } from "electron";
-import {
-  ChannelType,
-  DataCode,
-  ITouchClientChannel,
-  RawChannelSyncData,
-  RawStandardChannelData,
-  StandardChannelData,
-} from "../channel";
-
-class TouchChannel implements ITouchClientChannel {
-  channelMap: Map<string, Function[]> = new Map();
-
-  pendingMap: Map<string, Function> = new Map();
-
-  plugin: string;
-
-  constructor(pluginName: string) {
-    this.plugin = pluginName;
-
-    ipcRenderer.on("@plugin-process-message", this.__handle_main.bind(this));
-  }
-
-  __parse_raw_data(e: IpcRendererEvent, arg: any): RawStandardChannelData | null {
-    console.debug("Raw data: ", arg, e);
-    if (arg) {
-      const { name, header, code, data, sync } = arg;
-
-      if (header) {
-        return {
-          header: {
-            status: header.status || "request",
-            type: ChannelType.PLUGIN,
-            _originData: arg,
-            event: e
-          },
-          sync,
-          code,
-          data,
-          name: name as string,
-        };
-      }
-    }
-
-    console.error(e, arg);
-    return null;
-    // throw new Error("Invalid message!");
-  }
-
-  __handle_main(e: typeof IpcRendererEvent, _arg: any): any {
-    const arg = JSON.parse(_arg)
-    const rawData = this.__parse_raw_data(e, arg);
-    if ( !rawData ) return
-
-    if ( rawData.header.status === 'reply' && rawData.sync ) {
-      const { id } = rawData.sync;
-
-      return this.pendingMap.get(id)?.(rawData);
-    }
-
-    // if ( rawData.plugin !== this.plugin ) return
-
-    this.channelMap.get(rawData.name)?.forEach((func) => {
-      const handInData: StandardChannelData = {
-        reply: (code: DataCode, data: any) => {
-          e.sender.send(
-            "@plugin-process-message",
-            this.__parse_sender(code, rawData, data, rawData.sync)
-          );
-        },
-        ...rawData,
-      };
-
-      const res = func(handInData);
-
-      if (res && res instanceof Promise) return;
-
-      handInData.reply(DataCode.SUCCESS, res);
-    });
-  }
-
-  __parse_sender(
-    code: DataCode,
-    rawData: RawStandardChannelData,
-    data: any,
-    sync?: RawChannelSyncData
-  ): RawStandardChannelData {
-    return {
-      code,
-      data,
-      sync: !sync
-        ? undefined
-        : {
-            timeStamp: new Date().getTime(),
-            // reply sync timeout should follow the request timeout, unless user set it.
-            timeout: sync.timeout,
-            id: sync.id,
-          },
-      name: rawData.name,
-      header: {
-        status: "reply",
-        type: rawData.header.type,
-        _originData: rawData.header._originData,
-      },
-    };
-  }
-
-  regChannel(
-    eventName: string,
-    callback: Function
-  ): () => void {
-    const listeners = this.channelMap.get(eventName) || [];
-
-    if (!listeners.includes(callback)) {
-      listeners.push(callback);
-    } else return () => void 0;
-
-    this.channelMap.set(eventName, listeners);
-
-    return () => {
-      const index = listeners.indexOf(callback);
-
-      if (index !== -1) {
-        listeners.splice(index, 1);
-      }
-    };
-  }
-
-  send(eventName: string, arg: any): Promise<any> {
-    const uniqueId = `${new Date().getTime()}#${eventName}@${Math.random().toString(
-      12
-    )}`;
-
-    const data = {
-      code: DataCode.SUCCESS,
-      data: arg,
-      sync: {
-        timeStamp: new Date().getTime(),
-        timeout: 10000,
-        id: uniqueId,
-      },
-      name: eventName,
-      plugin: this.plugin,
-      header: {
-        status: "request",
-        type: ChannelType.PLUGIN,
-      },
-    } as RawStandardChannelData;
-
-    return new Promise((resolve) => {
-      
-      ipcRenderer.send("@plugin-process-message", data);
-
-      this.pendingMap.set(uniqueId, (res: any) => {
-        this.pendingMap.delete(uniqueId);
-
-        resolve(res.data);
-      });
-    });
-  }
-
-  sendSync(eventName: string, arg?: any): any {
-    const data = {
-      code: DataCode.SUCCESS,
-      data: arg,
-      name: eventName,
-      plugin: this.plugin,
-      header: {
-        status: "request",
-        type: ChannelType.PLUGIN,
-      },
-    } as RawStandardChannelData;
-
-    const res = this.__parse_raw_data(null, ipcRenderer.sendSync("@plugin-process-message", data))!
-    
-    if ( res.header.status === 'reply' ) return res.data;
-
-    return res;
-
-  }
-}
-
-let touchChannel: ITouchClientChannel
-
-export function genChannel() {
-  if (!touchChannel) {
-    // @ts-ignore
-    touchChannel = window.$channel = new TouchChannel(window.$plugin.name)
-  }
-
-  return touchChannel
-}
+// import { ipcRenderer } from 'electron';
+// const { ipcRenderer, IpcMainEvent } = require("electron");
+import { IpcRenderer, type IpcRendererEvent } from "electron";
+import {
+  ChannelType,
+  DataCode,
+  ITouchClientChannel,
+  RawChannelSyncData,
+  RawStandardChannelData,
+  StandardChannelData,
+} from "../channel";
+
+let ipcRenderer: IpcRenderer
+
+if (typeof window === 'undefined') {
+  ipcRenderer = require('electron').ipcRenderer
+} else {
+  ipcRenderer = window.electron.ipcRenderer as unknown as IpcRenderer
+}
+
+class TouchChannel implements ITouchClientChannel {
+  channelMap: Map<string, Function[]> = new Map();
+
+  pendingMap: Map<string, Function> = new Map();
+
+  plugin: string;
+
+  constructor(pluginName: string) {
+    this.plugin = pluginName;
+
+    ipcRenderer.on("@plugin-process-message", this.__handle_main.bind(this));
+  }
+
+  __parse_raw_data(e: IpcRendererEvent | undefined, arg: any): RawStandardChannelData | null {
+    console.debug("Raw data: ", arg, e);
+    if (arg) {
+      const { name, header, code, data, sync } = arg;
+
+      if (header) {
+        return {
+          header: {
+            status: header.status || "request",
+            type: ChannelType.PLUGIN,
+            _originData: arg,
+            event: e
+          },
+          sync,
+          code,
+          data,
+          name: name as string,
+        };
+      }
+    }
+
+    console.error(e, arg);
+    return null;
+    // throw new Error("Invalid message!");
+  }
+
+  __handle_main(e: IpcRendererEvent, _arg: any): any {
+    const arg = JSON.parse(_arg)
+    const rawData = this.__parse_raw_data(e, arg);
+    if ( !rawData ) return
+
+    if ( rawData.header.status === 'reply' && rawData.sync ) {
+      const { id } = rawData.sync;
+
+      return this.pendingMap.get(id)?.(rawData);
+    }
+
+    // if ( rawData.plugin !== this.plugin ) return
+
+    this.channelMap.get(rawData.name)?.forEach((func) => {
+      const handInData: StandardChannelData = {
+        reply: (code: DataCode, data: any) => {
+          e.sender.send(
+            "@plugin-process-message",
+            this.__parse_sender(code, rawData, data, rawData.sync)
+          );
+        },
+        ...rawData,
+      };
+
+      const res = func(handInData);
+
+      if (res && res instanceof Promise) return;
+
+      handInData.reply(DataCode.SUCCESS, res);
+    });
+  }
+
+  __parse_sender(
+    code: DataCode,
+    rawData: RawStandardChannelData,
+    data: any,
+    sync?: RawChannelSyncData
+  ): RawStandardChannelData {
+    return {
+      code,
+      data,
+      sync: !sync
+        ? undefined
+        : {
+            timeStamp: new Date().getTime(),
+            // reply sync timeout should follow the request timeout, unless user set it.
+            timeout: sync.timeout,
+            id: sync.id,
+          },
+      name: rawData.name,
+      header: {
+        event: rawData.header.event,
+        status: "reply",
+        type: rawData.header.type,
+        _originData: rawData.header._originData,
+      },
+    };
+  }
+
+  regChannel(
+    eventName: string,
+    callback: Function
+  ): () => void {
+    const listeners = this.channelMap.get(eventName) || [];
+
+    if (!listeners.includes(callback)) {
+      listeners.push(callback);
+    } else return () => void 0;
+
+    this.channelMap.set(eventName, listeners);
+
+    return () => {
+      const index = listeners.indexOf(callback);
+
+      if (index !== -1) {
+        listeners.splice(index, 1);
+      }
+    };
+  }
+
+  send(eventName: string, arg: any): Promise<any> {
+    const uniqueId = `${new Date().getTime()}#${eventName}@${Math.random().toString(
+      12
+    )}`;
+
+    const data = {
+      code: DataCode.SUCCESS,
+      data: arg,
+      sync: {
+        timeStamp: new Date().getTime(),
+        timeout: 10000,
+        id: uniqueId,
+      },
+      name: eventName,
+      plugin: this.plugin,
+      header: {
+        status: "request",
+        type: ChannelType.PLUGIN,
+      },
+    } as RawStandardChannelData;
+
+    return new Promise((resolve) => {
+
+      ipcRenderer.send("@plugin-process-message", data);
+
+      this.pendingMap.set(uniqueId, (res: any) => {
+        this.pendingMap.delete(uniqueId);
+
+        resolve(res.data);
+      });
+    });
+  }
+
+  sendSync(eventName: string, arg?: any): any {
+    const data = {
+      code: DataCode.SUCCESS,
+      data: arg,
+      name: eventName,
+      plugin: this.plugin,
+      header: {
+        status: "request",
+        type: ChannelType.PLUGIN,
+      },
+    } as RawStandardChannelData;
+
+    const res = this.__parse_raw_data(void 0, ipcRenderer.sendSync("@plugin-process-message", data))!
+
+    if ( res.header.status === 'reply' ) return res.data;
+
+    return res;
+
+  }
+}
+
+let touchChannel: ITouchClientChannel
+
+export function genChannel() {
+  if (!touchChannel) {
+    // @ts-ignore
+    touchChannel = window.$channel = new TouchChannel(window.$plugin.name)
+  }
+
+  return touchChannel
+}