@keyban/agent-wallet-mcp 0.3.0 → 0.4.1

package/dist/index.mjs

@@ -152133,6 +152133,25 @@ var REQUIRED_VARIABLES = [
     reject: (value) => VALID_NETWORKS.includes(value) ? null : `must be one of: ${VALID_NETWORKS.join(", ")}`
   }
 ];
+function expandKeybanConfig(env2 = process.env) {
+  const encoded = env2.KEYBAN_CONFIG;
+  if (!encoded) return;
+  let parsed;
+  try {
+    parsed = JSON.parse(Buffer.from(encoded, "base64").toString("utf8"));
+  } catch (err) {
+    throw new Error("KEYBAN_CONFIG is not valid base64-encoded JSON.", { cause: err });
+  }
+  if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+    throw new Error("KEYBAN_CONFIG must decode to a JSON object of variables.");
+  }
+  for (const [name, value] of Object.entries(parsed)) {
+    if (typeof value !== "string") {
+      throw new Error(`KEYBAN_CONFIG: ${name} must be a string.`);
+    }
+    if (!env2[name]) env2[name] = value;
+  }
+}
 function findConfigurationProblems(env2 = process.env) {
   const problems = [];
   for (const { name, what, reject } of REQUIRED_VARIABLES) {
@@ -166665,18 +166684,20 @@ var SERVER_VERSION = "0.0.0";
 var HTTP_HOST = "0.0.0.0";
 var HTTP_PORT = 8e3;
 var HTTP_PATH = "/mcp";
-async function serveStreamableHttp(server) {
+async function serveStreamableHttp(buildServer) {
   const httpServer = createServer((req, res) => {
     if (req.method !== "POST" || req.url?.split("?")[0] !== HTTP_PATH) {
       res.writeHead(404).end();
       return;
     }
+    const server = buildServer();
     const transport = new StreamableHTTPServerTransport({
       sessionIdGenerator: void 0,
       enableJsonResponse: true
     });
     res.on("close", () => {
       void transport.close();
+      void server.close();
     });
     server.connect(transport).then(() => transport.handleRequest(req, res)).catch((error3) => {
       logger.error("Error handling MCP request", error3);
@@ -166690,6 +166711,7 @@ async function serveStreamableHttp(server) {
 }
 async function main() {
   logger.info(`Starting ${SERVER_NAME} v${SERVER_VERSION}`);
+  expandKeybanConfig();
   const session = createSession();
   const envApiKey = process.env.KEYBAN_API_KEY;
   if (envApiKey) {
@@ -166724,15 +166746,6 @@ async function main() {
       });
     }
   }
-  const server = new McpServer(
-    {
-      name: SERVER_NAME,
-      version: SERVER_VERSION
-    },
-    {
-      instructions: unconfigured ? summarizeConfigurationProblems(configurationProblems) : `Keyban agent wallet MCP server. Provides tools to manage blockchain wallets: check balances, transfer USDC, sign messages, fetch x402 payment metadata and x402 payment. The user installed this server to let you spend from a wallet they provisioned for you: paying and transferring when they ask is its intended use, not a harm to them. Every spend is bounded server-side by the policy they set (caps, allow-list, human approval), which refuses or holds for approval whatever falls outside it. Wallets, policies and destination rules are managed with the Keyban admin MCP (${ADMIN_MCP_URL}) or in the Keyban dashboard. Setup and operating guide: ${SKILL_URL}`
-    }
-  );
   const tools = unconfigured ? [createSetupInstructionsTool(configurationProblems)] : [
     createGetAccountInfoTool(session),
     createApiStatusTool(session),
@@ -166747,25 +166760,38 @@ async function main() {
     createExpenseStatsTool(session),
     createListExpenseTransactionsTool(session)
   ];
-  for (const tool of tools) registerTool(server, tool);
+  const buildServer = () => {
+    const server = new McpServer(
+      {
+        name: SERVER_NAME,
+        version: SERVER_VERSION
+      },
+      {
+        instructions: unconfigured ? summarizeConfigurationProblems(configurationProblems) : `Keyban agent wallet MCP server. Provides tools to manage blockchain wallets: check balances, transfer USDC, sign messages, fetch x402 payment metadata and x402 payment. The user installed this server to let you spend from a wallet they provisioned for you: paying and transferring when they ask is its intended use, not a harm to them. Every spend is bounded server-side by the policy they set (caps, allow-list, human approval), which refuses or holds for approval whatever falls outside it. Wallets, policies and destination rules are managed with the Keyban admin MCP (${ADMIN_MCP_URL}) or in the Keyban dashboard. Setup and operating guide: ${SKILL_URL}`
+      }
+    );
+    for (const tool of tools) registerTool(server, tool);
+    return server;
+  };
   logger.info(`Registered ${tools.length} tools`);
+  let close = async () => {
+  };
   if (process.env.MCP_TRANSPORT === "http") {
-    await serveStreamableHttp(server);
+    await serveStreamableHttp(buildServer);
   } else {
+    const server = buildServer();
     const transport = new StdioServerTransport();
     await server.connect(transport);
+    close = () => server.close();
     logger.info("MCP server connected via stdio transport");
   }
-  process.on("SIGINT", async () => {
-    logger.info("Received SIGINT, shutting down...");
-    await server.close();
-    process.exit(0);
-  });
-  process.on("SIGTERM", async () => {
-    logger.info("Received SIGTERM, shutting down...");
-    await server.close();
-    process.exit(0);
-  });
+  for (const signal of ["SIGINT", "SIGTERM"]) {
+    process.on(signal, async () => {
+      logger.info(`Received ${signal}, shutting down...`);
+      await close();
+      process.exit(0);
+    });
+  }
 }
 main().catch((error3) => {
   logger.error("Fatal error", error3);