a2a-xmtp 1.4.6 → 2.0.0

package/src/index.ts

@@ -1,230 +0,0 @@
-// ============================================================
-// Module 1: Plugin 入口
-// 使用 OpenClaw Plugin SDK 真实 API
-// ============================================================
-
-import { definePluginEntry } from "openclaw/plugin-sdk/plugin-entry";
-import { Type } from "@sinclair/typebox";
-import { mkdirSync } from "node:fs";
-import { join } from "node:path";
-import { IdentityRegistry } from "./transport/identity-registry.js";
-import { PolicyEngine } from "./coordination/policy-engine.js";
-import { GroupScheduler } from "./coordination/group-scheduler.js";
-import { MessageOrchestrator } from "./coordination/message-orchestrator.js";
-import { XmtpBridge } from "./transport/xmtp-bridge.js";
-import { handleXmtpSend } from "./tools/xmtp-send.js";
-import { handleXmtpInbox } from "./tools/xmtp-inbox.js";
-import { handleDiscoverAgents } from "./tools/xmtp-agents.js";
-import { handleXmtpGroup } from "./tools/xmtp-group.js";
-import { DEFAULT_PLUGIN_CONFIG, type PluginConfig } from "./types.js";
-
-const AGENT_ID = "main"; // OpenClaw 默认 agent
-
-// Module-level singletons: OpenClaw may call register() multiple times from different
-// subsystems (gateway, plugins) within the same process. Shared state must live at
-// module scope so all closures (tools, HTTP routes, service) reference the same instances.
-const bridges = new Map<string, XmtpBridge>();
-let registry: IdentityRegistry;
-let policyEngine: PolicyEngine;
-
-export default definePluginEntry({
-  id: "a2a-xmtp",
-  name: "Agent-to-Agent IM (XMTP)",
-  description: "Decentralized Agent-to-Agent E2EE messaging powered by XMTP protocol",
-
-  register(api) {
-
-    // ── 1. 注册 Tools ──
-
-    api.registerTool({
-      name: "xmtp_send",
-      label: "Send XMTP Message",
-      description:
-        "Send an E2EE message to another agent via XMTP. " +
-        "Supports cross-gateway and cross-organization communication.",
-      parameters: Type.Object({
-        to: Type.Optional(Type.String({ description: "Target agent ID or XMTP address (0x...). Optional when conversationId is provided." })),
-        message: Type.String({ description: "Message content to send" }),
-        conversationId: Type.Optional(Type.String({ description: "Reuse existing conversation. When set, 'to' is optional." })),
-        contentType: Type.Optional(
-          Type.String({ description: "Message type: text or markdown", default: "text" }),
-        ),
-      }),
-      async execute(_toolCallId, params) {
-        const p = params as { to?: string; message: string; conversationId?: string; contentType?: "text" | "markdown" };
-        return await handleXmtpSend(bridges, registry, policyEngine, p, AGENT_ID);
-      },
-    });
-
-    api.registerTool({
-      name: "xmtp_inbox",
-      label: "XMTP Inbox",
-      description:
-        "Check your XMTP inbox for messages from other agents. " +
-        "Messages are E2E encrypted and only you can read them.",
-      parameters: Type.Object({
-        limit: Type.Optional(Type.Number({ description: "Max messages to return (default: 10)" })),
-        from: Type.Optional(Type.String({ description: "Filter by sender agent ID or address" })),
-      }),
-      async execute(_toolCallId, params) {
-        const p = params as { limit?: number; from?: string };
-        return await handleXmtpInbox(bridges, registry, p, AGENT_ID);
-      },
-    });
-
-    api.registerTool({
-      name: "xmtp_agents",
-      label: "Discover XMTP Agents",
-      description:
-        "Discover agents available for XMTP communication. " +
-        "Lists registered agents and their connection status.",
-      parameters: Type.Object({
-        includeExternal: Type.Optional(
-          Type.Boolean({ description: "Include ERC-8004 external registry (Phase 3)" }),
-        ),
-      }),
-      async execute(_toolCallId, params) {
-        const p = params as { includeExternal?: boolean };
-        return await handleDiscoverAgents(bridges, registry, p);
-      },
-    });
-
-    api.registerTool({
-      name: "xmtp_group",
-      label: "XMTP Group Management",
-      description:
-        "Manage XMTP group conversations. Actions: create (new group), list (all groups), " +
-        "members (view members), add_member, remove_member.",
-      parameters: Type.Object({
-        action: Type.String({ description: "Action: create | list | members | add_member | remove_member" }),
-        members: Type.Optional(Type.Array(Type.String(), { description: "Agent IDs or 0x addresses (for create/add_member/remove_member)" })),
-        conversationId: Type.Optional(Type.String({ description: "Group conversation ID (for members/add_member/remove_member)" })),
-        name: Type.Optional(Type.String({ description: "Group name (for create)" })),
-      }),
-      async execute(_toolCallId, params) {
-        const p = params as { action: string; members?: string[]; conversationId?: string; name?: string };
-        return await handleXmtpGroup(bridges, registry, p, AGENT_ID);
-      },
-    });
-
-    // ── 2. 注册 HTTP 状态路由 ──
-
-    api.registerHttpRoute({
-      path: "/a2a-xmtp/status",
-      auth: "gateway",
-      handler: async (_req, res) => {
-        const status = {
-          plugin: "a2a-xmtp",
-          bridgeCount: bridges.size,
-          agents: Array.from(bridges.entries()).map(([id, b]) => ({
-            agentId: id,
-            xmtpAddress: b.address,
-            connected: b.isConnected,
-            env: b.env,
-          })),
-          policy: policyEngine?.getPolicy(),
-        };
-        res.setHeader("Content-Type", "application/json");
-        res.end(JSON.stringify(status, null, 2));
-        return true;
-      },
-    });
-
-    // ── 3. 注册 Service（生命周期管理） ──
-
-    api.registerService({
-      id: "a2a-xmtp-bridge",
-      async start(ctx) {
-        ctx.logger.info("[a2a-xmtp] Starting XMTP Bridge Service...");
-
-        // 加载配置（所有选项均有默认值，用户无需手动编辑配置文件）
-        const pluginCfg = (ctx.config as any)?.plugins?.entries?.["a2a-xmtp"]?.config;
-        const defaultDbPath = join(ctx.stateDir, "xmtp-data");
-        const config: PluginConfig = {
-          xmtp: {
-            env: pluginCfg?.xmtp?.env ?? DEFAULT_PLUGIN_CONFIG.xmtp.env,
-            dbPath: pluginCfg?.xmtp?.dbPath ?? defaultDbPath,
-          },
-          policy: {
-            maxTurns: pluginCfg?.policy?.maxTurns ?? DEFAULT_PLUGIN_CONFIG.policy.maxTurns,
-            minIntervalMs: pluginCfg?.policy?.minIntervalMs ?? DEFAULT_PLUGIN_CONFIG.policy.minIntervalMs,
-            ttlMinutes: pluginCfg?.policy?.ttlMinutes ?? DEFAULT_PLUGIN_CONFIG.policy.ttlMinutes,
-            consentMode: pluginCfg?.policy?.consentMode ?? DEFAULT_PLUGIN_CONFIG.policy.consentMode,
-          },
-          groupScheduling: {
-            baseDelayMs: pluginCfg?.groupScheduling?.baseDelayMs ?? DEFAULT_PLUGIN_CONFIG.groupScheduling.baseDelayMs,
-            slotTimeoutMs: pluginCfg?.groupScheduling?.slotTimeoutMs ?? DEFAULT_PLUGIN_CONFIG.groupScheduling.slotTimeoutMs,
-            claimExpireMs: pluginCfg?.groupScheduling?.claimExpireMs ?? DEFAULT_PLUGIN_CONFIG.groupScheduling.claimExpireMs,
-          },
-          walletKey: pluginCfg?.walletKey,
-        };
-
-        // 自动创建数据目录
-        mkdirSync(config.xmtp.dbPath, { recursive: true });
-
-        // 初始化核心组件
-        registry = new IdentityRegistry(ctx.stateDir, config.xmtp.env);
-        policyEngine = new PolicyEngine(config.policy);
-
-        // 初始化群聊调度器和消息编排器
-        const groupScheduler = new GroupScheduler(config.groupScheduling);
-        const orchestrator = new MessageOrchestrator(
-          api.runtime.subagent,
-          ctx.logger,
-          policyEngine,
-          groupScheduler,
-        );
-
-        // 初始化主 Agent 的 XMTP Bridge
-        try {
-          const walletConfig = await registry.initAgent(AGENT_ID, config.walletKey);
-          policyEngine.registerLocalAgent(AGENT_ID);
-
-          const bridge = new XmtpBridge(
-            AGENT_ID,
-            walletConfig,
-            policyEngine,
-            registry,
-            config.xmtp.dbPath,
-          );
-
-          // 消息回调：委托给 orchestrator 处理
-          bridge.onMessage = (agentId, payload) =>
-            orchestrator.handleMessage(bridge, agentId, payload);
-
-          // Claim 回调：通知 orchestrator 取消 failover 等待
-          bridge.onClaim = (conversationId, senderAddress, messageId) =>
-            orchestrator.handleClaim(conversationId, senderAddress, messageId);
-
-          // 群聊 self 消息计数回调：保持 messageCount 同步
-          bridge.onSelfGroupMessage = (conversationId) =>
-            orchestrator.handleSelfGroupMessage(conversationId);
-
-          await bridge.start();
-          bridges.set(AGENT_ID, bridge);
-
-          ctx.logger.info(
-            `[a2a-xmtp] Bridge started: ${AGENT_ID} → ${bridge.address} (env: ${config.xmtp.env})`,
-          );
-        } catch (err) {
-          ctx.logger.error(
-            `[a2a-xmtp] Failed to start bridge: ${err instanceof Error ? err.message : String(err)}`,
-          );
-        }
-      },
-
-      async stop(ctx) {
-        ctx.logger.info("[a2a-xmtp] Stopping XMTP bridges...");
-        for (const [id, bridge] of bridges) {
-          try {
-            await bridge.stop();
-            ctx.logger.info(`[a2a-xmtp] Bridge stopped: ${id}`);
-          } catch (err) {
-            ctx.logger.error(`[a2a-xmtp] Error stopping bridge ${id}: ${err}`);
-          }
-        }
-        bridges.clear();
-      },
-    });
-  },
-});

package/src/tools/xmtp-agents.ts

@@ -1,38 +0,0 @@
-// ============================================================
-// Tool: xmtp_agents — 发现可通信的 Agent
-// ============================================================
-
-import type { XmtpBridge } from "../transport/xmtp-bridge.js";
-import type { IdentityRegistry } from "../transport/identity-registry.js";
-
-export async function handleDiscoverAgents(
-  bridges: Map<string, XmtpBridge>,
-  registry: IdentityRegistry,
-  params: { includeExternal?: boolean },
-) {
-  const localAgents = registry.listAgents();
-
-  const lines = localAgents.map((agent) => {
-    const bridge = bridges.get(agent.agentId);
-    const status = bridge?.isConnected ? "online" : "offline";
-    return `- ${agent.agentId} (${agent.address}) [${status}]`;
-  });
-
-  let text = `Available agents (${localAgents.length}):\n${lines.join("\n")}`;
-
-  if (params.includeExternal) {
-    text += "\n\nNote: ERC-8004 external registry lookup is planned for Phase 3.";
-  }
-
-  return {
-    content: [{ type: "text" as const, text }],
-    details: {
-      count: localAgents.length,
-      agents: localAgents.map((a) => ({
-        agentId: a.agentId,
-        address: a.address,
-        connected: bridges.get(a.agentId)?.isConnected ?? false,
-      })),
-    },
-  };
-}

package/src/tools/xmtp-group.ts

@@ -1,153 +0,0 @@
-// ============================================================
-// Tool: xmtp_group — 群聊管理（创建、列表、成员查看/添加/移除）
-// ============================================================
-
-import type { XmtpBridge } from "../transport/xmtp-bridge.js";
-import type { IdentityRegistry } from "../transport/identity-registry.js";
-
-type GroupAction = "create" | "list" | "members" | "add_member" | "remove_member";
-
-export async function handleXmtpGroup(
-  bridges: Map<string, XmtpBridge>,
-  registry: IdentityRegistry,
-  params: {
-    action: string;
-    members?: string[];
-    conversationId?: string;
-    name?: string;
-  },
-  agentId: string,
-) {
-  const bridge = bridges.get(agentId) ?? bridges.values().next().value;
-  if (!bridge) {
-    return { content: [{ type: "text" as const, text: "No XMTP bridge available. Plugin not initialized." }], details: null };
-  }
-
-  const action = params.action as GroupAction;
-
-  try {
-    switch (action) {
-      case "create": {
-        if (!params.members?.length) {
-          return { content: [{ type: "text" as const, text: "Parameter 'members' is required for create action. Provide agent IDs or 0x addresses." }], details: null };
-        }
-        // 解析成员地址
-        const addresses: string[] = [];
-        for (const member of params.members) {
-          if (member.startsWith("0x")) {
-            addresses.push(member);
-          } else {
-            const addr = registry.getAddress(member);
-            if (!addr) {
-              return { content: [{ type: "text" as const, text: `Agent "${member}" not found. Use xmtp_agents to discover available agents.` }], details: null };
-            }
-            addresses.push(addr);
-          }
-        }
-
-        const result = await bridge.createGroup(addresses, params.name);
-        return {
-          content: [{ type: "text" as const, text: `Group created${result.name ? ` "${result.name}"` : ""} with ${addresses.length} members.` }],
-          details: {
-            action: "create",
-            conversationId: result.conversationId,
-            name: result.name,
-            memberCount: addresses.length,
-          },
-        };
-      }
-
-      case "list": {
-        const groups = await bridge.listGroups();
-        if (groups.length === 0) {
-          return { content: [{ type: "text" as const, text: "No group conversations found." }], details: null };
-        }
-        const lines = groups.map((g) => {
-          const nameLabel = g.name ? ` "${g.name}"` : "";
-          return `- ${g.conversationId}${nameLabel} (${g.memberAddresses.length} members, created ${g.createdAt})`;
-        });
-        return {
-          content: [{ type: "text" as const, text: `Groups:\n${lines.join("\n")}` }],
-          details: { action: "list", count: groups.length, groups },
-        };
-      }
-
-      case "members": {
-        if (!params.conversationId) {
-          return { content: [{ type: "text" as const, text: "Parameter 'conversationId' is required for members action." }], details: null };
-        }
-        const members = await bridge.getGroupMembers(params.conversationId);
-        // 尝试解析 agentId
-        const resolved = members.map((addr) => {
-          const aid = registry.getAgentId(addr);
-          return aid ? `${aid} (${addr})` : addr;
-        });
-        return {
-          content: [{ type: "text" as const, text: `Members (${members.length}):\n${resolved.map((m) => `- ${m}`).join("\n")}` }],
-          details: { action: "members", conversationId: params.conversationId, count: members.length, members },
-        };
-      }
-
-      case "add_member": {
-        if (!params.conversationId) {
-          return { content: [{ type: "text" as const, text: "Parameter 'conversationId' is required for add_member action." }], details: null };
-        }
-        if (!params.members?.length) {
-          return { content: [{ type: "text" as const, text: "Parameter 'members' is required for add_member action." }], details: null };
-        }
-        const addresses: string[] = [];
-        for (const member of params.members) {
-          if (member.startsWith("0x")) {
-            addresses.push(member);
-          } else {
-            const addr = registry.getAddress(member);
-            if (!addr) {
-              return { content: [{ type: "text" as const, text: `Agent "${member}" not found.` }], details: null };
-            }
-            addresses.push(addr);
-          }
-        }
-        await bridge.addGroupMembers(params.conversationId, addresses);
-        return {
-          content: [{ type: "text" as const, text: `Added ${addresses.length} member(s) to group.` }],
-          details: { action: "add_member", conversationId: params.conversationId, added: addresses },
-        };
-      }
-
-      case "remove_member": {
-        if (!params.conversationId) {
-          return { content: [{ type: "text" as const, text: "Parameter 'conversationId' is required for remove_member action." }], details: null };
-        }
-        if (!params.members?.length) {
-          return { content: [{ type: "text" as const, text: "Parameter 'members' is required for remove_member action." }], details: null };
-        }
-        const addresses: string[] = [];
-        for (const member of params.members) {
-          if (member.startsWith("0x")) {
-            addresses.push(member);
-          } else {
-            const addr = registry.getAddress(member);
-            if (!addr) {
-              return { content: [{ type: "text" as const, text: `Agent "${member}" not found.` }], details: null };
-            }
-            addresses.push(addr);
-          }
-        }
-        await bridge.removeGroupMembers(params.conversationId, addresses);
-        return {
-          content: [{ type: "text" as const, text: `Removed ${addresses.length} member(s) from group.` }],
-          details: { action: "remove_member", conversationId: params.conversationId, removed: addresses },
-        };
-      }
-
-      default:
-        return {
-          content: [{ type: "text" as const, text: `Unknown action "${action}". Supported: create, list, members, add_member, remove_member.` }],
-          details: null,
-        };
-    }
-  } catch (err: unknown) {
-    const msg = err instanceof Error ? err.message : String(err);
-    return { content: [{ type: "text" as const, text: `Group operation failed: ${msg}` }], details: null };
-  }
-}

package/src/tools/xmtp-inbox.ts

@@ -1,47 +0,0 @@
-// ============================================================
-// Tool: xmtp_inbox — 查看 XMTP 收件箱消息
-// ============================================================
-
-import type { XmtpBridge } from "../transport/xmtp-bridge.js";
-import type { IdentityRegistry } from "../transport/identity-registry.js";
-
-export async function handleXmtpInbox(
-  bridges: Map<string, XmtpBridge>,
-  registry: IdentityRegistry,
-  params: { limit?: number; from?: string },
-  agentId: string,
-) {
-  const bridge = bridges.get(agentId) ?? bridges.values().next().value;
-  if (!bridge) {
-    return { content: [{ type: "text" as const, text: "No XMTP bridge available." }], details: null };
-  }
-
-  // 解析 from
-  let fromFilter = params.from;
-  if (fromFilter && !fromFilter.startsWith("0x")) {
-    const addr = registry.getAddress(fromFilter);
-    if (addr) fromFilter = addr;
-  }
-
-  try {
-    const messages = await bridge.getInbox({ limit: params.limit ?? 10, from: fromFilter });
-
-    if (messages.length === 0) {
-      return { content: [{ type: "text" as const, text: "No messages in inbox." }], details: null };
-    }
-
-    const lines = messages.map((m, i) => {
-      const sender = m.from.agentId ?? m.from.xmtpAddress;
-      const groupTag = m.isGroup ? " [group]" : "";
-      return `${i + 1}. [${m.timestamp}]${groupTag} from ${sender}: ${m.content}`;
-    });
-
-    return {
-      content: [{ type: "text" as const, text: lines.join("\n") }],
-      details: { count: messages.length, myAddress: bridge.address },
-    };
-  } catch (err: unknown) {
-    const msg = err instanceof Error ? err.message : String(err);
-    return { content: [{ type: "text" as const, text: `Failed to fetch inbox: ${msg}` }], details: null };
-  }
-}

package/src/tools/xmtp-send.ts

@@ -1,77 +0,0 @@
-// ============================================================
-// Tool: xmtp_send — 发送 E2EE 消息到另一个 Agent
-// ============================================================
-
-import type { XmtpBridge } from "../transport/xmtp-bridge.js";
-import type { IdentityRegistry } from "../transport/identity-registry.js";
-import type { PolicyEngine } from "../coordination/policy-engine.js";
-
-export async function handleXmtpSend(
-  bridges: Map<string, XmtpBridge>,
-  registry: IdentityRegistry,
-  policyEngine: PolicyEngine,
-  params: { to?: string; message: string; conversationId?: string; contentType?: "text" | "markdown" },
-  senderAgentId: string,
-) {
-  // 找到发送者 bridge（使用第一个可用的 bridge）
-  const bridge = bridges.get(senderAgentId) ?? bridges.values().next().value;
-  if (!bridge) {
-    return { content: [{ type: "text" as const, text: "No XMTP bridge available. Plugin not initialized." }], details: null };
-  }
-
-  // 当提供 conversationId 时，to 可省略（直接向已有会话发送）
-  if (!params.to && !params.conversationId) {
-    return { content: [{ type: "text" as const, text: "Either 'to' or 'conversationId' must be provided." }], details: null };
-  }
-
-  // 解析目标地址（仅在提供 to 时需要）
-  let targetAddress: string | undefined;
-  if (params.to) {
-    if (params.to.startsWith("0x")) {
-      targetAddress = params.to;
-    } else {
-      const addr = registry.getAddress(params.to);
-      if (!addr) {
-        return {
-          content: [{ type: "text" as const, text: `Agent "${params.to}" not found. Use xmtp_agents to discover available agents.` }],
-          details: null,
-        };
-      }
-      targetAddress = addr;
-    }
-
-    if (targetAddress.toLowerCase() === bridge.address.toLowerCase()) {
-      return { content: [{ type: "text" as const, text: "Cannot send message to yourself." }], details: null };
-    }
-  }
-
-  // Policy 检查
-  const toLabel = params.to ?? params.conversationId!;
-  const convId = params.conversationId ?? `dm:${senderAgentId}:${params.to}`;
-  const check = policyEngine.checkOutgoing({ from: senderAgentId, to: toLabel, conversationId: convId });
-  if (!check.allowed) {
-    return { content: [{ type: "text" as const, text: `Blocked: ${check.reason}` }], details: null };
-  }
-
-  try {
-    const result = await bridge.sendMessage(targetAddress ?? "", params.message, {
-      contentType: params.contentType,
-      conversationId: params.conversationId,
-    });
-
-    const recipientLabel = params.to ?? `conversation ${result.conversationId}`;
-    return {
-      content: [{ type: "text" as const, text: `Message sent to ${recipientLabel}.` }],
-      details: {
-        status: "sent",
-        conversationId: result.conversationId,
-        messageId: result.messageId,
-        to: params.to,
-        toAddress: targetAddress,
-      },
-    };
-  } catch (err: unknown) {
-    const msg = err instanceof Error ? err.message : String(err);
-    return { content: [{ type: "text" as const, text: `Failed to send: ${msg}` }], details: null };
-  }
-}

package/src/transport/identity-registry.ts

@@ -1,133 +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, readdirSync } 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;
-    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
-      }
-    }
-  }
-}