pi-zh-cn 0.2.3 → 0.2.4

package/README.md

@@ -18,6 +18,7 @@
 - 底部状态栏：当前目录 / 分支 / 会话名、token 与缓存用量、费用、上下文占比、模型和思考等级
 - 加载提示：「正在处理…」；thinking 折叠块的标签
 - 八个内置工具的执行结果：读取、执行（bash）、编辑、写入、搜索内容、查找文件、列出目录、PowerShell——状态行都是中文，比如「完成」「退出码 N」「输出过长已截断」
+- 编辑与写入的改动展示沿用 pi 内置渲染器（只有标题和展开提示是中文）：`edit` 显示行号 + 红绿 diff，`write` 预览文件内容（折叠 10 行，`ctrl+o` 展开全量）
 - 项目信任弹窗：信任/不信任选择，含「信任父文件夹」变体
 - 内置斜杠命令描述：core 全部 23 个命令
 

package/extensions/localized-tools.ts

@@ -13,20 +13,24 @@ import {
   createPowerShellTool,
   createReadTool,
   createWriteTool,
+  getLanguageFromPath,
+  highlightCode,
+  keyHint,
   type BashToolDetails,
-  type EditToolDetails,
   type ReadToolDetails,
 } from "@earendil-works/pi-coding-agent";
 import { Text } from "@earendil-works/pi-tui";
 import { zh } from "./zh.ts";
 import {
   countResultRender,
-  errorFirstLine,
   notePartial,
   shellRenders,
   withExpanded,
 } from "./tool-renders.ts";
 
+/** write 工具折叠态预览的行数（与内置 write 渲染器一致）。 */
+const WRITE_PREVIEW_LINES = 10;
+
 /**
  * 重注册所有内置工具的中文渲染。
  *
@@ -118,38 +122,7 @@ export function registerLocalizedTools(pi: ExtensionAPI, cwd: string) {
       return new Text(text, 0, 0);
     },
 
-    renderResult(
-      result: any,
-      { expanded, isPartial }: { expanded: boolean; isPartial: boolean },
-      theme: any,
-    ) {
-      if (isPartial) return notePartial(zh.tools.edit.editing, theme);
-
-      const err = errorFirstLine(result, theme);
-      if (err) return err;
-
-      const details = result.details as EditToolDetails | undefined;
-      if (!details?.diff) {
-        return new Text(theme.fg("success", zh.tools.edit.applied), 0, 0);
-      }
-
-      let additions = 0,
-        removals = 0;
-      for (const line of details.diff.split("\n")) {
-        if (line.startsWith("+") && !line.startsWith("+++")) additions++;
-        if (line.startsWith("-") && !line.startsWith("---")) removals++;
-      }
-
-      let text = theme.fg("success", zh.tools.edit.additions(additions));
-      text += theme.fg("dim", " / ");
-      text += theme.fg("error", zh.tools.edit.removals(removals));
-
-      return new Text(
-        withExpanded(expanded ? details.diff : "", text, theme),
-        0,
-        0,
-      );
-    },
+    // 不覆盖 renderResult：沿用内置渲染器（行号 + 红绿 + 词级高亮的 diff）。
   });
 
   // write
@@ -157,20 +130,38 @@ export function registerLocalizedTools(pi: ExtensionAPI, cwd: string) {
     ...createWriteTool(cwd),
     label: zh.tools.write.label,
 
-    renderCall(args: any, theme: any) {
+    renderCall(args: any, theme: any, context: any) {
+      const rawLines = String(args.content ?? "")
+        .replace(/\r/g, "")
+        .replace(/\t/g, "   ")
+        .split("\n");
+      let end = rawLines.length;
+      while (end > 0 && rawLines[end - 1] === "") end--;
+
       let text = theme.fg("toolTitle", theme.bold(zh.tools.write.title));
       text += theme.fg("accent", args.path);
-      const lineCount = String(args.content).split("\n").length;
-      text += theme.fg("dim", ` (${zh.tools.write.lines(lineCount)})`);
+      text += theme.fg("dim", ` (${zh.tools.write.lines(end)})`);
+
+      if (end > 0) {
+        const maxLines = context.expanded ? end : WRITE_PREVIEW_LINES;
+        const lang = args.path ? getLanguageFromPath(args.path) : undefined;
+        // ponytail: 只高亮预览到的行，大文件折叠态不跑全量高亮；跨行语法
+        // （块注释等）在预览边界可能染色偏移，需要精确时改全量高亮 + 增量缓存
+        // （上游 write.js 的做法）。
+        text += `\n\n${highlightCode(rawLines.slice(0, maxLines).join("\n"), lang).join("\n")}`;
+        if (end > maxLines) {
+          text += theme.fg(
+            "muted",
+            `\n${zh.tools.write.moreLines(end - maxLines, end)}`,
+          );
+          text += keyHint("app.tools.expand", zh.tools.write.expand);
+          text += theme.fg("muted", "）");
+        }
+      }
       return new Text(text, 0, 0);
     },
 
-    renderResult(result: any, { isPartial }: { isPartial: boolean }, theme: any) {
-      if (isPartial) return notePartial(zh.tools.write.writing, theme);
-      const err = errorFirstLine(result, theme);
-      if (err) return err;
-      return new Text(theme.fg("success", zh.tools.write.written), 0, 0);
-    },
+    // 不覆盖 renderResult：沿用内置渲染器（成功时不追加内容，出错时显示完整错误）。
   });
 
   // grep / find / ls：调用渲染各自略有差异，结果渲染走同一个计数模板。

package/extensions/tool-renders.ts

@@ -22,15 +22,6 @@ export const countNonEmpty = (s: string): number =>
 export const notePartial = (msg: string, theme: any): Text =>
   new Text(theme.fg("warning", msg), 0, 0);
 
-// 错误结果：内容以 "Error" 开头时只展示首行（edit/write 共用）。
-/** 展示错误首行（红色）；结果不是错误时返回 undefined。 */
-export const errorFirstLine = (result: any, theme: any): Text | undefined => {
-  const out = textOf(result);
-  return out.startsWith("Error")
-    ? new Text(theme.fg("error", out.split("\n")[0]), 0, 0)
-    : undefined;
-};
-
 // 展开态：在摘要后追加逐行弱化的原始输出。
 /**
  * 组装展开态文本：raw 为空时返回 summary，否则在 summary 之后

package/extensions/zh.ts

@@ -95,6 +95,9 @@ export const zh = {
       writing: "写入中…",
       written: "已写入",
       lines: (n: number) => `${n} 行`,
+      moreLines: (remaining: number, total: number) =>
+        `…（还有 ${remaining} 行，共 ${total} 行，`,
+      expand: "展开",
     },
     grep: {
       label: "搜索内容",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-zh-cn",
-  "version": "0.2.3",
+  "version": "0.2.4",
   "description": "pi 编码代理的简体中文界面汉化插件",
   "repository": {
     "type": "git",