@zubari/sdk 0.1.8 → 0.1.10

package/dist/react/index.mjs

@@ -1111,7 +1111,8 @@ async function getPriceForChain(chain) {
 }
 var STORAGE_KEYS = {
   ENCRYPTED_SEED: "encrypted_seed",
-  ACTIVE_WALLET: "active_wallet"
+  ACTIVE_WALLET: "active_wallet",
+  DERIVED_ADDRESSES: "derived_addresses"
 };
 var SUPPORTED_CHAINS = ["ethereum", "bitcoin", "ton", "tron", "solana", "spark"];
 var WalletManager = class _WalletManager {
@@ -1252,6 +1253,7 @@ var WalletManager = class _WalletManager {
    */
   async deleteWallet() {
     await this.storage.removeItem(STORAGE_KEYS.ENCRYPTED_SEED);
+    await this.storage.removeItem(STORAGE_KEYS.DERIVED_ADDRESSES);
     this.currentSeed = null;
     this.derivedAddress = null;
     this.derivedAddresses = {};
@@ -1465,6 +1467,7 @@ var WalletManager = class _WalletManager {
    * In browser: Uses the backend API (which has Tether WDK)
    * In Node.js: Uses native WDK directly
    * Returns REAL cryptographically valid addresses for all chains.
+   * Addresses are persisted to storage for future use.
    */
   async deriveAllAddressesWithWdk() {
     if (!this.currentSeed) {
@@ -1480,22 +1483,63 @@ var WalletManager = class _WalletManager {
         }
       }
       this.derivedAddresses = addresses;
+      await this.saveAddressesToStorage(addresses);
       return addresses;
     } catch (error) {
       console.error("WDK derivation failed:", error);
       throw error;
     }
   }
+  /**
+   * Save derived addresses to storage
+   */
+  async saveAddressesToStorage(addresses) {
+    try {
+      await this.storage.setItem(STORAGE_KEYS.DERIVED_ADDRESSES, JSON.stringify(addresses));
+      console.log("Saved derived addresses to storage:", Object.keys(addresses));
+    } catch (error) {
+      console.warn("Failed to save addresses to storage:", error);
+    }
+  }
+  /**
+   * Load derived addresses from storage
+   */
+  async loadAddressesFromStorage() {
+    try {
+      const stored = await this.storage.getItem(STORAGE_KEYS.DERIVED_ADDRESSES);
+      if (stored) {
+        const addresses = JSON.parse(stored);
+        console.log("Loaded derived addresses from storage:", Object.keys(addresses));
+        return addresses;
+      }
+    } catch (error) {
+      console.warn("Failed to load addresses from storage:", error);
+    }
+    return null;
+  }
   /**
    * Derive addresses for all enabled chains using Tether WDK
    * The unified WDK service handles all fallback strategies automatically:
    * - Browser: API backend -> Browser derivation
    * - Node.js: Native WDK -> API backend -> Browser derivation
+   *
+   * First tries to load from storage to avoid losing real WDK-derived addresses.
    */
   async deriveAllAddressesAsync() {
     if (!this.currentSeed) {
       throw new Error("Wallet is locked");
     }
+    const storedAddresses = await this.loadAddressesFromStorage();
+    if (storedAddresses && Object.keys(storedAddresses).length > 1) {
+      const expectedEthAddress = _WalletManager.deriveAddress(this.currentSeed);
+      if (storedAddresses.ethereum === expectedEthAddress) {
+        console.log("Using addresses from storage (verified by Ethereum address)");
+        this.derivedAddresses = storedAddresses;
+        return storedAddresses;
+      } else {
+        console.log("Stored addresses do not match current seed, re-deriving...");
+      }
+    }
     try {
       return await this.deriveAllAddressesWithWdk();
     } catch (error) {
@@ -1785,6 +1829,12 @@ function useWalletManager(options = {}) {
           manager.deriveAllAddresses();
         }
         updateState();
+        try {
+          const balances = await manager.fetchAllBalances();
+          setChainBalances(balances);
+        } catch (err) {
+          console.warn("Failed to fetch balances after create:", err);
+        }
         return result;
       } catch (err) {
         const message = err instanceof Error ? err.message : "Failed to create wallet";
@@ -1809,6 +1859,12 @@ function useWalletManager(options = {}) {
           manager.deriveAllAddresses();
         }
         updateState();
+        try {
+          const balances = await manager.fetchAllBalances();
+          setChainBalances(balances);
+        } catch (err) {
+          console.warn("Failed to fetch balances after import:", err);
+        }
       } catch (err) {
         const message = err instanceof Error ? err.message : "Failed to import wallet";
         setError(message);
@@ -1832,6 +1888,12 @@ function useWalletManager(options = {}) {
           manager.deriveAllAddresses();
         }
         updateState();
+        try {
+          const balances = await manager.fetchAllBalances();
+          setChainBalances(balances);
+        } catch (err) {
+          console.warn("Failed to fetch balances after unlock:", err);
+        }
       } catch (err) {
         const message = err instanceof Error ? err.message : "Invalid password";
         setError(message);