Function buildImmediateTextSegments

packages/cli/src/transform.ts:1135–1174

(parentNode: any, source?: string)

  | { type: "boundary"; text: string; leftChildIndex: number; rightChildIndex: number };

function buildImmediateTextSegments(parentNode: any, source?: string): ImmediateTextSegment[] {
  const children = parentNode.children;
  if (!children || children.length === 0) return [];

  const lineStarts = source ? getLineStarts(source) : undefined;
  const segments: ImmediateTextSegment[] = [];
  let previousChildIndex: number | null = null;
  let previousText = "";

  for (let i = 0; i < children.length; i++) {
    const childText = getRenderedChildText(children[i], source, lineStarts);
    if (!childText) continue;

    if (previousChildIndex != null) {
      const boundarySource = getSourceBetween(children[previousChildIndex], children[i], source, lineStarts);
      if (needsBoundarySpace(previousText, childText, boundarySource)) {
        segments.push({ type: "boundary", text: " ", leftChildIndex: previousChildIndex, rightChildIndex: i });
      }
    }

    segments.push({ type: "child", text: childText, childIndex: i });
    previousChildIndex = i;
    previousText = childText;
  }

  const collapsedSegments: ImmediateTextSegment[] = [];
  let previousEndsWithWhitespace = false;

  for (const segment of segments) {
    if (!segment.text) continue;
    const whitespaceOnly = /^[\t\n\f\r ]+$/.test(segment.text);
    if (whitespaceOnly && previousEndsWithWhitespace) {
      continue;
    }
    collapsedSegments.push(whitespaceOnly ? { ...segment, text: " " } : segment);
    previousEndsWithWhitespace = /\s$/.test(collapsedSegments[collapsedSegments.length - 1].text);
  }

  return collapsedSegments;
}

function getSegmentsText(segments: ImmediateTextSegment[]): string {
  return segments.map((segment) => segment.text).join("");