BetterSEQTA-Plus/src/seqta/utils/ReactFiber.ts

class ReactFiber {
  private selector: string;
  private debug: boolean;
  private messageIdCounter: number = 0; // Counter for unique message IDs

  constructor(selector: string, options: {
    debug ? : boolean
  } = {}) {
    this.selector = selector;
    this.debug = options.debug || false;
  }

  static find(selector: string, options: {
    debug ? : boolean
  } = {}) {
    return new ReactFiber(selector, options);
  }

  private async sendMessage(action: string, payload: any = {}): Promise < any > {
    return new Promise((resolve, _) => {
      const messageId = this.messageIdCounter++;
      const message = {
        type: "reactFiberRequest",
        selector: this.selector,
        action,
        payload,
        debug: this.debug,
        messageId,
      };

      const listener = (response: any) => {
        if (response.data?.type === 'reactFiberResponse' && response.data?.messageId === messageId) {
          if (this.debug) {
            console.log("Content Received Response:", response.data.response);
          }
          resolve(response.data.response);
          window.removeEventListener("message", listener)
        }
      };
      window.addEventListener('message', listener);
      window.postMessage(message, "*");
    });
  }


  async getState(key ? : string | string[]): Promise < any > {
    return this.sendMessage("getState", {
      key
    });
  }

  async setState(update: any | ((prevState: any) => any)): Promise < ReactFiber > {
    const updateFnString = typeof update === 'function' ? update.toString() : null;
    const updateObject = typeof update !== 'function' ? update : null;

    await this.sendMessage("setState", {
      updateFn: updateFnString,
      updateObject
    });
    return this;
  }


  async getProps(propName ? : string): Promise < any > {
    return this.sendMessage("getProp", {
      propName
    });
  }

  async setProp(propName: string, value: any): Promise < ReactFiber > {
    await this.sendMessage("setProp", {
      propName,
      value
    });
    return this;
  }

  async forceUpdate(): Promise < ReactFiber > {
    await this.sendMessage("forceUpdate");
    return this;
  }
}

export default ReactFiber;