16655856454

Committed 31 Jul 2025 05:31PM UTC coverage: 38.319% (-0.8%) from 39.148%

Build # 16655856454

Build Type

Pull #12697

github

Committed by

web-flow

Commit Message

Merge 3dc4909bd into febfac2fb

Pull Request Pull Request #12697: Topic Subscriptions

Run Details

1860 of 5234 branches covered (35.54%)

Branch coverage included in aggregate %.

3 of 31 new or added lines in 9 files covered. (9.68%)

107 existing lines in 6 files now uncovered.

3297 of 8224 relevant lines covered (40.09%)

35.05 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

1.67

/libs/model/src/services/tokenBalanceCache/tokenBalanceCache.ts

import { logger, stats } from '@hicommonwealth/core';
import { BalanceSourceType } from '@hicommonwealth/shared';
import { getCosmosBalances, getEvmBalances } from './providers';
import { getSolanaBalances } from './providers/getSolanaBalances';
import { getSuiBalances } from './providers/getSuiBalances';
import {
  Balances,
  GetBalancesOptions,
  GetCosmosBalancesOptions,
  GetErcBalanceOptions,
  GetEvmBalancesOptions,
  GetSPLBalancesOptions,
  GetSuiNativeBalanceOptions,
  GetSuiNftBalanceOptions,
  GetSuiTokenBalanceOptions,
} from './types';

const log = logger(import.meta);

/**
 * This is the main function through which all balances should be fetched.
 * This function supports all balance sources and is fully compatible with Redis caching.
 */
export async function getBalances(
  options: GetBalancesOptions,
  ttl?: number,
): Promise<Balances> {
  if (options.addresses.length === 0) return {};

  let balances: Balances = {};

  try {
    if (
      options.balanceSourceType === BalanceSourceType.CosmosNative ||
      options.balanceSourceType === BalanceSourceType.CW20 ||
      options.balanceSourceType === BalanceSourceType.CW721
    ) {
      balances = await getCosmosBalances(options, ttl);
    } else if (
      options.balanceSourceType == BalanceSourceType.SPL ||
      options.balanceSourceType == BalanceSourceType.SOLNFT
    ) {
      balances = await getSolanaBalances(options, ttl);
    } else if (
      options.balanceSourceType === BalanceSourceType.SuiNative ||
      options.balanceSourceType === BalanceSourceType.SuiToken ||
      options.balanceSourceType === BalanceSourceType.SuiNFT
    ) {
      balances = await getSuiBalances(
        options as
          | GetSuiNativeBalanceOptions
          | GetSuiTokenBalanceOptions
          | GetSuiNftBalanceOptions,
        ttl,
      );
    } else {
      balances = await getEvmBalances(options as GetEvmBalancesOptions, ttl);
    }
  } catch (e) {
    const chainId =
      options.balanceSourceType === BalanceSourceType.SPL
        ? 'solana'
        : options.balanceSourceType === BalanceSourceType.SuiNative ||
            options.balanceSourceType === BalanceSourceType.SuiToken ||
            options.balanceSourceType === BalanceSourceType.SuiNFT
          ? (
              options as
                | GetSuiNativeBalanceOptions
                | GetSuiTokenBalanceOptions
                | GetSuiNftBalanceOptions
            ).sourceOptions.suiNetwork
          : (options as GetEvmBalancesOptions).sourceOptions.evmChainId ||
            (options as GetCosmosBalancesOptions).sourceOptions.cosmosChainId;

    const contractAddress =
      (options as GetSPLBalancesOptions).mintAddress ||
      (options as GetErcBalanceOptions).sourceOptions.contractAddress ||
      (options as GetSuiNativeBalanceOptions).sourceOptions.objectId ||
      (options.balanceSourceType === BalanceSourceType.SuiToken
        ? (options as GetSuiTokenBalanceOptions).sourceOptions.coinType
        : options.balanceSourceType === BalanceSourceType.SuiNFT
          ? (options as GetSuiNftBalanceOptions).sourceOptions.collectionId
          : undefined);

    const msg =
      `Failed to fetch balance(s) for ${options.addresses.length} address(es)` +
      `on chain ${chainId}${contractAddress ? ' for contract ' : ''}${
        contractAddress || ''
      }`;

    log.error(msg, e instanceof Error ? e : undefined, {
      fingerprint: `TBC: ${chainId} - ${contractAddress}`,
      addresses: options.addresses.slice(0, 5),
      contractAddress: (options as GetErcBalanceOptions).sourceOptions
        ?.contractAddress,
      evmChainId: (options as GetEvmBalancesOptions).sourceOptions?.evmChainId,
      cosmosChainId: (options as GetCosmosBalancesOptions).sourceOptions
        ?.cosmosChainId,
      suiNetwork: (
        options as
          | GetSuiNativeBalanceOptions
          | GetSuiTokenBalanceOptions
          | GetSuiNftBalanceOptions
      ).sourceOptions?.suiNetwork,
      objectId: (options as GetSuiNativeBalanceOptions).sourceOptions?.objectId,
      coinType:
        options.balanceSourceType === BalanceSourceType.SuiToken
          ? (options as GetSuiTokenBalanceOptions).sourceOptions.coinType
          : undefined,
      collectionId:
        options.balanceSourceType === BalanceSourceType.SuiNFT
          ? (options as GetSuiNftBalanceOptions).sourceOptions.collectionId
          : undefined,
    });
  }

  stats().incrementBy(
    'tbc.successful.balance.fetch',
    Object.keys(balances).length,
    {
      balance_source_type: options.balanceSourceType,
    },
  );
  return balances;
}

1	import { logger, stats } from '@hicommonwealth/core';
2	import { BalanceSourceType } from '@hicommonwealth/shared';
3	import { getCosmosBalances, getEvmBalances } from './providers';
4	import { getSolanaBalances } from './providers/getSolanaBalances';
5	import { getSuiBalances } from './providers/getSuiBalances';
6	import {
7	Balances,
8	GetBalancesOptions,
9	GetCosmosBalancesOptions,
10	GetErcBalanceOptions,
11	GetEvmBalancesOptions,
12	GetSPLBalancesOptions,
13	GetSuiNativeBalanceOptions,
14	GetSuiNftBalanceOptions,
15	GetSuiTokenBalanceOptions,
16	} from './types';
17
18	const log = logger(import.meta);	28✔
19
20	/**
21	* This is the main function through which all balances should be fetched.
22	* This function supports all balance sources and is fully compatible with Redis caching.
23	*/
24	export async function getBalances(
25	options: GetBalancesOptions,
26	ttl?: number,
27	): Promise<Balances> {
UNCOV 28	if (options.addresses.length === 0) return {};	×
29
UNCOV 30	let balances: Balances = {};	×
31
32	try {	×
UNCOV 33	if (	×
34	options.balanceSourceType === BalanceSourceType.CosmosNative \|\|	×
35	options.balanceSourceType === BalanceSourceType.CW20 \|\|
36	options.balanceSourceType === BalanceSourceType.CW721
37	) {
38	balances = await getCosmosBalances(options, ttl);	×
UNCOV 39	} else if (	×
40	options.balanceSourceType == BalanceSourceType.SPL \|\|	×
41	options.balanceSourceType == BalanceSourceType.SOLNFT
42	) {
43	balances = await getSolanaBalances(options, ttl);	×
UNCOV 44	} else if (	×
45	options.balanceSourceType === BalanceSourceType.SuiNative \|\|	×
46	options.balanceSourceType === BalanceSourceType.SuiToken \|\|
47	options.balanceSourceType === BalanceSourceType.SuiNFT
48	) {
UNCOV 49	balances = await getSuiBalances(	×
50	options as
51	\| GetSuiNativeBalanceOptions
52	\| GetSuiTokenBalanceOptions
53	\| GetSuiNftBalanceOptions,
54	ttl,
55	);
56	} else {
UNCOV 57	balances = await getEvmBalances(options as GetEvmBalancesOptions, ttl);	×
58	}
59	} catch (e) {
60	const chainId =
UNCOV 61	options.balanceSourceType === BalanceSourceType.SPL	×
62	? 'solana'
63	: options.balanceSourceType === BalanceSourceType.SuiNative \|\|	×
64	options.balanceSourceType === BalanceSourceType.SuiToken \|\|
65	options.balanceSourceType === BalanceSourceType.SuiNFT
66	? (
67	options as
68	\| GetSuiNativeBalanceOptions
69	\| GetSuiTokenBalanceOptions
70	\| GetSuiNftBalanceOptions
71	).sourceOptions.suiNetwork
72	: (options as GetEvmBalancesOptions).sourceOptions.evmChainId \|\|	×
73	(options as GetCosmosBalancesOptions).sourceOptions.cosmosChainId;
74
75	const contractAddress =
UNCOV 76	(options as GetSPLBalancesOptions).mintAddress \|\|	×
77	(options as GetErcBalanceOptions).sourceOptions.contractAddress \|\|
78	(options as GetSuiNativeBalanceOptions).sourceOptions.objectId \|\|
79	(options.balanceSourceType === BalanceSourceType.SuiToken	×
80	? (options as GetSuiTokenBalanceOptions).sourceOptions.coinType
81	: options.balanceSourceType === BalanceSourceType.SuiNFT	×
82	? (options as GetSuiNftBalanceOptions).sourceOptions.collectionId
83	: undefined);
84
85	const msg =
UNCOV 86	`Failed to fetch balance(s) for ${options.addresses.length} address(es)` +	×
87	`on chain ${chainId}${contractAddress ? ' for contract ' : ''}${	×
88	contractAddress \|\| ''	×
89	}`;
90
UNCOV 91	log.error(msg, e instanceof Error ? e : undefined, {	×
92	fingerprint: `TBC: ${chainId} - ${contractAddress}`,
93	addresses: options.addresses.slice(0, 5),
94	contractAddress: (options as GetErcBalanceOptions).sourceOptions
95	?.contractAddress,
96	evmChainId: (options as GetEvmBalancesOptions).sourceOptions?.evmChainId,
97	cosmosChainId: (options as GetCosmosBalancesOptions).sourceOptions
98	?.cosmosChainId,
99	suiNetwork: (
100	options as
101	\| GetSuiNativeBalanceOptions
102	\| GetSuiTokenBalanceOptions
103	\| GetSuiNftBalanceOptions
104	).sourceOptions?.suiNetwork,
105	objectId: (options as GetSuiNativeBalanceOptions).sourceOptions?.objectId,
106	coinType:
107	options.balanceSourceType === BalanceSourceType.SuiToken	×
108	? (options as GetSuiTokenBalanceOptions).sourceOptions.coinType
109	: undefined,
110	collectionId:
111	options.balanceSourceType === BalanceSourceType.SuiNFT	×
112	? (options as GetSuiNftBalanceOptions).sourceOptions.collectionId
113	: undefined,
114	});
115	}
116
UNCOV 117	stats().incrementBy(	×
118	'tbc.successful.balance.fetch',
119	Object.keys(balances).length,
120	{
121	balance_source_type: options.balanceSourceType,
122	},
123	);
UNCOV 124	return balances;	×
125	}

hicommonwealth / commonwealth / 16655856454

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous