@zhushanwen/pi-system-prompt 1.3.2 → 1.4.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-system-prompt",
-  "version": "1.3.2",
+  "version": "1.4.1",
   "description": "System prompt injection extension for Pi — reads config and appends to system prompt",
   "type": "module",
   "main": "src/index.ts",
@@ -36,6 +36,6 @@
   },
   "scripts": {
     "typecheck": "npx tsc --noEmit",
-    "test": "vitest run"
+    "test": "vitest run --pool=threads"
   }
 }

package/src/__tests__/system-prompt.test.ts

@@ -8,7 +8,7 @@
  *   损坏形态（"enabled": "false" 字符串 / capability 非对象）→ 注入（与 readSection 解析方向相反）
  * - before_agent_start 注入顺序：base → global instructions → capability → append（indexOf 链锁定，
  *   swap 注入顺序两行的 mutant 会被顺序用例 kill）
- * - -nc / --no-context-files 守卫：global 注入跳过、append 不受影响
+ * - -nc / --no-context-files 检查：global 注入跳过、append 不受影响
  * - global 候选选择：候选序优先、空白内容跳过继续找、目录缺失降级
  * - fail-safe：handler 全程 throw → return undefined + logger.error 可观测；logger 自身抛错的
  *   终极兜底 process.stderr.write；systemPrompt 非法类型的旧 quirk 锚定
@@ -17,7 +17,7 @@
  * 在 config 里显式 "capability":{"enabled":false} 隔离关注点；无法塞字段的用例（畸形 JSON /
  * 文件缺失）期望值如实更新为含 capability 段。capability 自身行为由专项 describe 锚定。
  *
- * mock 策略（参照 msg-id-mapper 测试模式）：pi SDK import type 零运行时解析，
+ * mock 策略（参照 host-internals msg-id-mapper 测试模式）：pi SDK import type 零运行时解析，
  * ExtensionAPI 用结构化桩；node:fs mock 后按路径分流（env 指向假目录，不碰真实文件系统）。
  *
  * 运行：cd extensions/taiji/system-prompt && npx vitest run
@@ -38,6 +38,7 @@ import createExtension, {
   CAPABILITY_INLINE_TAG_FAMILIES,
   CAPABILITY_PRESENTATION_ATTRS,
   SESSION_ARTIFACTS_DIR_SEGMENT,
+  SESSION_STORE_DIR_SEGMENT,
   resolveSessionArtifactsDir,
 } from '../index'
 import type { ExtensionAPI, BeforeAgentStartEvent } from '@earendil-works/pi-coding-agent'
@@ -57,7 +58,7 @@ const CAP_HEADER = '# TaiJi capabilities'
 /** 既有用例隔离关注点用的「capability 显式关闭」config 片段 */
 const CAP_OFF = '"capability": {"enabled": false}'
 
-/** hook 注册表桩（参照 msg-id-mapper harness 模式）——handler 第二参 ctx 透传（路径注入测试用） */
+/** hook 注册表桩（参照 host-internals msg-id-mapper harness 模式）——handler 第二参 ctx 透传（路径注入测试用） */
 function createHarness(): { beforeAgentStart: (event: BeforeAgentStartEvent, ctx?: unknown) => unknown } {
   const handlers = new Map<string, (...args: unknown[]) => unknown>()
   const pi = {
@@ -136,7 +137,7 @@ const savedEnv: Record<string, string | undefined> = {}
 const savedArgv = process.argv
 
 beforeEach(() => {
-  vi.clearAllMocks() // 清跨用例的 fs mock 调用记录（「守卫不触发」类断言依赖零计数）
+  vi.clearAllMocks() // 清跨用例的 fs mock 调用记录（「检查不触发」类断言依赖零计数）
   for (const k of ENV_KEYS) savedEnv[k] = process.env[k]
   process.env.TAIJI_AGENT_DATA_DIR = DATA_DIR
   process.env.TAIJI_GLOBAL_AGENTS_DIR = GLOBAL_DIR
@@ -237,7 +238,7 @@ describe('before_agent_start 注入顺序（base → global → capability → a
     expect(prompt).toContain(path.join(GLOBAL_DIR, 'AGENTS.md'))
   })
 
-  it('capability 段文案覆盖四点能力面（HTML 白名单 / 相对图片 / 相对链接双通道 / 远程图片不渲染）', () => {
+  it('capability 段文案覆盖四点能力侧（HTML 白名单 / 相对图片 / 相对链接双通道 / 远程图片不渲染）', () => {
     setupFs({ config: '{"append": {"enabled": false, "prompt": ""}}' })
     const prompt = runHook('BASE-PROMPT')!.systemPrompt as string
     const capText = prompt.slice(prompt.indexOf(CAP_HEADER))
@@ -314,9 +315,9 @@ describe('capability 段三态锚点（设计 D6：仅显式布尔 false 关闭
   })
 })
 
-describe('-nc / --no-context-files 守卫（contextFilesDisabled）', () => {
-  // capability 与守卫正交（内置段不受 --no-context-files 影响，该旗标只管上下文文件
-  // 发现），用例显式关 capability 隔离守卫语义。
+describe('-nc / --no-context-files 检查（contextFilesDisabled）', () => {
+  // capability 与检查正交（内置段不受 --no-context-files 影响，该旗标只管上下文文件
+  // 发现），用例显式关 capability 隔离检查语义。
   it('argv 含 --no-context-files → global 不注入，append 仍生效（用户显式退出不得溜回来）', () => {
     setupFs({
       config: `{"append": {"enabled": true, "prompt": "APPEND-TEXT"}, ${CAP_OFF}}`,
@@ -325,11 +326,11 @@ describe('-nc / --no-context-files 守卫（contextFilesDisabled）', () => {
     })
     process.argv = ['node', 'pi', '--no-context-files']
     expect(runHook('BASE-PROMPT')).toEqual({ systemPrompt: 'BASE-PROMPT\n\nAPPEND-TEXT' })
-    // 守卫在读 global 文件之前：readdirSync 不应被调用（global 目录完全不被触碰）
+    // 检查在读 global 文件之前：readdirSync 不应被调用（global 目录完全不被触碰）
     expect(readdirSync).not.toHaveBeenCalled()
   })
 
-  it('argv 含 -nc 短形式 → 同样跳过 global 注入（pi CLI 等价短形式，守卫两种形式都命中）', () => {
+  it('argv 含 -nc 短形式 → 同样跳过 global 注入（pi CLI 等价短形式，检查两种形式都命中）', () => {
     setupFs({
       config: `{"append": {"enabled": true, "prompt": "APPEND-TEXT"}, ${CAP_OFF}}`,
       globalEntries: ['AGENTS.md'],
@@ -385,26 +386,8 @@ describe('fail-safe（外层 catch return undefined，永不阻断 agent loop）
     )
   })
 
-  it('logger 自身抛错的终极兜底 → stderr 兜底，仍 return undefined', () => {
-    const stderrSpy = vi.spyOn(process.stderr, 'write').mockImplementation(() => true)
-    loggerMock.error.mockImplementation(() => {
-      throw new Error('logger gone')
-    })
-    const h = createHarness()
-    const event = { type: 'before_agent_start', prompt: 'hi' } as unknown as BeforeAgentStartEvent
-    Object.defineProperty(event, 'systemPrompt', {
-      get() {
-        throw new Error('getter boom')
-      },
-    })
-    expect(h.beforeAgentStart(event)).toBeUndefined()
-    expect(stderrSpy).toHaveBeenCalledWith(
-      expect.stringContaining('logHookFailure also failed'),
-    )
-  })
-
   it('systemPrompt 非法类型（undefined）且无 append → 旧 quirk 锚定：返回仅含 capability 段的结果而非 undefined', () => {
-    // R3 复核锚定的返回值守卫 quirk（index.ts newPrompt === event.systemPrompt 比较）：
+    // R3 复核锚定的返回值检查 quirk（index.ts newPrompt === event.systemPrompt 比较）：
     // base 收敛 ''，capability 默认开 → newPrompt 非空且 !== undefined → 必然返回对象。
     // 与重构前行为一致（非回归）：quirk 在「有注入时必然返回对象」，schema v2 后 capability
     // 默认开使该形态成为常态。
@@ -529,17 +512,20 @@ describe('capability 段 ①② 渲染锁定（文案 = 常量渲染产物，手
   })
 })
 
-describe('产物目录镜像常量与推导 helper（设计 D1/D7，供 u-artifacts 公式对拍消费）', () => {
-  it('段名字面量为 artifacts', () => {
+describe('产物目录镜像常量与推导 helper（session 存储树布局，公式对拍机检消费）', () => {
+  it('段名字面量：树根段 ext-sessions + 域段 artifacts（与 shared SESSION_STORE_DIR_SEGMENT / getSessionArtifactsDir 对拍）', () => {
+    expect(SESSION_STORE_DIR_SEGMENT).toBe('ext-sessions')
     expect(SESSION_ARTIFACTS_DIR_SEGMENT).toBe('artifacts')
   })
 
-  it('合法 pi sid → <dataDir>/artifacts/<sessionId>', () => {
+  it('合法 pi sid → <dataDir>/ext-sessions/<sessionId>/artifacts', () => {
     expect(resolveSessionArtifactsDir('0198ab12-cdef-7000-8000-1234567890ab')).toBe(
-      path.join(DATA_DIR, 'artifacts', '0198ab12-cdef-7000-8000-1234567890ab'),
+      path.join(DATA_DIR, 'ext-sessions', '0198ab12-cdef-7000-8000-1234567890ab', 'artifacts'),
     )
     // 允许 `.` 与 `_`（与 isPiSessionId 同域）
-    expect(resolveSessionArtifactsDir('a.b-c_d')).toBe(path.join(DATA_DIR, 'artifacts', 'a.b-c_d'))
+    expect(resolveSessionArtifactsDir('a.b-c_d')).toBe(
+      path.join(DATA_DIR, 'ext-sessions', 'a.b-c_d', 'artifacts'),
+    )
   })
 
   it('非法 sessionId（含冒号 / 空串 / 首尾非字母数字）→ throw（防路径穿越）', () => {
@@ -551,7 +537,9 @@ describe('产物目录镜像常量与推导 helper（设计 D1/D7，供 u-artifa
   it('resolveDataDir 回退（无 TAIJI_AGENT_DATA_DIR 时取 PI_CODING_AGENT_DIR 上级）', () => {
     delete process.env.TAIJI_AGENT_DATA_DIR
     process.env.PI_CODING_AGENT_DIR = path.join('/tmp/fallback-data', 'agent')
-    expect(resolveSessionArtifactsDir('abc')).toBe(path.join('/tmp/fallback-data', 'artifacts', 'abc'))
+    expect(resolveSessionArtifactsDir('abc')).toBe(
+      path.join('/tmp/fallback-data', 'ext-sessions', 'abc', 'artifacts'),
+    )
   })
 })
 
@@ -576,9 +564,9 @@ describe('capability 段 ③④（HTML 交付约定 + 预览约束）与产物
 
   it('ctx 会话 id 在 → ③ 段含产物目录绝对路径（与会话 id 匹配）', () => {
     const text = capabilityTextWithCtx(ctxWithSessionId(() => SID))
-    // 路径 = <dataDir>/artifacts/<sid>（与 shared getSessionArtifactsDir 同公式）
+    // 路径 = <dataDir>/ext-sessions/<sid>/artifacts（与 shared getSessionArtifactsDir 同公式）
     expect(text).toContain(resolveSessionArtifactsDir(SID))
-    expect(text).toContain(path.join(DATA_DIR, 'artifacts', SID))
+    expect(text).toContain(path.join(DATA_DIR, 'ext-sessions', SID, 'artifacts'))
     // ③ 段交付约定与 ④ 段预览约束齐备
     expect(text).toContain('info string is html-preview')
     expect(text).toContain("that file's absolute path")
@@ -605,7 +593,7 @@ describe('capability 段 ③④（HTML 交付约定 + 预览约束）与产物
       expect(text).toContain('blocked by the browser CORS policy')
       expect(text).toContain('CDN links will not load')
       // 不出现半截/伪造路径
-      expect(text).not.toContain(path.join(DATA_DIR, 'artifacts'))
+      expect(text).not.toContain(path.join(DATA_DIR, 'ext-sessions'))
     }
   })
 
@@ -616,7 +604,7 @@ describe('capability 段 ③④（HTML 交付约定 + 预览约束）与产物
       const text = capabilityTextWithCtx(ctx)
       expect(text).toContain('(unavailable this turn)')
       expect(text).toContain('info string is html-preview')
-      expect(text).not.toContain(path.join(DATA_DIR, 'artifacts'))
+      expect(text).not.toContain(path.join(DATA_DIR, 'ext-sessions'))
     }
   })
 

package/src/index.ts

@@ -214,7 +214,18 @@ function resolveDataDir(): string {
 }
 
 /**
- * 会话产物目录镜像常量（设计 D1/D7）：`<dataDir>/artifacts/<sessionId>/`。
+ * session 存储树根段镜像常量（session-store-layout-inversion D1）：
+ * `<dataDir>/ext-sessions/<sessionId>/...` 树根段。
+ *
+ * shared 侧单点声明 = `packages/shared/src/paths.ts` 的 `SESSION_STORE_DIR_SEGMENT` /
+ * `getSessionStoreDir`；本包不 import `@taiji/shared`（包边界 + C-proc-26/C-proc-09
+ * 门禁组合，见下方产物目录镜像说明），以同名字面量镜像，一致性由
+ * `scripts/check-artifact-dir-formula-sync.mjs` 对拍机检守护。
+ */
+export const SESSION_STORE_DIR_SEGMENT = 'ext-sessions'
+
+/**
+ * 会话产物目录镜像常量（session 存储树域 2）：`<dataDir>/ext-sessions/<sessionId>/artifacts/`。
  *
  * shared 侧单点声明 = `packages/shared/src/paths.ts` 的 `getSessionArtifactsDir`；本包
  * 不 import `@taiji/shared`（① 包边界——extensions 包全体不依赖 @taiji/shared；② 运行时
@@ -226,22 +237,22 @@ function resolveDataDir(): string {
 export const SESSION_ARTIFACTS_DIR_SEGMENT = 'artifacts'
 
 /**
- * 会话产物目录绝对路径（镜像 shared `getSessionArtifactsDir` 公式）。
+ * 会话产物目录绝对路径（镜像 shared `getSessionArtifactsDir` 公式：
+ * `<dataDir>/ext-sessions/<sessionId>/artifacts`）。
  *
  * sessionId 校验与 pi `assertValidSessionId` / runtime `isPiSessionId` 同域：首尾字母
- * 数字，中间允许 `[A-Za-z0-9._-]`（允许 `.` / `_`，禁 `:`）——不用 `getImageCacheDir`
- * 的窄集 `/^[A-Za-z0-9_-]+$/`（合法含 `.` 的 pi sid 会被窄集误拒）。非法 sessionId
+ * 数字，中间允许 `[A-Za-z0-9._-]`（允许 `.` / `_`，禁 `:`）。非法 sessionId
  * 直接 throw（防路径穿越）。
  *
  * @param sessionId 会话 id（子目录分区，须与 `isPiSessionId` 同域）
  * @throws Error 当 sessionId 含路径分隔符、冒号、空串或首尾非字母数字
  */
 export function resolveSessionArtifactsDir(sessionId: string): string {
-  // 校验与 pi-paths.ts 的 isPiSessionId 同域（字面量与 shared getSessionArtifactsDir 一致）
+  // 校验与 pi-paths.ts 的 isPiSessionId 同域（字面量与 shared PI_SESSION_ID_PATTERN 一致）
   if (!/^[A-Za-z0-9](?:[A-Za-z0-9._-]*[A-Za-z0-9])?$/.test(sessionId)) {
     throw new Error(`invalid sessionId (path traversal blocked): ${sessionId}`)
   }
-  return path.join(resolveDataDir(), SESSION_ARTIFACTS_DIR_SEGMENT, sessionId)
+  return path.join(resolveDataDir(), SESSION_STORE_DIR_SEGMENT, sessionId, SESSION_ARTIFACTS_DIR_SEGMENT)
 }
 
 /**
@@ -392,8 +403,7 @@ function readSection(raw: unknown): { enabled: boolean; prompt: string } {
 /**
  * pi 是否以 --no-context-files / -nc 启动。用户显式退出 AGENTS.md
  * 自动发现时，全局文件不得从这条通路溜回来。pi CLI 把 -nc 视为 --no-context-files
- * 的等价短形式（cli/args.ts），两种形式都必须命中守卫——与镜像侧
- * （argv-mirror.ts 同样解析两种形式）保持一致。
+ * 的等价短形式（cli/args.ts），本守卫单侧同时覆盖两种 argv 形态。
  */
 function contextFilesDisabled(): boolean {
   return process.argv.includes('--no-context-files') || process.argv.includes('-nc')
@@ -442,22 +452,15 @@ function buildSystemPrompt(event: BeforeAgentStartEvent, ctx?: ExtensionContext)
  * 落盘诊断，不泄露配置内容。
  *
  * 通道：extension-logger 的 error → appendEntry 写入 session JSONL（需 setPiHandle
- * 注入 pi handle 后生效）；TAIJI_AGENT_DEBUG=1 时另落文件日志。仅当 logger 自身抛错
- * 时才兜底 process.stderr.write（下方 catch），不外泄到 agent loop。
+ * 注入 pi handle 后生效）；TAIJI_AGENT_DEBUG=1 时另落文件日志。logger 全程吞错
+ * （appendEntry 与 fileLog 通道内部各自整体 try/catch，实装见 extension-logger
+ * error 方法），不会抛错到本调用方——旧的两层嵌套兜底（stderr 终极兜底 + 静默
+ * catch）已删，恒不命中。若未来 logger 改为可抛实现，异常由外层 handler 的
+ * try/catch 兜住（同样不外泄到 agent loop）。
  */
 function logHookFailure(err: unknown): void {
-  try {
-    const msg = err instanceof Error ? `${err.name}: ${err.message}` : String(err)
-    logger.error(`before_agent_start hook failed: ${msg}`)
-  } catch (nestedErr) {
-    // best-effort：logger 抛错时的终极兜底 process.stderr.write——其内部吞错不会抛，仍不外泄到 agent loop。
-    try {
-      process.stderr.write(`[taiji-system-prompt-extension] logHookFailure also failed: ${String(nestedErr)}\n`)
-    } catch (finalErr) {
-      /* 完全静默：两层兜底都失败时无处可写 */
-      void finalErr
-    }
-  }
+  const msg = err instanceof Error ? `${err.name}: ${err.message}` : String(err)
+  logger.error(`before_agent_start hook failed: ${msg}`)
 }
 
 export default function (pi: ExtensionAPI): void {