Function getDiffLines

desktop/src/features/agents/ui/agentSessionFileEditDiff.ts:103–142

(text: string)

}

function getDiffLines(text: string): FileEditDiffLine[] {
  const rawLines = text.split(/\r?\n/);
  const hasUnifiedDiff = rawLines.some((line) =>
    /^(diff --git|--- |\+\+\+ |@@)/.test(line),
  );
  const lines: FileEditDiffLine[] = [];
  let inUnifiedDiff = false;

  for (const rawLine of rawLines) {
    const shikiKind = getShikiDiffKind(rawLine);
    if (shikiKind) {
      lines.push({
        kind: shikiKind,
        text: rawLine.replace(
          shikiKind === "add" ? SHIKI_ADD_RE : SHIKI_REMOVE_RE,
          "",
        ),
      });
      continue;
    }

    if (hasUnifiedDiff && !inUnifiedDiff) {
      if (!/^(diff --git|--- |\+\+\+ |@@)/.test(rawLine)) {
        continue;
      }
      inUnifiedDiff = true;
    }

    if (hasUnifiedDiff) {
      lines.push(classifyUnifiedDiffLine(rawLine));
      continue;
    }

    lines.push({ kind: "context", text: rawLine });
  }

  return trimTrailingEmptyContextLines(
    lines.filter((line) => line.text.length > 0 || line.kind !== "meta"),
  );
}

function trimTrailingEmptyContextLines(
  lines: FileEditDiffLine[],