@agentwonderland/mcp 0.1.46 → 0.1.48

package/src/tools/wallet.ts

@@ -3,10 +3,15 @@ import { z } from "zod";
 import {
   getWallets,
   getCardConfig,
+  getLinkConfig,
+  getPendingLinkSetup,
+  setPendingLinkSetup,
+  setLinkConfig,
   setCardConfig,
   addWallet,
   getPendingCardSetupToken,
   getSpendPolicy,
+  setDefaultPaymentMethod,
   setSpendPolicy,
 } from "../core/config.js";
 import { getWalletAddress, isCardPaymentEnabled } from "../core/payments.js";
@@ -18,6 +23,12 @@ import {
   getCardCapabilities,
   pollCardSetup,
 } from "../core/card-setup.js";
+import {
+  getLinkCliAuthStatus,
+  listLinkPaymentMethods,
+  openLinkPaymentMethodAdd,
+  startLinkCliLogin,
+} from "../core/link-cli.js";
 import {
   isOwsAvailable,
   createOwsWallet,
@@ -34,6 +45,49 @@ function text(t: string) {
   return { content: [{ type: "text" as const, text: t }] };
 }
 
+function formatLinkApprovalPrompt(pending: { verificationUrl: string; phrase: string }): string {
+  return [
+    "Approve Agent Wonderland in Link:",
+    "",
+    pending.verificationUrl,
+    "",
+    `Verification phrase: ${pending.phrase}`,
+    "",
+    "After approving, run wallet_setup({ action: \"add-link\" }) again and I will finish connecting the payment method.",
+  ].join("\n");
+}
+
+function isFreshLinkSetup(pending: { createdAt: string }): boolean {
+  const createdAt = Date.parse(pending.createdAt);
+  if (!Number.isFinite(createdAt)) return false;
+  return Date.now() - createdAt < 10 * 60 * 1000;
+}
+
+function formatPaymentSetupMenu(): string {
+  return [
+    "Choose a payment method to set up:",
+    "",
+    "1. Link card or bank account (recommended)",
+    "   wallet_setup({ action: \"add-link\" })",
+    "",
+    "2. Tempo USDC",
+    "   wallet_setup({ action: \"create\", chain: \"tempo\" })",
+    "",
+    "3. Base USDC",
+    "   wallet_setup({ action: \"create\", chain: \"base\" })",
+    "",
+    "4. Solana USDC",
+    "   wallet_setup({ action: \"create\", chain: \"solana\" })",
+    "",
+    "5. Import an existing wallet",
+    "   wallet_setup({ action: \"import\", chain: \"base\", key: \"<private key>\" })",
+  ].join("\n");
+}
+
+function setDefaultCryptoPaymentMethod(chain: string): void {
+  setDefaultPaymentMethod(chain === "solana" ? "solana" : chain === "base" ? "base" : "tempo");
+}
+
 export function registerWalletTools(server: McpServer): void {
   // ── wallet_status (extracted from check_wallet) ─────────────────
   server.tool(
@@ -43,12 +97,13 @@ export function registerWalletTools(server: McpServer): void {
     async () => {
       const wallets = getWallets();
       const card = getCardConfig();
+      const link = getLinkConfig();
       const pendingCardSetupToken = getPendingCardSetupToken();
       const consumerPrincipal = await getConsumerPrincipal();
 
-      if (wallets.length === 0 && !card && !pendingCardSetupToken) {
+      if (wallets.length === 0 && !card && !link && !pendingCardSetupToken) {
         return text(
-          "No payment methods configured.\nUse wallet_setup to create or import a wallet.",
+          `No payment methods configured.\n\n${formatPaymentSetupMenu()}`,
         );
       }
 
@@ -95,6 +150,15 @@ export function registerWalletTools(server: McpServer): void {
         }
       }
 
+      if (link) {
+        const auth = await getLinkCliAuthStatus();
+        const label = link.label ? ` (${link.label})` : "";
+        lines.push(`  Link${label}: ${link.paymentMethodId}`);
+        lines.push(auth.authenticated
+          ? "  Link CLI: authenticated"
+          : "  Link CLI: not authenticated — run npx @stripe/link-cli auth login");
+      }
+
       if (pendingCardSetupToken) {
         lines.push("  Card setup: pending confirmation");
       }
@@ -121,11 +185,11 @@ export function registerWalletTools(server: McpServer): void {
   // ── wallet_setup ────────────────────────────────────────────────
   server.tool(
     "wallet_setup",
-    "Set up or manage a payment wallet. 'create' makes a new crypto wallet (encrypted via OWS if available, otherwise plaintext — run 'enable-ows' to upgrade). 'import' takes an existing private key. 'enable-ows' installs the Open Wallet Standard native module for encrypted at-rest storage. Tempo/Base share one EVM key; Solana uses a separate ed25519 key. NEVER delete or rotate keys programmatically; direct users to edit ~/.agentwonderland/config.json or ~/.ows/ manually.",
+    "Set up or manage an Agent Wonderland payment method. Use 'start' for a guided setup menu. Link card/bank is recommended for most users. 'add-link' connects a Link card or bank account for agent payments. 'create' makes a new crypto wallet (encrypted via OWS if available, otherwise plaintext — run 'enable-ows' to upgrade). 'import' takes an existing private key. 'enable-ows' installs the Open Wallet Standard native module for encrypted at-rest storage. Tempo/Base share one EVM key; Solana uses a separate ed25519 key. NEVER delete or rotate keys programmatically; direct users to edit ~/.agentwonderland/config.json or ~/.ows/ manually.",
     {
       action: z
-        .enum(["create", "import", "add-card", "remove-card", "enable-ows"])
-        .describe("'create' a wallet, 'import' an existing key, 'enable-ows' to install encrypted key storage, 'add-card'/'remove-card' for credit card (may be disabled depending on Stripe SPT availability)"),
+        .enum(["start", "create", "import", "add-card", "remove-card", "add-link", "remove-link", "enable-ows"])
+        .describe("'start' shows the guided payment setup menu, 'add-link' connects Link card/bank, 'create' makes a crypto wallet, 'import' imports an existing key, 'enable-ows' installs encrypted key storage"),
       name: z
         .string()
         .optional()
@@ -138,8 +202,160 @@ export function registerWalletTools(server: McpServer): void {
         ),
       chain: z.enum(["tempo", "base", "solana"]).optional()
         .describe("Primary chain (default: tempo). Tempo/Base use a shared EVM wallet; Solana uses a separate OWS wallet."),
+      link_payment_method_id: z.string().optional()
+        .describe("Link payment method ID from `npx @stripe/link-cli payment-methods list`; used with action 'add-link'."),
+      link_payment_method: z.string().optional()
+        .describe("Friendly Link payment method selector, such as a card/bank name or last4; used with action 'add-link'."),
     },
-    async ({ action, name, key, chain }) => {
+    async ({ action, name, key, chain, link_payment_method_id, link_payment_method }) => {
+      if (action === "start") {
+        return text(formatPaymentSetupMenu());
+      }
+
+      // ── Link setup flow ──────────────────────────────────────
+      if (action === "add-link") {
+        let auth = await getLinkCliAuthStatus();
+        if (!auth.authenticated) {
+          const pending = getPendingLinkSetup();
+          if (pending && isFreshLinkSetup(pending)) {
+            return text(formatLinkApprovalPrompt(pending));
+          }
+          if (pending) {
+            setPendingLinkSetup(null);
+          }
+
+          try {
+            const login = await startLinkCliLogin();
+            setPendingLinkSetup({
+              verificationUrl: login.verificationUrl,
+              phrase: login.phrase,
+              createdAt: new Date().toISOString(),
+            });
+            auth = await getLinkCliAuthStatus();
+            if (!auth.authenticated) {
+              return text(formatLinkApprovalPrompt({
+                verificationUrl: login.verificationUrl,
+                phrase: login.phrase,
+              }));
+            }
+          } catch (err) {
+            return text(
+              [
+                "Link authentication could not start from the MCP session.",
+                `Reason: ${err instanceof Error ? err.message : String(err)}`,
+                "",
+                "Run wallet_setup({ action: \"add-link\" }) again to retry.",
+              ].join("\n"),
+            );
+          }
+        }
+        setPendingLinkSetup(null);
+
+        const explicitLinkSelector = link_payment_method_id ?? link_payment_method;
+        if (explicitLinkSelector) {
+          const methods = await listLinkPaymentMethods();
+          const normalizedSelector = explicitLinkSelector.toLowerCase().trim();
+          const exact = methods.find((method) => method.id === explicitLinkSelector);
+          const matches = exact
+            ? [exact]
+            : methods.filter((method) => method.searchText?.includes(normalizedSelector));
+          if (matches.length === 0 && !link_payment_method_id) {
+            return text(
+              [
+                `No Link payment method matched "${explicitLinkSelector}".`,
+                "",
+                "Available methods:",
+                ...methods.map((method) => `  ${method.label ?? method.id}${method.label ? ` (${method.id})` : ""}`),
+              ].join("\n"),
+            );
+          }
+          if (matches.length > 1) {
+            return text(
+              [
+                `Multiple Link payment methods matched "${explicitLinkSelector}". Choose one:`,
+                ...matches.map((method) => `  ${method.label ?? method.id}${method.label ? ` (${method.id})` : ""}`),
+                "",
+                "Then run:",
+                "  wallet_setup({ action: \"add-link\", link_payment_method: \"<name or last4>\" })",
+              ].join("\n"),
+            );
+          }
+          const selected = matches[0] ?? { id: link_payment_method_id!, label: name };
+          setLinkConfig({
+            paymentMethodId: selected.id,
+            label: name ?? selected.label,
+          });
+          const principal = await ensureConsumerPrincipal();
+          return text(
+            [
+              "Link payment method connected.",
+              `  Payment method: ${selected.label ?? selected.id}${selected.label ? ` (${selected.id})` : ""}`,
+              `  Consumer principal: ${principal}`,
+              "",
+              "Use pay_with: \"link\" for agent runs.",
+            ].join("\n"),
+          );
+        }
+
+        let methods = await listLinkPaymentMethods();
+        if (methods.length === 0) {
+          try {
+            await openLinkPaymentMethodAdd();
+            methods = await listLinkPaymentMethods();
+          } catch (err) {
+            return text(
+              [
+                "No Link payment methods are available yet.",
+                `Reason: ${err instanceof Error ? err.message : String(err)}`,
+                "",
+                "Run wallet_setup({ action: \"add-link\" }) again after adding a payment method in Link.",
+              ].join("\n"),
+            );
+          }
+          if (methods.length === 0) {
+            return text("Link payment-method setup was started. Run wallet_setup({ action: \"add-link\" }) again after adding a payment method in Link.");
+          }
+        }
+
+        if (methods.length === 1) {
+          const method = methods[0];
+          setLinkConfig({
+            paymentMethodId: method.id,
+            label: method.label,
+          });
+          const principal = await ensureConsumerPrincipal();
+          return text(
+            [
+              "Link payment method connected.",
+              `  Payment method: ${method.id}${method.label ? ` (${method.label})` : ""}`,
+              `  Consumer principal: ${principal}`,
+              "",
+              "Use pay_with: \"link\" for agent runs.",
+            ].join("\n"),
+          );
+        }
+
+        return text(
+          [
+            "Multiple Link payment methods found. Choose one:",
+            ...methods.map((method) => `  ${method.label ?? method.id}${method.label ? ` (${method.id})` : ""}`),
+            "",
+            "Then run:",
+            "  wallet_setup({ action: \"add-link\", link_payment_method: \"<name or last4>\" })",
+          ].join("\n"),
+        );
+      }
+
+      if (action === "remove-link") {
+        const existing = getLinkConfig();
+        if (!existing) {
+          return text("No Link payment method is currently connected.");
+        }
+
+        setLinkConfig(null);
+        return text(`Removed Link payment method ${existing.paymentMethodId}.`);
+      }
+
       // ── Card setup flow ──────────────────────────────────────
       if (action === "add-card") {
         if (!isCardPaymentEnabled()) {
@@ -273,6 +489,7 @@ export function registerWalletTools(server: McpServer): void {
                 chainStatus = "Already linked to Agent Wonderland.";
               }
             }
+            setDefaultCryptoPaymentMethod(defaultCh);
             const principal = await ensureConsumerPrincipal();
             return text([
               "Found existing OWS wallet:",
@@ -307,6 +524,7 @@ export function registerWalletTools(server: McpServer): void {
               defaultChain: "solana",
               label: walletName,
             });
+            setDefaultCryptoPaymentMethod("solana");
             const principal = await ensureConsumerPrincipal();
             const owsNudge = platformSupportsOws()
               ? "\n\n⚠ Key stored in plaintext at ~/.agentwonderland/config.json. Run wallet_setup({ action: \"enable-ows\" }) to install encrypted at-rest storage — takes ~30s."
@@ -336,6 +554,7 @@ export function registerWalletTools(server: McpServer): void {
             defaultChain: defaultCh === "solana" ? "tempo" : defaultCh,
             label: walletName,
           });
+          setDefaultCryptoPaymentMethod(defaultCh);
           const principal = await ensureConsumerPrincipal();
           const owsNudge = platformSupportsOws()
             ? "\n\n⚠ Key stored in plaintext at ~/.agentwonderland/config.json. Run wallet_setup({ action: \"enable-ows\" }) to install encrypted at-rest storage — takes ~30s."
@@ -367,6 +586,7 @@ export function registerWalletTools(server: McpServer): void {
           defaultChain: defaultCh,
           label: walletName,
         });
+        setDefaultCryptoPaymentMethod(defaultCh);
         const principal = await ensureConsumerPrincipal();
 
         return text(
@@ -400,6 +620,7 @@ export function registerWalletTools(server: McpServer): void {
           defaultChain: defaultCh,
           label: walletName,
         });
+        setDefaultCryptoPaymentMethod(defaultCh);
         const principal = await ensureConsumerPrincipal();
 
         return text(
@@ -432,6 +653,7 @@ export function registerWalletTools(server: McpServer): void {
             defaultChain: "solana",
             label: walletName,
           });
+          setDefaultCryptoPaymentMethod("solana");
           const principal = await ensureConsumerPrincipal();
           const owsNudge = platformSupportsOws()
             ? "\n\n⚠ Key stored in plaintext at ~/.agentwonderland/config.json. Run wallet_setup({ action: \"enable-ows\" }) to install encrypted at-rest storage — takes ~30s."
@@ -472,6 +694,7 @@ export function registerWalletTools(server: McpServer): void {
           defaultChain: defaultCh,
           label: walletName,
         });
+        setDefaultCryptoPaymentMethod(defaultCh);
         const principal = await ensureConsumerPrincipal();
 
         const owsNudge = platformSupportsOws()

package/dist/tools/observability.d.ts

@@ -1,2 +0,0 @@
-import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-export declare function registerObservabilityTools(server: McpServer): void;

package/dist/tools/observability.js

@@ -1,20 +0,0 @@
-import { apiPost } from "../core/api-client.js";
-function text(t) {
-    return { content: [{ type: "text", text: t }] };
-}
-export function registerObservabilityTools(server) {
-    server.tool("open_observability_dashboard", "Generate a secure one-click sign-in URL for the Agent Wonderland web observability dashboard. The dashboard shows your agent runs, spend, rebates, and recent activity.", {}, async () => {
-        const result = await apiPost("/observability/link", {}, { ensureConsumerPrincipal: true });
-        const lines = [
-            "Your secure observability link is ready:",
-            result.url,
-            "",
-            `Expires: ${result.expires_at}`,
-        ];
-        if (result.consumer_principal) {
-            lines.push(`Consumer principal: ${result.consumer_principal}`);
-        }
-        lines.push("", "Open the link in your browser to view usage metrics, spend, rebates, and recent runs.");
-        return text(lines.join("\n"));
-    });
-}