@talismn/balances-react 0.0.0-pr2075-20250710142256 → 0.0.0-pr2076-20250703121429

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -2,26 +2,29 @@
2
2
 
3
3
  var jotai = require('jotai');
4
4
  var react = require('react');
5
+ var balances = require('@talismn/balances');
5
6
  var tokenRates = require('@talismn/token-rates');
6
7
  var jsxRuntime = require('react/jsx-runtime');
7
8
  var chaindataProvider = require('@talismn/chaindata-provider');
8
- var balances = require('@talismn/balances');
9
+ var util = require('@talismn/util');
9
10
  var jotaiEffect = require('jotai-effect');
10
- var lodashEs = require('lodash-es');
11
+ var utils = require('jotai/utils');
12
+ var lodash = require('lodash');
13
+ var rxjs = require('rxjs');
14
+ var anylogger = require('anylogger');
11
15
  var chainConnector = require('@talismn/chain-connector');
12
16
  var chainConnectorEvm = require('@talismn/chain-connector-evm');
13
17
  var connectionMeta = require('@talismn/connection-meta');
14
- var util = require('@talismn/util');
15
- var utils = require('jotai/utils');
16
- var rxjs = require('rxjs');
17
18
  var dexie = require('dexie');
18
- var anylogger = require('anylogger');
19
+ var isEqual = require('lodash/isEqual');
19
20
  var utilCrypto = require('@polkadot/util-crypto');
20
21
 
21
22
  function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
22
23
 
23
24
  var anylogger__default = /*#__PURE__*/_interopDefault(anylogger);
25
+ var isEqual__default = /*#__PURE__*/_interopDefault(isEqual);
24
26
 
27
+ const balanceModuleCreatorsAtom = jotai.atom(balances.defaultBalanceModules);
25
28
  const innerCoinsApiConfigAtom = jotai.atom(tokenRates.DEFAULT_COINSAPI_CONFIG);
26
29
  const coinsApiConfigAtom = jotai.atom(get => get(innerCoinsApiConfigAtom), (_get, set, options) => set(innerCoinsApiConfigAtom, {
27
30
  apiUrl: options.apiUrl ?? tokenRates.DEFAULT_COINSAPI_CONFIG.apiUrl
@@ -33,6 +36,55 @@ const enabledTokensAtom = jotai.atom(undefined);
33
36
  /** Sets the list of addresses for which token balances will be fetched by the balances subscription */
34
37
  const allAddressesAtom = jotai.atom([]);
35
38
 
39
+ var packageJson = {
40
+ name: "@talismn/balances-react"};
41
+
42
+ var log = anylogger__default.default(packageJson.name);
43
+
44
+ /**
45
+ // Persistence backend for indexedDB
46
+ // Add a new backend by implementing the BalancesPersistBackend interface
47
+ // configureStore can be called with a different indexedDB table
48
+ */
49
+ const {
50
+ persistData,
51
+ retrieveData
52
+ } = balances.configureStore();
53
+
54
+ /**
55
+ // Persistence backend for localStorage
56
+ */
57
+ const localStoragePersist = async balances$1 => {
58
+ const storedBalances = balances$1.map(b => {
59
+ // eslint-disable-next-line @typescript-eslint/no-unused-vars
60
+ const {
61
+ status,
62
+ ...rest
63
+ } = b;
64
+ return rest;
65
+ });
66
+ const deflated = balances.compress(storedBalances);
67
+ localStorage.setItem("talismanBalances", deflated.toString());
68
+ };
69
+ const localStorageRetrieve = async () => {
70
+ const deflated = localStorage.getItem("talismanBalances");
71
+ if (deflated) {
72
+ // deflated will be a long string of numbers separated by commas
73
+ const deflatedArray = deflated.split(",").map(n => parseInt(n, 10));
74
+ const deflatedBytes = new Uint8Array(deflatedArray.length);
75
+ deflatedArray.forEach((n, i) => deflatedBytes[i] = n);
76
+ return balances.decompress(deflatedBytes).map(b => ({
77
+ ...b,
78
+ status: "cache"
79
+ }));
80
+ }
81
+ return [];
82
+ };
83
+ const localStorageBalancesPersistBackend = {
84
+ persist: localStoragePersist,
85
+ retrieve: localStorageRetrieve
86
+ };
87
+
36
88
  const chaindataProviderAtom = jotai.atom(() => {
37
89
  return new chaindataProvider.ChaindataProvider({});
38
90
  });
@@ -47,44 +99,19 @@ const chainConnectorsAtom = jotai.atom(get => {
47
99
  };
48
100
  });
49
101
 
50
- const balancesProviderAtom = jotai.atom(get => {
51
- return new balances.BalancesProvider(get(chaindataProviderAtom), get(chainConnectorsAtom)) // TODO pass storage
52
- ;
53
- });
54
-
55
- const chaindataAtom = utils.atomWithObservable(get => {
56
- return rxjs.combineLatest({
57
- networks: get(chaindataProviderAtom).networks$,
58
- tokens: get(chaindataProviderAtom).tokens$
59
- }).pipe(util.firstThenDebounce(1_000));
60
- });
61
- const filteredChaindataAtom = jotai.atom(async get => {
62
- const enabledNetworkIds = get(enabledChainsAtom);
63
- const enabledTokenIds = get(enabledTokensAtom);
64
- const enableTestnets = get(enableTestnetsAtom);
65
- const chaindata = await get(chaindataAtom);
66
- const networks = chaindata.networks.filter(n => (enabledNetworkIds?.includes(n.id) || n.isDefault) && (enableTestnets || !n.isTestnet));
67
- const networkById = lodashEs.keyBy(networks, n => n.id);
68
- const tokens = chaindata.tokens.filter(token => (enabledTokenIds?.includes(token.id) || token.isDefault) && networkById[token.networkId]);
69
- return {
70
- networks,
71
- tokens
72
- };
73
- });
74
- const tokensAtom = jotai.atom(async get => {
75
- const chaindata = await get(filteredChaindataAtom);
76
- return chaindata.tokens;
77
- });
78
- const networksAtom = jotai.atom(async get => {
79
- const chaindata = await get(filteredChaindataAtom);
80
- return chaindata.networks;
102
+ const balanceModulesAtom = jotai.atom(get => {
103
+ const balanceModuleCreators = get(balanceModuleCreatorsAtom);
104
+ const chainConnectors = get(chainConnectorsAtom);
105
+ const chaindataProvider = get(chaindataProviderAtom);
106
+ if (!chainConnectors.substrate) return [];
107
+ if (!chainConnectors.evm) return [];
108
+ if (!chaindataProvider) return [];
109
+ return balanceModuleCreators.map(mod => mod({
110
+ chainConnectors,
111
+ chaindataProvider
112
+ }));
81
113
  });
82
114
 
83
- var packageJson = {
84
- name: "@talismn/balances-react"};
85
-
86
- var log = anylogger__default.default(packageJson.name);
87
-
88
115
  /**
89
116
  * Converts a dexie Observable into an rxjs Observable.
90
117
  */
@@ -98,6 +125,54 @@ function dexieToRxjs(o) {
98
125
  });
99
126
  }
100
127
 
128
+ const chainsAtom = jotai.atom(async get => (await get(chaindataAtom)).chains);
129
+ const chainsByIdAtom = jotai.atom(async get => (await get(chaindataAtom)).chainsById);
130
+ const chainsByGenesisHashAtom = jotai.atom(async get => (await get(chaindataAtom)).chainsByGenesisHash);
131
+ const evmNetworksAtom = jotai.atom(async get => (await get(chaindataAtom)).evmNetworks);
132
+ const evmNetworksByIdAtom = jotai.atom(async get => (await get(chaindataAtom)).evmNetworksById);
133
+ const tokensAtom = jotai.atom(async get => (await get(chaindataAtom)).tokens);
134
+ const tokensByIdAtom = jotai.atom(async get => (await get(chaindataAtom)).tokensById);
135
+ const miniMetadatasAtom = jotai.atom(async get => (await get(chaindataAtom)).miniMetadatas);
136
+ const chaindataAtom = utils.atomWithObservable(get => {
137
+ const enableTestnets = get(enableTestnetsAtom);
138
+ const filterTestnets = items => enableTestnets ? items : items.filter(({
139
+ isTestnet
140
+ }) => !isTestnet);
141
+ const filterMapTestnets = items => enableTestnets ? items : Object.fromEntries(Object.entries(items).filter(([, {
142
+ isTestnet
143
+ }]) => !isTestnet));
144
+ const filterEnabledTokens = tokens => tokens.filter(token => token.isDefault || "isCustom" in token && token.isCustom);
145
+ const filterMapEnabledTokens = tokensById => Object.fromEntries(Object.entries(tokensById).filter(([, token]) => token.isDefault || "isCustom" in token && token.isCustom));
146
+ const distinctUntilIsEqual = rxjs.distinctUntilChanged((a, b) => isEqual__default.default(a, b));
147
+ const chains = get(chaindataProviderAtom).getNetworks$("polkadot").pipe(distinctUntilIsEqual, rxjs.map(filterTestnets), distinctUntilIsEqual);
148
+ const chainsById = get(chaindataProviderAtom).getNetworksMapById$("polkadot").pipe(distinctUntilIsEqual, rxjs.map(filterMapTestnets), distinctUntilIsEqual);
149
+ const chainsByGenesisHash = get(chaindataProviderAtom).getNetworksMapByGenesisHash$().pipe(distinctUntilIsEqual, rxjs.map(filterMapTestnets), distinctUntilIsEqual);
150
+ const evmNetworks = get(chaindataProviderAtom).getNetworks$("ethereum").pipe(distinctUntilIsEqual, rxjs.map(filterTestnets), distinctUntilIsEqual);
151
+ const networks = get(chaindataProviderAtom).getNetworks$().pipe(distinctUntilIsEqual, rxjs.map(filterTestnets), distinctUntilIsEqual);
152
+ const evmNetworksById = get(chaindataProviderAtom).getNetworksMapById$("ethereum").pipe(distinctUntilIsEqual, rxjs.map(filterMapTestnets), distinctUntilIsEqual);
153
+ const networksById = get(chaindataProviderAtom).getNetworksMapById$().pipe(distinctUntilIsEqual, rxjs.map(filterMapTestnets), distinctUntilIsEqual);
154
+ const tokens = get(chaindataProviderAtom).tokens$.pipe(distinctUntilIsEqual, rxjs.map(filterEnabledTokens), distinctUntilIsEqual);
155
+ const tokensById = get(chaindataProviderAtom).getTokensMapById$().pipe(distinctUntilIsEqual, rxjs.map(filterMapEnabledTokens), distinctUntilIsEqual);
156
+ const miniMetadatasObservable = dexieToRxjs(dexie.liveQuery(() => balances.db.miniMetadatas.toArray()));
157
+ const miniMetadatas = rxjs.combineLatest([miniMetadatasObservable.pipe(distinctUntilIsEqual), chainsById]).pipe(rxjs.map(([miniMetadatas, chainsById]) => miniMetadatas.filter(m => chainsById[m.chainId])), distinctUntilIsEqual);
158
+ return rxjs.combineLatest({
159
+ networks,
160
+ networksById,
161
+ chains,
162
+ chainsById,
163
+ chainsByGenesisHash,
164
+ evmNetworks,
165
+ evmNetworksById,
166
+ tokens,
167
+ tokensById,
168
+ miniMetadatas
169
+ }).pipe(
170
+ // debounce to prevent hammering UI with updates
171
+ util.firstThenDebounce(1_000), distinctUntilIsEqual);
172
+ });
173
+
174
+ const cryptoWaitReadyAtom = jotai.atom(async () => await utilCrypto.cryptoWaitReady());
175
+
101
176
  const tokenRatesAtom = jotai.atom(async get => {
102
177
  // runs a timer to keep tokenRates up to date
103
178
  get(tokenRatesFetcherAtomEffect);
@@ -118,9 +193,9 @@ const tokenRatesFetcherAtomEffect = jotaiEffect.atomEffect(get => {
118
193
 
119
194
  // we have to get these synchronously so that jotai knows to restart our timer when they change
120
195
  const coinsApiConfig = get(coinsApiConfigAtom);
121
- const tokensPromise = get(tokensAtom);
196
+ const tokensByIdPromise = get(tokensByIdAtom);
122
197
  (async () => {
123
- const tokensById = lodashEs.keyBy(await tokensPromise, "id");
198
+ const tokensById = await tokensByIdPromise;
124
199
  const tokenIds = Object.keys(tokensById);
125
200
  const loopMs = 300_000; // 300_000ms = 300s = 5 minutes
126
201
  const retryTimeout = 5_000; // 5_000ms = 5 seconds
@@ -161,42 +236,213 @@ const tokenRatesFetcherAtomEffect = jotaiEffect.atomEffect(get => {
161
236
  return () => abort.abort("Unsubscribed");
162
237
  });
163
238
 
164
- const addressesByTokenIdAtom = jotai.atom(async get => {
165
- const [tokens, addresses] = await Promise.all([get(tokensAtom), get(allAddressesAtom)]);
166
- return lodashEs.fromPairs(tokens.map(token => [token.id, addresses]));
239
+ const allBalancesAtom = jotai.atom(async get => {
240
+ // set up our subscription to fetch balances from the various blockchains
241
+ get(balancesSubscriptionAtomEffect);
242
+ const [balances$1, hydrateData] = await Promise.all([get(balancesObservableAtom), get(balancesHydrateDataAtom)]);
243
+ return new balances.Balances(Object.values(balances$1).filter(balance => !!hydrateData?.tokens?.[balance.tokenId]),
244
+ // hydrate balance chains, evmNetworks, tokens and tokenRates
245
+ hydrateData);
167
246
  });
168
- const rawBalancesAtom = jotai.atom([]);
169
- const subscribeBalancesAtom = jotaiEffect.atomEffect((get, set) => {
170
- const unsub = (async () => {
171
- const balancesProvider = get(balancesProviderAtom);
172
- const addressesByTokenId = await get(addressesByTokenIdAtom);
173
- const sub = balancesProvider.getBalances$(addressesByTokenId).subscribe(balances => {
174
- set(rawBalancesAtom, balances.balances);
175
- });
176
- return () => {
177
- return sub.unsubscribe();
178
- };
179
- })();
180
- return () => {
181
- unsub.then(unsubscribe => unsubscribe());
182
- };
247
+ const balancesObservable = new rxjs.BehaviorSubject({});
248
+ const balancesObservableAtom = utils.atomWithObservable(() => balancesObservable);
249
+ const balancesPersistBackendAtom = jotai.atom(localStorageBalancesPersistBackend);
250
+ const hydrateBalancesObservableAtom = jotai.atom(async get => {
251
+ const persistBackend = get(balancesPersistBackendAtom);
252
+ const balances$1 = await persistBackend.retrieve();
253
+ balancesObservable.next(Object.fromEntries(balances$1.map(b => [balances.getBalanceId(b), {
254
+ ...b,
255
+ status: "cache"
256
+ }])));
183
257
  });
184
258
  const balancesHydrateDataAtom = jotai.atom(async get => {
185
- const [chaindata, tokenRates] = await Promise.all([get(chaindataAtom), get(tokenRatesAtom)]);
186
- const networksById = lodashEs.keyBy(chaindata.networks, "id");
187
- const tokensById = lodashEs.keyBy(chaindata.tokens, "id");
259
+ const [{
260
+ networksById,
261
+ tokensById
262
+ }, tokenRates] = await Promise.all([get(chaindataAtom), get(tokenRatesAtom)]);
188
263
  return {
189
264
  networks: networksById,
190
265
  tokens: tokensById,
191
266
  tokenRates
192
267
  };
193
268
  });
194
- const balancesAtom = jotai.atom(async get => {
195
- // subscribe to balancesProvider getBalance with addressesByTokenIdAtom as param
196
- get(subscribeBalancesAtom);
197
- const hydrate = await get(balancesHydrateDataAtom);
198
- const rawBalances = get(rawBalancesAtom);
199
- return new balances.Balances(rawBalances, hydrate);
269
+ const balancesSubscriptionAtomEffect = jotaiEffect.atomEffect(get => {
270
+ // lets us tear down the existing subscriptions when the atomEffect is restarted
271
+ const abort = new AbortController();
272
+
273
+ // we have to specify these synchronously, otherwise jotai won't know
274
+ // that it needs to restart our subscriptions when they change
275
+ const atomDependencies = Promise.all([get(cryptoWaitReadyAtom), get(balanceModulesAtom), get(allAddressesAtom), get(chainsAtom), get(chainsByIdAtom), get(evmNetworksAtom), get(evmNetworksByIdAtom), get(tokensAtom), get(tokensByIdAtom), get(miniMetadatasAtom), get(enabledChainsAtom), get(enabledTokensAtom), get(hydrateBalancesObservableAtom)]);
276
+ const persistBackend = get(balancesPersistBackendAtom);
277
+ const unsubsPromise = (async () => {
278
+ const [_cryptoReady, balanceModules, allAddresses, chains, chainsById, evmNetworks, evmNetworksById, tokens, tokensById, _miniMetadatas, enabledChainsConfig, enabledTokensConfig] = await atomDependencies;
279
+ if (abort.signal.aborted) return;
280
+
281
+ // persist data every thirty seconds
282
+ balancesObservable.pipe(rxjs.debounceTime(10000)).subscribe(balancesUpdate => {
283
+ persistBackend.persist(Object.values(balancesUpdate));
284
+ });
285
+ const updateBalances = async balancesUpdates => {
286
+ if (abort.signal.aborted) return;
287
+ const updatesWithIds = new balances.Balances(balancesUpdates);
288
+ const existing = balancesObservable.value;
289
+
290
+ // update initialising set here - before filtering out zero balances
291
+ // while this may include stale balances, the important thing is that the balance is no longer "initialising"
292
+ // balancesUpdates.forEach((b) => this.#initialising.delete(getBalanceId(b)))
293
+
294
+ const newlyZeroBalances = [];
295
+ const changedBalances = Object.fromEntries(updatesWithIds.each.filter(newB => {
296
+ const isZero = newB.total.tokens === "0";
297
+ // Keep new balances which are not zeros
298
+ const existingB = existing[newB.id];
299
+ if (!existingB && !isZero) return true;
300
+ const hasChanged = !lodash.isEqual(existingB, newB.toJSON());
301
+ // Collect balances now confirmed to be zero separately, so they can be filtered out from the main set
302
+ if (existingB && hasChanged && isZero) newlyZeroBalances.push(newB.id);
303
+ // Keep changed balances, which are not known zeros
304
+ return hasChanged && !isZero;
305
+ }).map(b => [b.id, b.toJSON()]));
306
+ if (Object.keys(changedBalances).length === 0 && newlyZeroBalances.length === 0) return;
307
+ const nonZeroBalances = newlyZeroBalances.length > 0 ? Object.fromEntries(Object.entries(existing).filter(([id]) => !newlyZeroBalances.includes(id))) : existing;
308
+ const newBalancesState = {
309
+ ...nonZeroBalances,
310
+ ...changedBalances
311
+ };
312
+ if (Object.keys(newBalancesState).length === 0) return;
313
+ balancesObservable.next(newBalancesState);
314
+ };
315
+ const deleteBalances = async balancesFilter => {
316
+ if (abort.signal.aborted) return;
317
+ const balancesToKeep = Object.fromEntries(new balances.Balances(Object.values(await get(balancesObservableAtom))).each.filter(b => !balancesFilter(b)).map(b => [b.id, b.toJSON()]));
318
+ balancesObservable.next(balancesToKeep);
319
+ };
320
+ const enabledChainIds = enabledChainsConfig?.map(genesisHash => chains.find(chain => chain.genesisHash === genesisHash)?.id);
321
+ const enabledChainsFilter = enabledChainIds ? token => token.platform === "polkadot" && enabledChainIds?.includes(token.networkId) : () => true;
322
+ const enabledTokensFilter = enabledTokensConfig ? token => enabledTokensConfig.includes(token.id) : () => true;
323
+ const enabledTokenIds = tokens.filter(enabledChainsFilter).filter(enabledTokensFilter).map(({
324
+ id
325
+ }) => id);
326
+ if (enabledTokenIds.length < 1 || allAddresses.length < 1) return;
327
+ const addressesByTokenByModule = {};
328
+ enabledTokenIds.flatMap(tokenId => tokensById[tokenId]).forEach(token => {
329
+ // filter out tokens on chains/evmNetworks which have no rpcs
330
+ const hasRpcs = token.networkId && (chainsById[token.networkId]?.rpcs?.length ?? 0) > 0 || token.networkId && (evmNetworksById[token.networkId]?.rpcs?.length ?? 0) > 0;
331
+ if (!hasRpcs) return;
332
+ if (!addressesByTokenByModule[token.type]) addressesByTokenByModule[token.type] = {};
333
+ addressesByTokenByModule[token.type][token.id] = allAddresses.filter(address => {
334
+ // for each address, fetch balances only from compatible chains
335
+ return util.isEthereumAddress(address) ? token.networkId || chainsById[token.networkId ?? ""]?.account === "secp256k1" : token.networkId && chainsById[token.networkId ?? ""]?.account !== "secp256k1";
336
+ });
337
+ });
338
+
339
+ // Delete invalid cached balances
340
+ const chainIds = new Set(chains.map(chain => chain.id));
341
+ const evmNetworkIds = new Set(evmNetworks.map(evmNetwork => evmNetwork.id));
342
+ await deleteBalances(balance => {
343
+ // delete cached balances for accounts which don't exist anymore
344
+ if (!balance.address || !allAddresses.includes(balance.address)) return true;
345
+
346
+ // delete cached balances when chain/evmNetwork doesn't exist
347
+ if (!chainIds.has(balance.networkId) && !evmNetworkIds.has(balance.networkId)) return true;
348
+
349
+ // delete cached balance when token doesn't exist / is disabled
350
+ if (!enabledTokenIds.includes(balance.tokenId)) return true;
351
+
352
+ // delete cached balance when module doesn't exist
353
+ if (!balanceModules.find(module => module.type === balance.source)) return true;
354
+
355
+ // delete cached balance for accounts on incompatible chains
356
+ // (substrate accounts shouldn't have evm balances)
357
+ // (evm accounts shouldn't have substrate balances (unless the chain uses secp256k1 accounts))
358
+ const chain = chains.find(({
359
+ id
360
+ }) => id === balance.networkId) || null;
361
+ const hasChain = chainIds.has(balance.networkId);
362
+ const hasEvmNetwork = evmNetworkIds.has(balance.networkId);
363
+ const chainUsesSecp256k1Accounts = chain?.account === "secp256k1";
364
+ if (!util.isEthereumAddress(balance.address)) {
365
+ if (!hasChain) return true;
366
+ if (chainUsesSecp256k1Accounts) return true;
367
+ }
368
+ if (util.isEthereumAddress(balance.address)) {
369
+ if (!hasEvmNetwork && !chainUsesSecp256k1Accounts) return true;
370
+ }
371
+
372
+ // keep balance
373
+ return false;
374
+ });
375
+ if (abort.signal.aborted) return;
376
+
377
+ // after 30 seconds, change the status of all balances still initializing to stale
378
+ setTimeout(() => {
379
+ if (abort.signal.aborted) return;
380
+ const staleObservable = balancesObservable.pipe(rxjs.map(val => Object.values(val).filter(({
381
+ status
382
+ }) => status === "cache").map(balance => ({
383
+ ...balance,
384
+ status: "stale"
385
+ }))));
386
+ rxjs.firstValueFrom(staleObservable).then(v => {
387
+ if (v.length) updateBalances(v);
388
+ });
389
+ }, 30_000);
390
+ return balanceModules.map(balanceModule => {
391
+ const unsub = balances.balances(balanceModule, addressesByTokenByModule[balanceModule.type] ?? {}, (error, balances) => {
392
+ // log errors
393
+ if (error) {
394
+ if (error?.type === "STALE_RPC_ERROR" || error?.type === "WEBSOCKET_ALLOCATION_EXHAUSTED_ERROR") {
395
+ const addressesByModuleToken = addressesByTokenByModule[balanceModule.type] ?? {};
396
+ const staleObservable = balancesObservable.pipe(rxjs.map(val => Object.values(val).filter(balance => {
397
+ const {
398
+ tokenId,
399
+ address,
400
+ source
401
+ } = balance;
402
+ const chainComparison = error.chainId ? "chainId" in balance && error.chainId === balance.chainId : error.evmNetworkId ? "evmNetworkId" in balance && error.evmNetworkId === balance.evmNetworkId : true;
403
+ return chainComparison && addressesByModuleToken[tokenId]?.includes(address) && source === balanceModule.type;
404
+ }).map(balance => ({
405
+ ...balance,
406
+ status: "stale"
407
+ }))));
408
+ rxjs.firstValueFrom(staleObservable).then(v => {
409
+ if (v.length) updateBalances(v);
410
+ });
411
+ }
412
+ return log.error(`Failed to fetch ${balanceModule.type} balances`, error);
413
+ }
414
+
415
+ // ignore empty balance responses
416
+ if (!balances) return;
417
+ // ignore balances from old subscriptions which are still in the process of unsubscribing
418
+ if (abort.signal.aborted) return;
419
+
420
+ // good balances
421
+ if (balances) {
422
+ if ("status" in balances) {
423
+ // For modules using the new SubscriptionResultWithStatus pattern
424
+ //TODO fix initialisin
425
+ // if (result.status === "initialising") this.#initialising.add(balanceModule.type)
426
+ // else this.#initialising.delete(balanceModule.type)
427
+ updateBalances(balances.data);
428
+ } else {
429
+ // add good ones to initialisedBalances
430
+ updateBalances(Object.values(balances.toJSON()));
431
+ }
432
+ }
433
+ });
434
+ return () => unsub.then(unsubscribe => unsubscribe());
435
+ });
436
+ })();
437
+
438
+ // close the existing subscriptions when our effect unmounts
439
+ // (wait 2 seconds before actually unsubscribing, to allow the websocket to be reused in that time)
440
+ const unsubscribe = () => unsubsPromise.then(unsubs => {
441
+ persistBackend.persist(Object.values(balancesObservable.value));
442
+ unsubs?.forEach(unsub => unsub());
443
+ });
444
+ abort.signal.addEventListener("abort", () => setTimeout(unsubscribe, 2_000));
445
+ return () => abort.abort("Unsubscribed");
200
446
  });
201
447
 
202
448
  const useSetBalancesAddresses = addresses => {
@@ -213,8 +459,10 @@ const useSetBalancesAddresses = addresses => {
213
459
  * @returns a Balances object containing the current balances state.
214
460
  */
215
461
 
216
- const useBalances = () => {
217
- return jotai.useAtomValue(balancesAtom);
462
+ const useBalances = persistBackend => {
463
+ const [, setPersistBackend] = jotai.useAtom(balancesPersistBackendAtom);
464
+ if (persistBackend) setPersistBackend(persistBackend);
465
+ return jotai.useAtomValue(allBalancesAtom);
218
466
  };
219
467
 
220
468
  // TODO: Extract to shared definition between extension and @talismn/balances-react
@@ -251,41 +499,30 @@ const useChainConnectors = () => jotai.useAtomValue(chainConnectorsAtom);
251
499
 
252
500
  const useChaindataProvider = () => jotai.useAtomValue(chaindataProviderAtom);
253
501
  const useChaindata = () => jotai.useAtomValue(chaindataAtom);
254
- const useNetworks = () => useChaindata().networks;
255
- const useNetworksById = () => {
256
- const {
257
- networks
258
- } = useChaindata();
259
- return react.useMemo(() => lodashEs.keyBy(networks, n => n.id), [networks]);
260
- };
261
- const useNetwork = networkId => {
262
- const networksById = useNetworksById();
263
- return networksById[networkId ?? ""] ?? null;
264
- };
265
- const useTokens = () => useChaindata().tokens;
266
- const useTokensById = () => {
267
- const {
268
- tokens
269
- } = useChaindata();
270
- return react.useMemo(() => lodashEs.keyBy(tokens, t => t.id), [tokens]);
271
- };
272
- const useToken = tokenId => {
273
- const tokensById = useTokensById();
274
- return tokensById[tokenId ?? ""] ?? null;
275
- };
502
+ const useChains = () => jotai.useAtomValue(chainsByIdAtom);
503
+ const useChainsByGenesisHash = () => jotai.useAtomValue(chainsByGenesisHashAtom);
504
+ const useEvmNetworks = () => jotai.useAtomValue(evmNetworksByIdAtom);
505
+ const useTokens = () => jotai.useAtomValue(tokensByIdAtom);
506
+ const useMiniMetadatas = () => jotai.useAtomValue(miniMetadatasAtom);
507
+ const useChain = chainId => useChains()[chainId ?? ""] ?? undefined;
508
+ const useEvmNetwork = evmNetworkId => useEvmNetworks()[evmNetworkId ?? ""] ?? undefined;
509
+ const useToken = tokenId => useTokens()[tokenId ?? ""] ?? undefined;
276
510
 
277
511
  const useTokenRates = () => jotai.useAtomValue(tokenRatesAtom);
278
512
  const useTokenRate = tokenId => useTokenRates()[tokenId ?? ""] ?? undefined;
279
513
 
280
- const cryptoWaitReadyAtom = jotai.atom(async () => await utilCrypto.cryptoWaitReady());
281
-
282
514
  const BalancesProvider = ({
515
+ balanceModules,
283
516
  coinsApiUrl,
284
517
  withTestnets,
285
518
  enabledChains,
286
519
  enabledTokens,
287
520
  children
288
521
  }) => {
522
+ const setBalanceModules = jotai.useSetAtom(balanceModuleCreatorsAtom);
523
+ react.useEffect(() => {
524
+ if (balanceModules !== undefined) setBalanceModules(balanceModules);
525
+ }, [balanceModules, setBalanceModules]);
289
526
  const setCoinsApiConfig = jotai.useSetAtom(coinsApiConfigAtom);
290
527
  react.useEffect(() => {
291
528
  setCoinsApiConfig({
@@ -335,30 +572,41 @@ Object.defineProperty(exports, "subTokensTokenId", {
335
572
  });
336
573
  exports.BalancesProvider = BalancesProvider;
337
574
  exports.allAddressesAtom = allAddressesAtom;
338
- exports.balancesAtom = balancesAtom;
575
+ exports.allBalancesAtom = allBalancesAtom;
576
+ exports.balanceModuleCreatorsAtom = balanceModuleCreatorsAtom;
577
+ exports.balanceModulesAtom = balanceModulesAtom;
578
+ exports.balancesPersistBackendAtom = balancesPersistBackendAtom;
339
579
  exports.chainConnectorsAtom = chainConnectorsAtom;
340
580
  exports.chaindataAtom = chaindataAtom;
341
581
  exports.chaindataProviderAtom = chaindataProviderAtom;
582
+ exports.chainsAtom = chainsAtom;
583
+ exports.chainsByGenesisHashAtom = chainsByGenesisHashAtom;
584
+ exports.chainsByIdAtom = chainsByIdAtom;
342
585
  exports.coinsApiConfigAtom = coinsApiConfigAtom;
343
586
  exports.cryptoWaitReadyAtom = cryptoWaitReadyAtom;
344
587
  exports.enableTestnetsAtom = enableTestnetsAtom;
345
588
  exports.enabledChainsAtom = enabledChainsAtom;
346
589
  exports.enabledTokensAtom = enabledTokensAtom;
590
+ exports.evmNetworksAtom = evmNetworksAtom;
591
+ exports.evmNetworksByIdAtom = evmNetworksByIdAtom;
347
592
  exports.getStaleChains = getStaleChains;
348
- exports.networksAtom = networksAtom;
593
+ exports.miniMetadatasAtom = miniMetadatasAtom;
349
594
  exports.tokenRatesAtom = tokenRatesAtom;
350
595
  exports.tokensAtom = tokensAtom;
596
+ exports.tokensByIdAtom = tokensByIdAtom;
351
597
  exports.useBalances = useBalances;
352
598
  exports.useBalancesStatus = useBalancesStatus;
599
+ exports.useChain = useChain;
353
600
  exports.useChainConnectors = useChainConnectors;
354
601
  exports.useChaindata = useChaindata;
355
602
  exports.useChaindataProvider = useChaindataProvider;
356
- exports.useNetwork = useNetwork;
357
- exports.useNetworks = useNetworks;
358
- exports.useNetworksById = useNetworksById;
603
+ exports.useChains = useChains;
604
+ exports.useChainsByGenesisHash = useChainsByGenesisHash;
605
+ exports.useEvmNetwork = useEvmNetwork;
606
+ exports.useEvmNetworks = useEvmNetworks;
607
+ exports.useMiniMetadatas = useMiniMetadatas;
359
608
  exports.useSetBalancesAddresses = useSetBalancesAddresses;
360
609
  exports.useToken = useToken;
361
610
  exports.useTokenRate = useTokenRate;
362
611
  exports.useTokenRates = useTokenRates;
363
612
  exports.useTokens = useTokens;
364
- exports.useTokensById = useTokensById;