Function parseReplySegments

lib/utils.ts:112–173

(markdown: string)

    };

export function parseReplySegments(markdown: string): ReplySegment[] {
  const segments: ReplySegment[] = [];
  const lines = markdown.split("\n");
  const fenceRegex = /^```([^\n]*)$/; // opening or closing fence line

  let textBuffer: string[] = [];
  let codeBuffer: string[] = [];
  let openTag: string | null = null; // e.g. tsx{path=src/App.tsx}

  const flushText = () => {
    if (textBuffer.length > 0) {
      segments.push({ type: "text", content: textBuffer.join("\n") });
      textBuffer = [];
    }
  };

  const parseTag = parseFenceTag;

  for (const line of lines) {
    const match = line.match(fenceRegex);
    if (match && !openTag) {
      // Opening fence
      openTag = match[1] || "";
      flushText();
      codeBuffer = [];
    } else if (match && openTag) {
      // Closing fence
      const { language, path } = parseTag(openTag);
      segments.push({
        type: "file",
        code: codeBuffer.join("\n"),
        language,
        path,
        isPartial: false,
      });
      openTag = null;
      codeBuffer = [];
    } else if (openTag) {
      codeBuffer.push(line);
    } else {
      textBuffer.push(line);
    }
  }

  // If a code fence remains open, emit a partial file segment
  if (openTag) {
    const { language, path } = parseTag(openTag);
    segments.push({
      type: "file",
      code: codeBuffer.join("\n"),
      language,
      path,
      isPartial: true,
    });
  } else {
    flushText();
  }