pi-pretty-tui 0.1.1 → 0.1.3

package/README.md

@@ -9,6 +9,8 @@ A visual refinement extension for the [Pi coding agent](https://pi.dev/). It pro
 - Compact `Read`, `Bash`, `Edit`, `Write`, `Grep`, `Find`, and `List` calls
 - Gray running, green success, and red failure indicators
 - Concise result summaries with expandable output
+- Native-style `Write` previews: first 10 lines when collapsed, full content when expanded
+- Live `Bash` output: latest 5 lines when collapsed, all available output when expanded
 - Colored `+added` and `-removed` edit statistics and diffs
 - Correct hanging indentation for long paths and wrapped tool output
 - Bullet (`•`) markers for unordered Markdown lists
@@ -25,7 +27,7 @@ Pi may warn that built-in tools are being overridden. This is expected: pi-prett
 
 ## Expand tool output
 
-Use Pi's configured `app.tools.expand` keybinding (usually `Ctrl+E`) to show or hide detailed output and edit diffs.
+Press `Ctrl+O` (Pi's default `app.tools.expand` keybinding) to show or hide detailed output, edit diffs, and complete `Write` content.
 
 ## Compatibility notice
 

package/extensions/index.ts

@@ -101,20 +101,53 @@ export default function prettyTui(pi: ExtensionAPI) {
     context.state.compactToolStatus = status;
   };
 
+  const callRow = (theme: any, name: string, detail: string, state: any): DisplayRow => ({
+    prefix: () => {
+      const status = (state.compactToolStatus ?? "running") as ToolStatus;
+      const color = status === "success" ? "success" : status === "error" ? "error" : "dim";
+      return theme.fg(color, "● ");
+    },
+    continuation: "  ",
+    content:
+      theme.fg("accent", theme.bold(name)) +
+      theme.fg("dim", "(") +
+      theme.fg("text", detail) +
+      theme.fg("dim", ")"),
+  });
+
   const call = (theme: any, name: string, detail: string, state: any) =>
-    block([{
-      prefix: () => {
-        const status = (state.compactToolStatus ?? "running") as ToolStatus;
-        const color = status === "success" ? "success" : status === "error" ? "error" : "dim";
-        return theme.fg(color, "● ");
-      },
-      continuation: "  ",
-      content:
-        theme.fg("accent", theme.bold(name)) +
-        theme.fg("dim", "(") +
-        theme.fg("text", detail) +
-        theme.fg("dim", ")"),
-    }]);
+    block([callRow(theme, name, detail, state)]);
+
+  const writeCall = (theme: any, path: string, content: string, expanded: boolean, state: any) => {
+    const lines = content.replace(/\t/g, "    ").split("\n");
+    while (lines.length > 0 && lines[lines.length - 1] === "") lines.pop();
+
+    const total = lines.length;
+    const shown = lines.slice(0, expanded ? total : 10);
+    const remaining = total - shown.length;
+    const rows: DisplayRow[] = [
+      callRow(theme, "Write", `${path} · ${total} ${total === 1 ? "line" : "lines"}`, state),
+    ];
+
+    for (let index = 0; index < shown.length; index++) {
+      const isLast = index === shown.length - 1 && remaining === 0;
+      rows.push({
+        prefix: theme.fg("dim", isLast ? "   └ " : "   │ "),
+        continuation: theme.fg("dim", "   │ "),
+        content: theme.fg("toolOutput", shown[index] || " "),
+      });
+    }
+
+    if (remaining > 0) {
+      rows.push({
+        prefix: theme.fg("muted", "   └ "),
+        continuation: "     ",
+        content: theme.fg("muted", `… ${remaining} more ${remaining === 1 ? "line" : "lines"}`),
+      });
+    }
+
+    return block(rows);
+  };
 
   const result = (
     theme: any,
@@ -161,6 +194,72 @@ export default function prettyTui(pi: ExtensionAPI) {
     return result(theme, label);
   };
 
+  const terminalOutputLines = (output: string): string[] => {
+    const lines: string[] = [];
+    let current = "";
+
+    for (let index = 0; index < output.length; index++) {
+      const char = output[index];
+      if (char === "\r") {
+        if (output[index + 1] === "\n") {
+          lines.push(current);
+          current = "";
+          index++;
+        } else {
+          // A bare carriage return redraws the current terminal line. Progress
+          // bars such as tqdm use this to update in place.
+          current = "";
+        }
+      } else if (char === "\n") {
+        lines.push(current);
+        current = "";
+      } else {
+        current += char;
+      }
+    }
+
+    if (current) lines.push(current);
+    return lines;
+  };
+
+  const bashResult = (
+    context: any,
+    theme: any,
+    summary: string,
+    output: string,
+    expanded: boolean,
+    status: ToolStatus,
+  ): Component => {
+    setStatus(context, status);
+    const lines = terminalOutputLines(output);
+    const shown = expanded ? lines : lines.slice(-5);
+    const omitted = lines.length - shown.length;
+    const rows: DisplayRow[] = [{
+      prefix: theme.fg("dim", "└  "),
+      continuation: "   ",
+      content: theme.fg(status === "error" ? "error" : "toolOutput", summary),
+    }];
+
+    if (omitted > 0) {
+      rows.push({
+        prefix: theme.fg("muted", "   │ "),
+        continuation: theme.fg("muted", "   │ "),
+        content: theme.fg("muted", `… ${omitted} earlier ${omitted === 1 ? "line" : "lines"}`),
+      });
+    }
+
+    for (let index = 0; index < shown.length; index++) {
+      const isLast = index === shown.length - 1;
+      rows.push({
+        prefix: theme.fg("dim", isLast ? "   └ " : "   │ "),
+        continuation: theme.fg("dim", "   │ "),
+        content: theme.fg("toolOutput", shown[index] || " "),
+      });
+    }
+
+    return block(rows);
+  };
+
   const completeStatus = (context: any, output: string): boolean => {
     const failed = isError(context, output);
     setStatus(context, failed ? "error" : "success");
@@ -200,20 +299,31 @@ export default function prettyTui(pi: ExtensionAPI) {
     ...bash,
     renderShell: "self",
     renderCall(args: any, theme: any, context: any) {
-      const command = String(args.command).replace(/\s*\n\s*/g, " ↵ ");
+      const command = typeof args.command === "string" ? args.command : "";
       return call(theme, "Bash", command, context.state);
     },
     renderResult(toolResult: any, options: any, theme: any, context: any) {
-      if (options.isPartial) return partialResult(context, theme, "Running…");
       const output = textContent(toolResult);
-      const failed = completeStatus(context, output);
-      const lines = nonEmptyLines(output);
+      if (options.isPartial) {
+        return bashResult(context, theme, "Running…", output, options.expanded, "running");
+      }
+
+      const failed = isError(context, output);
+      const outputLines = terminalOutputLines(output);
+      const lineCount = outputLines.filter((line) => line.length > 0).length;
       const summary = failed
-        ? output.split("\n")[0] || "Command failed"
-        : output
-          ? `Done · ${lines} output ${lines === 1 ? "line" : "lines"}`
+        ? outputLines[0] || "Command failed"
+        : outputLines.length > 0
+          ? `Done · ${lineCount} output ${lineCount === 1 ? "line" : "lines"}`
           : "Done";
-      return result(theme, summary, output, options.expanded, failed);
+      return bashResult(
+        context,
+        theme,
+        summary,
+        output,
+        options.expanded,
+        failed ? "error" : "success",
+      );
     },
   } as any);
 
@@ -261,8 +371,8 @@ export default function prettyTui(pi: ExtensionAPI) {
     ...write,
     renderShell: "self",
     renderCall(args: any, theme: any, context: any) {
-      const content = String(args.content ?? "");
-      return call(theme, "Write", `${args.path} · ${nonEmptyLines(content)} lines`, context.state);
+      const content = typeof args.content === "string" ? args.content : "";
+      return writeCall(theme, String(args.path ?? ""), content, context.expanded, context.state);
     },
     renderResult(toolResult: any, options: any, theme: any, context: any) {
       if (options.isPartial) return partialResult(context, theme, "Writing…");

package/package.json

@@ -1,10 +1,18 @@
 {
   "name": "pi-pretty-tui",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Polished tool calls and list rendering for the Pi coding agent TUI",
   "license": "MIT",
   "author": "Kainan Yang <ykn0309@whu.edu.cn>",
   "type": "module",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/ykn0309/pi-pretty-tui.git"
+  },
+  "homepage": "https://github.com/ykn0309/pi-pretty-tui#readme",
+  "bugs": {
+    "url": "https://github.com/ykn0309/pi-pretty-tui/issues"
+  },
   "keywords": [
     "pi-package",
     "pi-extension",
@@ -32,6 +40,6 @@
     "extensions": [
       "./extensions/index.ts"
     ],
-    "image": "https://unpkg.com/pi-pretty-tui@0.1.1/assets/screenshot.png"
+    "image": "https://unpkg.com/pi-pretty-tui@0.1.3/assets/screenshot.png"
   }
 }