@agentwonderland/mcp 0.1.24 → 0.1.25

package/src/tools/wallet.ts

@@ -6,6 +6,8 @@ import {
   setCardConfig,
   addWallet,
   getPendingCardSetupToken,
+  getSpendPolicy,
+  setSpendPolicy,
 } from "../core/config.js";
 import { getWalletAddress } from "../core/payments.js";
 import {
@@ -352,8 +354,13 @@ export function registerWalletTools(server: McpServer): void {
         .positive()
         .optional()
         .describe("Maximum USD per day across all transactions"),
+      require_confirmation_above: z
+        .number()
+        .positive()
+        .optional()
+        .describe("Require manual confirmation above this USD amount even when auto-spend is enabled"),
     },
-    async ({ wallet_id, max_per_tx, max_per_day }) => {
+    async ({ wallet_id, max_per_tx, max_per_day, require_confirmation_above }) => {
       const wallets = getWallets();
       const wallet = wallets.find((w) => w.id === wallet_id);
 
@@ -364,23 +371,36 @@ export function registerWalletTools(server: McpServer): void {
         );
       }
 
-      if (max_per_tx == null && max_per_day == null) {
+      if (
+        max_per_tx == null &&
+        max_per_day == null &&
+        require_confirmation_above == null
+      ) {
         return text(
-          "No policy changes specified. Provide max_per_tx and/or max_per_day.",
+          "No policy changes specified. Provide at least one policy field.",
         );
       }
 
+      const existing = getSpendPolicy(wallet_id) ?? {};
+      const nextPolicy = {
+        ...existing,
+        ...(max_per_tx != null ? { maxPerTxUsd: max_per_tx } : {}),
+        ...(max_per_day != null ? { maxPerDayUsd: max_per_day } : {}),
+        ...(require_confirmation_above != null ? { requireConfirmationAboveUsd: require_confirmation_above } : {}),
+      };
+      setSpendPolicy(wallet_id, nextPolicy);
+
       // Build policy summary
       const policies: string[] = [];
-      if (max_per_tx != null) {
-        policies.push(`Max per transaction: $${max_per_tx.toFixed(2)}`);
+      if (nextPolicy.maxPerTxUsd != null) {
+        policies.push(`Max per transaction: $${nextPolicy.maxPerTxUsd.toFixed(2)}`);
       }
-      if (max_per_day != null) {
-        policies.push(`Max per day: $${max_per_day.toFixed(2)}`);
+      if (nextPolicy.maxPerDayUsd != null) {
+        policies.push(`Max per day: $${nextPolicy.maxPerDayUsd.toFixed(2)}`);
+      }
+      if (nextPolicy.requireConfirmationAboveUsd != null) {
+        policies.push(`Require confirmation above: $${nextPolicy.requireConfirmationAboveUsd.toFixed(2)}`);
       }
-
-      // Note: actual persistence depends on the config module supporting policy fields.
-      // For now, we report what would be set. The core/config module will handle storage.
       return text(
         [
           `Spending policy set for wallet "${wallet_id}":`,