Streaming Code Patches: Apply Validated Hunks as Tokens Arrive

// patch/stream-frames.ts
export type HunkFrame = {
  path: string;
  oldStart: number;
  oldLines: string[];   // exact context expected on disk
  newLines: string[];
  baseSha: string;      // blob hash at plan time
  seq: number;
};

export function parseFrame(line: string): HunkFrame | null {
  try {
    const f = JSON.parse(line) as HunkFrame;
    if (!f.path || !Array.isArray(f.oldLines) || !Array.isArray(f.newLines)) return null;
    return f;
  } catch {
    return null; // ✅ ignore incomplete SSE chunks
  }
}

import { createHash } from "node:crypto";
import { readFileSync } from "node:fs";

export function validateHunk(f: HunkFrame): "ok" | "stale" | "mismatch" {
  const disk = readFileSync(f.path, "utf8");
  const sha = createHash("sha1").update(disk).digest("hex");
  if (sha !== f.baseSha) return "stale";

  const lines = disk.split("\n");
  const slice = lines.slice(f.oldStart - 1, f.oldStart - 1 + f.oldLines.length);
  // ✅ Exact context match — fuzzy apply is how you corrupt JSX
  if (slice.join("\n") !== f.oldLines.join("\n")) return "mismatch";
  return "ok";
}

export function applyHunk(f: HunkFrame) {
  if (validateHunk(f) !== "ok") throw new Error(`hunk_rejected:${f.path}:${f.seq}`);
  const lines = readFileSync(f.path, "utf8").split("\n");
  lines.splice(f.oldStart - 1, f.oldLines.length, ...f.newLines);
  // write atomically via tmp+rename in production
}

// agent/stream-apply.ts — illustrative Node consumer
async function consumePatchStream(readable: AsyncIterable<string>) {
  let buf = "";
  for await (const chunk of readable) {
    buf += chunk;
    const parts = buf.split("\n");
    buf = parts.pop() ?? "";
    for (const line of parts) {
      const frame = parseFrame(line);
      if (!frame) continue;
      const v = validateHunk(frame);
      if (v === "ok") {
        applyHunk(frame);
        ui.paintDiff(frame); // ✅ first paint in <1s typical
      } else {
        // ❌ Don't invent offsets — re-request this path only
        await agent.rerequestHunk(frame.path, frame.seq, reason: v);
      }
    }
  }
}