Class WindowMessageConnect

packages/message/window_message.ts:135–214

}

export class WindowMessageConnect implements MessageConnect {
  private readonly listenerId = `${uuidv4()}`; // 使用 uuidv4 确保唯一
  private target: PostMessage | null;
  private isSelfDisconnected = false;

  constructor(
    private messageId: string,
    EE: EventEmitter<string, any>,
    target: PostMessage
  ) {
    this.target = target; // 强引用
    const handler = (msg: TMessage) => {
      listenerMgr.emit(`onMessage:${this.listenerId}`, msg);
    };
    const cleanup = () => {
      if (this.target) {
        this.target = null;
        listenerMgr.removeAllListeners(`cleanup:${this.listenerId}`);
        EE.removeAllListeners("connectMessage:" + this.messageId); // 模拟 con.onMessage.removeListener
        EE.removeAllListeners("disconnect:" + this.messageId); // 模拟 con.onDisconnect.removeListener
        listenerMgr.emit(`onDisconnect:${this.listenerId}`, this.isSelfDisconnected);
        listenerMgr.removeAllListeners(`onDisconnect:${this.listenerId}`);
        listenerMgr.removeAllListeners(`onMessage:${this.listenerId}`);
      }
    };
    EE.addListener(`connectMessage:${this.messageId}`, handler); // 模拟 con.onMessage.addListener
    EE.addListener(`disconnect:${this.messageId}`, cleanup); // 模拟 con.onDisconnect.addListener
    listenerMgr.once(`cleanup:${this.listenerId}`, cleanup);
  }

  sendMessage(data: TMessage) {
    if (!this.target) {
      console.error("Attempted to sendMessage on a disconnected Target.");
      // 無法 sendMessage 不应该屏蔽错误
      throw new Error("Attempted to sendMessage on a disconnected Target.");
    }
    const body: WindowMessageBody<TMessage> = {
      messageId: this.messageId,
      type: "connectMessage",
      data,
    };
    this.target.postMessage(body);
  }

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

  disconnect(ignoreAlreadyDisconnected: boolean) {
    if (!this.target) {
      if (ignoreAlreadyDisconnected) return;
      console.warn("Attempted to disconnect on a disconnected Target.");
      // 重复 disconnect() 不应该屏蔽错误