Method insertExtraNodes

src/TextSnapshot.ts:154–324

(
    page: McpPage,
    idToNode: Map<string, TextSnapshotNode>,
    seenUniqueIds: Set<string>,
    snapshotId: number,
    idCounter: number,
    rootNodeWithId: TextSnapshotNode,
    seenBackendNodeIds: Set<number>,
    extraHandles: ElementHandle[],
  )

  // returned by third-party developer tools. We insert them into the tree by finding the closest ancestor
  // in the tree and inserting the node as a child. The ancestor's child nodes are re-parented if necessary.
  private static async insertExtraNodes(
    page: McpPage,
    idToNode: Map<string, TextSnapshotNode>,
    seenUniqueIds: Set<string>,
    snapshotId: number,
    idCounter: number,
    rootNodeWithId: TextSnapshotNode,
    seenBackendNodeIds: Set<number>,
    extraHandles: ElementHandle[],
  ): Promise<void> {
    const {uniqueBackendNodeIdToMcpId} = page;

    const createExtraNode = async (
      handle: ElementHandle,
    ): Promise<TextSnapshotNode | null> => {
      const backendNodeId = await handle.backendNodeId();
      if (!backendNodeId || seenBackendNodeIds.has(backendNodeId)) {
        return null;
      }
      const uniqueBackendId = `custom_${backendNodeId}`;
      if (seenUniqueIds.has(uniqueBackendId)) {
        return null;
      }
      seenBackendNodeIds.add(backendNodeId);

      let id = '';
      const mcpId = uniqueBackendNodeIdToMcpId.get(uniqueBackendId);
      if (mcpId !== undefined) {
        id = mcpId;
      } else {
        id = `${snapshotId}_${idCounter++}`;
        uniqueBackendNodeIdToMcpId.set(uniqueBackendId, id);
      }
      seenUniqueIds.add(uniqueBackendId);

      const tagHandle = await handle.getProperty('localName');
      const tagValue = await tagHandle.jsonValue();
      const extraNode: TextSnapshotNode = {
        role: tagValue,
        id,
        backendNodeId,
        children: [],
        elementHandle: async () => handle,
      };
      return extraNode;
    };

    const findAncestorNode = async (
      handle: ElementHandle,
    ): Promise<TextSnapshotNode | null> => {
      let ancestorHandle = await handle.evaluateHandle(el => el.parentElement);

      while (ancestorHandle) {
        const ancestorElement = ancestorHandle.asElement();
        if (!ancestorElement) {
          await ancestorHandle.dispose();
          return null;
        }