@owney/sdk 0.7.5 → 0.7.6

package/dist/index.cjs

@@ -825,6 +825,9 @@ function debugLog(scope, message, data) {
 }
 
 // src/agents/zyfai/zyfai.agent.ts
+var ERC7579_IS_MODULE_INSTALLED_ABI = (0, import_viem.parseAbi)([
+  "function isModuleInstalled(uint256 moduleTypeId, address module, bytes additionalContext) view returns (bool)"
+]);
 var DEFAULT_ZYFAI_RPC_URLS = {
   8453: "https://base-mainnet.g.alchemy.com/v2/ZWyVU-9XfS3z8Rn-xkq7V",
   42161: "https://arb-mainnet.g.alchemy.com/v2/ZWyVU-9XfS3z8Rn-xkq7V",
@@ -1142,7 +1145,12 @@ var ZyfaiAgent = class {
       const validChainId = isValidChainId(chainId);
       await Promise.all([
         this.ensureProtocolsSynced(validChainId, "USDC"),
-        this.ensureProtocolsSynced(validChainId, "WETH")
+        this.ensureProtocolsSynced(validChainId, "WETH"),
+        // Retrofit a session key for old wallets that are already deployed and
+        // funded but never got one — otherwise their balance stays idle until
+        // they happen to deposit again. Only touches Safes that are missing the
+        // validator, so pool-wallet users see no prompt. Non-fatal.
+        this.ensureSessionKeyIfDeployed(validChainId)
       ]);
     } catch (error) {
       throw error;
@@ -1340,6 +1348,65 @@ var ZyfaiAgent = class {
       return false;
     }
   }
+  /**
+   * Ensure Zyfai's automation can actually act on the user's Safe for this
+   * chain, by making sure a session key exists.
+   *
+   * Old wallets (pre-0.2.44) are deployed WITHOUT the smart-sessions validator,
+   * so Zyfai's rebalancer has no authorised path to move funds: deposits land
+   * and then sit as raw tokens forever. Pre-deployed pool wallets ship with the
+   * module already installed, which is why they never needed this.
+   *
+   * The gate is the on-chain module, not the wallet's class: it is the exact
+   * condition that matters, it needs no guessing from API shapes, and it means
+   * pool-wallet users are never prompted for a signature they don't need.
+   *
+   * Non-fatal by design. A failure here must not block a deposit — the funds
+   * still arrive safely in the Safe, they just won't be deployed into a pool
+   * until a session key exists.
+   */
+  /**
+   * Activation-time variant: only acts when the Safe already has code on this
+   * chain. Activation must never deploy anything — that is deposit's job, and it
+   * would mean surprise on-chain work just for opening the app.
+   */
+  async ensureSessionKeyIfDeployed(chainId) {
+    try {
+      const ownerAddress = this.getAddress();
+      const wallet = await this.sdk.getSmartWalletAddress(
+        ownerAddress,
+        chainId
+      );
+      if (!wallet.isDeployed) return;
+      await this.ensureSessionKey(ownerAddress, chainId, wallet.address);
+    } catch (error) {
+      console.warn(
+        "[zyfai] session-key check at activation failed (non-fatal):",
+        error instanceof Error ? error.message : String(error)
+      );
+    }
+  }
+  async ensureSessionKey(ownerAddress, chainId, smartWallet) {
+    try {
+      const installed = await this.getPublicClient(chainId).readContract({
+        address: smartWallet,
+        abi: ERC7579_IS_MODULE_INSTALLED_ABI,
+        functionName: "isModuleInstalled",
+        // moduleTypeId 1 = validator (ERC-7579)
+        args: [1n, import_sdk.SMART_SESSIONS_VALIDATOR, "0x"]
+      });
+      if (installed) return;
+      console.warn(
+        `[zyfai] Safe ${smartWallet} has no smart-sessions validator on chain ${chainId}; creating a session key so the agent can allocate.`
+      );
+      await this.sdk.createSessionKey(ownerAddress, chainId);
+    } catch (error) {
+      console.warn(
+        "[zyfai] createSessionKey skipped (non-fatal):",
+        error instanceof Error ? error.message : String(error)
+      );
+    }
+  }
   /**
    * Guarantee the user's Safe has code on `chainId` before anything transfers
    * funds into it, and return its address.
@@ -1360,17 +1427,21 @@ var ZyfaiAgent = class {
    */
   async ensureWalletDeployed(ownerAddress, chainId) {
     const wallet = await this.sdk.getSmartWalletAddress(ownerAddress, chainId);
-    if (wallet.isDeployed) return wallet.address;
+    if (wallet.isDeployed) {
+      await this.ensureSessionKey(ownerAddress, chainId, wallet.address);
+      return wallet.address;
+    }
     const poolFirst = await this.isPredeployedPoolWallet(
       ownerAddress,
       wallet.address
     );
     const deployAsPool = () => this.sdk.deployOnChains([chainId]);
     const deployAsLegacy = () => (
-      // `createSessionKey: false` — session keys are backend-owned since the
-      // predeployed-pool rollout; asking for one here costs a signature prompt
-      // and is rejected anyway.
-      this.sdk.deploySafe(ownerAddress, chainId, "aggressive", false)
+      // `createSessionKey: true` — per Zyfai, the old-wallet flow is
+      // deploySafe(true) -> createSessionKey() -> deposit. Passing false (which
+      // this did originally) deploys a Safe with no smart-sessions validator, so
+      // deposits land and are never allocated into a pool.
+      this.sdk.deploySafe(ownerAddress, chainId, "aggressive", true)
     );
     const [first, second] = poolFirst ? [deployAsPool, deployAsLegacy] : [deployAsLegacy, deployAsPool];
     try {
@@ -1394,6 +1465,7 @@ var ZyfaiAgent = class {
         this.id
       );
     }
+    await this.ensureSessionKey(ownerAddress, chainId, redeployed.address);
     return redeployed.address;
   }
   async deposit(state, chainId, amount, asset, depositCallback) {

package/dist/index.js

@@ -1,6 +1,6 @@
 // src/agents/zyfai/zyfai.agent.ts
-import { ZyfaiSDK } from "@zyfai/sdk";
-import { createPublicClient, http } from "viem";
+import { SMART_SESSIONS_VALIDATOR, ZyfaiSDK } from "@zyfai/sdk";
+import { createPublicClient, http, parseAbi } from "viem";
 import { base, arbitrum, mainnet } from "viem/chains";
 
 // src/types/config.ts
@@ -792,6 +792,9 @@ function debugLog(scope, message, data) {
 }
 
 // src/agents/zyfai/zyfai.agent.ts
+var ERC7579_IS_MODULE_INSTALLED_ABI = parseAbi([
+  "function isModuleInstalled(uint256 moduleTypeId, address module, bytes additionalContext) view returns (bool)"
+]);
 var DEFAULT_ZYFAI_RPC_URLS = {
   8453: "https://base-mainnet.g.alchemy.com/v2/ZWyVU-9XfS3z8Rn-xkq7V",
   42161: "https://arb-mainnet.g.alchemy.com/v2/ZWyVU-9XfS3z8Rn-xkq7V",
@@ -1109,7 +1112,12 @@ var ZyfaiAgent = class {
       const validChainId = isValidChainId(chainId);
       await Promise.all([
         this.ensureProtocolsSynced(validChainId, "USDC"),
-        this.ensureProtocolsSynced(validChainId, "WETH")
+        this.ensureProtocolsSynced(validChainId, "WETH"),
+        // Retrofit a session key for old wallets that are already deployed and
+        // funded but never got one — otherwise their balance stays idle until
+        // they happen to deposit again. Only touches Safes that are missing the
+        // validator, so pool-wallet users see no prompt. Non-fatal.
+        this.ensureSessionKeyIfDeployed(validChainId)
       ]);
     } catch (error) {
       throw error;
@@ -1307,6 +1315,65 @@ var ZyfaiAgent = class {
       return false;
     }
   }
+  /**
+   * Ensure Zyfai's automation can actually act on the user's Safe for this
+   * chain, by making sure a session key exists.
+   *
+   * Old wallets (pre-0.2.44) are deployed WITHOUT the smart-sessions validator,
+   * so Zyfai's rebalancer has no authorised path to move funds: deposits land
+   * and then sit as raw tokens forever. Pre-deployed pool wallets ship with the
+   * module already installed, which is why they never needed this.
+   *
+   * The gate is the on-chain module, not the wallet's class: it is the exact
+   * condition that matters, it needs no guessing from API shapes, and it means
+   * pool-wallet users are never prompted for a signature they don't need.
+   *
+   * Non-fatal by design. A failure here must not block a deposit — the funds
+   * still arrive safely in the Safe, they just won't be deployed into a pool
+   * until a session key exists.
+   */
+  /**
+   * Activation-time variant: only acts when the Safe already has code on this
+   * chain. Activation must never deploy anything — that is deposit's job, and it
+   * would mean surprise on-chain work just for opening the app.
+   */
+  async ensureSessionKeyIfDeployed(chainId) {
+    try {
+      const ownerAddress = this.getAddress();
+      const wallet = await this.sdk.getSmartWalletAddress(
+        ownerAddress,
+        chainId
+      );
+      if (!wallet.isDeployed) return;
+      await this.ensureSessionKey(ownerAddress, chainId, wallet.address);
+    } catch (error) {
+      console.warn(
+        "[zyfai] session-key check at activation failed (non-fatal):",
+        error instanceof Error ? error.message : String(error)
+      );
+    }
+  }
+  async ensureSessionKey(ownerAddress, chainId, smartWallet) {
+    try {
+      const installed = await this.getPublicClient(chainId).readContract({
+        address: smartWallet,
+        abi: ERC7579_IS_MODULE_INSTALLED_ABI,
+        functionName: "isModuleInstalled",
+        // moduleTypeId 1 = validator (ERC-7579)
+        args: [1n, SMART_SESSIONS_VALIDATOR, "0x"]
+      });
+      if (installed) return;
+      console.warn(
+        `[zyfai] Safe ${smartWallet} has no smart-sessions validator on chain ${chainId}; creating a session key so the agent can allocate.`
+      );
+      await this.sdk.createSessionKey(ownerAddress, chainId);
+    } catch (error) {
+      console.warn(
+        "[zyfai] createSessionKey skipped (non-fatal):",
+        error instanceof Error ? error.message : String(error)
+      );
+    }
+  }
   /**
    * Guarantee the user's Safe has code on `chainId` before anything transfers
    * funds into it, and return its address.
@@ -1327,17 +1394,21 @@ var ZyfaiAgent = class {
    */
   async ensureWalletDeployed(ownerAddress, chainId) {
     const wallet = await this.sdk.getSmartWalletAddress(ownerAddress, chainId);
-    if (wallet.isDeployed) return wallet.address;
+    if (wallet.isDeployed) {
+      await this.ensureSessionKey(ownerAddress, chainId, wallet.address);
+      return wallet.address;
+    }
     const poolFirst = await this.isPredeployedPoolWallet(
       ownerAddress,
       wallet.address
     );
     const deployAsPool = () => this.sdk.deployOnChains([chainId]);
     const deployAsLegacy = () => (
-      // `createSessionKey: false` — session keys are backend-owned since the
-      // predeployed-pool rollout; asking for one here costs a signature prompt
-      // and is rejected anyway.
-      this.sdk.deploySafe(ownerAddress, chainId, "aggressive", false)
+      // `createSessionKey: true` — per Zyfai, the old-wallet flow is
+      // deploySafe(true) -> createSessionKey() -> deposit. Passing false (which
+      // this did originally) deploys a Safe with no smart-sessions validator, so
+      // deposits land and are never allocated into a pool.
+      this.sdk.deploySafe(ownerAddress, chainId, "aggressive", true)
     );
     const [first, second] = poolFirst ? [deployAsPool, deployAsLegacy] : [deployAsLegacy, deployAsPool];
     try {
@@ -1361,6 +1432,7 @@ var ZyfaiAgent = class {
         this.id
       );
     }
+    await this.ensureSessionKey(ownerAddress, chainId, redeployed.address);
     return redeployed.address;
   }
   async deposit(state, chainId, amount, asset, depositCallback) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@owney/sdk",
-  "version": "0.7.5",
+  "version": "0.7.6",
   "type": "module",
   "main": "dist/index.cjs",
   "module": "dist/index.js",