@zyfai/sdk 0.2.37 → 0.2.39

package/dist/index.js

@@ -47,6 +47,7 @@ var import_axios = __toESM(require("axios"));
 // src/config/endpoints.ts
 var API_ENDPOINT = "https://api.zyf.ai";
 var DATA_API_ENDPOINT = "https://defiapi.zyf.ai";
+var WS_ENDPOINT = "wss://defiapi.zyf.ai/ws/events";
 var API_VERSION = "/api/v1";
 var DATA_API_VERSION = "/api/v2";
 var ENDPOINTS = {
@@ -80,6 +81,21 @@ var ENDPOINTS = {
   SDK_TVL: "/data/sdk-tvl",
   // Agent Identity Registry
   AGENT_TOKEN_URI: "/users/me/agent-token-uri",
+  // Simulation
+  SIMULATE_BEST_POSITIONS: (params) => {
+    const networks = Array.isArray(params.networks) ? params.networks.join(",") : params.networks;
+    const query = [
+      `amount=${params.amount}`,
+      `token=${params.token}`,
+      `networks=${networks}`,
+      `strategy=${params.strategy}`
+    ];
+    if (params.minSplit !== void 0) query.push(`minSplit=${params.minSplit}`);
+    if (params.protocols?.length) query.push(`protocols=${params.protocols.join(",")}`);
+    if (params.pools?.length) query.push(`pools=${params.pools.join(",")}`);
+    if (params.userPositions?.length) query.push(`userPositions=${encodeURIComponent(JSON.stringify(params.userPositions))}`);
+    return `/simulate/best-positions?${query.join("&")}`;
+  },
   // Customization
   CUSTOMIZE_BATCH: "/customization/customize-batch",
   CUSTOMIZATION_POOLS: (protocolId, strategy) => `/customization/pools?protocolId=${protocolId}${strategy ? `&strategy=${strategy}` : ""}`,
@@ -476,41 +492,34 @@ var VAULT_ABI = [
 ];
 var VAULT_ADDRESS = "0xD580071c47d4a667858B5FafAb85BC9C609beC5D";
 
+// src/config/constants.ts
+var MIN_PORTFOLIO_BALANCE = {
+  1: {
+    USDC: 10000n * 10n ** 6n,
+    // 10,000 USDC (6 decimals)
+    WETH: 5n * 10n ** 18n
+    // 5 WETH (18 decimals)
+  }
+};
+var formatMinPortfolioLabel = (raw, decimals, symbol) => {
+  const whole = raw / 10n ** BigInt(decimals);
+  return `${whole.toString()} ${symbol}`;
+};
+
 // src/core/ZyfaiSDK.ts
 var import_accounts2 = require("viem/accounts");
-var import_viem5 = require("viem");
+var import_viem4 = require("viem");
 
 // src/config/chains.ts
 var import_viem2 = require("viem");
 var import_chains = require("viem/chains");
-var import_viem3 = require("viem");
-var plasma = (0, import_viem3.defineChain)({
-  id: 9745,
-  name: "Plasma",
-  nativeCurrency: {
-    decimals: 18,
-    name: "Plasma",
-    symbol: "PLSM"
-  },
-  rpcUrls: {
-    default: {
-      http: ["https://rpc.plasma.to"]
-    }
-  },
-  blockExplorers: {
-    default: {
-      name: "Plasma Explorer",
-      url: "https://explorer.plasma.io"
-    }
-  }
-});
 var DEFAULT_TOKEN_ADDRESSES = {
+  1: "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48",
+  // USDC on Ethereum Mainnet
   8453: "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913",
   // USDC on Base
-  42161: "0xaf88d065e77c8cc2239327c5edb3a432268e5831",
+  42161: "0xaf88d065e77c8cc2239327c5edb3a432268e5831"
   // USDC on Arbitrum
-  9745: "0xB8CE59FC3717ada4C02eaDF9682A9e934F625ebb"
-  // USDT on Plasma
 };
 var ASSET_CONFIGS = {
   USDC: {
@@ -521,25 +530,23 @@ var ASSET_CONFIGS = {
     decimals: 6,
     tokenSymbols: ["USDC", "USDC.e", "USDT", "USDT0"],
     tokenSymbolsByChainId: {
+      1: "USDC",
       8453: "USDC",
       42161: "USDC",
-      9745: "USDT0",
       146: "USDC.e",
-      1: "USDC"
+      59144: "USDC"
     },
     addresses: {
+      1: "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48",
+      // Ethereum
       8453: "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913",
       // Base
       42161: "0xaf88d065e77c8cC2239327C5EDb3A432268e5831",
       // Arbitrum
-      9745: "0xb8ce59fc3717ada4c02eadf9682a9e934f625ebb",
-      // Plasma
       146: "0x29219dd400f2bf60e5a23d13be72b486d4038894",
       // Sonic
-      59144: "0x176211869ca2b568f2a7d4ee941e073a821ee1ff",
+      59144: "0x176211869ca2b568f2a7d4ee941e073a821ee1ff"
       // Linea
-      1: "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48"
-      // Ethereum
     },
     enabled: true
   },
@@ -551,26 +558,23 @@ var ASSET_CONFIGS = {
     decimals: 18,
     tokenSymbols: ["WETH", "ETH"],
     tokenSymbolsByChainId: {
+      1: "WETH",
       8453: "WETH",
       42161: "WETH",
-      9745: "WETH",
       146: "WETH",
-      59144: "WETH",
-      1: "WETH"
+      59144: "WETH"
     },
     addresses: {
+      1: "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
+      // Ethereum
       8453: "0x4200000000000000000000000000000000000006",
       // Base
       42161: "0x82aF49447D8a07e3bd95BD0d56f35241523fBab1",
       // Arbitrum
-      9745: "0x4200000000000000000000000000000000000006",
-      // Plasma
       146: "0x039e64f90d4199560e7533692f69448878db85c7",
       // Sonic
-      59144: "0xe5d7c2a44ffddf6b295a15c148167daaaf5cf34f",
+      59144: "0xe5d7c2a44ffddf6b295a15c148167daaaf5cf34f"
       // Linea
-      1: "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"
-      // Ethereum
     },
     enabled: true
   }
@@ -585,14 +589,14 @@ var getDefaultTokenAddress = (chainId, asset) => {
   return address;
 };
 var DEFAULT_RPC_URLS = {
+  1: "https://mainnet.infura.io/v3/8e6cdd06e30d40ac9990bf61bed3a3d0",
   8453: "https://mainnet.base.org",
-  42161: "https://arb1.arbitrum.io/rpc",
-  9745: "https://rpc.plasma.to"
+  42161: "https://arb1.arbitrum.io/rpc"
 };
 var CHAINS = {
+  1: import_chains.mainnet,
   8453: import_chains.base,
-  42161: import_chains.arbitrum,
-  9745: plasma
+  42161: import_chains.arbitrum
 };
 var getChainConfig = (chainId, rpcUrls) => {
   const chain = CHAINS[chainId];
@@ -620,7 +624,7 @@ var getSupportedChainIds = () => {
 // src/utils/safe-account.ts
 var import_module_sdk = require("@rhinestone/module-sdk");
 var import_accounts = require("permissionless/accounts");
-var import_viem4 = require("viem");
+var import_viem3 = require("viem");
 var import_account_abstraction = require("viem/account-abstraction");
 var SAFE_7579_ADDRESS = "0x7579EE8307284F293B1927136486880611F20002";
 var ERC7579_LAUNCHPAD_ADDRESS = "0x7579011aB74c46090561ea277Ba79D510c6C00ff";
@@ -630,12 +634,12 @@ var getSafeAccount = async (config) => {
   if (!safeOwnerAddress) {
     throw new Error("Safe owner address is required");
   }
-  const formattedOwnerAddress = (0, import_viem4.getAddress)(safeOwnerAddress);
+  const formattedOwnerAddress = (0, import_viem3.getAddress)(safeOwnerAddress);
   const ownableValidator = (0, import_module_sdk.getOwnableValidator)({
     owners: [formattedOwnerAddress],
     threshold: 1
   });
-  const saltHex = (0, import_viem4.fromHex)((0, import_viem4.toHex)(ACCOUNT_SALT), "bigint");
+  const saltHex = (0, import_viem3.fromHex)((0, import_viem3.toHex)(ACCOUNT_SALT), "bigint");
   const tempOwner = {
     address: formattedOwnerAddress,
     type: "json-rpc"
@@ -939,7 +943,7 @@ var _ZyfaiSDK = class _ZyfaiSDK {
         return;
       }
       const walletClient = this.getWalletClient();
-      const userAddress = (0, import_viem5.getAddress)(walletClient.account.address);
+      const userAddress = (0, import_viem4.getAddress)(walletClient.account.address);
       const chainId = this.currentChainId || walletClient.chain?.id || 8453;
       const challengeResponse = await this.httpClient.post(ENDPOINTS.AUTH_CHALLENGE, {});
       let uri;
@@ -1293,10 +1297,10 @@ var _ZyfaiSDK = class _ZyfaiSDK {
         this.walletClient.chain?.id || 8453,
         this.rpcUrls
       );
-      this.walletClient = (0, import_viem5.createWalletClient)({
+      this.walletClient = (0, import_viem4.createWalletClient)({
         account: newAddress,
         chain: chainConfig.chain,
-        transport: (0, import_viem5.custom)(this.currentProvider)
+        transport: (0, import_viem4.custom)(this.currentProvider)
       });
       this.currentChainId = this.walletClient.chain?.id || null;
       try {
@@ -1348,10 +1352,10 @@ var _ZyfaiSDK = class _ZyfaiSDK {
       }
       this.signer = (0, import_accounts2.privateKeyToAccount)(privateKey);
       this.currentChainId = chainId;
-      this.walletClient = (0, import_viem5.createWalletClient)({
+      this.walletClient = (0, import_viem4.createWalletClient)({
         account: this.signer,
         chain: chainConfig.chain,
-        transport: (0, import_viem5.http)(chainConfig.rpcUrl)
+        transport: (0, import_viem4.http)(chainConfig.rpcUrl)
       });
       connectedAddress = this.signer.address;
       this.currentProvider = null;
@@ -1369,10 +1373,10 @@ var _ZyfaiSDK = class _ZyfaiSDK {
         if (!accounts || accounts.length === 0) {
           throw new Error("No accounts found in wallet provider");
         }
-        this.walletClient = (0, import_viem5.createWalletClient)({
+        this.walletClient = (0, import_viem4.createWalletClient)({
           account: accounts[0],
           chain: chainConfig.chain,
-          transport: (0, import_viem5.custom)(provider)
+          transport: (0, import_viem4.custom)(provider)
         });
         connectedAddress = accounts[0];
         this.currentProvider = provider;
@@ -1381,7 +1385,7 @@ var _ZyfaiSDK = class _ZyfaiSDK {
           provider.on("accountsChanged", this.handleAccountsChanged.bind(this));
         }
       } else if (provider.account && provider.transport) {
-        this.walletClient = (0, import_viem5.createWalletClient)({
+        this.walletClient = (0, import_viem4.createWalletClient)({
           account: provider.account,
           chain: chainConfig.chain,
           transport: provider.transport
@@ -1434,10 +1438,10 @@ var _ZyfaiSDK = class _ZyfaiSDK {
         targetChainId,
         this.rpcUrls
       );
-      return (0, import_viem5.createWalletClient)({
+      return (0, import_viem4.createWalletClient)({
         account: this.signer,
         chain: targetChainConfig.chain,
-        transport: (0, import_viem5.http)(targetChainConfig.rpcUrl)
+        transport: (0, import_viem4.http)(targetChainConfig.rpcUrl)
       });
     } else {
       if (!this.walletClient) {
@@ -1800,22 +1804,30 @@ var _ZyfaiSDK = class _ZyfaiSDK {
    * Deposit funds from EOA to Safe smart wallet
    * Transfers tokens from the connected wallet to the user's Safe and logs the deposit
    *
-   * Token is automatically selected based on chain:
-   * - Base (8453) and Arbitrum (42161): USDC
-   * - Plasma (9745): USDT
+   * Token is automatically selected based on chain (USDC by default):
+   * - Ethereum Mainnet (1), Base (8453), Arbitrum (42161)
+   *
+   * Minimum portfolio balance enforced (Safe balance + deposit amount),
+   * configured per-chain in `MIN_PORTFOLIO_BALANCE`:
+   * - Ethereum Mainnet (1): 5 USDC (test threshold)
+   * - Base (8453) & Arbitrum (42161): no minimum
+   *
+   * If no minimum is configured for a given (chainId, asset) pair,
+   * the check is skipped.
    *
    * @param userAddress - User's address (owner of the Safe)
    * @param chainId - Target chain ID
    * @param amount - Amount in least decimal units (e.g., "100000000" for 100 USDC with 6 decimals)
+   * @param asset - Optional asset symbol ("USDC" or "WETH"). Defaults to "USDC".
    * @returns Deposit response with transaction hash
    *
    * @example
    * ```typescript
-   * // Deposit 100 USDC (6 decimals) to Safe on Base
+   * // Deposit 10,000 USDC (6 decimals) to Safe on Base
    * const result = await sdk.depositFunds(
    *   "0xUser...",
    *   8453,
-   *   "100000000" // 100 USDC = 100 * 10^6
+   *   "10000000000" // 10,000 USDC = 10_000 * 10^6
    * );
    * ```
    */
@@ -1830,7 +1842,15 @@ var _ZyfaiSDK = class _ZyfaiSDK {
       if (!amount || isNaN(Number(amount)) || Number(amount) <= 0) {
         throw new Error("Valid amount is required");
       }
-      const token = getDefaultTokenAddress(chainId, asset);
+      const assetSymbol = (asset || "USDC").toUpperCase();
+      const assetConfig = ASSET_CONFIGS[assetSymbol];
+      if (!assetConfig) {
+        throw new Error(
+          `Unsupported asset: ${assetSymbol}. Supported: USDC, WETH.`
+        );
+      }
+      const minRequired = MIN_PORTFOLIO_BALANCE[chainId]?.[assetSymbol];
+      const token = getDefaultTokenAddress(chainId, assetSymbol);
       const walletClient = this.getWalletClient();
       const chainConfig = getChainConfig(chainId, this.rpcUrls);
       const safeAddress = await getDeterministicSafeAddress({
@@ -1848,6 +1868,25 @@ var _ZyfaiSDK = class _ZyfaiSDK {
         );
       }
       const amountBigInt = BigInt(amount);
+      if (minRequired !== void 0) {
+        const currentSafeBalance = await chainConfig.publicClient.readContract({
+          address: token,
+          abi: ERC20_ABI,
+          functionName: "balanceOf",
+          args: [safeAddress]
+        });
+        const totalAfterDeposit = currentSafeBalance + amountBigInt;
+        if (totalAfterDeposit < minRequired) {
+          const minLabel = formatMinPortfolioLabel(
+            minRequired,
+            assetConfig.decimals,
+            assetSymbol
+          );
+          throw new Error(
+            `Minimum portfolio balance not met on chain ${chainId}. Total portfolio must be at least ${minLabel}. Current Safe balance: ${currentSafeBalance.toString()}, deposit amount: ${amountBigInt.toString()}, total after deposit: ${totalAfterDeposit.toString()} (raw units, ${assetConfig.decimals} decimals).`
+          );
+        }
+      }
       const txHash = await walletClient.writeContract({
         address: token,
         abi: ERC20_ABI,
@@ -1895,8 +1934,7 @@ var _ZyfaiSDK = class _ZyfaiSDK {
    * - Need more control over transaction execution
    *
    * Token is automatically selected based on chain if not provided:
-   * - Base (8453) and Arbitrum (42161): USDC
-   * - Plasma (9745): USDT
+   * - Ethereum Mainnet (1), Base (8453), Arbitrum (42161): USDC by default
    *
    * @param chainId - Chain ID where the deposit was made
    * @param txHash - Transaction hash of the deposit
@@ -3073,6 +3111,54 @@ var _ZyfaiSDK = class _ZyfaiSDK {
       );
     }
   }
+  // ============================================================================
+  // Simulation
+  // ============================================================================
+  /**
+   * Simulate the best yield positions for a given amount, splitting it across
+   * the top-ranked pools (up to minSplit positions) for the requested chains/strategy.
+   *
+   * Returns ready-to-execute calldata (approve + deposit) per position. The deposit
+   * calldata contains a "<RECEIVER>" placeholder that must be replaced with the
+   * actual receiving address (e.g. the user's Safe) before sending the transaction.
+   *
+   * @param params - Simulation parameters
+   * @returns Chain-keyed map of simulated positions with calldata
+   *
+   * @example
+   * ```typescript
+   * const result = await sdk.simulateBestPositions({
+   *   amount: 3000,
+   *   token: "USDC",
+   *   networks: 8453,
+   *   strategy: "conservative",
+   *   minSplit: 3,
+   *   protocols: ["aave", "compound", "morpho"],
+   * });
+   * console.log(result.data["8453"]);
+   * ```
+   */
+  async simulateBestPositions(params) {
+    try {
+      if (!isValidPublicStrategy(params.strategy)) {
+        throw new Error(
+          `Invalid strategy: ${params.strategy}. Must be "conservative" or "aggressive".`
+        );
+      }
+      const internalStrategy = toInternalStrategy(params.strategy);
+      const response = await this.httpClient.get(
+        ENDPOINTS.SIMULATE_BEST_POSITIONS({
+          ...params,
+          strategy: internalStrategy
+        })
+      );
+      return response;
+    } catch (error) {
+      throw new Error(
+        `Failed to simulate best positions: ${error.message}`
+      );
+    }
+  }
   /**
    * Get currently selected pools for a protocol on a specific chain.
    *
@@ -3149,7 +3235,7 @@ var _ZyfaiSDK = class _ZyfaiSDK {
       if (!tokenUriResponse.tokenUri) {
         throw new Error("API did not return a tokenUri");
       }
-      const callData = (0, import_viem5.encodeFunctionData)({
+      const callData = (0, import_viem4.encodeFunctionData)({
         abi: IDENTITY_REGISTRY_ABI,
         functionName: "register",
         args: [tokenUriResponse.tokenUri]
@@ -3515,6 +3601,78 @@ var _ZyfaiSDK = class _ZyfaiSDK {
       symbol: tokenSymbol
     };
   }
+  // ============================================================================
+  // WebSocket Event Stream
+  // ============================================================================
+  /**
+   * Subscribe to real-time Zyfai risk and liquidity events via WebSocket.
+   *
+   * Events: depeg, liquidity_trap, liquidity_restored, pool_status_change,
+   * new_collateral_detected, liquidity_drop.
+   *
+   * @param handlers - Callback per event type, plus onError
+   * @param filters - Optional protocol/pool filter sent to the server on subscribe
+   * @returns Cleanup function — call it to close the connection
+   *
+   * @example
+   * ```typescript
+   * const unsubscribe = sdk.subscribeToEvents(
+   *   {
+   *     onDepeg: (data) => console.log("Depeg detected:", data.token, data.severity),
+   *     onLiquidityDrop: (data) => console.log("Liquidity drop:", data.pool, data.dropPercent),
+   *   },
+   *   { protocols: ["morpho"], pools: ["gauntlet usdc core"] }
+   * );
+   *
+   * // Later, close the connection:
+   * unsubscribe();
+   * ```
+   */
+  subscribeToEvents(handlers, filters) {
+    let ws = null;
+    let closed = false;
+    const connect = () => {
+      ws = new globalThis.WebSocket(WS_ENDPOINT);
+      ws.onopen = () => {
+        const msg = { type: "subscribe" };
+        if (filters && (filters.chains?.length || filters.protocols?.length || filters.pools?.length)) {
+          msg.filters = filters;
+        }
+        ws.send(JSON.stringify(msg));
+      };
+      ws.onmessage = (event) => {
+        try {
+          const msg = JSON.parse(event.data);
+          if (msg.type !== "event") return;
+          switch (msg.eventType) {
+            case "depeg":
+              handlers.onDepeg?.(msg.data);
+              break;
+            case "new_collateral_detected":
+              handlers.onNewCollateralDetected?.(msg.data);
+              break;
+            case "liquidity_drop":
+              handlers.onLiquidityDrop?.(msg.data);
+              break;
+          }
+        } catch {
+        }
+      };
+      ws.onerror = (error) => {
+        handlers.onError?.(error);
+      };
+      ws.onclose = () => {
+        if (!closed) {
+          setTimeout(connect, 3e3);
+        }
+      };
+    };
+    connect();
+    return () => {
+      closed = true;
+      ws?.close();
+    };
+  }
 };
 // ============================================================================
 // Agent Identity Registry