@talismn/balances 1.3.2 → 1.3.4

package/dist/index.mjs

@@ -38,7 +38,7 @@ import anylogger from "anylogger";
 // package.json
 var package_default = {
   name: "@talismn/balances",
-  version: "1.3.2",
+  version: "1.3.4",
   author: "Talisman",
   homepage: "https://talisman.xyz",
   license: "GPL-3.0-or-later",
@@ -81,7 +81,7 @@ var package_default = {
     "bignumber.js": "^9.1.2",
     "lodash-es": "4.17.21",
     "p-queue": "8.1.0",
-    "polkadot-api": "1.13.1",
+    "polkadot-api": "1.23.3",
     rxjs: "^7.8.1",
     "scale-ts": "^1.6.1",
     viem: "^2.27.3",
@@ -3413,12 +3413,15 @@ var Balances = class _Balances {
   //
   // Properties
   //
-  #balances = [];
+  #balancesMap = /* @__PURE__ */ new Map();
+  #cachedArray = null;
+  #cachedFilteredMirrorTokens = null;
+  #cachedSumFormatter = null;
   //
   // Methods
   //
   constructor(balances, hydrate) {
-    if (balances instanceof _Balances) return new _Balances(balances.each, hydrate);
+    if (balances instanceof _Balances) return new _Balances([...balances], hydrate);
     if (balances instanceof Balance) return new _Balances([balances], hydrate);
     if (!Array.isArray(balances)) return new _Balances(Object.values(balances), hydrate);
     if (balances.length === 0) return this;
@@ -3427,14 +3430,14 @@ var Balances = class _Balances {
         balances.map((storage) => new Balance(storage)),
         hydrate
       );
-    this.#balances = balances;
+    this.#balancesMap = new Map(balances.map((b) => [b.id, b]));
     if (hydrate !== void 0) this.hydrate(hydrate);
   }
   /**
    * Calling toJSON on a collection of balances will return the underlying BalanceJsonList.
    */
   toJSON = () => Object.fromEntries(
-    this.#balances.map((balance) => {
+    [...this.#balancesMap.values()].map((balance) => {
       try {
         return [balance.id, balance.toJSON()];
       } catch (error) {
@@ -3454,17 +3457,17 @@ var Balances = class _Balances {
    *   // do something
    * }
    */
-  [Symbol.iterator] = () => (
-    // Create an array of the balances in this collection and return the result of its iterator.
-    this.#balances[Symbol.iterator]()
-  );
+  [Symbol.iterator] = () => this.#balancesMap.values()[Symbol.iterator]();
   /**
    * Hydrates all balances in this collection.
    *
    * @param sources - The sources to hydrate from.
    */
   hydrate = (sources) => {
-    this.#balances.map((balance) => balance.hydrate(sources));
+    this.#cachedArray = null;
+    this.#cachedFilteredMirrorTokens = null;
+    this.#cachedSumFormatter = null;
+    for (const balance of this.#balancesMap.values()) balance.hydrate(sources);
   };
   /**
    * Retrieve a balance from this collection by id.
@@ -3472,7 +3475,7 @@ var Balances = class _Balances {
    * @param id - The id of the balance to fetch.
    * @returns The balance if one exists, or none.
    */
-  get = (id) => this.#balances.find((balance) => balance.id === id) ?? null;
+  get = (id) => this.#balancesMap.get(id) ?? null;
   /**
    * Retrieve balances from this collection by search query.
    *
@@ -3487,13 +3490,28 @@ var Balances = class _Balances {
     const filter2 = (balance) => orQueries.some(
       (query2) => typeof query2 === "function" ? query2(balance) : query2.every(([key, value]) => balance[key] === value)
     );
-    return new _Balances([...this].filter(filter2));
+    return new _Balances(this.#toArray().filter(filter2));
   };
   /**
    * Filters this collection to exclude token balances where the token has a `mirrorOf` field
    * and another balance exists in this collection for the token specified by the `mirrorOf` field.
    */
-  filterMirrorTokens = () => new _Balances([...this].filter(filterMirrorTokens));
+  filterMirrorTokens = () => {
+    if (!this.#cachedFilteredMirrorTokens) {
+      const balances = this.#toArray();
+      const tokenIds = /* @__PURE__ */ new Set();
+      for (const b of balances) {
+        if (b.tokenId) tokenIds.add(b.tokenId);
+      }
+      this.#cachedFilteredMirrorTokens = new _Balances(
+        balances.filter((balance) => {
+          const mirrorOf = balance.token?.mirrorOf;
+          return !mirrorOf || !tokenIds.has(mirrorOf);
+        })
+      );
+    }
+    return this.#cachedFilteredMirrorTokens;
+  };
   /**
    * Filters this collection to only include balances which are not zero AND have a fiat conversion rate.
    */
@@ -3512,11 +3530,9 @@ var Balances = class _Balances {
    */
   add = (balances) => {
     if (balances instanceof Balance) return this.add(new _Balances(balances));
-    const mergedBalances = Object.fromEntries(
-      this.#balances.map((balance) => [balance.id, balance])
-    );
-    balances.each.forEach((balance) => mergedBalances[balance.id] = balance);
-    return new _Balances(Object.values(mergedBalances));
+    const mergedMap = new Map(this.#balancesMap);
+    for (const balance of balances) mergedMap.set(balance.id, balance);
+    return new _Balances([...mergedMap.values()]);
   };
   /**
    * Remove balances from this collection by id.
@@ -3528,11 +3544,16 @@ var Balances = class _Balances {
    */
   remove = (ids) => {
     if (!Array.isArray(ids)) return this.remove([ids]);
-    return new _Balances(this.#balances.filter((balance) => !ids.includes(balance.id)));
+    const idSet = new Set(ids);
+    return new _Balances(this.#toArray().filter((balance) => !idSet.has(balance.id)));
   };
   // TODO: Add some more useful aggregator methods
+  #toArray = () => {
+    if (!this.#cachedArray) this.#cachedArray = [...this.#balancesMap.values()];
+    return this.#cachedArray;
+  };
   get each() {
-    return [...this];
+    return this.#toArray();
   }
   /** @deprecated use each instead */
   get sorted() {
@@ -3544,7 +3565,7 @@ var Balances = class _Balances {
    * @returns The number of balances in this collection.
    */
   get count() {
-    return [...this].length;
+    return this.#balancesMap.size;
   }
   /**
    * Get the summed value of balances in this collection.
@@ -3555,7 +3576,8 @@ var Balances = class _Balances {
    * balances.sum.fiat('usd').transferable
    */
   get sum() {
-    return new SumBalancesFormatter(this);
+    if (!this.#cachedSumFormatter) this.#cachedSumFormatter = new SumBalancesFormatter(this);
+    return this.#cachedSumFormatter;
   }
 };
 var getBalanceId = (balance) => {
@@ -7715,12 +7737,13 @@ var BalancesProvider = class {
   updateStorage$(balanceIds, balancesResult) {
     if (balancesResult.status !== "live") return;
     const storage = this.#storage.getValue();
+    const failedIds = new Set(balancesResult.failedBalanceIds);
     const balances = assign11(
       {},
       storage.balances,
       // delete all balances expected in the result set (except the ones that failed). because if they are not present it means they are empty.
       fromPairs2(
-        balanceIds.filter((bid) => !balancesResult.failedBalanceIds.includes(bid)).map((balanceId) => [balanceId, void 0])
+        balanceIds.filter((bid) => !failedIds.has(bid)).map((balanceId) => [balanceId, void 0])
       ),
       keyBy8(
         // storage balances must have status "cache", because they are used as start value when initialising subsequent subscriptions