@chainlesschain/personal-data-hub 0.1.0 → 0.2.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/__tests__/adapters/ai-chat-cookie-capture-spec.test.js +211 -0
- package/__tests__/adapters/ai-chat-health-checker.test.js +262 -0
- package/__tests__/adapters/ai-chat-history.test.js +396 -0
- package/__tests__/adapters/ai-chat-http-client.test.js +242 -0
- package/__tests__/adapters/ai-chat-vendors.test.js +874 -0
- package/__tests__/adapters/alipay-bill-adapter.test.js +538 -0
- package/__tests__/adapters/email-adapter.test.js +138 -1
- package/__tests__/adapters/email-classifier.test.js +347 -0
- package/__tests__/adapters/email-pdf-extractor.test.js +529 -0
- package/__tests__/adapters/email-retry-progress.test.js +294 -0
- package/__tests__/adapters/email-templates.test.js +699 -0
- package/__tests__/adapters/social-toutiao-kuaishou-scaffold.test.js +269 -0
- package/__tests__/adapters/system-data-adapter.test.js +440 -0
- package/__tests__/adapters/system-data-android-ingest.test.js +144 -0
- package/__tests__/adapters/system-data-android.test.js +387 -0
- package/__tests__/adapters/system-data-disclosure.test.js +153 -0
- package/__tests__/adapters/wechat-bootstrap.test.js +240 -0
- package/__tests__/adapters/wechat-env-probe.test.js +162 -0
- package/__tests__/adapters/wechat-frida-agent.test.js +191 -0
- package/__tests__/adapters/wechat-frida-integration.test.js +149 -0
- package/__tests__/adapters/wechat-frida-key-provider.test.js +188 -0
- package/__tests__/adapters/wechat-md5-key-provider.test.js +101 -0
- package/__tests__/analysis-skills.test.js +556 -0
- package/__tests__/analysis.test.js +329 -1
- package/__tests__/e2e/ai-chat-cross-source-journey.test.js +213 -0
- package/__tests__/e2e/full-user-journey.test.js +188 -0
- package/__tests__/entity-resolver-ingest-hook.test.js +177 -0
- package/__tests__/entity-resolver-stages.test.js +411 -0
- package/__tests__/entity-resolver-vault.test.js +246 -0
- package/__tests__/entity-resolver.test.js +526 -0
- package/__tests__/fixtures/entity-resolver-200-mock.json +96 -0
- package/__tests__/integration/ai-chat-history-registry.test.js +228 -0
- package/__tests__/integration/aichat-wizard-end-to-end.test.js +282 -0
- package/__tests__/integration/cross-adapter-pipelines.test.js +396 -0
- package/__tests__/integration/wechat-bootstrap-end-to-end.test.js +390 -0
- package/__tests__/longtail-adapters.test.js +217 -0
- package/__tests__/mobile-extractor.test.js +288 -0
- package/__tests__/registry.test.js +4 -2
- package/__tests__/shopping-adapters.test.js +296 -0
- package/__tests__/sidecar-contacts-cross-validate.test.js +163 -0
- package/__tests__/sidecar-supervisor.test.js +120 -0
- package/__tests__/social-adapters.test.js +206 -0
- package/__tests__/travel-adapters.test.js +325 -0
- package/__tests__/vault.test.js +3 -3
- package/__tests__/wechat-adapter.test.js +476 -0
- package/__tests__/whatsapp-adapter.test.js +135 -0
- package/lib/adapter-spec.js +12 -0
- package/lib/adapters/_python-sidecar-base.js +207 -0
- package/lib/adapters/ai-chat-history/ai-chat-adapter.js +374 -0
- package/lib/adapters/ai-chat-history/cookie-auth.js +109 -0
- package/lib/adapters/ai-chat-history/cookie-capture-spec.js +331 -0
- package/lib/adapters/ai-chat-history/health-checker.js +210 -0
- package/lib/adapters/ai-chat-history/http-client.js +211 -0
- package/lib/adapters/ai-chat-history/index.js +28 -0
- package/lib/adapters/ai-chat-history/schema-map.js +258 -0
- package/lib/adapters/ai-chat-history/vendor-spec.js +86 -0
- package/lib/adapters/ai-chat-history/vendors/coze.js +179 -0
- package/lib/adapters/ai-chat-history/vendors/deepseek.js +199 -0
- package/lib/adapters/ai-chat-history/vendors/doubao.js +255 -0
- package/lib/adapters/ai-chat-history/vendors/dreamina.js +174 -0
- package/lib/adapters/ai-chat-history/vendors/hunyuan.js +176 -0
- package/lib/adapters/ai-chat-history/vendors/kimi.js +182 -0
- package/lib/adapters/ai-chat-history/vendors/qianfan.js +160 -0
- package/lib/adapters/ai-chat-history/vendors/tongyi.js +193 -0
- package/lib/adapters/ai-chat-history/vendors/zhipu.js +202 -0
- package/lib/adapters/ai-chat-history/wizard-controller.js +473 -0
- package/lib/adapters/alipay-bill/alipay-bill-adapter.js +311 -0
- package/lib/adapters/alipay-bill/counterparty.js +129 -0
- package/lib/adapters/alipay-bill/csv-parser.js +217 -0
- package/lib/adapters/alipay-bill/index.js +41 -0
- package/lib/adapters/alipay-bill/zip-decryptor.js +111 -0
- package/lib/adapters/email-imap/classifier.js +495 -0
- package/lib/adapters/email-imap/email-adapter.js +419 -8
- package/lib/adapters/email-imap/index.js +42 -0
- package/lib/adapters/email-imap/pdf-extractor.js +192 -0
- package/lib/adapters/email-imap/templates/bill.js +232 -0
- package/lib/adapters/email-imap/templates/government.js +120 -0
- package/lib/adapters/email-imap/templates/index.js +78 -0
- package/lib/adapters/email-imap/templates/order.js +186 -0
- package/lib/adapters/email-imap/templates/other.js +114 -0
- package/lib/adapters/email-imap/templates/register.js +113 -0
- package/lib/adapters/email-imap/templates/travel.js +157 -0
- package/lib/adapters/email-imap/templates/utils.js +275 -0
- package/lib/adapters/email-imap/transactions.js +234 -0
- package/lib/adapters/messaging-qq/index.js +158 -0
- package/lib/adapters/messaging-telegram/index.js +142 -0
- package/lib/adapters/messaging-whatsapp/index.js +189 -0
- package/lib/adapters/shopping-base/index.js +208 -0
- package/lib/adapters/shopping-jd/index.js +150 -0
- package/lib/adapters/shopping-meituan/index.js +154 -0
- package/lib/adapters/shopping-taobao/index.js +176 -0
- package/lib/adapters/social-bilibili/index.js +171 -0
- package/lib/adapters/social-douyin/index.js +116 -0
- package/lib/adapters/social-kuaishou/index.js +237 -0
- package/lib/adapters/social-toutiao/index.js +236 -0
- package/lib/adapters/social-weibo/index.js +164 -0
- package/lib/adapters/social-xiaohongshu/index.js +96 -0
- package/lib/adapters/system-data/disclosure.js +166 -0
- package/lib/adapters/system-data/index.js +34 -0
- package/lib/adapters/system-data/system-data-adapter.js +344 -0
- package/lib/adapters/system-data-android/adapter.js +348 -0
- package/lib/adapters/system-data-android/index.js +76 -0
- package/lib/adapters/travel-12306/index.js +151 -0
- package/lib/adapters/travel-amap/index.js +164 -0
- package/lib/adapters/travel-baidu-map/index.js +162 -0
- package/lib/adapters/travel-base/index.js +240 -0
- package/lib/adapters/travel-ctrip/index.js +151 -0
- package/lib/adapters/wechat/bootstrap.js +146 -0
- package/lib/adapters/wechat/content-parser.js +326 -0
- package/lib/adapters/wechat/db-reader.js +209 -0
- package/lib/adapters/wechat/env-probe.js +218 -0
- package/lib/adapters/wechat/frida-agent/loader.js +67 -0
- package/lib/adapters/wechat/frida-agent/wechat-key-hook.js +126 -0
- package/lib/adapters/wechat/index.js +37 -0
- package/lib/adapters/wechat/key-extractor.js +158 -0
- package/lib/adapters/wechat/key-providers/frida-key-provider.js +244 -0
- package/lib/adapters/wechat/key-providers/index.js +22 -0
- package/lib/adapters/wechat/key-providers/key-provider-base.js +44 -0
- package/lib/adapters/wechat/key-providers/md5-key-provider.js +81 -0
- package/lib/adapters/wechat/normalize.js +220 -0
- package/lib/adapters/wechat/wechat-adapter.js +205 -0
- package/lib/analysis-skills/base.js +113 -0
- package/lib/analysis-skills/footprint.js +167 -0
- package/lib/analysis-skills/index.js +58 -0
- package/lib/analysis-skills/interests.js +161 -0
- package/lib/analysis-skills/relations.js +226 -0
- package/lib/analysis-skills/spending.js +219 -0
- package/lib/analysis-skills/timeline.js +167 -0
- package/lib/analysis.js +191 -2
- package/lib/entity-resolver/embedding-stage.js +198 -0
- package/lib/entity-resolver/entity-resolver.js +384 -0
- package/lib/entity-resolver/index.js +42 -0
- package/lib/entity-resolver/llm-stage.js +191 -0
- package/lib/entity-resolver/rule-stage.js +208 -0
- package/lib/entity-resolver/worker.js +149 -0
- package/lib/index.js +131 -0
- package/lib/migrations.js +73 -0
- package/lib/mobile-extractor/android.js +193 -0
- package/lib/mobile-extractor/index.js +9 -0
- package/lib/mobile-extractor/ios.js +223 -0
- package/lib/prompt-builder.js +11 -1
- package/lib/query-parser.js +7 -1
- package/lib/registry.js +42 -0
- package/lib/sidecar/index.js +15 -0
- package/lib/sidecar/supervisor.js +359 -0
- package/lib/vault.js +343 -0
- package/package.json +36 -3
- package/scripts/_make-fixture-all.js +126 -0
- package/scripts/_make-fixture-contacts.js +84 -0
- package/scripts/evaluate-entity-resolver.js +213 -0
- package/scripts/smoke-phase-5-5.js +196 -0
- package/scripts/smoke-phase-5-7.js +181 -0
- package/scripts/smoke-system-data-contacts.js +309 -0
- package/scripts/smoke-system-data.js +312 -0
|
@@ -0,0 +1,188 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
/**
|
|
4
|
+
* E2E — full Personal Data Hub user journey.
|
|
5
|
+
*
|
|
6
|
+
* Simulates a user from zero to "ask a question and get an answer":
|
|
7
|
+
* 1. Initialize fresh vault
|
|
8
|
+
* 2. Register + sync EmailAdapter (mocked IMAP session)
|
|
9
|
+
* 3. Register + import Alipay CSV
|
|
10
|
+
* 4. EntityResolver auto-merges same-email persons
|
|
11
|
+
* 5. Worker drains uncertain pairs (rule-stage only, no LLM needed)
|
|
12
|
+
* 6. SpendingSkill answers "how much did I spend?" with combined totals
|
|
13
|
+
* 7. TimelineSkill weaves events into chronological story
|
|
14
|
+
* 8. EventDetail surfaces the source raw row for a citation
|
|
15
|
+
*
|
|
16
|
+
* Avoids real network / real LLM / real device — every external boundary
|
|
17
|
+
* is mocked via DI seams (sessionFactory, csvParser, embeddingStage).
|
|
18
|
+
*/
|
|
19
|
+
|
|
20
|
+
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
21
|
+
|
|
22
|
+
const fs = require("node:fs");
|
|
23
|
+
const path = require("node:path");
|
|
24
|
+
const os = require("node:os");
|
|
25
|
+
|
|
26
|
+
const {
|
|
27
|
+
LocalVault, generateKeyHex, AdapterRegistry,
|
|
28
|
+
EntityResolver,
|
|
29
|
+
EmailAdapter, AlipayBillAdapter,
|
|
30
|
+
SpendingSkill, TimelineSkill, RelationsSkill,
|
|
31
|
+
} = require("../../lib");
|
|
32
|
+
|
|
33
|
+
describe("E2E — full Personal Data Hub user journey", () => {
|
|
34
|
+
let dir, vault, registry, resolver;
|
|
35
|
+
|
|
36
|
+
beforeEach(() => {
|
|
37
|
+
dir = fs.mkdtempSync(path.join(os.tmpdir(), "pdh-e2e-"));
|
|
38
|
+
vault = new LocalVault({ path: path.join(dir, "v.db"), key: generateKeyHex() });
|
|
39
|
+
vault.open();
|
|
40
|
+
resolver = new EntityResolver({ vault });
|
|
41
|
+
registry = new AdapterRegistry({ vault, entityResolver: resolver });
|
|
42
|
+
});
|
|
43
|
+
|
|
44
|
+
afterEach(() => {
|
|
45
|
+
try { vault.close(); } catch (_e) {}
|
|
46
|
+
try { fs.rmSync(dir, { recursive: true, force: true }); } catch (_e) {}
|
|
47
|
+
});
|
|
48
|
+
|
|
49
|
+
it("zero → email + alipay → entity merge → spending answer", async () => {
|
|
50
|
+
// ─── Step 1-2: Register + sync EmailAdapter with mocked IMAP ───
|
|
51
|
+
const emailEnv = {
|
|
52
|
+
uid: 1, internalDate: new Date("2026-05-15T10:00:00Z"),
|
|
53
|
+
flags: ["\\Seen"], messageId: "<msg-1@x>",
|
|
54
|
+
subject: "招商银行 11 月对账单",
|
|
55
|
+
from: [{ name: "招商银行", address: "ebank@cmbchina.com" }],
|
|
56
|
+
to: [{ address: "me@example.com" }], cc: [],
|
|
57
|
+
date: new Date("2026-05-15T10:00:00Z"), size: 1024,
|
|
58
|
+
source: Buffer.from("RAW", "utf-8"),
|
|
59
|
+
};
|
|
60
|
+
const emailSessionFactory = () => ({
|
|
61
|
+
async connect() {},
|
|
62
|
+
async openMailbox() { return { uidValidity: 1, uidNext: 9999, exists: 1 }; },
|
|
63
|
+
async *fetchFullSince() { yield emailEnv; },
|
|
64
|
+
async close() {},
|
|
65
|
+
});
|
|
66
|
+
const emailAdapter = new EmailAdapter({
|
|
67
|
+
account: { provider: "qq", email: "me@example.com", authCode: "x", folders: ["INBOX"] },
|
|
68
|
+
sessionFactory: emailSessionFactory,
|
|
69
|
+
parser: async () => ({
|
|
70
|
+
textBody: "尊敬的客户:\n您的招商银行信用卡 应还金额 ¥3,256.78 元",
|
|
71
|
+
attachments: [],
|
|
72
|
+
}),
|
|
73
|
+
});
|
|
74
|
+
registry.register(emailAdapter);
|
|
75
|
+
const emailReport = await registry.syncAdapter("email-imap");
|
|
76
|
+
expect(emailReport.status).toBe("ok");
|
|
77
|
+
expect(emailReport.entityCounts.events).toBeGreaterThanOrEqual(1);
|
|
78
|
+
|
|
79
|
+
// ─── Step 3: Register + import Alipay CSV ───
|
|
80
|
+
const alipayCsv = [
|
|
81
|
+
"支付宝交易记录明细查询",
|
|
82
|
+
"账号:[me@example.com]",
|
|
83
|
+
"起始日期:[2026-05-01 00:00:00] 终止日期:[2026-05-31 00:00:00]",
|
|
84
|
+
"---------------------------------交易记录明细列表------------------------------",
|
|
85
|
+
"交易号,商家订单号,交易创建时间,付款时间,最近修改时间,交易来源地,类型,交易对方,商品名称,金额(元),收/支,交易状态,服务费(元),成功退款(元),备注,资金状态",
|
|
86
|
+
"2026051022001112345001,T1,2026-05-10 14:00:00,2026-05-10 14:00:02,2026-05-10 14:00:02,客户端,即时到账交易,美团,美团外卖订单,38.50,支出,交易成功,0.00,0.00,,已支出",
|
|
87
|
+
"2026051522001112345002,T2,2026-05-15 09:00:00,2026-05-15 09:00:05,2026-05-15 09:00:05,客户端,即时到账交易,淘宝,运动鞋,299.00,支出,交易成功,0.00,0.00,,已支出",
|
|
88
|
+
"2026052022001112345003,,2026-05-20 12:00:00,2026-05-20 12:00:02,2026-05-20 12:00:02,客户端,转账,张三,生日礼物,500.00,支出,交易成功,0.00,0.00,生日快乐,已支出",
|
|
89
|
+
"---------------------------------交易记录明细列表结束------------------------------",
|
|
90
|
+
].join("\n");
|
|
91
|
+
const csvPath = path.join(dir, "alipay.csv");
|
|
92
|
+
fs.writeFileSync(csvPath, alipayCsv, "utf-8");
|
|
93
|
+
|
|
94
|
+
const alipayAdapter = new AlipayBillAdapter({
|
|
95
|
+
account: { email: "me@example.com" },
|
|
96
|
+
});
|
|
97
|
+
registry.register(alipayAdapter);
|
|
98
|
+
const alipayReport = await registry.syncAdapter("alipay-bill", { csvPath });
|
|
99
|
+
expect(alipayReport.status).toBe("ok");
|
|
100
|
+
expect(alipayReport.entityCounts.events).toBe(3);
|
|
101
|
+
|
|
102
|
+
// ─── Step 4-5: EntityResolver drained (rule stage only) ───
|
|
103
|
+
// Email's sender person (招商银行) is a separate identity; Alipay's
|
|
104
|
+
// 美团/淘宝/张三 are 3 new persons. None should auto-merge unless
|
|
105
|
+
// they share identifiers. The resolver should record decisions.
|
|
106
|
+
const queueStatsBefore = vault.resolveQueueStats();
|
|
107
|
+
expect(queueStatsBefore.pending + queueStatsBefore.done).toBeGreaterThanOrEqual(0);
|
|
108
|
+
|
|
109
|
+
// ─── Step 6: SpendingSkill answers "how much did I spend?" ───
|
|
110
|
+
const spendingSkill = new SpendingSkill({ vault });
|
|
111
|
+
const spending = await spendingSkill.run({});
|
|
112
|
+
expect(spending.summary.totalSpend).toBeCloseTo(38.50 + 299 + 500, 2);
|
|
113
|
+
expect(spending.summary.eventCount).toBeGreaterThanOrEqual(3);
|
|
114
|
+
expect(spending.breakdown.length).toBeGreaterThanOrEqual(3);
|
|
115
|
+
expect(spending.breakdown[0].key).toBe("张三"); // biggest
|
|
116
|
+
|
|
117
|
+
// ─── Step 7: TimelineSkill weaves into story ───
|
|
118
|
+
const timelineSkill = new TimelineSkill({ vault });
|
|
119
|
+
const timeline = await timelineSkill.run({
|
|
120
|
+
since: Date.parse("2026-05-01T00:00:00Z"),
|
|
121
|
+
until: Date.parse("2026-06-01T00:00:00Z"),
|
|
122
|
+
});
|
|
123
|
+
expect(timeline.entries.length).toBeGreaterThanOrEqual(4); // email + 3 alipay
|
|
124
|
+
expect(timeline.summary.byAdapter["email-imap"]).toBeGreaterThanOrEqual(1);
|
|
125
|
+
expect(timeline.summary.byAdapter["alipay-bill"]).toBe(3);
|
|
126
|
+
// Chronological: 5/10 first, 5/15 next, 5/20 last
|
|
127
|
+
expect(timeline.entries[0].occurredAt).toBeLessThan(timeline.entries[timeline.entries.length - 1].occurredAt);
|
|
128
|
+
|
|
129
|
+
// ─── Step 8: RelationsSkill (ranked mode) ───
|
|
130
|
+
const relationsSkill = new RelationsSkill({ vault });
|
|
131
|
+
const relations = await relationsSkill.run({}); // ranked mode
|
|
132
|
+
expect(relations.ranked.length).toBeGreaterThanOrEqual(3);
|
|
133
|
+
// Top should be 张三 (biggest transfer)
|
|
134
|
+
const zhangsan = relations.ranked.find((p) => p.name === "张三");
|
|
135
|
+
expect(zhangsan).toBeDefined();
|
|
136
|
+
expect(zhangsan.totalSpend).toBe(500);
|
|
137
|
+
|
|
138
|
+
// ─── Step 9: event-detail deep-link works ───
|
|
139
|
+
// Pull one event and verify the source row is recoverable
|
|
140
|
+
const events = vault.queryEvents({ adapter: "alipay-bill", limit: 10 });
|
|
141
|
+
expect(events.length).toBe(3);
|
|
142
|
+
// Find the Taobao transaction by counterparty (title is item name, e.g. "运动鞋")
|
|
143
|
+
const tx = events.find((e) => e.extra && e.extra.counterparty === "淘宝");
|
|
144
|
+
expect(tx).toBeDefined();
|
|
145
|
+
expect(tx.source.adapter).toBe("alipay-bill");
|
|
146
|
+
expect(tx.extra.counterpartyKind).toBe("merchant");
|
|
147
|
+
});
|
|
148
|
+
|
|
149
|
+
it("idempotent re-sync — same alipay CSV produces 0 duplicate events", async () => {
|
|
150
|
+
const alipayCsv = [
|
|
151
|
+
"支付宝交易记录明细查询",
|
|
152
|
+
"账号:[me@example.com]",
|
|
153
|
+
"---------------------------------交易记录明细列表------------------------------",
|
|
154
|
+
"交易号,商家订单号,交易创建时间,付款时间,最近修改时间,交易来源地,类型,交易对方,商品名称,金额(元),收/支,交易状态,服务费(元),成功退款(元),备注,资金状态",
|
|
155
|
+
"2026051022001112345001,T1,2026-05-10 14:00:00,2026-05-10 14:00:02,2026-05-10 14:00:02,客户端,即时到账交易,美团,外卖,38.50,支出,交易成功,0.00,0.00,,已支出",
|
|
156
|
+
].join("\n");
|
|
157
|
+
const csvPath = path.join(dir, "alipay.csv");
|
|
158
|
+
fs.writeFileSync(csvPath, alipayCsv, "utf-8");
|
|
159
|
+
|
|
160
|
+
const adapter = new AlipayBillAdapter({ account: { email: "me@example.com" } });
|
|
161
|
+
registry.register(adapter);
|
|
162
|
+
await registry.syncAdapter("alipay-bill", { csvPath });
|
|
163
|
+
await registry.syncAdapter("alipay-bill", { csvPath });
|
|
164
|
+
|
|
165
|
+
const events = vault.queryEvents({ adapter: "alipay-bill", limit: 100 });
|
|
166
|
+
expect(events.length).toBe(1); // dedup by originalId
|
|
167
|
+
});
|
|
168
|
+
|
|
169
|
+
it("vault stats reflect post-sync entity counts", async () => {
|
|
170
|
+
const csvPath = path.join(dir, "alipay.csv");
|
|
171
|
+
fs.writeFileSync(csvPath, [
|
|
172
|
+
"支付宝交易记录明细查询",
|
|
173
|
+
"账号:[me@example.com]",
|
|
174
|
+
"---------------------------------交易记录明细列表------------------------------",
|
|
175
|
+
"交易号,商家订单号,交易创建时间,付款时间,最近修改时间,交易来源地,类型,交易对方,商品名称,金额(元),收/支,交易状态,服务费(元),成功退款(元),备注,资金状态",
|
|
176
|
+
"2026051022001112345001,T1,2026-05-10 14:00:00,2026-05-10 14:00:02,2026-05-10 14:00:02,客户端,即时到账交易,美团,外卖,38.50,支出,交易成功,0.00,0.00,,已支出",
|
|
177
|
+
].join("\n"), "utf-8");
|
|
178
|
+
|
|
179
|
+
const adapter = new AlipayBillAdapter({ account: { email: "me@example.com" } });
|
|
180
|
+
registry.register(adapter);
|
|
181
|
+
await registry.syncAdapter("alipay-bill", { csvPath });
|
|
182
|
+
|
|
183
|
+
const stats = vault.stats();
|
|
184
|
+
expect(stats.events).toBe(1);
|
|
185
|
+
expect(stats.persons).toBe(1); // 美团 merchant
|
|
186
|
+
expect(stats.rawEvents).toBeGreaterThanOrEqual(1);
|
|
187
|
+
});
|
|
188
|
+
});
|
|
@@ -0,0 +1,177 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
4
|
+
|
|
5
|
+
const path = require("node:path");
|
|
6
|
+
const fs = require("node:fs");
|
|
7
|
+
const os = require("node:os");
|
|
8
|
+
const { LocalVault } = require("../lib/vault");
|
|
9
|
+
const { generateKeyHex } = require("../lib/key-providers");
|
|
10
|
+
const { AdapterRegistry } = require("../lib/registry");
|
|
11
|
+
const { EntityResolver } = require("../lib/entity-resolver");
|
|
12
|
+
|
|
13
|
+
// Minimal adapter that yields raw rows then normalizes to UnifiedSchema
|
|
14
|
+
class FakeAdapter {
|
|
15
|
+
constructor(name, persons) {
|
|
16
|
+
this.name = name;
|
|
17
|
+
this.version = "0.1.0";
|
|
18
|
+
this.capabilities = ["sync:test"];
|
|
19
|
+
this.rateLimits = {};
|
|
20
|
+
this.dataDisclosure = { fields: [], sensitivity: "low", legalGate: false };
|
|
21
|
+
this._persons = persons;
|
|
22
|
+
}
|
|
23
|
+
async authenticate() { return { ok: true }; }
|
|
24
|
+
async healthCheck() { return { ok: true, lastChecked: Date.now() }; }
|
|
25
|
+
async *sync() {
|
|
26
|
+
for (const p of this._persons) {
|
|
27
|
+
yield { adapter: this.name, originalId: p.id, capturedAt: Date.now(), payload: { person: p } };
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
normalize(raw) {
|
|
31
|
+
const p = raw.payload.person;
|
|
32
|
+
const now = Date.now();
|
|
33
|
+
const source = {
|
|
34
|
+
adapter: this.name, adapterVersion: this.version,
|
|
35
|
+
originalId: raw.originalId, capturedAt: raw.capturedAt, capturedBy: "api",
|
|
36
|
+
};
|
|
37
|
+
return {
|
|
38
|
+
events: [{
|
|
39
|
+
id: `evt-${raw.originalId}`,
|
|
40
|
+
type: "event",
|
|
41
|
+
subtype: "interaction",
|
|
42
|
+
occurredAt: now,
|
|
43
|
+
actor: p.id,
|
|
44
|
+
ingestedAt: now,
|
|
45
|
+
source,
|
|
46
|
+
}],
|
|
47
|
+
persons: [{
|
|
48
|
+
id: p.id,
|
|
49
|
+
type: "person",
|
|
50
|
+
subtype: "contact",
|
|
51
|
+
names: p.names || [],
|
|
52
|
+
identifiers: p.identifiers || {},
|
|
53
|
+
ingestedAt: now,
|
|
54
|
+
source,
|
|
55
|
+
}],
|
|
56
|
+
places: [], items: [], topics: [],
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
function makeRig() {
|
|
62
|
+
const dir = fs.mkdtempSync(path.join(os.tmpdir(), "er-hook-"));
|
|
63
|
+
const vault = new LocalVault({ path: path.join(dir, "v.db"), key: generateKeyHex() });
|
|
64
|
+
vault.open();
|
|
65
|
+
const resolver = new EntityResolver({ vault });
|
|
66
|
+
const registry = new AdapterRegistry({ vault, entityResolver: resolver });
|
|
67
|
+
return { vault, resolver, registry, dir };
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
function cleanup({ vault, dir }) {
|
|
71
|
+
try { vault.close(); } catch (_e) {}
|
|
72
|
+
try { fs.rmSync(dir, { recursive: true, force: true }); } catch (_e) {}
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
// ─── tests ────────────────────────────────────────────────────────────────
|
|
76
|
+
|
|
77
|
+
describe("Phase 8.6 — registry → entity resolver ingest hook", () => {
|
|
78
|
+
let rig;
|
|
79
|
+
afterEach(() => cleanup(rig));
|
|
80
|
+
|
|
81
|
+
it("registry.syncAdapter fires resolveOnIngest with new Persons", async () => {
|
|
82
|
+
rig = makeRig();
|
|
83
|
+
const adapter = new FakeAdapter("a1", [
|
|
84
|
+
{ id: "p-1", names: ["Alice"], identifiers: { email: ["alice@x.com"] } },
|
|
85
|
+
{ id: "p-2", names: ["Bob"], identifiers: { email: ["bob@x.com"] } },
|
|
86
|
+
]);
|
|
87
|
+
rig.registry.register(adapter);
|
|
88
|
+
const report = await rig.registry.syncAdapter("a1");
|
|
89
|
+
expect(report.status).toBe("ok");
|
|
90
|
+
expect(report.entityResolver).toBeDefined();
|
|
91
|
+
expect(report.entityResolver.newPersons).toBe(2);
|
|
92
|
+
});
|
|
93
|
+
|
|
94
|
+
it("cross-adapter R1 match: same email → auto-merged", async () => {
|
|
95
|
+
rig = makeRig();
|
|
96
|
+
// Adapter 1 imports Alice via email
|
|
97
|
+
rig.registry.register(new FakeAdapter("email", [
|
|
98
|
+
{ id: "p-email-alice", names: ["Alice"], identifiers: { email: ["alice@x.com"] } },
|
|
99
|
+
]));
|
|
100
|
+
await rig.registry.syncAdapter("email");
|
|
101
|
+
|
|
102
|
+
// Adapter 2 imports same Alice via different name + same email
|
|
103
|
+
rig.registry.register(new FakeAdapter("alipay", [
|
|
104
|
+
{ id: "p-alipay-alice-X", names: ["A. Alice"], identifiers: { email: ["alice@x.com"] } },
|
|
105
|
+
]));
|
|
106
|
+
const report = await rig.registry.syncAdapter("alipay");
|
|
107
|
+
expect(report.entityResolver.sameImmediate).toBeGreaterThanOrEqual(1);
|
|
108
|
+
|
|
109
|
+
// Verify merge group exists
|
|
110
|
+
const members = rig.vault.getMergeGroupMembers("p-email-alice").sort();
|
|
111
|
+
expect(members).toEqual(["p-alipay-alice-X", "p-email-alice"]);
|
|
112
|
+
});
|
|
113
|
+
|
|
114
|
+
it("uncertain pairs (name overlap only) → enqueued for async", async () => {
|
|
115
|
+
rig = makeRig();
|
|
116
|
+
rig.registry.register(new FakeAdapter("email", [
|
|
117
|
+
{ id: "p-1", names: ["张三"] }, // no identifiers
|
|
118
|
+
]));
|
|
119
|
+
await rig.registry.syncAdapter("email");
|
|
120
|
+
|
|
121
|
+
rig.registry.register(new FakeAdapter("alipay", [
|
|
122
|
+
{ id: "p-2", names: ["张三"] },
|
|
123
|
+
]));
|
|
124
|
+
const report = await rig.registry.syncAdapter("alipay");
|
|
125
|
+
expect(report.entityResolver.enqueued).toBeGreaterThanOrEqual(1);
|
|
126
|
+
expect(rig.vault.resolveQueueStats().pending).toBeGreaterThanOrEqual(1);
|
|
127
|
+
});
|
|
128
|
+
|
|
129
|
+
it("zero-overlap pair → no merge, no resolver decisions, person still ingested", async () => {
|
|
130
|
+
rig = makeRig();
|
|
131
|
+
rig.registry.register(new FakeAdapter("email", [
|
|
132
|
+
{ id: "p-1", names: ["Alice"], identifiers: { email: ["a@x.com"] } },
|
|
133
|
+
]));
|
|
134
|
+
await rig.registry.syncAdapter("email");
|
|
135
|
+
|
|
136
|
+
rig.registry.register(new FakeAdapter("alipay", [
|
|
137
|
+
{ id: "p-2", names: ["Bob"], identifiers: { phone: ["13999998888"] } },
|
|
138
|
+
]));
|
|
139
|
+
const report = await rig.registry.syncAdapter("alipay");
|
|
140
|
+
expect(report.status).toBe("ok");
|
|
141
|
+
expect(rig.vault.stats().mergeGroups).toBe(0);
|
|
142
|
+
});
|
|
143
|
+
|
|
144
|
+
it("resolver failure does NOT break ingest (audit-logged)", async () => {
|
|
145
|
+
rig = makeRig();
|
|
146
|
+
// Mock resolver that throws on resolveOnIngest
|
|
147
|
+
rig.registry.entityResolver = {
|
|
148
|
+
resolveOnIngest: () => { throw new Error("resolver boom"); },
|
|
149
|
+
};
|
|
150
|
+
rig.registry.register(new FakeAdapter("a1", [
|
|
151
|
+
{ id: "p-1", names: ["Alice"], identifiers: { email: ["alice@x.com"] } },
|
|
152
|
+
]));
|
|
153
|
+
const report = await rig.registry.syncAdapter("a1");
|
|
154
|
+
expect(report.status).toBe("ok"); // sync completes
|
|
155
|
+
expect(report.entityCounts.persons).toBe(1); // person still ingested
|
|
156
|
+
const audits = rig.vault.queryAudit({ action: "adapter.sync.entity_resolver_failed", limit: 5 });
|
|
157
|
+
expect(audits.length).toBeGreaterThanOrEqual(1);
|
|
158
|
+
});
|
|
159
|
+
|
|
160
|
+
it("no entityResolver = registry works as before (backward-compat)", async () => {
|
|
161
|
+
const dir = fs.mkdtempSync(path.join(os.tmpdir(), "er-hook-"));
|
|
162
|
+
const vault = new LocalVault({ path: path.join(dir, "v.db"), key: generateKeyHex() });
|
|
163
|
+
vault.open();
|
|
164
|
+
try {
|
|
165
|
+
const registry = new AdapterRegistry({ vault }); // no entityResolver
|
|
166
|
+
registry.register(new FakeAdapter("a1", [
|
|
167
|
+
{ id: "p-1", names: ["x"], identifiers: { email: ["a@x.com"] } },
|
|
168
|
+
]));
|
|
169
|
+
const report = await registry.syncAdapter("a1");
|
|
170
|
+
expect(report.status).toBe("ok");
|
|
171
|
+
expect(report.entityResolver).toBeUndefined();
|
|
172
|
+
} finally {
|
|
173
|
+
vault.close();
|
|
174
|
+
fs.rmSync(dir, { recursive: true, force: true });
|
|
175
|
+
}
|
|
176
|
+
});
|
|
177
|
+
});
|