Class ExtensionMessage

packages/message/extension_message.ts:7–156

const listenerMgr = new EventEmitter<string, any>(); // 单一管理器

export class ExtensionMessage implements Message {
  constructor(private backgroundPrimary = false) {}

  connect(data: TMessage): Promise<MessageConnect> {
    return new Promise((resolve) => {
      const con = chrome.runtime.connect();
      con.postMessage(data);
      resolve(new ExtensionMessageConnect(con));
    });
  }

  // 发送消息 注意不进行回调的内存泄漏
  sendMessage<T = any>(data: TMessage): Promise<T> {
    return new Promise((resolve: ((value: T) => void) | null) => {
      chrome.runtime.sendMessage(data, (resp: T) => {
        const lastError = chrome.runtime.lastError;
        if (lastError) {
          console.error("chrome.runtime.lastError in chrome.runtime.sendMessage:", lastError);
          // 通信API出错不回继续对话
        }
        resolve!(resp);
        resolve = null;
      });
    });
  }

  tryEnableUserScriptConnectionListener = (..._args: any) => {
    // empty function
  };
  tryEnableUserScriptMessageListener = (..._args: any) => {
    // empty function
  };

  onConnect(callback: (data: TMessage, con: MessageConnect) => void) {
    chrome.runtime.onConnect.addListener((port: chrome.runtime.Port) => {
      let myPort: chrome.runtime.Port | null = port;
      const lastError = chrome.runtime.lastError;
      if (lastError) {
        console.error("chrome.runtime.lastError in chrome.runtime.onConnect", lastError);
        // 消息API发生错误因此不继续执行
      }
      const handler = (msg: TMessage) => {
        const port = myPort;
        if (port !== null) {
          myPort = null;
          port.onMessage.removeListener(handler);
          callback(msg, new ExtensionMessageConnect(port));
        }
      };
      myPort.onMessage.addListener(handler);
    });

    if (this.backgroundPrimary) {
      let addUserScriptConnectionListener: (() => void) | null = () => {
        try {
          // 监听用户脚本的连接
          chrome.runtime.onUserScriptConnect.addListener((port: chrome.runtime.Port) => {
            let myPort: chrome.runtime.Port | null = port;