@talismn/balances 0.0.0-pr2075-20250708110002 → 0.0.0-pr2075-20250708143919

package/dist/talismn-balances.esm.js

@@ -11,9 +11,9 @@ import pako from 'pako';
 import { parseAbi, erc20Abi, getContract, ContractFunctionExecutionError, hexToString, erc20Abi_bytes32, encodeFunctionData, isHex, hexToBigInt, withRetry } from 'viem';
 import { assign, omit, isEqual, fromPairs, toPairs, keys, keyBy, uniq, values, groupBy as groupBy$1 } from 'lodash';
 import z from 'zod/v4';
-import { Observable, distinctUntilChanged, of, timer, switchMap, from, firstValueFrom, scan, share, map, switchAll, combineLatest, mergeMap, toArray, interval, startWith, exhaustMap, BehaviorSubject, debounceTime, takeUntil, withLatestFrom, concatMap, tap } from 'rxjs';
+import { Observable, distinctUntilChanged, of, map, timer, switchMap, from, firstValueFrom, combineLatest, scan, share, switchAll, mergeMap, toArray, interval, startWith, exhaustMap, BehaviorSubject, debounceTime, takeUntil, withLatestFrom, concatMap, filter, tap } from 'rxjs';
 import isEqual$1 from 'lodash/isEqual';
-import { unifyMetadata, decAnyMetadata, getDynamicBuilder, getLookupFn, decodeScale, parseMetadataRpc, getStorageKeyPrefix, compactMetadata, encodeMetadata, papiParse, papiStringify, toHex, getMetadataVersion, encodeStateKey } from '@talismn/scale';
+import { unifyMetadata, decAnyMetadata, getDynamicBuilder, getLookupFn, decodeScale, parseMetadataRpc, getStorageKeyPrefix, compactMetadata, encodeMetadata, toHex, papiParse, papiStringify, getMetadataVersion, encodeStateKey } from '@talismn/scale';
 import { Metadata, TypeRegistry } from '@polkadot/types';
 import groupBy from 'lodash/groupBy';
 import { mergeUint8, toHex as toHex$1 } from '@polkadot-api/utils';
@@ -1766,7 +1766,7 @@ const getTransferCallData$8 = ({
   };
 };
 
-const SUBSCRIPTION_INTERVAL$8 = 6_000;
+const SUBSCRIPTION_INTERVAL$6 = 6_000;
 const subscribeBalances$8 = ({
   networkId,
   tokensWithAddresses,
@@ -1784,7 +1784,7 @@ const subscribeBalances$8 = ({
         });
         if (abortController.signal.aborted) return;
         subscriber.next(balances);
-        setTimeout(poll, SUBSCRIPTION_INTERVAL$8);
+        setTimeout(poll, SUBSCRIPTION_INTERVAL$6);
       } catch (error) {
         log.error("Error", {
           module: MODULE_TYPE$8,
@@ -1980,7 +1980,7 @@ const getTransferCallData$7 = ({
   };
 };
 
-const SUBSCRIPTION_INTERVAL$7 = 6_000;
+const SUBSCRIPTION_INTERVAL$5 = 6_000;
 const subscribeBalances$7 = ({
   networkId,
   tokensWithAddresses,
@@ -1998,7 +1998,7 @@ const subscribeBalances$7 = ({
         });
         if (abortController.signal.aborted) return;
         subscriber.next(balances);
-        setTimeout(poll, SUBSCRIPTION_INTERVAL$7);
+        setTimeout(poll, SUBSCRIPTION_INTERVAL$5);
       } catch (error) {
         log.error("Error", {
           module: MODULE_TYPE$7,
@@ -2313,7 +2313,7 @@ const getTransferCallData$6 = ({
   };
 };
 
-const SUBSCRIPTION_INTERVAL$6 = 6_000;
+const SUBSCRIPTION_INTERVAL$4 = 6_000;
 const subscribeBalances$6 = ({
   networkId,
   tokensWithAddresses,
@@ -2331,7 +2331,7 @@ const subscribeBalances$6 = ({
         });
         if (abortController.signal.aborted) return;
         subscriber.next(balances);
-        setTimeout(poll, SUBSCRIPTION_INTERVAL$6);
+        setTimeout(poll, SUBSCRIPTION_INTERVAL$4);
       } catch (error) {
         log.error("Error", {
           module: MODULE_TYPE$6,
@@ -3464,7 +3464,7 @@ const fetchBalances$6 = async ({
 }) => {
   const balanceDefs = getBalanceDefs(tokensWithAddresses);
   if (!miniMetadata?.data) {
-    log.warn("MiniMetadata is required for fetching balances");
+    log.warn(`MiniMetadata is required for fetching ${MODULE_TYPE$5} balances on ${networkId}.`);
     return {
       success: [],
       errors: balanceDefs.map(def => ({
@@ -3496,7 +3496,7 @@ const fetchBalances$6 = async ({
       }))
     };
   }
-  const queries = buildQueries$6(networkId, balanceDefs, miniMetadata);
+  const queries = buildQueries$7(networkId, balanceDefs, miniMetadata);
   const balances = await new RpcStateQueryHelper(connector, queries).fetch();
   return balanceDefs.reduce((acc, def) => {
     const balance = balances.find(b => b?.address === def.address && b?.tokenId === def.token.id);
@@ -3524,7 +3524,7 @@ const fetchBalances$6 = async ({
     errors: []
   });
 };
-const buildQueries$6 = (networkId, balanceDefs, miniMetadata) => {
+const buildQueries$7 = (networkId, balanceDefs, miniMetadata) => {
   const networkStorageCoders = buildNetworkStorageCoders(networkId, miniMetadata, {
     storage: ["Assets", "Account"]
   });
@@ -3533,9 +3533,9 @@ const buildQueries$6 = (networkId, balanceDefs, miniMetadata) => {
     address
   }) => {
     const scaleCoder = networkStorageCoders?.storage;
-    const stateKey = tryEncode$1(scaleCoder, Number(token.assetId), address) ??
+    const stateKey = tryEncode$2(scaleCoder, Number(token.assetId), address) ??
     // Asset Hub
-    tryEncode$1(scaleCoder, BigInt(token.assetId), address); // Astar
+    tryEncode$2(scaleCoder, BigInt(token.assetId), address); // Astar
 
     if (!stateKey) {
       log.warn(`Invalid assetId / address in ${networkId} storage query ${token.assetId} / ${address}`);
@@ -3591,7 +3591,7 @@ const buildQueries$6 = (networkId, balanceDefs, miniMetadata) => {
     };
   }).filter(isNotNil);
 };
-const tryEncode$1 = (scaleCoder, ...args) => {
+const tryEncode$2 = (scaleCoder, ...args) => {
   try {
     return scaleCoder?.keys?.enc?.(...args);
   } catch {
@@ -3834,46 +3834,163 @@ const getTransferAllEncodedArgs$2 = (assetId, to, codec) => {
   })]);
 };
 
-const SUBSCRIPTION_INTERVAL$5 = 6_000;
+const fetchRuntimeCallResult = async (connector, networkId, metadataRpc, apiName, method, args) => {
+  const {
+    builder
+  } = parseMetadataRpc(metadataRpc);
+  const call = builder.buildRuntimeCall(apiName, method);
+  const hex = await connector.send(networkId, "state_call", [`${apiName}_${method}`, toHex(call.args.enc(args))]);
+  return call.value.dec(hex);
+};
+
+const tryGetConstantValue = (metadataRpc, pallet, constant) => {
+  const {
+    unifiedMetadata,
+    builder
+  } = parseMetadataRpc(metadataRpc);
+  const encodedValue = unifiedMetadata.pallets.find(({
+    name
+  }) => name === pallet)?.constants.find(({
+    name
+  }) => name === constant)?.value;
+  if (!encodedValue) return null;
+  const codec = builder.buildConstant(pallet, constant);
+  return codec.dec(encodedValue);
+};
+
+const fetchRpcQueryPack = async (connector, networkId, queries) => {
+  const allStateKeys = queries.flatMap(({
+    stateKeys
+  }) => stateKeys).filter(isNotNil);
+
+  // doing a query without keys would throw an error => return early
+  if (!allStateKeys.length) return queries.map(({
+    stateKeys,
+    decodeResult
+  }) => decodeResult(stateKeys.map(() => null)));
+  const [result] = await connector.send(networkId, "state_queryStorageAt", [allStateKeys]);
+  return decodeRpcQueryPack(queries, result);
+};
+const getRpcQueryPack$ = (connector, networkId, queries, timeout = false) => {
+  const allStateKeys = queries.flatMap(({
+    stateKeys
+  }) => stateKeys).filter(isNotNil);
+
+  // doing a query without keys would throw an error => return early
+  if (!allStateKeys.length) return of(queries.map(({
+    stateKeys,
+    decodeResult
+  }) => decodeResult(stateKeys.map(() => null))));
+  return new Observable(subscriber => {
+    const promUnsub = connector.subscribe(networkId, "state_subscribeStorage", "state_storage", [allStateKeys], (error, result) => {
+      if (error) subscriber.error(error);else subscriber.next(decodeRpcQueryPack(queries, result));
+    }, timeout);
+    return () => {
+      promUnsub.then(unsub => unsub("state_unsubscribeStorage"));
+    };
+  });
+};
+const decodeRpcQueryPack = (queries, result) => {
+  return queries.reduce((acc, {
+    stateKeys,
+    decodeResult
+  }) => {
+    const changes = stateKeys.map(stateKey => {
+      if (!stateKey) return null;
+      const change = result.changes.find(([key]) => key === stateKey);
+      if (!change) return null;
+      return change[1];
+    });
+    acc.push(decodeResult(changes));
+    return acc;
+  }, []);
+};
+
 const subscribeBalances$5 = ({
   networkId,
   tokensWithAddresses,
   connector,
   miniMetadata
 }) => {
-  return new Observable(subscriber => {
-    const abortController = new AbortController();
+  const balanceDefs = getBalanceDefs(tokensWithAddresses);
+  const queries = buildQueries$6(networkId, balanceDefs, miniMetadata);
+  return getRpcQueryPack$(connector, networkId, queries).pipe(map(balances => ({
+    success: balances,
+    errors: []
+  })));
+};
+const buildQueries$6 = (networkId, balanceDefs, miniMetadata) => {
+  const networkStorageCoders = buildNetworkStorageCoders(networkId, miniMetadata, {
+    storage: ["Assets", "Account"]
+  });
+  return balanceDefs.map(({
+    token,
+    address
+  }) => {
+    const scaleCoder = networkStorageCoders?.storage;
+    const stateKey = tryEncode$1(scaleCoder, Number(token.assetId), address) ??
+    // Asset Hub
+    tryEncode$1(scaleCoder, BigInt(token.assetId), address); // Astar
 
-    // on hydration balances are fetched using a runtimeApi, which can't be subscribed to.
-    // => poll values for each block
-    const poll = async () => {
-      try {
-        if (abortController.signal.aborted) return;
-        const balances = await fetchBalances$6({
-          networkId,
-          tokensWithAddresses: tokensWithAddresses,
-          connector,
-          miniMetadata
-        });
-        if (abortController.signal.aborted) return;
-        subscriber.next(balances);
-        setTimeout(poll, SUBSCRIPTION_INTERVAL$5);
-      } catch (error) {
-        log.error("Error", {
-          module: MODULE_TYPE$5,
-          networkId,
-          miniMetadata,
-          addressesByToken: tokensWithAddresses,
-          error
-        });
-        subscriber.error(error);
-      }
+    if (!stateKey) {
+      log.warn(`Invalid assetId / address in ${networkId} storage query ${token.assetId} / ${address}`);
+      return null;
+    } else log.log(`VALID assetId / address in ${networkId} storage query ${token.assetId} / ${address}`);
+    const decodeResult = changes => {
+      /** NOTE: This type is only a hint for typescript, the chain can actually return whatever it wants to */
+
+      const decoded = decodeScale(scaleCoder, changes[0], `Failed to decode substrate-assets balance on chain ${networkId}`) ?? {
+        balance: 0n,
+        status: {
+          type: "Liquid"
+        }};
+      const isFrozen = decoded?.status?.type === "Frozen";
+      const amount = (decoded?.balance ?? 0n).toString();
+
+      // due to the following balance calculations, which are made in the `Balance` type:
+      //
+      // total balance        = (free balance) + (reserved balance)
+      // transferable balance = (free balance) - (frozen balance)
+      //
+      // when `isFrozen` is true we need to set **both** the `free` and `frozen` amounts
+      // of this balance to the value we received from the RPC.
+      //
+      // if we only set the `frozen` amount, then the `total` calculation will be incorrect!
+      const free = amount;
+      const frozen = token.isFrozen || isFrozen ? amount : "0";
+
+      // include balance values even if zero, so that newly-zero values overwrite old values
+      const balanceValues = [{
+        type: "free",
+        label: "free",
+        amount: free.toString()
+      }, {
+        type: "locked",
+        label: "frozen",
+        amount: frozen.toString()
+      }];
+      const balance = {
+        source: "substrate-assets",
+        status: "live",
+        address,
+        networkId,
+        tokenId: token.id,
+        values: balanceValues
+      };
+      return balance;
     };
-    poll();
-    return () => {
-      abortController.abort();
+    return {
+      stateKeys: [stateKey],
+      decodeResult
     };
-  }).pipe(distinctUntilChanged(isEqual));
+  }).filter(isNotNil);
+};
+const tryEncode$1 = (scaleCoder, ...args) => {
+  try {
+    return scaleCoder?.keys?.enc?.(...args);
+  } catch {
+    return null;
+  }
 };
 
 const SubAssetsBalanceModule = {
@@ -3903,7 +4020,7 @@ const fetchBalances$5 = async ({
 }) => {
   const balanceDefs = getBalanceDefs(tokensWithAddresses);
   if (!miniMetadata?.data) {
-    log.warn("MiniMetadata is required for fetching balances");
+    log.warn(`MiniMetadata is required for fetching ${MODULE_TYPE$4} balances on ${networkId}.`);
     return {
       success: [],
       errors: balanceDefs.map(def => ({
@@ -4232,7 +4349,7 @@ const getTransferAllEncodedArgs$1 = (onChainId, to, codec) => {
   })]);
 };
 
-const SUBSCRIPTION_INTERVAL$4 = 6_000;
+const SUBSCRIPTION_INTERVAL$3 = 6_000;
 const subscribeBalances$4 = ({
   networkId,
   tokensWithAddresses,
@@ -4255,7 +4372,7 @@ const subscribeBalances$4 = ({
         });
         if (abortController.signal.aborted) return;
         subscriber.next(balances);
-        setTimeout(poll, SUBSCRIPTION_INTERVAL$4);
+        setTimeout(poll, SUBSCRIPTION_INTERVAL$3);
       } catch (error) {
         log.error("Error", {
           module: MODULE_TYPE$4,
@@ -4299,30 +4416,6 @@ const SubHydrationTokenConfigSchema = z.strictObject({
 const MODULE_TYPE$3 = SubHydrationTokenSchema.shape.type.value;
 const PLATFORM$3 = SubHydrationTokenSchema.shape.platform.value;
 
-const fetchRuntimeCallResult = async (connector, networkId, metadataRpc, apiName, method, args) => {
-  const {
-    builder
-  } = parseMetadataRpc(metadataRpc);
-  const call = builder.buildRuntimeCall(apiName, method);
-  const hex = await connector.send(networkId, "state_call", [`${apiName}_${method}`, toHex(call.args.enc(args))]);
-  return call.value.dec(hex);
-};
-
-const tryGetConstantValue = (metadataRpc, pallet, constant) => {
-  const {
-    unifiedMetadata,
-    builder
-  } = parseMetadataRpc(metadataRpc);
-  const encodedValue = unifiedMetadata.pallets.find(({
-    name
-  }) => name === pallet)?.constants.find(({
-    name
-  }) => name === constant)?.value;
-  if (!encodedValue) return null;
-  const codec = builder.buildConstant(pallet, constant);
-  return codec.dec(encodedValue);
-};
-
 const fetchBalances$4 = async ({
   networkId,
   tokensWithAddresses,
@@ -4331,7 +4424,7 @@ const fetchBalances$4 = async ({
 }) => {
   const balanceDefs = getBalanceDefs(tokensWithAddresses);
   if (!miniMetadata?.data) {
-    log.warn("MiniMetadata is required for fetching balances");
+    log.warn(`MiniMetadata is required for fetching ${MODULE_TYPE$3} balances on ${networkId}.`);
     return {
       success: [],
       errors: balanceDefs.map(def => ({
@@ -4578,7 +4671,7 @@ const getTransferCallData$3 = ({
   };
 };
 
-const SUBSCRIPTION_INTERVAL$3 = 6_000;
+const SUBSCRIPTION_INTERVAL$2 = 6_000;
 const subscribeBalances$3 = ({
   networkId,
   tokensWithAddresses,
@@ -4601,7 +4694,7 @@ const subscribeBalances$3 = ({
         });
         if (abortController.signal.aborted) return;
         subscriber.next(balances);
-        setTimeout(poll, SUBSCRIPTION_INTERVAL$3);
+        setTimeout(poll, SUBSCRIPTION_INTERVAL$2);
       } catch (error) {
         log.error("Error", {
           module: MODULE_TYPE$3,
@@ -4633,140 +4726,6 @@ const SubHydrationBalanceModule = {
 const MODULE_TYPE$2 = SubNativeTokenSchema.shape.type.value;
 const PLATFORM$2 = SubNativeTokenSchema.shape.platform.value;
 
-/**
- * Pass some these into an `RpcStateQueryHelper` in order to easily batch multiple state queries into the one rpc call.
- */
-
-const fetchQueriesPack = async (connector, networkId, queries) => {
-  const allStateKeys = queries.flatMap(({
-    stateKeys
-  }) => stateKeys).filter(isNotNil);
-
-  // doing a query with only null keys would throw an error => return early
-  if (!allStateKeys.length) return queries.map(({
-    stateKeys,
-    decodeResult
-  }) => decodeResult(stateKeys.map(() => null)));
-  const response = await connector.send(networkId, "state_queryStorageAt", [allStateKeys]);
-  const results = queries.reduce((acc, {
-    stateKeys,
-    decodeResult
-  }) => {
-    const changes = stateKeys.map(stateKey => {
-      if (!stateKey) return null;
-      const change = response[0].changes.find(([key]) => key === stateKey);
-      if (!change) return null;
-      return change[1];
-    });
-    acc.push(decodeResult(changes));
-    return acc;
-  }, []);
-  return results;
-};
-
-/**
- * Used by a variety of balance modules to help batch multiple state queries into the one rpc call.
- */
-// export class RpcStateQueriesHelper<T> {
-//   #connector: ChainConnector
-//   #queries: Array<RpcStateQueries<T>>
-
-//   constructor(connector: ChainConnector, queries: Array<RpcStateQueries<T>>) {
-//     this.#connector = connector
-//     this.#queries = queries
-//   }
-
-//   async subscribe(
-//     networkId: DotNetworkId,
-//     callback: SubscriptionCallback<T[]>,
-//     timeout: number | false = false,
-//     subscribeMethod = "state_subscribeStorage",
-//     responseMethod = "state_storage",
-//     unsubscribeMethod = "state_unsubscribeStorage",
-//   ): Promise<UnsubscribeFn> {
-//     const params = [this.#queries.flatMap(({ stateKeys }) => stateKeys)]
-
-//     const unsub = this.#connector.subscribe(
-//       networkId,
-//       subscribeMethod,
-//       responseMethod,
-//       params,
-//       (error, result) => {
-//         error
-//           ? callback(error)
-//           : callback(null, this.#distributeChangesToQueryDecoders.call(this, chainId, result))
-//       },
-//       timeout,
-//     )
-
-//     const subscriptions = queries.map(([chainId, queries]) => {
-//       const params = [queries.map(({ stateKey }) => stateKey)]
-
-//       const unsub = this.#connector.subscribe(
-//         networkId,
-//         subscribeMethod,
-//         responseMethod,
-//         params,
-//         (error, result) => {
-//           error
-//             ? callback(error)
-//             : callback(null, this.#distributeChangesToQueryDecoders.call(this, chainId, result))
-//         },
-//         timeout,
-//       )
-
-//       return () => unsub.then((unsubscribe) => unsubscribe(unsubscribeMethod))
-//     })
-
-//     return () => subscriptions.forEach((unsubscribe) => unsubscribe())
-//   }
-
-//   async fetch(method = "state_queryStorageAt"): Promise<T[]> {
-//     const queriesByChain = groupBy(this.#queries, "chainId")
-
-//     const resultsByChain = await Promise.all(
-//       Object.entries(queriesByChain).map(async ([chainId, queries]) => {
-//         const params = [queries.map(({ stateKey }) => stateKey)]
-
-//         const result = (await this.#connector.send(chainId, method, params))[0]
-//         return this.#distributeChangesToQueryDecoders.call(this, chainId, result)
-//       }),
-//     )
-
-//     return resultsByChain.flatMap((result) => result)
-//   }
-
-//   #distributeChangesToQueryDecoders(chainId: DotNetworkId, result: unknown): T[] {
-//     if (typeof result !== "object" || result === null) return []
-//     if (!hasOwnProperty(result, "changes") || typeof result.changes !== "object") return []
-//     if (!Array.isArray(result.changes)) return []
-
-//     return result.changes.flatMap(([reference, change]: [unknown, unknown]): [T] | [] => {
-//       if (typeof reference !== "string") {
-//         log.warn(`Received non-string reference in RPC result: ${reference}`)
-//         return []
-//       }
-
-//       if (typeof change !== "string" && change !== null) {
-//         log.warn(`Received non-string and non-null change in RPC result: ${reference} | ${change}`)
-//         return []
-//       }
-
-//       const query = this.#queries.find(
-//         ({ chainId: cId, stateKey }) => cId === chainId && stateKey === reference,
-//       )
-//       if (!query) {
-//         log.warn(
-//           `Failed to find query:\n${reference} in\n${this.#queries.map(({ stateKey }) => stateKey)}`,
-//         )
-//         return []
-//       }
-
-//       return [query.decodeResult(change)]
-//     })
-//   }
-// }
-
 const SUBTENSOR_ROOT_NETUID$1 = 0;
 const SUBTENSOR_MIN_STAKE_AMOUNT_PLANK$1 = 1000000n;
 const TAO_DECIMALS$1 = 9n;
@@ -5486,7 +5445,7 @@ const fetchBalances$3 = async ({
 }) => {
   const balanceDefs = getBalanceDefs(tokensWithAddresses);
   if (!miniMetadata?.data) {
-    log.warn("MiniMetadata is required for fetching balances");
+    log.warn(`MiniMetadata is required for fetching ${MODULE_TYPE$2} balances on ${networkId}.`);
     return {
       success: [],
       errors: balanceDefs.map(def => ({
@@ -5518,12 +5477,12 @@ const fetchBalances$3 = async ({
       }))
     };
   }
-  const queries = buildBaseQueries(networkId, balanceDefs, miniMetadata);
-  const partialBalances = await fetchQueriesPack(connector, networkId, queries);
+  const baseQueries = buildBaseQueries(networkId, balanceDefs, miniMetadata);
+  const partialBalances = await fetchRpcQueryPack(connector, networkId, baseQueries);
 
   // now for each balance that includes nomPoolStaking, we need to fetch the metadata for the pool
   const nomPoolQueries = buildNomPoolQueries(networkId, partialBalances, miniMetadata);
-  const balances = await fetchQueriesPack(connector, networkId, nomPoolQueries);
+  const balances = await fetchRpcQueryPack(connector, networkId, nomPoolQueries);
 
   // TODO ⚠️ dedupe locks
 
@@ -5759,46 +5718,32 @@ const getTransferAllEncodedArgs = (to, codec) => {
   })]);
 };
 
-const SUBSCRIPTION_INTERVAL$2 = 6_000;
 const subscribeBalances$2 = ({
   networkId,
   tokensWithAddresses,
   connector,
   miniMetadata
 }) => {
-  return new Observable(subscriber => {
-    const abortController = new AbortController();
-
-    // on hydration balances are fetched using a runtimeApi, which can't be subscribed to.
-    // => poll values for each block
-    const poll = async () => {
-      try {
-        if (abortController.signal.aborted) return;
-        const balances = await fetchBalances$3({
-          networkId,
-          tokensWithAddresses: tokensWithAddresses,
-          connector,
-          miniMetadata
-        });
-        if (abortController.signal.aborted) return;
-        subscriber.next(balances);
-        setTimeout(poll, SUBSCRIPTION_INTERVAL$2);
-      } catch (error) {
-        log.error("Error", {
-          module: MODULE_TYPE$2,
-          networkId,
-          miniMetadata,
-          addressesByToken: tokensWithAddresses,
-          error
-        });
-        subscriber.error(error);
-      }
-    };
-    poll();
-    return () => {
-      abortController.abort();
+  // could be use as shared observable key if we decide to cache the sub
+  const balanceDefs = getBalanceDefs(tokensWithAddresses);
+  const baseQueries = buildBaseQueries(networkId, balanceDefs, miniMetadata);
+  const baseBalances$ = getRpcQueryPack$(connector, networkId, baseQueries).pipe(switchMap(partialBalances => {
+    // now for each balance that includes nomPoolStaking, we need to fetch the metadata for the pool
+    const nomPoolQueries = buildNomPoolQueries(networkId, partialBalances, miniMetadata);
+    return getRpcQueryPack$(connector, networkId, nomPoolQueries);
+  }));
+  const subtensorBalancesByAddress$ = getSubtensorStakingBalances$(connector, networkId, balanceDefs, miniMetadata);
+  return combineLatest([baseBalances$, subtensorBalancesByAddress$]).pipe(map(([baseBalances, subtensorBalancesByAddress]) => {
+    // add subtensor balances to base balances
+    for (const [address, subtensorBalances] of toPairs(subtensorBalancesByAddress)) {
+      const balance = baseBalances.find(b => b.address === address);
+      if (balance?.values) balance.values.push(...subtensorBalances);
+    }
+    return {
+      success: baseBalances,
+      errors: []
     };
-  }).pipe(distinctUntilChanged(isEqual));
+  }));
 };
 
 const SubNativeBalanceModule = {
@@ -7238,7 +7183,7 @@ const fetchBalances$1 = async ({
 }) => {
   const balanceDefs = getBalanceDefs(tokensWithAddresses);
   if (!miniMetadata?.data) {
-    log.warn("MiniMetadata is required for fetching balances");
+    log.warn(`MiniMetadata is required for fetching ${MODULE_TYPE} balances on ${networkId}.`);
     return {
       success: [],
       errors: balanceDefs.map(def => ({
@@ -10458,9 +10403,17 @@ class BalancesProvider {
       miniMetadatas: values(miniMetadatas).filter(isNotNil)
     })));
   }
-  getBalances$(addressesByToken) {
-    const networkIds = uniq(keys(addressesByToken).map(tokenId => parseTokenId(tokenId).networkId));
-    return combineLatest(networkIds.map(networkId => this.getNetworkBalances$(networkId, addressesByToken))).pipe(map(results => {
+
+  // this is the only public method
+  getBalances$(addressesByTokenId) {
+    // split by network
+    const addressesByTokenIdByNetworkId = toPairs(addressesByTokenId).reduce((acc, [tokenId, addresses]) => {
+      const networkId = parseTokenId(tokenId).networkId;
+      if (!acc[networkId]) acc[networkId] = {};
+      acc[networkId][tokenId] = addresses;
+      return acc;
+    }, {});
+    return combineLatest(toPairs(addressesByTokenIdByNetworkId).map(([networkId]) => this.getNetworkBalances$(networkId, addressesByTokenIdByNetworkId[networkId]))).pipe(map(results => {
       return {
         status: results.some(({
           status
@@ -10469,9 +10422,17 @@ class BalancesProvider {
       };
     }), startWith({
       status: "initialising",
-      balances: this.getStoredBalances(addressesByToken)
+      balances: this.getStoredBalances(addressesByTokenId)
     }), distinctUntilChanged(isEqual));
   }
+  fetchBalances(addressesByTokenId) {
+    // TODO: better
+    return firstValueFrom(this.getBalances$(addressesByTokenId).pipe(filter(({
+      status
+    }) => status === "live"), map(({
+      balances
+    }) => balances)));
+  }
   getNetworkBalances$(networkId, addressesByTokenId) {
     const network$ = this.#chaindataProvider.getNetworkById$(networkId);
     const tokensMapById$ = this.#chaindataProvider.getTokensMapById$();
@@ -10543,7 +10504,42 @@ class BalancesProvider {
     }));
   }
   getNetworkMiniMetadatas$(networkId) {
-    return this.#chaindataProvider.getNetworkById$(networkId).pipe(switchMap(network => isNetworkDot(network) && this.#chainConnectors.substrate ? from(getMiniMetadatas(this.#chainConnectors.substrate, this.#chaindataProvider, networkId)) : of([])));
+    return this.#chaindataProvider.getNetworkById$(networkId).pipe(switchMap(network => isNetworkDot(network) && this.#chainConnectors.substrate ? from(getSpecVersion(this.#chainConnectors.substrate, networkId)).pipe(switchMap(specVersion => this.getMiniMetadatas$(networkId, specVersion))) : of([])));
+  }
+  getMiniMetadatas$(networkId, specVersion) {
+    return combineLatest({
+      defaultMiniMetadatas: this.getDefaultMiniMetadatas$(networkId, specVersion),
+      storedMiniMetadatas: this.getStoredMiniMetadatas$(networkId, specVersion)
+    }).pipe(switchMap(({
+      storedMiniMetadatas,
+      defaultMiniMetadatas
+    }) => {
+      if (defaultMiniMetadatas.length) return of(defaultMiniMetadatas);
+      if (storedMiniMetadatas.length) return of(storedMiniMetadatas);
+      if (!this.#chainConnectors.substrate) return of([]);
+      return from(
+      // fetch them from the chain
+      getMiniMetadatas(this.#chainConnectors.substrate, this.#chaindataProvider, networkId)).pipe(
+      // and persist in storage for later reuse
+      tap(newMiniMetadatas => {
+        if (!newMiniMetadatas.length) return;
+        const storage = this.#storage.getValue();
+        const miniMetadatas = assign(
+        // keep minimetadatas of other networks
+        keyBy(values(storage.miniMetadatas).filter(m => m.chainId !== networkId), m => m.id),
+        // add the ones for our network
+        keyBy(newMiniMetadatas, m => m.id));
+        this.#storage.next(assign({}, storage, {
+          miniMetadatas
+        }));
+      }));
+    }));
+  }
+  getStoredMiniMetadatas$(networkId, specVersion) {
+    return this.storage$.pipe(map(storage => storage.miniMetadatas.filter(m => m.chainId === networkId && m.specVersion === specVersion && m.version === MINIMETADATA_VERSION)), distinctUntilChanged(isEqual));
+  }
+  getDefaultMiniMetadatas$(networkId, specVersion) {
+    return this.#chaindataProvider.miniMetadatas$.pipe(map(miniMetadatas => miniMetadatas.filter(m => m.chainId === networkId && m.specVersion === specVersion && m.version === MINIMETADATA_VERSION)), distinctUntilChanged(isEqual));
   }
   getStoredBalances(addressesByToken) {
     const balanceDefs = toPairs(addressesByToken).flatMap(([tokenId, addresses]) => addresses.map(address => [tokenId, address]));
@@ -10554,11 +10550,4 @@ class BalancesProvider {
   }
 }
 
-// const getStoredBalances = (
-//   storedBalances: Record<string, IBalance>,
-//   addressesByToken: Record<TokenId, Address[]>,
-// ): IBalance[] => {
-
-// }
-
 export { BALANCE_MODULES, Balance, BalanceFormatter, BalanceValueGetter, Balances, BalancesProvider, Change24hCurrencyFormatter, DefaultBalanceModule, EvmErc20BalanceModule, EvmErc20TokenConfigSchema, EvmNativeBalanceModule, EvmNativeTokenConfigSchema, EvmUniswapV2BalanceModule, EvmUniswapV2TokenConfigSchema, FiatSumBalancesFormatter, ONE_ALPHA_TOKEN$1 as ONE_ALPHA_TOKEN, PlanckSumBalancesFormatter, RpcStateQueryHelper, SCALE_FACTOR$1 as SCALE_FACTOR, SUBTENSOR_MIN_STAKE_AMOUNT_PLANK$1 as SUBTENSOR_MIN_STAKE_AMOUNT_PLANK, SUBTENSOR_ROOT_NETUID$1 as SUBTENSOR_ROOT_NETUID, SubAssetsBalanceModule, SubAssetsTokenConfigSchema, SubForeignAssetsBalanceModule, SubForeignAssetsTokenConfigSchema, SubHydrationBalanceModule, SubHydrationTokenConfigSchema, SubNativeBalanceModule, SubNativeMiniMetadataExtraSchema, SubNativeModuleConfigSchema, SubNativeTokenConfigSchema, SubPsp22BalanceModule, SubPsp22TokenConfigSchema, SubTokensBalanceModule, SubTokensMiniMetadataExtraSchema, SubTokensModuleConfigSchema, SubTokensTokenConfigSchema, SumBalancesFormatter, TalismanBalancesDatabase, abiMulticall, balances, buildNetworkStorageCoders, buildStorageCoders, calculateAlphaPrice$1 as calculateAlphaPrice, calculateTaoAmountFromAlpha$1 as calculateTaoAmountFromAlpha, calculateTaoFromDynamicInfo$1 as calculateTaoFromDynamicInfo, compress, configureStore, db, decodeOutput, decompress, defaultBalanceModules, deriveMiniMetadataId, detectTransferMethod, erc20BalancesAggregatorAbi, excludeFromFeePayableLocks, excludeFromTransferableAmount, filterBaseLocks, filterMirrorTokens, getBalanceId, getLockTitle, getLockedType$1 as getLockedType, getUniqueChainIds, getValueId, includeInTotalExtraAmount, makeContractCaller, uniswapV2PairAbi };