Class ExtensionMessageConnect

packages/message/extension_message.ts:158–232

}

export class ExtensionMessageConnect implements MessageConnect {
  private readonly listenerId = `${uuidv4()}`; // 使用 uuidv4 确保唯一
  private con: chrome.runtime.Port | null;
  private isSelfDisconnected = false;

  constructor(con: chrome.runtime.Port) {
    this.con = con; // 强引用
    const handler = (msg: TMessage, _con: chrome.runtime.Port) => {
      listenerMgr.emit(`onMessage:${this.listenerId}`, msg);
    };
    const cleanup = () => {
      const con = this.con;
      if (con !== null) {
        this.con = null;
        listenerMgr.removeAllListeners(`cleanup:${this.listenerId}`);
        con.onMessage.removeListener(handler);
        con.onDisconnect.removeListener(cleanup);
        listenerMgr.emit(`onDisconnect:${this.listenerId}`, this.isSelfDisconnected);
        listenerMgr.removeAllListeners(`onDisconnect:${this.listenerId}`);
        listenerMgr.removeAllListeners(`onMessage:${this.listenerId}`);
      }
    };
    con.onMessage.addListener(handler);
    con.onDisconnect.addListener(cleanup);
    listenerMgr.once(`cleanup:${this.listenerId}`, cleanup);
  }

  sendMessage(data: TMessage) {
    if (!this.con) {
      console.warn("Attempted to sendMessage on a disconnected port.");
      // 無法 sendMessage 不应该屏蔽错误
      throw new Error("Attempted to sendMessage on a disconnected port.");
    }
    this.con.postMessage(data);
  }

  onMessage(callback: (data: TMessage) => void) {
    if (!this.con) {
      console.error("onMessage Invalid Port");
      // 無法監聽的話不应该屏蔽错误
      throw new Error("onMessage Invalid Port");
    }
    listenerMgr.addListener(`onMessage:${this.listenerId}`, callback);
  }

  disconnect(ignoreAlreadyDisconnected?: boolean) {
    if (!this.con) {
      if (ignoreAlreadyDisconnected) return;
      console.warn("Attempted to disconnect on a disconnected port.");
      // 重复 disconnect() 不应该屏蔽错误
      throw new Error("Attempted to disconnect on a disconnected port.");
    }
    this.isSelfDisconnected = true;
    this.con?.disconnect();
    // Note: .disconnect() will NOT automatically trigger the 'cleanup' listener
    listenerMgr.emit(`cleanup:${this.listenerId}`);
  }