@zhushanwen/pi-msg-id-mapper 1.0.2 → 1.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-msg-id-mapper",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Client UUID ↔ user entry ID mapping extension for Pi — enables metadata preservation across sessions",
   "type": "module",
   "main": "src/index.ts",
@@ -32,6 +32,9 @@
     "@vitest/coverage-v8": "^4.1.9",
     "vitest": "^4.1.8"
   },
+  "dependencies": {
+    "@zhushanwen/pi-extension-logger": "0.3.0"
+  },
   "scripts": {
     "typecheck": "npx tsc --noEmit",
     "test": "vitest run"

package/src/__tests__/msg-id-mapper.test.ts

@@ -14,6 +14,14 @@
  * 运行：cd extensions/taiji/msg-id-mapper && npx vitest run
  */
 import { describe, it, expect, vi, afterEach } from 'vitest'
+const { loggerMock } = vi.hoisted(() => ({
+  loggerMock: { debug: vi.fn(), warn: vi.fn(), error: vi.fn() },
+}))
+vi.mock('@zhushanwen/pi-extension-logger', () => ({
+  getLogger: () => loggerMock,
+  createLogger: () => loggerMock,
+}))
+
 import createMapper from '../index'
 import type {
   ExtensionAPI,
@@ -147,12 +155,12 @@ describe('input hook · clientUuid 提取（extractClientUuid）', () => {
 
   it('畸形 event（handler 内抛错）→ 兜底 return undefined 不外抛（console.error 可观测）', () => {
     const h = createHarness()
-    const errSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
+    const errSpy = loggerMock.error
     // event 为 null → 取 event.source 即 throw，走 catch 兜底
     expect(h.inputRaw(null)).toBeUndefined()
     expect(errSpy).toHaveBeenCalledWith(
-      '[xyz-client-msg-id-mapper] input hook error:',
-      expect.any(TypeError),
+      'input hook error',
+      { detail: expect.stringContaining('TypeError') },
     )
   })
 })
@@ -240,7 +248,7 @@ describe('flush · 映射写入 / 幂等 / 重试 / 异常兜底（writeMapping
 
   it('appendEntry 抛错 → flush 吞错不外抛，flag 未清，下次 hook 幂等重试', () => {
     const h = createHarness()
-    const errSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
+    const errSpy = loggerMock.error
     h.appendEntry.mockImplementation(() => {
       throw new Error('jsonl write failed')
     })
@@ -248,7 +256,7 @@ describe('flush · 映射写入 / 幂等 / 重试 / 异常兜底（writeMapping
     h.input(`task ${marker(UUID)}`)
     h.messageEnd('user')
     expect(() => h.flush('message_start')).not.toThrow()
-    expect(errSpy).toHaveBeenCalledWith('[xyz-client-msg-id-mapper] flush error:', expect.any(Error))
+    expect(errSpy).toHaveBeenCalledWith('flush error', { detail: expect.stringContaining('jsonl write failed') })
 
     // 重试成功：appendEntry 恢复 → turn_end 补写（flag 未清的幂等重试语义）
     h.appendEntry.mockImplementation(() => {})
@@ -258,7 +266,7 @@ describe('flush · 映射写入 / 幂等 / 重试 / 异常兜底（writeMapping
 
   it('getLeafId 抛错 → flush 吞错不外抛（catch 兜底覆盖 writeMapping 全程）', () => {
     const h = createHarness()
-    const errSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
+    const errSpy = loggerMock.error
     h.input(`task ${marker(UUID)}`)
     h.messageEnd('user')
 
@@ -267,7 +275,7 @@ describe('flush · 映射写入 / 幂等 / 重试 / 异常兜底（writeMapping
         throw new Error('session manager gone')
       }),
     ).not.toThrow()
-    expect(errSpy).toHaveBeenCalledWith('[xyz-client-msg-id-mapper] flush error:', expect.any(Error))
+    expect(errSpy).toHaveBeenCalledWith('flush error', { detail: expect.stringContaining('session manager gone') })
   })
 
   it('无 user message_end（未置 flag）→ 三重安全网全不写', () => {

package/src/index.ts

@@ -29,6 +29,9 @@
  */
 
 import type { ExtensionAPI, ExtensionContext, InputEvent, MessageEndEvent, MessageStartEvent, TurnEndEvent, AgentEndEvent } from '@earendil-works/pi-coding-agent'
+import { getLogger } from '@zhushanwen/pi-extension-logger'
+
+const logger = getLogger('xyz-client-msg-id-mapper')
 
 // uuid = appendUser 生成的 user message id（`u-` + 36 字符 hex uuid，共 38 字符）。
 // 与 shared SegmentsMetadataEntry.clientUuid 严格一致（同 appendUser 返回值）：
@@ -67,7 +70,7 @@ export default function (pi: ExtensionAPI): void {
       return { action: 'transform' as const, text: event.text.replace(TAG_STRIP, '').trimEnd() }
     } catch (err) {
       // 吞错，不阻断主流程（pi runner 也会 try/catch，但显式兜底避免意外 return）。
-      console.error('[xyz-client-msg-id-mapper] input hook error:', err)
+      logger.error('input hook error', { detail: String(err) })
       return undefined
     }
   })
@@ -81,7 +84,7 @@ export default function (pi: ExtensionAPI): void {
       }
     } catch (err) {
       // best-effort：置 flag 失败不阻断消息流——仅丢失该条 clientUuid↔entryId 映射，对话不受影响。
-      console.error('[xyz-client-msg-id-mapper] message_end hook error:', err)
+      logger.error('message_end hook error', { detail: String(err) })
     }
     return undefined
   })
@@ -109,7 +112,7 @@ export default function (pi: ExtensionAPI): void {
       writeMapping(ctx)
     } catch (err) {
       // best-effort：映射写入失败不阻断消息流——丢的是本条映射，下次 hook 会因 flag 未清而幂等重试。
-      console.error('[xyz-client-msg-id-mapper] flush error:', err)
+      logger.error('flush error', { detail: String(err) })
     }
   }