@trim21/personal-pi-extensions 0.0.430 → 0.0.431

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trim21/personal-pi-extensions",
-  "version": "0.0.430",
+  "version": "0.0.431",
   "type": "module",
   "description": "Custom pi coding-agent extensions: bwrap sandbox, workspace guard, opencode edit, and more",
   "keywords": [
@@ -37,6 +37,7 @@
     "@earendil-works/pi-ai": "^0.84.1",
     "@earendil-works/pi-coding-agent": "^0.84.1",
     "@eslint/js": "10.0.1",
+    "@types/lodash-es": "^4.17.12",
     "@types/node": "^24.13.3",
     "@types/turndown": "^5.0.6",
     "@typescript/native": "npm:typescript@^7.0.2",
@@ -94,6 +95,7 @@
     "@vscode/tree-sitter-wasm": "^0.3.1",
     "jsonc-parser": "^3.3.1",
     "linkedom": "^0.18.0",
+    "lodash-es": "^4.18.1",
     "minimatch": "^10.2.6",
     "turndown": "^7.2.0",
     "vscode-jsonrpc": "^9.0.1",

package/src/bwrap/runtime.ts

@@ -13,6 +13,7 @@ import {
   truncateTail,
   type TruncationResult,
 } from "@earendil-works/pi-coding-agent";
+import { throttle } from "lodash-es";
 import { type TObject, Type } from "typebox";
 
 import { type CommandSpec, parseCommand } from "../lib/cli.js";
@@ -379,42 +380,24 @@ export class BwrapRuntime {
     await using output = new BashOutput(request.ctx.sessionManager.getSessionId());
     const { onUpdate } = request;
 
-    // 流式进度：节流推送尾部快照（对齐 pi 内置 bash 的实时输出体验）
-    let updateTimer: ReturnType<typeof setTimeout> | undefined;
-    let dirty = false;
-    let lastUpdateAt = 0;
-    const emitUpdate = () => {
-      if (!onUpdate || !dirty) return;
-      dirty = false;
-      lastUpdateAt = Date.now();
-      onUpdate({
-        content: [{ type: "text", text: output.tailSnapshot() }],
-        details: undefined,
-      });
-    };
-    const scheduleUpdate = () => {
-      if (!onUpdate) return;
-      dirty = true;
-      const delay = BASH_UPDATE_THROTTLE_MS - (Date.now() - lastUpdateAt);
-      if (delay <= 0) {
-        if (updateTimer) clearTimeout(updateTimer);
-        updateTimer = undefined;
-        emitUpdate();
-        return;
-      }
-      if (updateTimer) return;
-      updateTimer = setTimeout(() => {
-        updateTimer = undefined;
-        emitUpdate();
-      }, delay);
-    };
+    // 流式进度：限流推送尾部快照（对齐 pi 内置 bash 的实时输出体验）
+    const emitUpdate = throttle(
+      () => {
+        onUpdate?.({
+          content: [{ type: "text", text: output.tailSnapshot() }],
+          details: undefined,
+        });
+      },
+      BASH_UPDATE_THROTTLE_MS,
+      { trailing: true },
+    );
 
     try {
       if (onUpdate) onUpdate({ content: [], details: undefined });
       const { exitCode } = await operations.exec(request.command, execCwd, {
         onData: (data) => {
           output.append(data);
-          scheduleUpdate();
+          emitUpdate();
         },
         signal: request.signal,
         timeout: request.timeout,
@@ -451,8 +434,8 @@ export class BwrapRuntime {
       }
       throw error;
     } finally {
-      if (updateTimer) clearTimeout(updateTimer);
-      if (onUpdate && dirty) emitUpdate();
+      emitUpdate.flush();
+      emitUpdate.cancel();
     }
   }