@zyfai/sdk 0.1.7 → 0.1.8

package/dist/index.d.mts

@@ -437,6 +437,21 @@ declare class ZyfaiSDK {
      * ```
      */
     updateUserProfile(request: UpdateUserProfileRequest): Promise<UpdateUserProfileResponse>;
+    /**
+     * Initialize user after Safe deployment
+     * This method is automatically called after deploySafe to initialize user state
+     *
+     * @param smartWallet - Safe smart wallet address
+     * @param chainId - Target chain ID
+     * @returns Initialization response
+     *
+     * @example
+     * ```typescript
+     * await sdk.initializeUser("0x1396730...", 8453);
+     * ```
+     * @internal
+     */
+    private initializeUser;
     /**
      * Handle account changes from wallet provider
      * Resets authentication state when wallet is switched
@@ -515,6 +530,14 @@ declare class ZyfaiSDK {
      * @private
      */
     private signSessionKey;
+    /**
+     * Update user protocols with available protocols from the chain
+     * This method is automatically called before activating session key
+     *
+     * @param chainId - Target chain ID
+     * @internal
+     */
+    private updateUserProtocols;
     /**
      * Activate session key via ZyFAI API
      */

package/dist/index.d.ts

@@ -437,6 +437,21 @@ declare class ZyfaiSDK {
      * ```
      */
     updateUserProfile(request: UpdateUserProfileRequest): Promise<UpdateUserProfileResponse>;
+    /**
+     * Initialize user after Safe deployment
+     * This method is automatically called after deploySafe to initialize user state
+     *
+     * @param smartWallet - Safe smart wallet address
+     * @param chainId - Target chain ID
+     * @returns Initialization response
+     *
+     * @example
+     * ```typescript
+     * await sdk.initializeUser("0x1396730...", 8453);
+     * ```
+     * @internal
+     */
+    private initializeUser;
     /**
      * Handle account changes from wallet provider
      * Resets authentication state when wallet is switched
@@ -515,6 +530,14 @@ declare class ZyfaiSDK {
      * @private
      */
     private signSessionKey;
+    /**
+     * Update user protocols with available protocols from the chain
+     * This method is automatically called before activating session key
+     *
+     * @param chainId - Target chain ID
+     * @internal
+     */
+    private updateUserProtocols;
     /**
      * Activate session key via ZyFAI API
      */

package/dist/index.js

@@ -57,6 +57,7 @@ var ENDPOINTS = {
   AUTH_CHALLENGE: "/auth/challenge",
   // User
   USER_ME: "/users/me",
+  USER_INITIALIZE: "/users/initialize",
   USER_WITHDRAW: "/users/withdraw",
   PARTIAL_WITHDRAW: "/users/partial-withdraw",
   // Session Keys
@@ -733,6 +734,43 @@ var ZyfaiSDK = class {
       );
     }
   }
+  /**
+   * Initialize user after Safe deployment
+   * This method is automatically called after deploySafe to initialize user state
+   *
+   * @param smartWallet - Safe smart wallet address
+   * @param chainId - Target chain ID
+   * @returns Initialization response
+   *
+   * @example
+   * ```typescript
+   * await sdk.initializeUser("0x1396730...", 8453);
+   * ```
+   * @internal
+   */
+  async initializeUser(smartWallet, chainId) {
+    try {
+      await this.authenticateUser();
+      const response = await this.httpClient.post(
+        ENDPOINTS.USER_INITIALIZE,
+        {
+          smartWallet,
+          chainId
+        }
+      );
+      return {
+        success: true,
+        userId: response.userId || response.id,
+        smartWallet: response.smartWallet,
+        chainId: response.chainId,
+        message: response.message
+      };
+    } catch (error) {
+      throw new Error(
+        `Failed to initialize user: ${error.message}`
+      );
+    }
+  }
   /**
    * Handle account changes from wallet provider
    * Resets authentication state when wallet is switched
@@ -1001,6 +1039,14 @@ var ZyfaiSDK = class {
           updateError.message
         );
       }
+      try {
+        await this.initializeUser(deploymentResult.safeAddress, chainId);
+      } catch (initError) {
+        console.warn(
+          "Failed to initialize user after Safe deployment:",
+          initError.message
+        );
+      }
       return {
         success: true,
         safeAddress: deploymentResult.safeAddress,
@@ -1061,6 +1107,7 @@ var ZyfaiSDK = class {
       if (!signatureResult.signature) {
         throw new Error("Failed to obtain session key signature");
       }
+      await this.updateUserProtocols(chainId);
       const activation = await this.activateSessionKey(
         signatureResult.signature,
         signatureResult.sessionNonces
@@ -1136,6 +1183,30 @@ var ZyfaiSDK = class {
       );
     }
   }
+  /**
+   * Update user protocols with available protocols from the chain
+   * This method is automatically called before activating session key
+   *
+   * @param chainId - Target chain ID
+   * @internal
+   */
+  async updateUserProtocols(chainId) {
+    try {
+      const protocolsResponse = await this.getAvailableProtocols(chainId);
+      if (!protocolsResponse.protocols || protocolsResponse.protocols.length === 0) {
+        console.warn(`No protocols available for chain ${chainId}`);
+        return;
+      }
+      const protocolIds = protocolsResponse.protocols.map((p) => p.id);
+      await this.updateUserProfile({
+        protocols: protocolIds
+      });
+    } catch (error) {
+      console.warn(
+        `Failed to update user protocols: ${error.message}`
+      );
+    }
+  }
   /**
    * Activate session key via ZyFAI API
    */

package/dist/index.mjs

@@ -18,6 +18,7 @@ var ENDPOINTS = {
   AUTH_CHALLENGE: "/auth/challenge",
   // User
   USER_ME: "/users/me",
+  USER_INITIALIZE: "/users/initialize",
   USER_WITHDRAW: "/users/withdraw",
   PARTIAL_WITHDRAW: "/users/partial-withdraw",
   // Session Keys
@@ -712,6 +713,43 @@ var ZyfaiSDK = class {
       );
     }
   }
+  /**
+   * Initialize user after Safe deployment
+   * This method is automatically called after deploySafe to initialize user state
+   *
+   * @param smartWallet - Safe smart wallet address
+   * @param chainId - Target chain ID
+   * @returns Initialization response
+   *
+   * @example
+   * ```typescript
+   * await sdk.initializeUser("0x1396730...", 8453);
+   * ```
+   * @internal
+   */
+  async initializeUser(smartWallet, chainId) {
+    try {
+      await this.authenticateUser();
+      const response = await this.httpClient.post(
+        ENDPOINTS.USER_INITIALIZE,
+        {
+          smartWallet,
+          chainId
+        }
+      );
+      return {
+        success: true,
+        userId: response.userId || response.id,
+        smartWallet: response.smartWallet,
+        chainId: response.chainId,
+        message: response.message
+      };
+    } catch (error) {
+      throw new Error(
+        `Failed to initialize user: ${error.message}`
+      );
+    }
+  }
   /**
    * Handle account changes from wallet provider
    * Resets authentication state when wallet is switched
@@ -980,6 +1018,14 @@ var ZyfaiSDK = class {
           updateError.message
         );
       }
+      try {
+        await this.initializeUser(deploymentResult.safeAddress, chainId);
+      } catch (initError) {
+        console.warn(
+          "Failed to initialize user after Safe deployment:",
+          initError.message
+        );
+      }
       return {
         success: true,
         safeAddress: deploymentResult.safeAddress,
@@ -1040,6 +1086,7 @@ var ZyfaiSDK = class {
       if (!signatureResult.signature) {
         throw new Error("Failed to obtain session key signature");
       }
+      await this.updateUserProtocols(chainId);
       const activation = await this.activateSessionKey(
         signatureResult.signature,
         signatureResult.sessionNonces
@@ -1115,6 +1162,30 @@ var ZyfaiSDK = class {
       );
     }
   }
+  /**
+   * Update user protocols with available protocols from the chain
+   * This method is automatically called before activating session key
+   *
+   * @param chainId - Target chain ID
+   * @internal
+   */
+  async updateUserProtocols(chainId) {
+    try {
+      const protocolsResponse = await this.getAvailableProtocols(chainId);
+      if (!protocolsResponse.protocols || protocolsResponse.protocols.length === 0) {
+        console.warn(`No protocols available for chain ${chainId}`);
+        return;
+      }
+      const protocolIds = protocolsResponse.protocols.map((p) => p.id);
+      await this.updateUserProfile({
+        protocols: protocolIds
+      });
+    } catch (error) {
+      console.warn(
+        `Failed to update user protocols: ${error.message}`
+      );
+    }
+  }
   /**
    * Activate session key via ZyFAI API
    */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zyfai/sdk",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "TypeScript SDK for ZyFAI Yield Optimization Engine - Deploy Safe smart wallets, manage session keys, and interact with DeFi protocols",
   "main": "dist/index.js",
   "module": "dist/index.mjs",