@fairblock/stabletrust 1.0.11 → 1.0.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fairblock/stabletrust",
-  "version": "1.0.11",
+  "version": "1.0.13",
   "description": "SDK for confidential transfers using homomorphic encryption and zero-knowledge proofs",
   "main": "src/index.js",
   "types": "src/index.d.ts",

package/src/client.js

@@ -74,6 +74,9 @@ export class ConfidentialTransferClient {
     // WASM will be auto-initialized on first use
     this._wasmModule = null;
 
+    // Cache for derived keys — deterministic per wallet+chain+contract, safe to reuse
+    this._keyCache = new Map();
+
     try {
       this.provider = new ethers.JsonRpcProvider(this.config.rpcUrl);
       this.contract = new ethers.Contract(
@@ -118,8 +121,12 @@ export class ConfidentialTransferClient {
       if (!wallet) {
         throw new Error("Wallet is required");
       }
+      const address = await wallet.getAddress();
+      if (this._keyCache.has(address)) {
+        return this._keyCache.get(address);
+      }
       const wasm = await this._getWasm();
-      return await deriveKeys(
+      const keys = await deriveKeys(
         wallet,
         {
           chainId: this.config.chainId,
@@ -127,6 +134,8 @@ export class ConfidentialTransferClient {
         },
         wasm.generate_deterministic_keypair,
       );
+      this._keyCache.set(address, keys);
+      return keys;
     } catch (error) {
       throw new Error(`Failed to derive keys: ${error.message}`);
     }
@@ -155,11 +164,11 @@ export class ConfidentialTransferClient {
    * @param {ethers.Wallet|ethers.Signer} wallet - The wallet to create account for
    * @param {Object} [options] - Options
    * @param {boolean} [options.waitForFinalization=true] - Wait for account finalization
-   * @param {number} [options.maxAttempts=30] - Max attempts to wait for finalization
+   * @param {number} [options.maxAttempts=225] - Max attempts to wait for finalization
    * @returns {Promise<{publicKey: string, privateKey: string}>} The derived keys
    */
   async ensureAccount(wallet, options = {}) {
-    const { waitForFinalization = true, maxAttempts = 30 } = options;
+    const { waitForFinalization = true, maxAttempts = 225 } = options;
 
     try {
       const address = await wallet.getAddress();
@@ -183,7 +192,7 @@ export class ConfidentialTransferClient {
       if (waitForFinalization) {
         let attempts = 0;
         while (!accountInfo.finalized && attempts < maxAttempts) {
-          await sleep(2000);
+          await sleep(400);
           accountInfo = await this.getAccountInfo(address);
           attempts++;
         }
@@ -354,20 +363,17 @@ export class ConfidentialTransferClient {
       const depositAmount = BigInt(amount);
       const tokenContract = this._getTokenContract(tokenAddress);
 
-      // Check token balance
-      const tokenBalance = await tokenContract.balanceOf(address);
+      // Check token balance and allowance in parallel
+      const [tokenBalance, allowance] = await Promise.all([
+        tokenContract.balanceOf(address),
+        tokenContract.allowance(address, this.config.contractAddress),
+      ]);
       if (tokenBalance < depositAmount) {
         throw new Error(
           `Insufficient token balance. Required: ${depositAmount}, Available: ${tokenBalance}`,
         );
       }
 
-      // Check and approve if needed
-      const allowance = await tokenContract.allowance(
-        address,
-        this.config.contractAddress,
-      );
-
       if (allowance < depositAmount) {
         const approveTx = await tokenContract
           .connect(wallet)
@@ -439,14 +445,15 @@ export class ConfidentialTransferClient {
 
       const senderAddress = await senderWallet.getAddress();
 
-      // Auto-derive sender keys
-      const derivedSenderKeys = await this._deriveKeys(senderWallet);
+      // Derive sender keys and fetch recipient account info in parallel
+      const [derivedSenderKeys, recipientAccountInfo] = await Promise.all([
+        this._deriveKeys(senderWallet),
+        this.getAccountInfo(recipientAddress),
+      ]);
       if (!derivedSenderKeys?.privateKey) {
         throw new Error("Failed to derive sender keys");
       }
 
-      // Auto-derive recipient public key
-      const recipientAccountInfo = await this.getAccountInfo(recipientAddress);
       if (!recipientAccountInfo.exists) {
         throw new Error(
           `Recipient account does not exist. Address: ${recipientAddress}`,
@@ -474,12 +481,14 @@ export class ConfidentialTransferClient {
         "transfer",
       );
 
-      const balanceSummary = await this.getConfidentialBalance(
-        senderAddress,
-        derivedSenderKeys.privateKey,
-        tokenAddress,
-      );
-      const fee = await this.getFeeAmount();
+      const [balanceSummary, fee] = await Promise.all([
+        this.getConfidentialBalance(
+          senderAddress,
+          derivedSenderKeys.privateKey,
+          tokenAddress,
+        ),
+        this.getFeeAmount(),
+      ]);
       const derivedCurrentBalanceCiphertext =
         balanceSummary.available.ciphertext;
       const derivedCurrentBalance = balanceSummary.available.amount;
@@ -881,7 +890,8 @@ export class ConfidentialTransferClient {
    */
   async _waitForGlobalState(address, actionLabel) {
     let attempts = 0;
-    const maxAttempts = 300;
+    const maxAttempts = 450; // 450 * 200ms = 90 seconds max wait
+    await sleep(2000);
 
     while (attempts < maxAttempts) {
       try {
@@ -912,8 +922,8 @@ export class ConfidentialTransferClient {
    * @param {string} privateKey - Private key for decryption
    * @param {string} tokenAddress - Token address
    * @param {Object} [options] - Options
-   * @param {number} [options.maxAttempts=60] - Maximum polling attempts
-   * @param {number} [options.intervalMs=3000] - Polling interval in milliseconds
+   * @param {number} [options.maxAttempts=225] - Maximum polling attempts
+   * @param {number} [options.intervalMs=400] - Polling interval in milliseconds
    * @returns {Promise<{amount: number, ciphertext: string}>}
    */
   async _waitForPendingBalance(
@@ -922,7 +932,7 @@ export class ConfidentialTransferClient {
     tokenAddress,
     options = {},
   ) {
-    const { maxAttempts = 60, intervalMs = 3000 } = options;
+    const { maxAttempts = 225, intervalMs = 400 } = options;
 
     try {
       for (let i = 0; i < maxAttempts; i++) {