@heyhuynhgiabuu/pi-diff 0.4.1 → 0.5.1

package/dist/index.js

@@ -1,4 +1,3 @@
-"use strict";
 /**
  * pi-diff — Shiki-powered terminal diff renderer for pi.
  *
@@ -21,52 +20,16 @@
  *   • Large-diff fallback (skip highlighting, still show diff)
  *   • Async rendering with invalidate() for non-blocking preview
  */
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.__testing = void 0;
-exports.default = diffRendererExtension;
-const node_fs_1 = require("node:fs");
-const node_path_1 = require("node:path");
-const cli_1 = require("@shikijs/cli");
-const Diff = __importStar(require("diff"));
-const diff_js_1 = require("./core/diff.js");
-const command_js_1 = require("./review/command.js");
-const export_js_1 = require("./review/export.js");
-const git_js_1 = require("./review/git.js");
-const hunk_preview_js_1 = require("./review/hunk-preview.js");
-const interactive_js_1 = require("./review/interactive.js");
+import { existsSync, readFileSync } from "node:fs";
+import { extname, relative } from "node:path";
+import { codeToANSI } from "@shikijs/cli";
+import * as Diff from "diff";
+import { parseDiff } from "./core/diff.js";
+import { registerReviewDiffCommand } from "./review/command.js";
+import { formatReviewMarkdown } from "./review/export.js";
+import { countReviewDiffLines, readGitDiff } from "./review/git.js";
+import { applyDiffPalette as applySharedDiffPalette, lang as detectDiffLanguage, renderSplit as renderSharedSplit, resolveDiffColors as resolveSharedDiffColors, themeCacheKey as sharedThemeCacheKey, } from "./review/hunk-preview.js";
+import { formatInteractiveReviewPanel } from "./review/interactive.js";
 const DIFF_PRESETS = {
     default: {
         name: "default",
@@ -245,8 +208,8 @@ function loadDiffConfig() {
     const paths = [`${process.cwd()}/.pi/settings.json`, `${process.env.HOME ?? ""}/.pi/settings.json`];
     for (const p of paths) {
         try {
-            if ((0, node_fs_1.existsSync)(p)) {
-                const raw = JSON.parse((0, node_fs_1.readFileSync)(p, "utf-8"));
+            if (existsSync(p)) {
+                const raw = JSON.parse(readFileSync(p, "utf-8"));
                 if (raw.diffTheme || raw.diffColors) {
                     return { diffTheme: raw.diffTheme, diffColors: raw.diffColors };
                 }
@@ -681,7 +644,7 @@ function lnum(n, w, fg = FG_LNUM) {
 function shortPath(cwd, home, p) {
     if (!p)
         return "";
-    const r = (0, node_path_1.relative)(cwd, p);
+    const r = relative(cwd, p);
     if (!r.startsWith("..") && !r.startsWith("/"))
         return r;
     return p.replace(home, "~");
@@ -777,14 +740,14 @@ const EXT_LANG = {
     vue: "vue",
 };
 function lang(fp) {
-    return EXT_LANG[(0, node_path_1.extname)(fp).slice(1).toLowerCase()];
+    return EXT_LANG[extname(fp).slice(1).toLowerCase()];
 }
 // ---------------------------------------------------------------------------
 // Shiki ANSI cache + pre-warm
 // ---------------------------------------------------------------------------
 // Pre-warm the Shiki singleton (loads WASM grammars + theme) so the first
 // diff render doesn't pay the ~200-500ms startup cost.
-(0, cli_1.codeToANSI)("", "typescript", THEME).catch(() => { });
+codeToANSI("", "typescript", THEME).catch(() => { });
 const _cache = new Map();
 function _touch(k, v) {
     _cache.delete(k);
@@ -807,7 +770,7 @@ async function hlBlock(code, language) {
     if (hit)
         return _touch(k, hit);
     try {
-        const ansi = normalizeShikiContrast(await (0, cli_1.codeToANSI)(code, language, THEME));
+        const ansi = normalizeShikiContrast(await codeToANSI(code, language, THEME));
         const out = (ansi.endsWith("\n") ? ansi.slice(0, -1) : ansi).split("\n");
         return _touch(k, out);
     }
@@ -1175,9 +1138,9 @@ async function renderSplit(diff, language, max = MAX_PREVIEW_LINES, dc = DEFAULT
 // ---------------------------------------------------------------------------
 // Extension
 // ---------------------------------------------------------------------------
-exports.__testing = {
+export const __testing = {
     normalizeShikiContrast,
-    parseDiff: diff_js_1.parseDiff,
+    parseDiff,
     renderSplit,
     renderUnified,
 };
@@ -1202,9 +1165,9 @@ function normalizeOptionalPositiveInteger(value, name) {
         throw new Error(`${name} must be a positive integer`);
     return number;
 }
-async function diffRendererExtension(pi) {
+export default async function diffRendererExtension(pi) {
     // Apply diff theme palette from settings/presets before rendering
-    (0, hunk_preview_js_1.applyDiffPalette)();
+    applySharedDiffPalette();
     let createWriteTool, createEditTool, getMarkdownTheme, TextComponent, MarkdownComponent;
     try {
         const sdk = await import("@earendil-works/pi-coding-agent");
@@ -1224,8 +1187,45 @@ async function diffRendererExtension(pi) {
     const cwd = process.cwd();
     const home = process.env.HOME ?? "";
     const sp = (p) => shortPath(cwd, home, p);
-    const reviewComments = [];
-    (0, command_js_1.registerReviewDiffCommand)(pi, cwd);
+    /** Wrap a Text component so its render(width) kicks off async diff rendering
+     *  using the real TUI width (which accounts for the sidebar). */
+    function getWidthAwareText(lastComponent) {
+        const text = (lastComponent ?? new TextComponent("", 0, 0));
+        if (text.__piDiffWidthAware)
+            return text;
+        const baseRender = typeof text.render === "function" ? text.render.bind(text) : null;
+        if (!baseRender)
+            return text;
+        text.__piDiffWidthAware = true;
+        text.__piDiffRender = baseRender;
+        text.render = (width) => {
+            const task = text.__piDiffTask;
+            if (task) {
+                const renderWidth = Math.max(1, Math.floor(width || termW()));
+                const key = task.key(renderWidth);
+                if (text.__piDiffRenderedKey !== key) {
+                    text.__piDiffRenderedKey = key;
+                    text.setText(task.placeholder);
+                    Promise.resolve(task.render(renderWidth))
+                        .then((rendered) => {
+                        if (text.__piDiffRenderedKey !== key)
+                            return;
+                        text.setText(rendered);
+                        task.invalidate?.();
+                    })
+                        .catch(() => {
+                        if (text.__piDiffRenderedKey !== key)
+                            return;
+                        text.setText(task.fallback);
+                        task.invalidate?.();
+                    });
+                }
+            }
+            return text.__piDiffRender(width);
+        };
+        return text;
+    }
+    registerReviewDiffCommand(pi, cwd);
     pi.registerTool({
         name: "review_git_diff",
         label: "Review Git Diff",
@@ -1279,31 +1279,32 @@ Examples:
         },
         async execute(_tid, params = {}) {
             try {
-                const safeParams = params ?? {};
-                const diff = await (0, git_js_1.readGitDiff)(cwd, reviewGitDiffMode(safeParams));
+                const safeParams = (params ?? {});
+                const diff = await readGitDiff(cwd, reviewGitDiffMode(safeParams));
                 const maxLinesPerHunk = reviewGitDiffMaxLines(safeParams);
                 const markdown = safeParams.includeRawDiff || safeParams.raw
-                    ? (0, export_js_1.formatReviewMarkdown)(diff, { includeRawDiff: true, maxLinesPerHunk })
-                    : (0, interactive_js_1.formatInteractiveReviewPanel)(diff, reviewComments, {
+                    ? formatReviewMarkdown(diff, { includeRawDiff: true, maxLinesPerHunk })
+                    : formatInteractiveReviewPanel(diff, [], {
                         file: safeParams.file,
                         hunkId: safeParams.hunkId,
                         maxFiles: normalizeOptionalPositiveInteger(safeParams.maxFiles, "maxFiles"),
                         maxHunks: normalizeOptionalPositiveInteger(safeParams.maxHunks, "maxHunks"),
                         maxLinesPerHunk,
                     });
-                const counts = (0, git_js_1.countReviewDiffLines)(diff);
+                const counts = countReviewDiffLines(diff);
                 return {
                     content: [{ type: "text", text: markdown }],
                     details: {
                         _type: "reviewGitDiff",
                         markdown,
-                        mode: diff.mode,
+                        mode: String(diff.mode),
                         fileCount: diff.files.length,
                         insertions: counts.insertions,
                         deletions: counts.deletions,
-                        commentCount: reviewComments.length,
+                        commentCount: 0,
                         focusedFile: safeParams.file,
                         focusedHunk: safeParams.hunkId,
+                        error: undefined,
                     },
                 };
             }
@@ -1311,14 +1312,15 @@ Examples:
                 const message = error instanceof Error ? error.message : String(error);
                 return {
                     content: [{ type: "text", text: `Error: ${message}` }],
-                    details: { _type: "reviewGitDiff", error: message },
+                    details: { _type: "reviewGitDiff", markdown: "", mode: "error", fileCount: 0, insertions: 0, deletions: 0, commentCount: 0, focusedFile: undefined, focusedHunk: undefined, error: message },
                 };
             }
         },
         renderCall(args, theme, ctx) {
             const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
-            const base = typeof args?.base === "string" && args.base.trim() ? ` vs ${args.base.trim()}` : " working tree";
-            const focus = typeof args?.file === "string" && args.file.trim() ? ` • ${args.file.trim()}` : "";
+            const safeArgs = args;
+            const base = typeof safeArgs?.base === "string" && safeArgs.base.trim() ? ` vs ${safeArgs.base.trim()}` : " working tree";
+            const focus = typeof safeArgs?.file === "string" && safeArgs.file.trim() ? ` • ${safeArgs.file.trim()}` : "";
             text.setText(`${theme.fg("toolTitle", theme.bold("review_git_diff"))}${theme.fg("muted", `${base}${focus}`)}`);
             return text;
         },
@@ -1342,113 +1344,6 @@ Examples:
             return text;
         },
     });
-    pi.registerTool({
-        name: "review_git_comment",
-        label: "Draft Review Comment",
-        description: "Draft an inline code-review comment for the current interactive Git review. This stores comments in memory for the current Pi session and does not modify files or submit anything externally.",
-        promptSnippet: "Draft an inline comment for the current Git review session.",
-        parameters: {
-            type: "object",
-            properties: {
-                file: { type: "string", description: "Changed file path to comment on." },
-                line: { type: "number", description: "Optional old/new line number to comment on." },
-                hunkId: { type: "string", description: "Optional hunk id from review_git_diff output." },
-                body: { type: "string", description: "Comment body." },
-            },
-            required: ["file", "body"],
-            additionalProperties: false,
-        },
-        async execute(_tid, params) {
-            const file = typeof params?.file === "string" ? params.file.trim() : "";
-            const body = typeof params?.body === "string" ? params.body.trim() : "";
-            if (!file)
-                return {
-                    content: [{ type: "text", text: "Error: file is required" }],
-                    details: { error: "file required" },
-                };
-            if (!body)
-                return {
-                    content: [{ type: "text", text: "Error: body is required" }],
-                    details: { error: "body required" },
-                };
-            const comment = (0, interactive_js_1.createReviewComment)({
-                comments: reviewComments,
-                file,
-                body,
-                line: normalizeOptionalPositiveInteger(params.line, "line"),
-                hunkId: typeof params.hunkId === "string" && params.hunkId.trim() ? params.hunkId.trim() : undefined,
-            });
-            reviewComments.push(comment);
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: `Drafted ${comment.id} on ${comment.file}${comment.line ? `:${comment.line}` : ""}\n\n${comment.body}`,
-                    },
-                ],
-                details: { _type: "reviewGitComment", comment, commentCount: reviewComments.length },
-            };
-        },
-        renderCall(args, theme, ctx) {
-            const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
-            text.setText(`${theme.fg("toolTitle", theme.bold("review_git_comment"))} ${theme.fg("accent", args?.file ?? "")}`);
-            return text;
-        },
-        renderResult(result, _opt, theme, ctx) {
-            const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
-            if (ctx.isError || result.details?.error) {
-                text.setText(`\n${theme.fg("error", result.details?.error ?? "review_git_comment failed")}`);
-                return text;
-            }
-            const comment = result.details?.comment;
-            text.setText(`  ${theme.fg("success", `✓ drafted ${comment?.id ?? "comment"}`)}${theme.fg("muted", ` (${result.details?.commentCount ?? 0} total)`)}`);
-            return text;
-        },
-    });
-    pi.registerTool({
-        name: "review_git_comments",
-        label: "Review Comments",
-        description: "List or clear drafted interactive Git review comments for the current Pi session.",
-        promptSnippet: "List or clear drafted review comments.",
-        parameters: {
-            type: "object",
-            properties: {
-                clear: { type: "boolean", description: "Clear all drafted comments when true." },
-            },
-            additionalProperties: false,
-        },
-        async execute(_tid, params = {}) {
-            if (params?.clear) {
-                const cleared = reviewComments.length;
-                reviewComments.length = 0;
-                return {
-                    content: [{ type: "text", text: `Cleared ${cleared} drafted review comments.` }],
-                    details: { _type: "reviewGitComments", cleared, commentCount: 0 },
-                };
-            }
-            const markdown = (0, interactive_js_1.formatReviewComments)(reviewComments);
-            return {
-                content: [{ type: "text", text: markdown }],
-                details: { _type: "reviewGitComments", markdown, commentCount: reviewComments.length },
-            };
-        },
-        renderCall(_args, theme, ctx) {
-            const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
-            text.setText(theme.fg("toolTitle", theme.bold("review_git_comments")));
-            return text;
-        },
-        renderResult(result, _opt, theme, ctx) {
-            if (MarkdownComponent && getMarkdownTheme && typeof result.details?.markdown === "string") {
-                return new MarkdownComponent(result.details.markdown, 0, 0, getMarkdownTheme());
-            }
-            const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
-            const cleared = typeof result.details?.cleared === "number"
-                ? `cleared ${result.details.cleared}`
-                : `${result.details?.commentCount ?? 0} drafted`;
-            text.setText(`  ${theme.fg("muted", cleared)} review comments`);
-            return text;
-        },
-    });
     // =======================================================================
     // write
     // =======================================================================
@@ -1460,8 +1355,8 @@ Examples:
             const fp = params.path ?? params.file_path ?? "";
             let old = null;
             try {
-                if (fp && (0, node_fs_1.existsSync)(fp))
-                    old = (0, node_fs_1.readFileSync)(fp, "utf-8");
+                if (fp && existsSync(fp))
+                    old = readFileSync(fp, "utf-8");
             }
             catch {
                 old = null;
@@ -1470,8 +1365,8 @@ Examples:
             const content = params.content ?? "";
             // Store in details — the only custom field TUI preserves in renderResult
             if (old !== null && old !== content) {
-                const diff = (0, diff_js_1.parseDiff)(old, content);
-                const lg = (0, hunk_preview_js_1.lang)(fp);
+                const diff = parseDiff(old, content);
+                const lg = detectDiffLanguage(fp);
                 result.details = { _type: "diff", summary: summarize(diff.added, diff.removed), diff, language: lg };
             }
             else if (old === null) {
@@ -1485,7 +1380,7 @@ Examples:
         },
         renderCall(args, theme, ctx) {
             const fp = args?.path ?? args?.file_path ?? "";
-            const isNew = !fp || !(0, node_fs_1.existsSync)(fp);
+            const isNew = !fp || !existsSync(fp);
             const label = isNew ? "create" : "write";
             const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
             const hdr = `${theme.fg("toolTitle", theme.bold(label))} ${theme.fg("accent", sp(fp))}`;
@@ -1497,11 +1392,11 @@ Examples:
             }
             // New file preview with Shiki
             if (args?.content && ctx.argsComplete && isNew) {
-                const previewKey = `create:${(0, hunk_preview_js_1.themeCacheKey)(theme)}:${fp}:${String(args.content).length}`;
+                const previewKey = `create:${sharedThemeCacheKey(theme)}:${fp}:${String(args.content).length}`;
                 if (ctx.state._previewKey !== previewKey) {
                     ctx.state._previewKey = previewKey;
                     ctx.state._previewText = hdr;
-                    const lg = (0, hunk_preview_js_1.lang)(fp);
+                    const lg = detectDiffLanguage(fp);
                     hlBlock(args.content, lg)
                         .then((lines) => {
                         if (ctx.state._previewKey !== previewKey)
@@ -1524,51 +1419,41 @@ Examples:
             return text;
         },
         renderResult(result, _opt, theme, ctx) {
-            const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
+            const text = getWidthAwareText(ctx.lastComponent);
             if (ctx.isError) {
                 const e = result.content
                     ?.filter((c) => c.type === "text")
                     .map((c) => c.text || "")
                     .join("\n") ?? "Error";
+                text.__piDiffTask = undefined;
                 text.setText(`\n${theme.fg("error", e)}`);
                 return text;
             }
             const d = result.details;
             if (d?._type === "diff") {
-                const w = termW();
-                const key = `wd:${(0, hunk_preview_js_1.themeCacheKey)(theme)}:${w}:${d.summary}:${d.diff?.lines?.length ?? 0}:${d.language ?? ""}`;
-                if (ctx.state._wdk !== key) {
-                    ctx.state._wdk = key;
-                    ctx.state._wdt = `  ${d.summary}\n${theme.fg("muted", "  rendering diff…")}`;
-                    const dc = (0, hunk_preview_js_1.resolveDiffColors)(theme);
-                    (0, hunk_preview_js_1.renderSplit)(d.diff, d.language, MAX_RENDER_LINES, dc, w)
-                        .then((rendered) => {
-                        if (ctx.state._wdk !== key)
-                            return;
-                        ctx.state._wdt = `  ${d.summary}\n${rendered}`;
-                        ctx.invalidate();
-                    })
-                        .catch(() => {
-                        if (ctx.state._wdk !== key)
-                            return;
-                        ctx.state._wdt = `  ${d.summary}`;
-                        ctx.invalidate();
-                    });
-                }
-                text.setText(ctx.state._wdt ?? `  ${d.summary}`);
+                const themeKey = sharedThemeCacheKey(theme);
+                const dc = resolveSharedDiffColors(theme);
+                text.__piDiffTask = {
+                    placeholder: `  ${d.summary}\n${theme.fg("muted", "  rendering diff…")}`,
+                    fallback: `  ${d.summary}`,
+                    invalidate: ctx.invalidate,
+                    key: (width) => `wd:${themeKey}:${width}:${d.summary}:${d.diff?.lines?.length ?? 0}:${d.language ?? ""}`,
+                    render: async (width) => `  ${d.summary}\n${await renderSharedSplit(d.diff, d.language, MAX_RENDER_LINES, dc, width)}`,
+                };
                 return text;
             }
             if (d?._type === "noChange") {
+                text.__piDiffTask = undefined;
                 text.setText(`  ${theme.fg("muted", "✓ no changes")}`);
                 return text;
             }
             if (d?._type === "new") {
                 const { lines: lineCount, content: rawContent, filePath: fp } = d;
-                const pk = `nf:${(0, hunk_preview_js_1.themeCacheKey)(theme)}:${fp}:${lineCount}`;
+                const pk = `nf:${sharedThemeCacheKey(theme)}:${fp}:${lineCount}`;
                 if (ctx.state._nfk !== pk) {
                     ctx.state._nfk = pk;
                     ctx.state._nft = `  ${theme.fg("success", `✓ new file (${lineCount} lines)`)}`;
-                    const lg = (0, hunk_preview_js_1.lang)(fp);
+                    const lg = detectDiffLanguage(fp);
                     if (rawContent) {
                         hlBlock(rawContent, lg)
                             .then((hlLines) => {
@@ -1611,7 +1496,7 @@ Examples:
         return oldText && oldText !== newText ? [{ oldText, newText }] : [];
     }
     function summarizeEditOperations(operations) {
-        const diffs = operations.map((edit) => (0, diff_js_1.parseDiff)(edit.oldText, edit.newText));
+        const diffs = operations.map((edit) => parseDiff(edit.oldText, edit.newText));
         const totalAdded = diffs.reduce((sum, diff) => sum + diff.added, 0);
         const totalRemoved = diffs.reduce((sum, diff) => sum + diff.removed, 0);
         return {
@@ -1631,11 +1516,12 @@ Examples:
             if (operations.length === 0)
                 return result;
             const { diffs, summary } = summarizeEditOperations(operations);
+            const lg = detectDiffLanguage(fp);
             if (operations.length === 1) {
                 let editLine = 0;
                 try {
-                    if (fp && (0, node_fs_1.existsSync)(fp)) {
-                        const f = (0, node_fs_1.readFileSync)(fp, "utf-8");
+                    if (fp && existsSync(fp)) {
+                        const f = readFileSync(fp, "utf-8");
                         const idx = f.indexOf(operations[0].newText);
                         if (idx >= 0)
                             editLine = f.slice(0, idx).split("\n").length;
@@ -1644,7 +1530,7 @@ Examples:
                 catch {
                     editLine = 0;
                 }
-                result.details = { _type: "editInfo", summary, editLine };
+                result.details = { _type: "editInfo", summary, editLine, diff: diffs[0], language: lg };
                 return result;
             }
             result.details = {
@@ -1660,85 +1546,54 @@ Examples:
             const operations = getEditOperations(args);
             const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
             const hdr = `${theme.fg("toolTitle", theme.bold("edit"))} ${theme.fg("accent", sp(fp))}`;
-            if (!(ctx.argsComplete && operations.length > 0)) {
-                text.setText(hdr);
-                return text;
+            if (ctx.argsComplete && operations.length > 0) {
+                const { totalAdded, totalRemoved } = summarizeEditOperations(operations);
+                text.setText(`${hdr}  ${theme.fg("muted", summarize(totalAdded, totalRemoved))}`);
             }
-            const pk = JSON.stringify({ fp, operations, theme: (0, hunk_preview_js_1.themeCacheKey)(theme), w: termW() });
-            if (ctx.state._pk !== pk) {
-                ctx.state._pk = pk;
-                ctx.state._pt = `${hdr}  ${theme.fg("muted", "(rendering…)")}`;
-                const lg = (0, hunk_preview_js_1.lang)(fp);
-                const dc = (0, hunk_preview_js_1.resolveDiffColors)(theme);
-                if (operations.length === 1) {
-                    const diff = (0, diff_js_1.parseDiff)(operations[0].oldText, operations[0].newText);
-                    (0, hunk_preview_js_1.renderSplit)(diff, lg, MAX_PREVIEW_LINES, dc, termW())
-                        .then((rendered) => {
-                        if (ctx.state._pk !== pk)
-                            return;
-                        ctx.state._pt = `${hdr}\n${summarize(diff.added, diff.removed)}\n${rendered}`;
-                        ctx.invalidate();
-                    })
-                        .catch(() => {
-                        if (ctx.state._pk !== pk)
-                            return;
-                        ctx.state._pt = `${hdr}  ${summarize(diff.added, diff.removed)}`;
-                        ctx.invalidate();
-                    });
-                }
-                else {
-                    const { diffs, summary } = summarizeEditOperations(operations);
-                    const maxShown = Math.min(operations.length, 3);
-                    const previewLines = Math.max(8, Math.floor(MAX_PREVIEW_LINES / maxShown));
-                    Promise.all(diffs.slice(0, maxShown).map((diff, index) => (0, hunk_preview_js_1.renderSplit)(diff, lg, previewLines, dc, termW())
-                        .then((rendered) => `Edit ${index + 1}/${operations.length}\n${rendered}`)
-                        .catch(() => `Edit ${index + 1}/${operations.length}  ${summarize(diff.added, diff.removed)}`)))
-                        .then((sections) => {
-                        if (ctx.state._pk !== pk)
-                            return;
-                        const remainder = operations.length - maxShown;
-                        const suffix = remainder > 0 ? `\n${theme.fg("muted", `… ${remainder} more edit blocks`)}` : "";
-                        ctx.state._pt = `${hdr}\n${operations.length} edits ${summary}\n\n${sections.join("\n\n")}${suffix}`;
-                        ctx.invalidate();
-                    })
-                        .catch(() => {
-                        if (ctx.state._pk !== pk)
-                            return;
-                        ctx.state._pt = `${hdr}  ${operations.length} edits ${summary}`;
-                        ctx.invalidate();
-                    });
-                }
+            else {
+                text.setText(hdr);
             }
-            text.setText(ctx.state._pt ?? hdr);
             return text;
         },
         renderResult(result, _opt, theme, ctx) {
-            const text = ctx.lastComponent ?? new TextComponent("", 0, 0);
+            const text = getWidthAwareText(ctx.lastComponent);
             if (ctx.isError) {
                 const e = result.content
                     ?.filter((c) => c.type === "text")
                     .map((c) => c.text || "")
                     .join("\n") ?? "Error";
+                text.__piDiffTask = undefined;
                 text.setText(`\n${theme.fg("error", e)}`);
                 return text;
             }
-            if (result.details?._type === "editInfo") {
-                const { summary: s, editLine } = result.details;
+            const d = result.details;
+            if (d?._type === "editInfo" && d.diff) {
+                const themeKey = sharedThemeCacheKey(theme);
+                const dc = resolveSharedDiffColors(theme);
+                const loc = d.editLine > 0 ? ` ${theme.fg("muted", `at line ${d.editLine}`)}` : "";
+                text.__piDiffTask = {
+                    placeholder: `  ${d.summary}${loc}\n${theme.fg("muted", "  rendering diff…")}`,
+                    fallback: `  ${d.summary}${loc}`,
+                    invalidate: ctx.invalidate,
+                    key: (width) => `ed:${themeKey}:${width}:${d.summary}:${d.diff?.lines?.length ?? 0}:${d.language ?? ""}`,
+                    render: async (width) => `  ${d.summary}${loc}\n${await renderSharedSplit(d.diff, d.language, MAX_PREVIEW_LINES, dc, width)}`,
+                };
+                return text;
+            }
+            if (d?._type === "editInfo") {
+                const { summary: s, editLine } = d;
                 const loc = editLine > 0 ? ` ${theme.fg("muted", `at line ${editLine}`)}` : "";
-                const content = `  ${s}${loc}`;
-                const vis = content.replace(ANSI_RE, "").length;
-                const pad = Math.max(0, termW() - vis);
-                text.setText(`${content}${" ".repeat(pad)}`);
+                text.__piDiffTask = undefined;
+                text.setText(`  ${s}${loc}`);
                 return text;
             }
-            if (result.details?._type === "multiEditInfo") {
-                const { summary: s, editCount, diffLineCount } = result.details;
-                const content = `  ${editCount} edits ${s}${typeof diffLineCount === "number" ? ` ${theme.fg("muted", `(${diffLineCount} diff lines)`)}` : ""}`;
-                const vis = content.replace(ANSI_RE, "").length;
-                const pad = Math.max(0, termW() - vis);
-                text.setText(`${content}${" ".repeat(pad)}`);
+            if (d?._type === "multiEditInfo") {
+                const { summary: s, editCount, diffLineCount } = d;
+                text.__piDiffTask = undefined;
+                text.setText(`  ${editCount} edits ${s}${typeof diffLineCount === "number" ? ` ${theme.fg("muted", `(${diffLineCount} diff lines)`)}` : ""}`);
                 return text;
             }
+            text.__piDiffTask = undefined;
             text.setText(`  ${theme.fg("dim", String(result?.content?.[0]?.text ?? "edited").slice(0, 120))}`);
             return text;
         },