@trim21/personal-pi-extensions 0.1.660 → 0.1.662

package/README.md

@@ -45,8 +45,9 @@
 - **脚本接口**：`tools` / `ALL_TOOLS` / `text` / `image` / `exit` / `console.*` /
   `store(key, value)` / `load(key)`；首行可选 `// @options: {"max_output_tokens": 10000}`。
   脚本没有超时：死循环由调用方中止（Esc）结束，等嵌套调用返回（含用户审批弹窗）多久都不算超时。
-- **store** 写在会话的自定义 entry 上（跟随分支），输出超过 `max_output_tokens` 时头尾
-  截断并把全文落到 `$TMPDIR/pi-codemode-*.txt`。
+- **store** 记在每次成功调用工具结果的 `details.store` 上（与 `src/lib/file-reads.ts` 的
+  已读记账同一套做法），下一次调用从当前分支的 toolResult 重放；输出超过 `max_output_tokens`
+  时头尾截断并把全文落到 `$TMPDIR/pi-codemode-*.txt`。
 - **构建**：worker 入口是 esbuild 产物 `src/codemode/worker.js`（随仓库提交，
   `pnpm run build:codemode-worker` 重新生成，pre-commit 会自动跑）；`quickjs-wasi` 的
   wasm 在注册工具时编译一次，之后每次执行复用。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trim21/personal-pi-extensions",
-  "version": "0.1.660",
+  "version": "0.1.662",
   "type": "module",
   "description": "Custom pi coding-agent extensions: bwrap sandbox, workspace guard, opencode edit, and more",
   "keywords": [

package/src/codemode/sandbox.ts

@@ -3,10 +3,10 @@
  *
  * 沙箱就是 QuickJS VM 本身（VM 里没有 node、文件、网络、timer），脚本唯一的出口是
  * 注入的工具，而这些工具由主线程执行——所以这里只做三件事：把注册时编译好的 wasm
- * 与脚本交给 worker、转发脚本的嵌套调用与输出、在结束/超时/中止时 `terminate()`。
+ * 与脚本交给 worker、转发脚本的嵌套调用与输出、在结束或中止时 `terminate()`。
  *
- * worker 一次执行一个：死循环脚本由主线程 terminate 掉，连同它那份 VM 一起丢弃，
- * 不会污染后续执行，也不需要取消协议。
+ * worker 一次执行一个：脚本没有超时，死循环由调用方中止后 terminate 掉，连同它那份
+ * VM 一起丢弃，不会污染后续执行，也不需要取消协议。
  */
 
 import { Worker } from "node:worker_threads";

package/src/codemode/tool.ts

@@ -26,9 +26,6 @@ import { CODEMODE_SOURCE_GRAMMAR, DEFAULT_OUTPUT_TOKENS, parseCodemodeSource } f
 
 export const CODEMODE_TOOL_NAME = "codemode";
 
-/** store 落在 session 的自定义 entry 上：随分支保存与恢复。 */
-const STORE_ENTRY_TYPE = "codemode-store";
-
 /** 估计 token 用的字符数（与 pi 一致）。 */
 const CHARS_PER_TOKEN = 4;
 
@@ -89,17 +86,29 @@ function buildDescription(tools: readonly CallableTool[]): string {
   ].join("\n");
 }
 
+/**
+ * store 的恢复：把当前分支上每次 codemode 调用记在工具结果 `details.store` 里的写入
+ * 按顺序重放（与 `src/lib/file-reads.ts` 从 toolResult 的 details 重放已读同一套做法）。
+ * 只认分支上的结果，所以 rewind / fork / resume 后 store 跟着分支走。
+ */
 function readStore(ctx: ExtensionContext): Record<string, unknown> {
   const store = new Map<string, unknown>();
   for (const entry of ctx.sessionManager.getBranch()) {
-    if (entry.type !== "custom" || entry.customType !== STORE_ENTRY_TYPE) {
+    if (entry.type !== "message" || entry.message.role !== "toolResult") {
+      continue;
+    }
+    if (entry.message.toolName !== CODEMODE_TOOL_NAME) {
+      continue;
+    }
+    const details = entry.message.details;
+    if (typeof details !== "object" || details === null) {
       continue;
     }
-    const data = (entry as { data?: unknown }).data;
-    if (typeof data !== "object" || data === null) {
+    const record = (details as { store?: unknown }).store;
+    if (typeof record !== "object" || record === null || Array.isArray(record)) {
       continue;
     }
-    const writes = data as Partial<StoreWrites>;
+    const writes = record as Partial<StoreWrites>;
     for (const key of writes.delete ?? []) {
       store.delete(key);
     }
@@ -294,9 +303,7 @@ export function createCodemodeTools(pi: ExtensionAPI): CodemodeTools {
           const images = imagesOf(outcome.output);
           if (outcome.ok) {
             const writes = outcome.writes;
-            if (Object.keys(writes.set).length > 0 || writes.delete.length > 0) {
-              pi.appendEntry(STORE_ENTRY_TYPE, writes);
-            }
+            const hasWrites = Object.keys(writes.set).length > 0 || writes.delete.length > 0;
             const value =
               outcome.value === undefined ? "" : `\n\n${JSON.stringify(outcome.value, null, 2)}`;
             const truncated = await truncateOutput(
@@ -315,6 +322,8 @@ export function createCodemodeTools(pi: ExtensionAPI): CodemodeTools {
               ],
               details: {
                 calls: outcome.calls,
+                // store 的写入随工具结果持久化，下一次调用从这里重放恢复
+                ...(hasWrites && { store: writes }),
                 pendant: scriptPendant(code, `${outcome.calls.length} tool call(s)`),
                 ...(truncated.fullOutputPath && { fullOutputPath: truncated.fullOutputPath }),
               },