@zhushanwen/pi-subagent-workflow 0.3.1 → 0.3.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-subagent-workflow",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "type": "module",
   "main": "index.ts",
   "description": "Unified subagent execution and multi-agent workflow orchestration for Pi — spawned-process agent runtime with sync/background modes, stateful workflow management with persistence, state machine, and execution tracing.",

package/src/execution/__tests__/run-spawn-integration.test.ts

@@ -828,4 +828,84 @@ describe("runSpawn", () => {
 
   // 注：C1（orphan 进程兜底）与 M8（stdout 边界）describe 块已移至 run-spawn-edges.test.ts，
   // 拆分以保持本文件 < 1000 行（pre-commit hook 限制）。两文件各自独立声明文件级 mock。
+
+  // ============================================================
+  // E2E：runSpawn 从主进程 process.argv 镜像 extension/approve flag 到子进程
+  // 验证完整链路：runSpawn 真实执行 → 读 process.argv → buildSpawnArgs 拼参 →
+  // spawn 收到的 args 含镜像 flag。spawn 被 mock（FakeChild），但 runSpawn 体内逻辑
+  // 全真实跑，process.argv 用真实进程变量（测试中临时覆写 + 还原）。
+  // ============================================================
+  describe("E2E: 镜像主进程 argv flag 到子进程 spawn args", () => {
+    const originalArgv = process.argv;
+
+    afterEach(() => {
+      process.argv = originalArgv;
+    });
+
+    it("主进程 argv 含 --extension/--no-extensions/--approve → 子进程 spawn args 全部镜像", async () => {
+      // 模拟主 pi 进程启动参数（xyz-agent runtime 启动 pi 的真实形态）
+      process.argv = [
+        "bun", "/path/to/pi",
+        "--mode", "rpc",
+        "--no-extensions",
+        "--approve",
+        "--extension", "/exts/goal",
+        "--extension", "/exts/todo",
+        "--session-dir", "/sessions",
+      ];
+
+      const record = makeRecord();
+      const promise = runSpawn(record, "Task", makeOpts(), makeCtx());
+      await waitForSpawn();
+      const child = lastSpawnedChild();
+      mockSessionFileExists(
+        "/tmp/test/agents/subagents/--tmp-test--/sessions/2026-07-03T12-00-00-000Z_sess-mirror.jsonl",
+      );
+      emitStdoutLine(child, sessionHeader("sess-mirror"));
+      child.stdout.end();
+      child.stderr.end();
+      child.emit("close", 0);
+      await promise;
+
+      // 断言 spawn 收到的调用参数含全部镜像 flag
+      const spawnCall = mockSpawn.mock.calls[0];
+      // spawn(command, args, options) → args 是第二个参数
+      const spawnArgs = spawnCall[1] as string[];
+
+      expect(spawnArgs).toContain("--no-extensions");
+      expect(spawnArgs).toContain("--approve");
+      // 两个 extension 路径都镜像，顺序保留
+      const extIdxs = spawnArgs
+        .map((a, i) => (a === "--extension" ? i : -1))
+        .filter((i) => i >= 0);
+      expect(extIdxs).toHaveLength(2);
+      expect(spawnArgs[extIdxs[0] + 1]).toBe("/exts/goal");
+      expect(spawnArgs[extIdxs[1] + 1]).toBe("/exts/todo");
+    });
+
+    it("主进程 argv 无目标 flag → 子进程 spawn args 不含镜像 flag（向后兼容）", async () => {
+      // 模拟纯 pi CLI 直接跑（无 extension/approve 配置）
+      process.argv = ["bun", "/path/to/pi", "--mode", "rpc"];
+
+      const record = makeRecord();
+      const promise = runSpawn(record, "Task", makeOpts(), makeCtx());
+      await waitForSpawn();
+      const child = lastSpawnedChild();
+      mockSessionFileExists(
+        "/tmp/test/agents/subagents/--tmp-test--/sessions/2026-07-03T12-00-00-000Zsess-noflag.jsonl",
+      );
+      emitStdoutLine(child, sessionHeader("sess-noflag"));
+      child.stdout.end();
+      child.stderr.end();
+      child.emit("close", 0);
+      await promise;
+
+      const spawnCall = mockSpawn.mock.calls[0];
+      const spawnArgs = spawnCall[1] as string[];
+
+      expect(spawnArgs).not.toContain("--no-extensions");
+      expect(spawnArgs).not.toContain("--approve");
+      expect(spawnArgs).not.toContain("--extension");
+    });
+  });
 });

package/src/execution/__tests__/spawn-args.test.ts

@@ -7,6 +7,7 @@ import * as path from "node:path";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 
 import { MAX_FORK_DEPTH } from "../session-context-resolver.ts";
+import { mirrorMainProcessFlags } from "../argv-mirror.ts";
 import { buildEnvBlock, buildSpawnArgs } from "../session-runner.ts";
 
 describe("buildSpawnArgs", () => {
@@ -131,6 +132,105 @@ describe("buildSpawnArgs", () => {
     );
     expect(args).not.toContain("--tools");
   });
+
+  // ============================================================
+  // mirrorFlags 透传：子进程镜像主进程 extension/approve flag
+  // ============================================================
+
+  it("mirrorFlags 透传：noExtensions+approve+extensionPaths 全量 push（TC5）", () => {
+    const args = buildSpawnArgs({
+      ...baseParams,
+      mirrorFlags: { noExtensions: true, approve: true, extensionPaths: ["/e1", "/e2"] },
+    });
+    expect(args).toContain("--no-extensions");
+    expect(args).toContain("--approve");
+    // 每个 extension 独立 token，顺序保留
+    const extIdxs = args.map((a, i) => (a === "--extension" ? i : -1)).filter((i) => i >= 0);
+    expect(extIdxs).toHaveLength(2);
+    expect(args[extIdxs[0] + 1]).toBe("/e1");
+    expect(args[extIdxs[1] + 1]).toBe("/e2");
+  });
+
+  it("mirrorFlags 全 false/空 → 不追加任何目标 flag（TC6）", () => {
+    const args = buildSpawnArgs({
+      ...baseParams,
+      mirrorFlags: { noExtensions: false, approve: false, extensionPaths: [] },
+    });
+    expect(args).not.toContain("--no-extensions");
+    expect(args).not.toContain("--approve");
+    expect(args).not.toContain("--extension");
+    // 仅基础参数
+    expect(args).toEqual(["--mode", "rpc", "--session-dir", "/sessions/dir"]);
+  });
+
+  it("mirrorFlags undefined → 行为等同旧版（TC7）", () => {
+    const args = buildSpawnArgs(baseParams);
+    expect(args).toEqual(["--mode", "rpc", "--session-dir", "/sessions/dir"]);
+    expect(args).not.toContain("--extension");
+    expect(args).not.toContain("--no-extensions");
+    expect(args).not.toContain("--approve");
+  });
+});
+
+// ============================================================
+// mirrorMainProcessFlags：从主进程 argv 解析可镜像的 flag
+// ============================================================
+
+describe("mirrorMainProcessFlags", () => {
+  it("--extension 多次出现（空格分隔）+ 布尔 flag（TC1）", () => {
+    const r = mirrorMainProcessFlags([
+      "bun", "/pi", "--mode", "rpc", "--no-extensions", "--approve",
+      "--extension", "/a", "--extension", "/b",
+    ]);
+    expect(r).toEqual({ noExtensions: true, approve: true, extensionPaths: ["/a", "/b"] });
+  });
+
+  it("--extension=path 等号形式 + 短形式 -e/-ne/-a（TC2）", () => {
+    const r = mirrorMainProcessFlags([
+      "bun", "/pi", "--extension=/x", "-ne", "-a",
+    ]);
+    expect(r).toEqual({ noExtensions: true, approve: true, extensionPaths: ["/x"] });
+  });
+
+  it("混合形式（空格 + 等号），顺序保留（TC3）", () => {
+    const r = mirrorMainProcessFlags([
+      "bun", "/pi", "--extension", "/a", "--extension=/b", "--extension", "/c",
+    ]);
+    expect(r.extensionPaths).toEqual(["/a", "/b", "/c"]);
+    expect(r.noExtensions).toBe(false);
+    expect(r.approve).toBe(false);
+  });
+
+  it("无目标 flag → 全空/全 false（向后兼容，TC4）", () => {
+    const r = mirrorMainProcessFlags(["bun", "/pi", "--mode", "rpc"]);
+    expect(r).toEqual({ noExtensions: false, approve: false, extensionPaths: [] });
+  });
+
+  it("不误吃其他 flag 值与 positional 参数（TC8）", () => {
+    const r = mirrorMainProcessFlags([
+      "bun", "/pi", "--no-extensions", "--skill", "/sk", "some prompt text",
+    ]);
+    expect(r.noExtensions).toBe(true);
+    expect(r.extensionPaths).toEqual([]);
+    // --skill 的 /sk 不混入 extensionPaths；positional prompt 被忽略
+  });
+
+  it("空 argv / 仅前导两项 → 全空", () => {
+    expect(mirrorMainProcessFlags([])).toEqual({ noExtensions: false, approve: false, extensionPaths: [] });
+    expect(mirrorMainProcessFlags(["bun", "/pi"])).toEqual({
+      noExtensions: false, approve: false, extensionPaths: [],
+    });
+  });
+
+  it("--extension 末尾无值 → 跳过（不越界、不误吃下一个 token）", () => {
+    const r = mirrorMainProcessFlags(["bun", "/pi", "--extension"]);
+    expect(r.extensionPaths).toEqual([]);
+  });
+
+  it("-e 短形式多次 + 等号混用", () => {
+    const r = mirrorMainProcessFlags(["bun", "/pi", "-e", "/a", "-e=/b", "-e", "/c"]);
+    expect(r.extensionPaths).toEqual(["/a", "/b", "/c"]);
+  });
 });
 
 // ============================================================

package/src/execution/argv-mirror.ts

@@ -0,0 +1,90 @@
+/**
+ * 从主进程 argv 解析可镜像给子进程的 flag。
+ *
+ * 独立模块：argv 解析是独立职责，从 session-runner.ts 拆出避免该文件超行数上限。
+ */
+
+/**
+ * 镜像 flag 集合：从主进程 argv 解析出的可透传给子进程的 flag。
+ * 面向 buildSpawnArgs 的入参形态（解析后）。undefined 字段语义与解析为空一致。
+ */
+export interface MirrorFlags {
+  noExtensions: boolean;
+  approve: boolean;
+  extensionPaths: string[];
+}
+
+/**
+ * 有值 flag 的解析规则表：成员是「后跟一个值」的 flag 名（含长短形式）。
+ * 用于 mirrorMainProcessFlags 跳过其他 flag 的值时不误吃。
+ */
+const VALUED_FLAGS = new Set<string>([
+  "--extension", "-e",
+  "--skill",
+  "--model", "--system-prompt", "--append-system-prompt",
+  "--tools", "-t", "--exclude-tools", "-xt",
+  "--fork", "--session-dir", "--mode",
+  "--thinking", "--models",
+]);
+
+/** argv 中 flag 的起始索引：argv[0]=runtime, argv[1]=binary 路径。 */
+const ARGV_FLAG_START = 2;
+
+/**
+ * 从主进程 argv 解析可镜像的 flag（--no-extensions/--approve/--extension）。
+ *
+ * 数据源是主 pi 进程的 process.argv：子进程 spawn 的父就是主进程，
+ * 主进程 argv 完整保留启动时收到的全部 flag（已运行时验证）。这让子进程
+ * extension/approve 加载行为与主进程一致，且对任意 pi 宿主通用（不止 xyz-agent）。
+ *
+ * 解析规则：
+ * - --no-extensions / -ne、--approve / -a：布尔 flag
+ * - --extension / -e：支持 `--extension <path>`（空格）与 `--extension=<path>`（等号）
+ * - 其他 flag（在 VALUED_FLAGS 中）的值不被误当 extension 路径
+ * - positional 参数被忽略
+ * - argv[0]/argv[1] 是 bun/pi binary 路径，从 argv[2:] 开始扫
+ */
+export function mirrorMainProcessFlags(argv: readonly string[]): MirrorFlags {
+  let hasNoExtensions = false;
+  let hasApprove = false;
+  const extensionPaths: string[] = [];
+
+  // argv[0]=runtime(bun), argv[1]=pi binary 路径；flag 从 argv[2] 起
+  const flagArgs = argv.length > ARGV_FLAG_START ? argv.slice(ARGV_FLAG_START) : [];
+
+  for (let i = 0; i < flagArgs.length; i++) {
+    const tok = flagArgs[i];
+    if (tok === "--no-extensions" || tok === "-ne") {
+      hasNoExtensions = true;
+      continue;
+    }
+    if (tok === "--approve" || tok === "-a") {
+      hasApprove = true;
+      continue;
+    }
+    // 等号形式 --extension=path / -e=path
+    const eqMatch = /^(--extension|-e)=(.*)$/.exec(tok);
+    if (eqMatch) {
+      const val = eqMatch[2];
+      if (val) extensionPaths.push(val);
+      continue;
+    }
+    // 空格形式 --extension <path> / -e <path>：值在下一个 token
+    if (tok === "--extension" || tok === "-e") {
+      const next = flagArgs[i + 1];
+      if (next !== undefined && !next.startsWith("-") && next.length > 0) {
+        extensionPaths.push(next);
+        i++; // 跳过值
+      }
+      continue;
+    }
+    // 其他有值 flag：跳过其值，避免误吃（如 --skill /a 的 /a）
+    if (VALUED_FLAGS.has(tok)) {
+      i++;
+      continue;
+    }
+    // 其他情况（未知 flag、--flag=val 形式、positional）忽略
+  }
+
+  return { noExtensions: hasNoExtensions, approve: hasApprove, extensionPaths };
+}

package/src/execution/session-runner.ts

@@ -11,6 +11,7 @@ import * as fs from "node:fs";
 
 import type { ExtensionMode } from "@mariozechner/pi-coding-agent";
 
+import { type MirrorFlags, mirrorMainProcessFlags } from "./argv-mirror.ts";
 import { writeAliveMarker } from "./alive-store.ts";
 import { type DialogGlobalQueue, type UiRequestHandler } from "./dialog-queue.ts";
 import { updateFromEvent } from "./execution-record.ts";
@@ -422,6 +423,11 @@ export function buildSpawnArgs(
     sessionDir: string;
     forkSource: string | undefined;
     skillPaths: string[] | undefined;
+    /**
+     * 镜像自主进程 argv 的 flag（--no-extensions/--approve/--extension）。
+     * undefined 或全空/全 false 时行为不变（向后兼容）。
+     */
+    mirrorFlags?: MirrorFlags;
   },
 ): string[] {
   // task 不通过命令行传——pi 的 runRpcMode 只消费 stdin RpcCommand，
@@ -451,6 +457,16 @@ export function buildSpawnArgs(
       args.push("--skill", sp);
     }
   }
+  // 镜像主进程的 extension/approve flag：让子进程 extension 加载行为与主进程一致。
+  // undefined/空值时不追加（向后兼容）。顺序紧跟 skill 之后，注入类 flag 集中。
+  const mf = params.mirrorFlags;
+  if (mf) {
+    if (mf.noExtensions) args.push("--no-extensions");
+    if (mf.approve) args.push("--approve");
+    for (const ep of mf.extensionPaths) {
+      args.push("--extension", ep);
+    }
+  }
   return args;
 }
 
@@ -625,6 +641,8 @@ export async function runSpawn(
       sessionDir,
       forkSource,
       skillPaths: skillPaths.length > 0 ? skillPaths : undefined,
+      // 镜像主进程 argv 的 extension/approve flag，让子进程加载行为对齐主进程
+      mirrorFlags: mirrorMainProcessFlags(process.argv),
     },
   );
   const invocation = getPiInvocation(spawnArgs);