feat: add WeChat 4.1.13 macOS key capture

2026-10-06 05:27:48 +08:00 · 2026-09-14 18:20:11 +08:00
parent ffbe78f74a
commit 00095ae627
5 changed files with 218 additions and 58 deletions
@@ -0,0 +1,96 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+vi.mock('electron', () => ({
+  app: {
+    getPath: vi.fn(() => '/tmp/tracememo-key-service-test')
+  }
+}))
+
+import {
+  buildXkeyHelperArguments,
+  mapXkeyHelperFailure,
+  parseXkeyHelperOutput
+} from '../../src/main/key-service-mac'
+
+describe('parseXkeyHelperOutput', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('returns a valid helper key without exposing progress output', () => {
+    const key = 'ab'.repeat(32)
+    expect(
+      parseXkeyHelperOutput(`progress line\n${JSON.stringify({ success: true, key })}`)
+    ).toEqual({ success: true, key })
+  })
+
+  it('maps a login-time capture timeout to an actionable message', () => {
+    const result = parseXkeyHelperOutput(
+      JSON.stringify({
+        success: false,
+        result:
+          'sink found at 0x1234\nsoftware breakpoint unavailable\nhardware breakpoint armed\ntimeout waiting for breakpoint hit'
+      })
+    )
+
+    expect(result).toEqual({
+      success: false,
+      code: 'CAPTURE_TIMEOUT',
+      error:
+        '已完成管理员授权，但监听期间微信没有触发账号密钥派生。请先停留在微信登录界面，在 TraceMemo 点击“自动获取密钥”，授权后点击微信“登录”；已有登录凭据时通常不需要扫码。'
+    })
+  })
+
+  it('does not expose raw helper diagnostics for an unknown failure', () => {
+    const result = parseXkeyHelperOutput(
+      JSON.stringify({
+        success: false,
+        result: 'private diagnostic /Users/example/source/xkey.mm:123'
+      })
+    )
+
+    expect(result).toEqual({
+      success: false,
+      code: 'HELPER_RESULT_INVALID',
+      error: '密钥工具未返回有效密钥，请确认微信仍在运行后重试。'
+    })
+    expect(result.error).not.toContain('/Users/example')
+  })
+
+  it('sanitizes diagnostics returned through an AppleScript command error', () => {
+    const result = mapXkeyHelperFailure(
+      'sink found at 0x1234\nhardware breakpoint armed on 81/81 threads\nWAIT_FAILED:no_breakpoint_hit',
+      'OSASCRIPT_1'
+    )
+
+    expect(result).toEqual({
+      success: false,
+      code: 'CAPTURE_TIMEOUT',
+      error:
+        '已完成管理员授权，但监听期间微信没有触发账号密钥派生。请先停留在微信登录界面，在 TraceMemo 点击“自动获取密钥”，授权后点击微信“登录”；已有登录凭据时通常不需要扫码。'
+    })
+  })
+
+  it('uses the verified 4.1.13 account-key capture contract', () => {
+    expect(buildXkeyHelperArguments(60037, 120_000)).toEqual([
+      '60037',
+      '120000',
+      '--profile',
+      'wechat-4.1.13',
+      '--account'
+    ])
+  })
+
+  it('extracts a helper error embedded in AppleScript diagnostics', () => {
+    expect(
+      mapXkeyHelperFailure(
+        'hardware breakpoint armed\n{"success":false,"result":"ERROR:HOOK_FAILED:ptrace"}',
+        'OSASCRIPT_1'
+      )
+    ).toEqual({
+      success: false,
+      code: 'HOOK_FAILED',
+      error: '密钥工具执行未完成（HOOK_FAILED），请确认微信仍在运行后重试。'
+    })
+  })
+})