@trim21/personal-pi-extensions 0.1.679 → 0.1.681

package/src/codemode/worker.ts

@@ -17,15 +17,13 @@ import { PRELUDE_SOURCE } from "./prelude.js";
 import {
   decodeHostMessage,
   type HostMessage,
+  MEMORY_LIMIT_BYTES,
   type ScriptError,
   type StoreWrites,
   type WorkerBootstrap,
   type WorkerMessage,
 } from "./protocol.js";
 
-/** QuickJS VM 的堆上限：超量分配在脚本里变成 InternalError，而不是拖垮宿主。 */
-const MEMORY_LIMIT_BYTES = 512 * 1024 * 1024;
-
 /**
  * QuickJS 把引擎诊断写到 fd 1 / 2，那会直接进 pi 的 TUI；按写入长度回报并丢弃内容，
  * 避免 libc 重试。

package/src/index.ts

@@ -120,8 +120,11 @@ export default function personalExtensions(pi: ExtensionAPI): void {
     for (const { name, module } of modules) {
       runModule(name, () => module.register(bus));
     }
-    // codemode 最后注册：它把总线上已有的工具写进自己的描述，并在注册时编译 wasm
-    await runModuleAsync("codemode", () => codemode.register(bus));
+    // codemode 最后注册：它把总线上已有的工具写进自己的描述，并在注册时编译 wasm。
+    // 脚本的 fs 原语与文件工具共用请求策略与已读记账，所以这里把工具集那两份注入过去。
+    await runModuleAsync("codemode", () =>
+      codemode.register(bus, { policy: services.policy, reads: fileToolset.reads }),
+    );
 
     for (const warning of warnings) {
       ctx.ui.notify(warning, "warning");

package/src/opencode/files.ts

@@ -407,8 +407,14 @@ async function formatDirectoryEntries(dirPath: string): Promise<string[]> {
 
 // ── read-before-edit 记账 ────────────────────────────────────────────────────
 
-/** 会更新 reads 记账（src/lib/file-reads.ts）并随 details 持久化快照的工具名。 */
-const READS_TOOL_NAMES = new Set(["read", "edit", "write", "lsp-rename"]);
+/**
+ * 会更新 reads 记账（src/lib/file-reads.ts）并随 details 持久化快照的工具名。
+ *
+ * 含 `codemode`：脚本的 `fs.read` / `fs.write`（src/codemode/fs.ts）与本工具集共用
+ * 同一个 ReadsState，脚本记下的已读落在 codemode 自己的工具结果里，重放分支时要一起
+ * 收回来（见 `restoreReads`）。
+ */
+const READS_TOOL_NAMES = new Set(["read", "edit", "write", "lsp-rename", "codemode"]);
 
 function registerReadTool(bus: ToolBus, getService: () => LspService, state: ReadsState): void {
   bus.register({
@@ -897,6 +903,8 @@ export interface OpencodeFileToolset {
   register(bus: ToolBus): void;
   onLspEnabled(bus: ToolBus, service: LspService): void;
   restoreReads(ctx: ExtensionContext): void;
+  /** 与工具共用的已读记账：codemode 的 fs 原语拿它做 stale 保护（两边互通）。 */
+  readonly reads: ReadsState;
 }
 
 /**
@@ -939,6 +947,8 @@ export function createOpencodeFileTools(
     restoreReads(ctx) {
       restoreReads(state, ctx.sessionManager, READS_TOOL_NAMES);
     },
+
+    reads: state,
   };
 
   const manager =