@agentwonderland/mcp 0.1.51 → 0.1.53

package/src/tools/rebates.ts

@@ -0,0 +1,102 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { apiGet } from "../core/api-client.js";
+
+type RebateStatus = {
+  consumer_principal: string;
+  principal_type: string;
+  principal_value: string;
+  payout_address: string | null;
+  payout_chain: string;
+  auto_payout_threshold_usd: string;
+  totals: {
+    lifetime_earned_usd: string;
+    pending_usd: string;
+    paid_usd: string;
+    blocked_usd: string;
+    total_count: number;
+    pending_count: number;
+    paid_count: number;
+    blocked_count: number;
+  };
+  by_source: Array<{
+    source_type: string;
+    status: string;
+    count: number;
+    rebate_usd: string;
+  }>;
+  recent_payouts: Array<{
+    id: string;
+    status: string;
+    amount_usd: string;
+    earning_count: number;
+    tx_hash: string | null;
+    processed_at: string | null;
+  }>;
+};
+
+function text(t: string) {
+  return { content: [{ type: "text" as const, text: t }] };
+}
+
+function money(value: string | number): string {
+  const numeric = Number(value);
+  return Number.isFinite(numeric) ? `$${numeric.toFixed(4)}` : `$${value}`;
+}
+
+function sourceLabel(value: string): string {
+  return value.replace(/_/g, " ");
+}
+
+export function registerRebateTools(server: McpServer): void {
+  server.tool(
+    "rebate_status",
+    "Show your Agent Wonderland consumer rebate balance, payout threshold, blocked rebates, and recent payout transactions.",
+    {},
+    async () => {
+      const status = await apiGet<RebateStatus>("/rebates/status", { ensureConsumerPrincipal: true });
+      const threshold = Number(status.auto_payout_threshold_usd);
+      const pending = Number(status.totals.pending_usd);
+      const remaining = Math.max(0, threshold - pending);
+
+      const lines = [
+        "Rebate status",
+        `Consumer principal: ${status.consumer_principal}`,
+        `Base payout address: ${status.payout_address ?? "not configured"}`,
+        "",
+        `Lifetime earned: ${money(status.totals.lifetime_earned_usd)} across ${status.totals.total_count} rebate(s)`,
+        `Paid out: ${money(status.totals.paid_usd)} across ${status.totals.paid_count} rebate(s)`,
+        `Pending: ${money(status.totals.pending_usd)} across ${status.totals.pending_count} rebate(s)`,
+        `Blocked: ${money(status.totals.blocked_usd)} across ${status.totals.blocked_count} rebate(s)`,
+      ];
+
+      if (Number.isFinite(threshold) && threshold > 0) {
+        lines.push(
+          pending >= threshold
+            ? `Auto payout: eligible for the next processor run (threshold ${money(threshold)})`
+            : `Auto payout: ${money(remaining)} until the ${money(threshold)} threshold`,
+        );
+      }
+
+      if (status.by_source.length > 0) {
+        lines.push("", "By source:");
+        for (const row of status.by_source) {
+          lines.push(`  ${sourceLabel(row.source_type)} / ${row.status}: ${money(row.rebate_usd)} (${row.count})`);
+        }
+      }
+
+      if (status.recent_payouts.length > 0) {
+        lines.push("", "Recent payouts:");
+        for (const payout of status.recent_payouts.slice(0, 5)) {
+          const tx = payout.tx_hash ? ` tx=${payout.tx_hash}` : "";
+          lines.push(`  ${payout.status}: ${money(payout.amount_usd)} (${payout.earning_count})${tx}`);
+        }
+      }
+
+      if (!status.payout_address) {
+        lines.push("", "No Base payout address is configured, so new rebates may be blocked.");
+      }
+
+      return text(lines.join("\n"));
+    },
+  );
+}

package/src/tools/run.ts

@@ -220,6 +220,10 @@ export function registerRunTools(server: McpServer): void {
           `  Payment: ${formatPaymentLabel(method)}`,
         ];
 
+        if (method === "link") {
+          quoteLines.push("  Link: after confirming here, approve the Link spend request and rerun the confirmed call.");
+        }
+
         const creditPackLines = buildCreditPackOfferLines(agent);
         if (creditPackLines.length > 0) {
           quoteLines.push("", ...creditPackLines);
@@ -299,6 +303,9 @@ export function registerRunTools(server: McpServer): void {
           );
         }
         const msg = apiErr?.message ?? "Failed to run agent";
+        if (msg.includes("Link approval required.")) {
+          return text(msg);
+        }
         if (msg.includes("Missing required field") || msg.includes("validation failed")) {
           return text(`Error: ${msg}\n\nUse get_agent("${agent_id}") to see the required input fields.`);
         }

package/src/tools/solve.ts

@@ -292,6 +292,9 @@ export function registerSolveTools(server: McpServer): void {
           `Best match: ${selected.name}`,
           `Cost: $${estimatedCost.toFixed(2)}`,
           `Payment: ${formatPaymentLabel(method)}`,
+          ...(method === "link"
+            ? ["Link: after confirming here, approve the Link spend request and rerun the confirmed call."]
+            : []),
           ...(() => {
             const summary = buildCreditPackSummary(selected);
             return summary.length > 0 ? ["", ...summary] : [];
@@ -361,7 +364,11 @@ export function registerSolveTools(server: McpServer): void {
             ].join("\n"),
           );
         }
-        return text(`Error: ${apiErr?.message ?? "Failed to run agent"}`);
+        const msg = apiErr?.message ?? "Failed to run agent";
+        if (msg.includes("Link approval required.")) {
+          return text(msg);
+        }
+        return text(`Error: ${msg}`);
       }
 
       pendingSolves.delete(pendingKey);

package/src/tools/wallet.ts

@@ -39,7 +39,7 @@ import {
   installOws,
   platformSupportsOws,
 } from "../core/ows-adapter.js";
-import { ensureConsumerPrincipal, getConsumerPrincipal } from "../core/principal.js";
+import { ensureConsumerPrincipal, ensureConsumerPrincipalForMethod, getConsumerPrincipal } from "../core/principal.js";
 import { MCP_PACKAGE_VERSION } from "../core/version.js";
 
 function text(t: string) {
@@ -292,7 +292,7 @@ export function registerWalletTools(server: McpServer): void {
             paymentMethodId: selected.id,
             label: name ?? selected.label,
           });
-          const principal = await ensureConsumerPrincipal();
+          const principal = await ensureConsumerPrincipalForMethod("link");
           return text(
             [
               "Link payment method connected.",
@@ -330,7 +330,7 @@ export function registerWalletTools(server: McpServer): void {
             paymentMethodId: method.id,
             label: method.label,
           });
-          const principal = await ensureConsumerPrincipal();
+          const principal = await ensureConsumerPrincipalForMethod("link");
           return text(
             [
               "Link payment method connected.",
@@ -380,7 +380,7 @@ export function registerWalletTools(server: McpServer): void {
           const result = await pollCardSetup(pendingToken, 250);
 
           if (result) {
-            const principal = await ensureConsumerPrincipal();
+            const principal = await ensureConsumerPrincipalForMethod("card");
             return text(
               `Connected! ${result.brand} ****${result.last4} is ready for payments.\n\n` +
               `Consumer principal: ${principal}`,