buildTreeNode() — react Function Reference

Source Code

compiler/packages/babel-plugin-react-compiler/src/Validation/ValidateNoDerivedComputationsInEffects_exp.ts lines 501–567

function buildTreeNode(
  sourceId: IdentifierId,
  context: ValidationContext,
  visited: Set<string> = new Set(),
): Array<TreeNode> {
  const sourceMetadata = context.derivationCache.cache.get(sourceId);
  if (!sourceMetadata) {
    return [];
  }

  if (sourceMetadata.isStateSource && isNamedIdentifier(sourceMetadata.place)) {
    return [
      {
        name: sourceMetadata.place.identifier.name.value,
        typeOfValue: sourceMetadata.typeOfValue,
        isSource: sourceMetadata.isStateSource,
        children: [],
      },
    ];
  }

  const children: Array<TreeNode> = [];

  const namedSiblings: Set<string> = new Set();
  for (const childId of sourceMetadata.sourcesIds) {
    CompilerError.invariant(childId !== sourceId, {
      reason:
        'Unexpected self-reference: a value should not have itself as a source',
      loc: sourceMetadata.place.loc,
    });

    const childNodes = buildTreeNode(
      childId,
      context,
      new Set([
        ...visited,
        ...(isNamedIdentifier(sourceMetadata.place)
          ? [sourceMetadata.place.identifier.name.value]
          : []),
      ]),
    );
    if (childNodes) {
      for (const childNode of childNodes) {
        if (!namedSiblings.has(childNode.name)) {
          children.push(childNode);
          namedSiblings.add(childNode.name);
        }
      }
    }
  }

  if (
    isNamedIdentifier(sourceMetadata.place) &&
    !visited.has(sourceMetadata.place.identifier.name.value)
  ) {
    return [
      {
        name: sourceMetadata.place.identifier.name.value,
        typeOfValue: sourceMetadata.typeOfValue,
        isSource: sourceMetadata.isStateSource,
        children: children,
      },
    ];
  }

  return children;
}