a2a-xmtp 1.2.4 → 1.3.0

package/dist/transport/xmtp-bridge.js

@@ -0,0 +1,265 @@
+// ============================================================
+// Module 2: XMTP Bridge Service
+// 每个 Agent 对应一个 XMTP Client，消息 stream + 收件箱缓存
+// ============================================================
+import { Agent, createUser, createSigner } from "@xmtp/agent-sdk";
+import { createA2APayload } from "../types.js";
+/** IdentifierKind.Ethereum = 0 (const enum, 不能在 isolatedModules 下直接访问) */
+const IDENTIFIER_KIND_ETHEREUM = 0;
+export class XmtpBridge {
+    agentId;
+    walletConfig;
+    policyEngine;
+    registry;
+    dbPath;
+    agent = null;
+    running = false;
+    syncTimer = null;
+    /** 收件箱缓存 */
+    inboxBuffer = [];
+    maxInboxBuffer = 100;
+    /** 消息回调（由 plugin 入口设置，用于触发 subagent 处理） */
+    onMessage;
+    constructor(agentId, walletConfig, policyEngine, registry, dbPath) {
+        this.agentId = agentId;
+        this.walletConfig = walletConfig;
+        this.policyEngine = policyEngine;
+        this.registry = registry;
+        this.dbPath = dbPath;
+    }
+    async start() {
+        if (this.running)
+            return;
+        // Reconstruct full user from stored key — createUser() accepts an existing private key
+        const user = createUser(this.walletConfig.privateKey);
+        const signer = createSigner(user);
+        this.agent = await Agent.create(signer, {
+            env: this.walletConfig.env,
+            dbPath: `${this.dbPath}/${this.agentId}`,
+        });
+        await this.registry.updateInboxId(this.agentId, this.agent.address);
+        this.agent.on("text", async (ctx) => {
+            await this.handleIncoming(ctx, "text");
+        });
+        this.agent.on("markdown", async (ctx) => {
+            await this.handleIncoming(ctx, "markdown");
+        });
+        await this.agent.start();
+        this.running = true;
+        // 定期同步会话列表，发现被外部添加到的新群聊（每 30 秒）
+        this.syncTimer = setInterval(async () => {
+            try {
+                await this.agent?.client.conversations.sync();
+            }
+            catch { /* 同步失败不影响正常运行 */ }
+        }, 30_000);
+    }
+    async stop() {
+        if (!this.running || !this.agent)
+            return;
+        if (this.syncTimer) {
+            clearInterval(this.syncTimer);
+            this.syncTimer = null;
+        }
+        await this.agent.stop();
+        this.running = false;
+    }
+    async sendMessage(toAddress, content, opts) {
+        if (!this.agent)
+            throw new Error(`Bridge for ${this.agentId} not started`);
+        let conversation;
+        if (opts?.conversationId) {
+            conversation = await this.agent.client.conversations.getConversationById(opts.conversationId);
+            if (!conversation)
+                throw new Error(`Conversation ${opts.conversationId} not found`);
+        }
+        else {
+            conversation = await this.agent.createDmWithAddress(toAddress);
+        }
+        const messageId = opts?.contentType === "markdown"
+            ? await conversation.sendMarkdown(content)
+            : await conversation.sendText(content);
+        this.policyEngine.recordTurn(conversation.id);
+        return { conversationId: conversation.id, messageId: String(messageId) };
+    }
+    async getInbox(opts) {
+        const limit = opts?.limit ?? 10;
+        let messages = [...this.inboxBuffer];
+        if (opts?.from) {
+            const f = opts.from.toLowerCase();
+            messages = messages.filter((m) => m.from.agentId === opts.from || m.from.xmtpAddress.toLowerCase() === f);
+        }
+        return messages.slice(0, limit);
+    }
+    /**
+     * 获取指定会话的最近消息（供 coordination 层查询）
+     */
+    getRecentMessages(conversationId, limit) {
+        return this.inboxBuffer
+            .filter((m) => m.conversationId === conversationId)
+            .slice(0, limit);
+    }
+    async createGroup(memberAddresses, name) {
+        if (!this.agent)
+            throw new Error(`Bridge for ${this.agentId} not started`);
+        const opts = name ? { groupName: name } : undefined;
+        const group = await this.agent.createGroupWithAddresses(memberAddresses, opts);
+        if (name) {
+            try {
+                await group.updateName(name);
+            }
+            catch { /* 部分 SDK 版本 createGroupWithAddresses 不支持 opts */ }
+        }
+        return { conversationId: group.id, name: group.name ?? name ?? "" };
+    }
+    async listGroups() {
+        if (!this.agent)
+            throw new Error(`Bridge for ${this.agentId} not started`);
+        const conversations = await this.agent.client.conversations.list();
+        const groups = [];
+        for (const conv of conversations) {
+            const meta = await conv.metadata();
+            if (meta.conversationType !== 1)
+                continue; // 1 = Group, 0 = Dm
+            const members = await conv.members();
+            groups.push({
+                conversationId: conv.id,
+                name: conv.name ?? "",
+                memberAddresses: members.map((m) => m.accountAddresses?.[0]?.toLowerCase() ?? m.inboxId),
+                createdAt: conv.createdAt.toISOString(),
+            });
+        }
+        return groups;
+    }
+    async getGroupMembers(conversationId) {
+        if (!this.agent)
+            throw new Error(`Bridge for ${this.agentId} not started`);
+        const conv = await this.agent.client.conversations.getConversationById(conversationId);
+        if (!conv)
+            throw new Error(`Conversation ${conversationId} not found`);
+        const members = await conv.members();
+        return members.map((m) => m.accountAddresses?.[0]?.toLowerCase() ?? m.inboxId);
+    }
+    async addGroupMembers(conversationId, addresses) {
+        if (!this.agent)
+            throw new Error(`Bridge for ${this.agentId} not started`);
+        const conv = await this.agent.client.conversations.getConversationById(conversationId);
+        if (!conv)
+            throw new Error(`Conversation ${conversationId} not found`);
+        const identifiers = addresses.map((addr) => ({
+            identifier: addr,
+            identifierKind: IDENTIFIER_KIND_ETHEREUM,
+        }));
+        await conv.addMembersByIdentifiers(identifiers);
+    }
+    async removeGroupMembers(conversationId, addresses) {
+        if (!this.agent)
+            throw new Error(`Bridge for ${this.agentId} not started`);
+        const conv = await this.agent.client.conversations.getConversationById(conversationId);
+        if (!conv)
+            throw new Error(`Conversation ${conversationId} not found`);
+        const identifiers = addresses.map((addr) => ({
+            identifier: addr,
+            identifierKind: IDENTIFIER_KIND_ETHEREUM,
+        }));
+        await conv.removeMembersByIdentifiers(identifiers);
+    }
+    /**
+     * 检查群聊中在 afterTimestamp 之后是否有新消息（排除自身和指定 sender）
+     * 用于防止多 agent 同时回复同一条消息
+     */
+    async hasNewGroupReplies(conversationId, afterTimestamp, excludeAddresses) {
+        if (!this.agent)
+            return false;
+        try {
+            const conv = await this.agent.client.conversations.getConversationById(conversationId);
+            if (!conv)
+                return false;
+            await conv.sync();
+            const messages = await conv.messages({ limit: BigInt(10) });
+            const cutoffMs = new Date(afterTimestamp).getTime();
+            const excluded = new Set(excludeAddresses.map((a) => a.toLowerCase()));
+            return messages.some((m) => {
+                const sentMs = Number(BigInt(m.sentAtNs) / 1000000n);
+                const sender = (m.senderInboxId ?? m.senderAddress ?? "").toLowerCase();
+                return sentMs > cutoffMs && !excluded.has(sender);
+            });
+        }
+        catch {
+            return false;
+        }
+    }
+    async canMessage(address) {
+        if (!this.agent)
+            return false;
+        const result = await this.agent.client.canMessage([
+            { identifier: address, identifierKind: IDENTIFIER_KIND_ETHEREUM },
+        ]);
+        return result.get(address.toLowerCase()) ?? false;
+    }
+    async handleIncoming(ctx, contentType) {
+        const senderAddress = await ctx.getSenderAddress();
+        if (senderAddress.toLowerCase() === this.address.toLowerCase())
+            return;
+        const senderAgentId = await this.registry.resolveAgentId(senderAddress);
+        const policyResult = this.policyEngine.checkIncoming({
+            from: senderAgentId || senderAddress,
+            to: this.agentId,
+            conversationId: ctx.conversation.id,
+        });
+        if (!policyResult.allowed)
+            return;
+        const convState = this.policyEngine.getConversationState(ctx.conversation.id);
+        const turn = convState?.turn ?? 0;
+        const depth = convState?.depth ?? 0;
+        this.policyEngine.recordTurn(ctx.conversation.id, depth + 1);
+        // 群聊时获取参与者列表
+        let participants = [];
+        const isGroup = ctx.conversation.isGroup ?? false;
+        if (isGroup) {
+            try {
+                const members = await ctx.conversation.members();
+                participants = members.map((m) => m.accountAddresses?.[0]?.toLowerCase() ?? m.inboxId);
+            }
+            catch { /* 获取成员失败时保持空数组 */ }
+        }
+        const payload = createA2APayload({
+            fromAgentId: senderAgentId,
+            fromAddress: senderAddress,
+            conversationId: ctx.conversation.id,
+            isGroup,
+            participants,
+            messageId: ctx.message.id ?? crypto.randomUUID(),
+            content: String(ctx.message.content),
+            contentType,
+            turn: turn + 1,
+            depth: depth + 1,
+        });
+        // 缓存到 inbox
+        this.inboxBuffer.unshift({
+            id: payload.message.id,
+            from: { agentId: senderAgentId, xmtpAddress: senderAddress },
+            conversationId: ctx.conversation.id,
+            isGroup,
+            content: String(ctx.message.content),
+            contentType,
+            timestamp: payload.timestamp,
+        });
+        if (this.inboxBuffer.length > this.maxInboxBuffer)
+            this.inboxBuffer.pop();
+        // 通知 plugin 层（传递结构化 payload，由调用方决定如何处理）
+        if (this.onMessage) {
+            this.onMessage(this.agentId, payload);
+        }
+    }
+    get address() {
+        return this.agent?.address ?? this.walletConfig.address;
+    }
+    get isConnected() {
+        return this.running;
+    }
+    get env() {
+        return this.walletConfig.env;
+    }
+}
+//# sourceMappingURL=xmtp-bridge.js.map

package/dist/transport/xmtp-bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"xmtp-bridge.js","sourceRoot":"","sources":["../../src/transport/xmtp-bridge.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,gCAAgC;AAChC,8CAA8C;AAC9C,+DAA+D;AAE/D,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAIlE,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE/C,0EAA0E;AAC1E,MAAM,wBAAwB,GAAG,CAAC,CAAC;AAEnC,MAAM,OAAO,UAAU;IAaV;IACD;IACA;IACA;IACA;IAhBF,KAAK,GAAsC,IAAI,CAAC;IAChD,OAAO,GAAG,KAAK,CAAC;IAChB,SAAS,GAA0C,IAAI,CAAC;IAEhE,YAAY;IACJ,WAAW,GAAmB,EAAE,CAAC;IACxB,cAAc,GAAG,GAAG,CAAC;IAEtC,2CAA2C;IAC3C,SAAS,CAAwD;IAEjE,YACW,OAAe,EAChB,YAA8B,EAC9B,YAA0B,EAC1B,QAA0B,EAC1B,MAAc;QAJb,YAAO,GAAP,OAAO,CAAQ;QAChB,iBAAY,GAAZ,YAAY,CAAkB;QAC9B,iBAAY,GAAZ,YAAY,CAAc;QAC1B,aAAQ,GAAR,QAAQ,CAAkB;QAC1B,WAAM,GAAN,MAAM,CAAQ;IACrB,CAAC;IAEJ,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QAEzB,uFAAuF;QACvF,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,UAA2B,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QAElC,IAAI,CAAC,KAAK,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE;YACtC,GAAG,EAAE,IAAI,CAAC,YAAY,CAAC,GAAG;YAC1B,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE;SACzC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAQ,CAAC,CAAC;QAErE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,GAAQ,EAAE,EAAE;YACvC,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,KAAK,EAAE,GAAQ,EAAE,EAAE;YAC3C,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,gCAAgC;QAChC,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;YACtC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;YAChD,CAAC;YAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC;QAC/B,CAAC,EAAE,MAAM,CAAC,CAAC;IACb,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,OAAO;QACzC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,WAAW,CACf,SAAiB,EACjB,OAAe,EACf,IAAqE;QAErE,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,OAAO,cAAc,CAAC,CAAC;QAE3E,IAAI,YAAiB,CAAC;QACtB,IAAI,IAAI,EAAE,cAAc,EAAE,CAAC;YACzB,YAAY,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC9F,IAAI,CAAC,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,cAAc,YAAY,CAAC,CAAC;QACtF,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,MAAO,IAAI,CAAC,KAAa,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,EAAE,WAAW,KAAK,UAAU;YAChD,CAAC,CAAC,MAAM,YAAY,CAAC,YAAY,CAAC,OAAO,CAAC;YAC1C,CAAC,CAAC,MAAM,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAEzC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAE9C,OAAO,EAAE,cAAc,EAAE,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAwC;QACrD,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;QAChC,IAAI,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,IAAI,IAAI,EAAE,IAAI,EAAE,CAAC;YACf,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YAClC,QAAQ,GAAG,QAAQ,CAAC,MAAM,CACxB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,KAAK,CAAC,CAC9E,CAAC;QACJ,CAAC;QACD,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,cAAsB,EAAE,KAAa;QACrD,OAAO,IAAI,CAAC,WAAW;aACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,KAAK,cAAc,CAAC;aAClD,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,eAAyB,EAAE,IAAa;QACxD,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,OAAO,cAAc,CAAC,CAAC;QAC3E,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QACpD,MAAM,KAAK,GAAG,MAAO,IAAI,CAAC,KAAa,CAAC,wBAAwB,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QACxF,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBAAC,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,iDAAiD,CAAC,CAAC;QACnG,CAAC;QACD,OAAO,EAAE,cAAc,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,EAAE,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,OAAO,cAAc,CAAC,CAAC;QAC3E,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QACnE,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,IAAI,IAAI,CAAC,gBAAgB,KAAK,CAAC;gBAAE,SAAS,CAAC,oBAAoB;YAC/D,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC;gBACV,cAAc,EAAE,IAAI,CAAC,EAAE;gBACvB,IAAI,EAAG,IAAY,CAAC,IAAI,IAAI,EAAE;gBAC9B,eAAe,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CACtC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,OAAO,CACpD;gBACD,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE;aACxC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,cAAsB;QAC1C,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,OAAO,cAAc,CAAC,CAAC;QAC3E,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;QACvF,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,cAAc,YAAY,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC;IACtF,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,cAAsB,EAAE,SAAmB;QAC/D,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,OAAO,cAAc,CAAC,CAAC;QAC3E,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;QACvF,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,cAAc,YAAY,CAAC,CAAC;QACvE,MAAM,WAAW,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC3C,UAAU,EAAE,IAAI;YAChB,cAAc,EAAE,wBAAwB;SACzC,CAAC,CAAC,CAAC;QACJ,MAAO,IAAY,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,cAAsB,EAAE,SAAmB;QAClE,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,OAAO,cAAc,CAAC,CAAC;QAC3E,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;QACvF,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,cAAc,YAAY,CAAC,CAAC;QACvE,MAAM,WAAW,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC3C,UAAU,EAAE,IAAI;YAChB,cAAc,EAAE,wBAAwB;SACzC,CAAC,CAAC,CAAC;QACJ,MAAO,IAAY,CAAC,0BAA0B,CAAC,WAAW,CAAC,CAAC;IAC9D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,kBAAkB,CACtB,cAAsB,EACtB,cAAsB,EACtB,gBAA0B;QAE1B,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;YACvF,IAAI,CAAC,IAAI;gBAAE,OAAO,KAAK,CAAC;YACxB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE,CAAQ,EAAE,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC;YACpD,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;YACvE,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE;gBAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,QAAU,CAAC,CAAC;gBACvD,MAAM,MAAM,GAAG,CAAC,CAAC,CAAC,aAAa,IAAI,CAAC,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;gBACxE,OAAO,MAAM,GAAG,QAAQ,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAe;QAC9B,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC;YAChD,EAAE,UAAU,EAAE,OAAO,EAAE,cAAc,EAAE,wBAAwB,EAAE;SAClE,CAAC,CAAC;QACH,OAAO,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,IAAI,KAAK,CAAC;IACpD,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,GAAQ,EAAE,WAA2B;QAChE,MAAM,aAAa,GAAW,MAAM,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAE3D,IAAI,aAAa,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;YAAE,OAAO;QAEvE,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAExE,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;YACnD,IAAI,EAAE,aAAa,IAAI,aAAa;YACpC,EAAE,EAAE,IAAI,CAAC,OAAO;YAChB,cAAc,EAAE,GAAG,CAAC,YAAY,CAAC,EAAE;SACpC,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,CAAC,OAAO;YAAE,OAAO;QAElC,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAC9E,MAAM,IAAI,GAAG,SAAS,EAAE,IAAI,IAAI,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,SAAS,EAAE,KAAK,IAAI,CAAC,CAAC;QAEpC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QAE7D,aAAa;QACb,IAAI,YAAY,GAAa,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,GAAG,CAAC,YAAY,CAAC,OAAO,IAAI,KAAK,CAAC;QAClD,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;gBACjD,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CACpC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,OAAO,CACpD,CAAC;YACJ,CAAC;YAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,OAAO,GAAG,gBAAgB,CAAC;YAC/B,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,aAAa;YAC1B,cAAc,EAAE,GAAG,CAAC,YAAY,CAAC,EAAE;YACnC,OAAO;YACP,YAAY;YACZ,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE;YAChD,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC;YACpC,WAAW;YACX,IAAI,EAAE,IAAI,GAAG,CAAC;YACd,KAAK,EAAE,KAAK,GAAG,CAAC;SACjB,CAAC,CAAC;QAEH,YAAY;QACZ,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;YACvB,EAAE,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;YACtB,IAAI,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE;YAC5D,cAAc,EAAE,GAAG,CAAC,YAAY,CAAC,EAAE;YACnC,OAAO;YACP,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC;YACpC,WAAW;YACX,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,cAAc;YAAE,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;QAE1E,wCAAwC;QACxC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;IAC1D,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;IAC/B,CAAC;CACF"}

package/dist/types.d.ts

@@ -0,0 +1,100 @@
+/** Agent 的 XMTP 钱包配置 */
+export interface XmtpWalletConfig {
+    privateKey: string;
+    address: string;
+    xmtpInboxId: string;
+    env: XmtpEnv;
+}
+export type XmtpEnv = "dev" | "production";
+/** 注入 Agent session 的消息格式 */
+export interface A2AInjectPayload {
+    type: "a2a-xmtp";
+    from: {
+        agentId: string | null;
+        xmtpAddress: string;
+        displayName?: string;
+    };
+    conversation: {
+        id: string;
+        isGroup: boolean;
+        participants: string[];
+    };
+    message: {
+        id: string;
+        content: string;
+        contentType: A2AContentType;
+        turn: number;
+        depth: number;
+        replyTo?: string;
+    };
+    timestamp: string;
+}
+export type A2AContentType = "text" | "markdown" | "transaction" | "action";
+/** 防循环策略配置 */
+export interface ConversationPolicy {
+    maxTurns: number;
+    maxDepth: number;
+    minIntervalMs: number;
+    ttlMinutes: number;
+    consentMode: ConsentMode;
+}
+export type ConsentMode = "auto-allow-local" | "explicit-only";
+export type ConsentState = "allow" | "deny" | "unknown";
+/** 插件运行时配置（从 openclaw.json entries.a2a-xmtp.config 解析） */
+export interface PluginConfig {
+    xmtp: {
+        env: XmtpEnv;
+        dbPath: string;
+    };
+    policy: ConversationPolicy;
+    walletKey?: string;
+}
+/** 会话状态（Policy Engine 内部使用） */
+export interface ConversationState {
+    turn: number;
+    depth: number;
+    lastSendTime: number;
+    createdAt: number;
+}
+/** 收件箱消息 */
+export interface InboxMessage {
+    id: string;
+    from: {
+        agentId: string | null;
+        xmtpAddress: string;
+    };
+    conversationId: string;
+    isGroup: boolean;
+    content: string;
+    contentType: A2AContentType;
+    timestamp: string;
+}
+/** 群聊信息 */
+export interface GroupInfo {
+    conversationId: string;
+    name: string;
+    memberAddresses: string[];
+    createdAt: string;
+}
+/** 默认策略配置 */
+export declare const DEFAULT_POLICY: ConversationPolicy;
+/** 默认插件配置 */
+export declare const DEFAULT_PLUGIN_CONFIG: PluginConfig;
+/** 格式化 A2A 消息（注入 session 时的文本表示） */
+export declare function formatA2AMessage(payload: A2AInjectPayload): string;
+/** 创建 A2AInjectPayload */
+export declare function createA2APayload(params: {
+    fromAgentId: string | null;
+    fromAddress: string;
+    conversationId: string;
+    isGroup: boolean;
+    participants: string[];
+    messageId: string;
+    content: string;
+    contentType: A2AContentType;
+    turn: number;
+    depth: number;
+    replyTo?: string;
+    displayName?: string;
+}): A2AInjectPayload;
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAIA,wBAAwB;AACxB,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,GAAG,EAAE,OAAO,CAAC;CACd;AAED,MAAM,MAAM,OAAO,GAAG,KAAK,GAAG,YAAY,CAAC;AAE3C,6BAA6B;AAC7B,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE;QACJ,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,YAAY,EAAE;QACZ,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,EAAE,OAAO,CAAC;QACjB,YAAY,EAAE,MAAM,EAAE,CAAC;KACxB,CAAC;IACF,OAAO,EAAE;QACP,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,cAAc,CAAC;QAC5B,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,cAAc,GAAG,MAAM,GAAG,UAAU,GAAG,aAAa,GAAG,QAAQ,CAAC;AAE5E,cAAc;AACd,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,WAAW,CAAC;CAC1B;AAED,MAAM,MAAM,WAAW,GAAG,kBAAkB,GAAG,eAAe,CAAC;AAC/D,MAAM,MAAM,YAAY,GAAG,OAAO,GAAG,MAAM,GAAG,SAAS,CAAC;AAExD,0DAA0D;AAC1D,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE;QACJ,GAAG,EAAE,OAAO,CAAC;QACb,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,MAAM,EAAE,kBAAkB,CAAC;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,+BAA+B;AAC/B,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,YAAY;AACZ,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACJ,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,cAAc,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,WAAW;AACX,MAAM,WAAW,SAAS;IACxB,cAAc,EAAE,MAAM,CAAC;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,aAAa;AACb,eAAO,MAAM,cAAc,EAAE,kBAM5B,CAAC;AAEF,aAAa;AACb,eAAO,MAAM,qBAAqB,EAAE,YAMnC,CAAC;AAEF,oCAAoC;AACpC,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,gBAAgB,GAAG,MAAM,CAIlE;AAED,0BAA0B;AAC1B,wBAAgB,gBAAgB,CAAC,MAAM,EAAE;IACvC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,cAAc,CAAC;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,GAAG,gBAAgB,CAuBnB"}

package/dist/types.js

@@ -0,0 +1,51 @@
+// ============================================================
+// Module 6: 类型定义 & 消息协议适配
+// ============================================================
+/** 默认策略配置 */
+export const DEFAULT_POLICY = {
+    maxTurns: 10,
+    maxDepth: 5,
+    minIntervalMs: 5000,
+    ttlMinutes: 60,
+    consentMode: "auto-allow-local",
+};
+/** 默认插件配置 */
+export const DEFAULT_PLUGIN_CONFIG = {
+    xmtp: {
+        env: "dev",
+        dbPath: "./xmtp-data",
+    },
+    policy: DEFAULT_POLICY,
+};
+/** 格式化 A2A 消息（注入 session 时的文本表示） */
+export function formatA2AMessage(payload) {
+    const sender = payload.from.agentId || payload.from.xmtpAddress;
+    const prefix = payload.conversation.isGroup ? "[A2A Group]" : "[A2A]";
+    return `${prefix} from ${sender} (turn ${payload.message.turn}): ${payload.message.content}`;
+}
+/** 创建 A2AInjectPayload */
+export function createA2APayload(params) {
+    return {
+        type: "a2a-xmtp",
+        from: {
+            agentId: params.fromAgentId,
+            xmtpAddress: params.fromAddress,
+            displayName: params.displayName,
+        },
+        conversation: {
+            id: params.conversationId,
+            isGroup: params.isGroup,
+            participants: params.participants,
+        },
+        message: {
+            id: params.messageId,
+            content: params.content,
+            contentType: params.contentType,
+            turn: params.turn,
+            depth: params.depth,
+            replyTo: params.replyTo,
+        },
+        timestamp: new Date().toISOString(),
+    };
+}
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,0BAA0B;AAC1B,+DAA+D;AA0F/D,aAAa;AACb,MAAM,CAAC,MAAM,cAAc,GAAuB;IAChD,QAAQ,EAAE,EAAE;IACZ,QAAQ,EAAE,CAAC;IACX,aAAa,EAAE,IAAI;IACnB,UAAU,EAAE,EAAE;IACd,WAAW,EAAE,kBAAkB;CAChC,CAAC;AAEF,aAAa;AACb,MAAM,CAAC,MAAM,qBAAqB,GAAiB;IACjD,IAAI,EAAE;QACJ,GAAG,EAAE,KAAK;QACV,MAAM,EAAE,aAAa;KACtB;IACD,MAAM,EAAE,cAAc;CACvB,CAAC;AAEF,oCAAoC;AACpC,MAAM,UAAU,gBAAgB,CAAC,OAAyB;IACxD,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC;IAChE,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC;IACtE,OAAO,GAAG,MAAM,SAAS,MAAM,UAAU,OAAO,CAAC,OAAO,CAAC,IAAI,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;AAC/F,CAAC;AAED,0BAA0B;AAC1B,MAAM,UAAU,gBAAgB,CAAC,MAahC;IACC,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE;YACJ,OAAO,EAAE,MAAM,CAAC,WAAW;YAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC;QACD,YAAY,EAAE;YACZ,EAAE,EAAE,MAAM,CAAC,cAAc;YACzB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,YAAY,EAAE,MAAM,CAAC,YAAY;SAClC;QACD,OAAO,EAAE;YACP,EAAE,EAAE,MAAM,CAAC,SAAS;YACpB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB;QACD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,21 +1,9 @@
 {
   "name": "a2a-xmtp",
-  "version": "1.2.4",
+  "version": "1.3.0",
   "type": "module",
-  "main": "src/index.ts",
-  "types": "src/index.ts",
-  "dependencies": {
-    "@xmtp/agent-sdk": "^2.3.0",
-    "@sinclair/typebox": "^0.34.0"
-  },
-  "peerDependencies": {
-    "openclaw": "*"
-  },
-  "openclaw": {
-    "extensions": [
-      "./src/index.ts"
-    ]
-  },
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
   "description": "Decentralized Agent-to-Agent E2EE messaging for OpenClaw via XMTP",
   "keywords": [
     "openclaw",
@@ -27,8 +15,31 @@
     "web3"
   ],
   "license": "MIT",
+  "scripts": {
+    "build": "tsc",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "lint": "tsc --noEmit"
+  },
+  "dependencies": {
+    "@xmtp/agent-sdk": "^2.3.0",
+    "@sinclair/typebox": "^0.34.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.7.0",
+    "vitest": "^3.0.0",
+    "@types/node": "^22.0.0"
+  },
+  "peerDependencies": {
+    "openclaw": "*"
+  },
+  "openclaw": {
+    "extensions": [
+      "./dist/index.js"
+    ]
+  },
   "files": [
-    "src/",
+    "dist/",
     "openclaw.plugin.json",
     "package.json",
     "README.md"

package/src/identity-registry.ts

@@ -1,134 +0,0 @@
-// ============================================================
-// Module 3: Identity Registry
-// agentId ↔ XMTP wallet address 双向映射，wallet key 生成/加载
-// 使用文件系统持久化（stateDir）
-// ============================================================
-
-import { createUser, createSigner } from "@xmtp/agent-sdk";
-import { privateKeyToAccount } from "viem/accounts";
-import { readFileSync, writeFileSync, mkdirSync, existsSync } from "node:fs";
-import { join } from "node:path";
-import type { XmtpWalletConfig, XmtpEnv } from "./types.js";
-
-export class IdentityRegistry {
-  private agentToConfig = new Map<string, XmtpWalletConfig>();
-  private addressToAgent = new Map<string, string>();
-  private storePath: string;
-
-  constructor(
-    stateDir: string,
-    private env: XmtpEnv = "dev",
-  ) {
-    this.storePath = join(stateDir, "identities");
-    mkdirSync(this.storePath, { recursive: true });
-    this.loadFromDisk();
-  }
-
-  /**
-   * 为 Agent 初始化 XMTP wallet。
-   * 如果已有配置则加载，否则生成新的 wallet。
-   */
-  async initAgent(agentId: string, existingKey?: string): Promise<XmtpWalletConfig> {
-    const cached = this.agentToConfig.get(agentId);
-    if (cached) return cached;
-
-    // 尝试从磁盘加载
-    const filePath = join(this.storePath, `${agentId}.json`);
-    if (existsSync(filePath)) {
-      const config: XmtpWalletConfig = JSON.parse(readFileSync(filePath, "utf-8"));
-      this.cacheMapping(agentId, config);
-      return config;
-    }
-
-    // 生成新 wallet 或使用提供的 key
-    let privateKey: string;
-    let address: string;
-
-    if (existingKey) {
-      privateKey = existingKey;
-      const account = privateKeyToAccount(existingKey as `0x${string}`);
-      address = account.address;
-    } else {
-      const user = createUser();
-      privateKey = user.key;            // createUser() returns { key, account, wallet }
-      address = user.account.address;
-    }
-
-    const config: XmtpWalletConfig = {
-      privateKey,
-      address,
-      xmtpInboxId: "",
-      env: this.env,
-    };
-
-    // 持久化到磁盘
-    writeFileSync(filePath, JSON.stringify(config, null, 2));
-    this.cacheMapping(agentId, config);
-
-    return config;
-  }
-
-  /** 更新 inboxId */
-  async updateInboxId(agentId: string, inboxId: string): Promise<void> {
-    const config = this.agentToConfig.get(agentId);
-    if (!config) return;
-    config.xmtpInboxId = inboxId;
-    const filePath = join(this.storePath, `${agentId}.json`);
-    writeFileSync(filePath, JSON.stringify(config, null, 2));
-  }
-
-  getAddress(agentId: string): string | undefined {
-    return this.agentToConfig.get(agentId)?.address;
-  }
-
-  getConfig(agentId: string): XmtpWalletConfig | undefined {
-    return this.agentToConfig.get(agentId);
-  }
-
-  getAgentId(address: string): string | undefined {
-    return this.addressToAgent.get(address.toLowerCase());
-  }
-
-  async resolveAgentId(address: string): Promise<string | null> {
-    return this.addressToAgent.get(address.toLowerCase()) ?? null;
-  }
-
-  listAgents(): Array<{ agentId: string; address: string; env: XmtpEnv }> {
-    return Array.from(this.agentToConfig.entries()).map(([id, config]) => ({
-      agentId: id,
-      address: config.address,
-      env: config.env,
-    }));
-  }
-
-  has(agentId: string): boolean {
-    return this.agentToConfig.has(agentId);
-  }
-
-  registerExternal(address: string, agentId: string): void {
-    this.addressToAgent.set(address.toLowerCase(), agentId);
-  }
-
-  private cacheMapping(agentId: string, config: XmtpWalletConfig): void {
-    this.agentToConfig.set(agentId, config);
-    this.addressToAgent.set(config.address.toLowerCase(), agentId);
-  }
-
-  /** 从磁盘加载所有已有 identity */
-  private loadFromDisk(): void {
-    if (!existsSync(this.storePath)) return;
-    const { readdirSync } = require("node:fs") as typeof import("node:fs");
-    for (const file of readdirSync(this.storePath)) {
-      if (!file.endsWith(".json")) continue;
-      const agentId = file.replace(".json", "");
-      try {
-        const config: XmtpWalletConfig = JSON.parse(
-          readFileSync(join(this.storePath, file), "utf-8"),
-        );
-        this.cacheMapping(agentId, config);
-      } catch {
-        // skip corrupt files
-      }
-    }
-  }
-}