7741747152

Committed 01 Feb 2024 01:01PM UTC coverage: 93.245% (-0.07%) from 93.311%

Build # 7741747152

Build Type

Pull #1096

github

Committed by

hectorgomezv

Commit Message

Fix ZerionBalances response

Pull Request Pull Request #1096: Add Zerion Balances provider

Run Details

1502 of 1798 branches covered (0.0%)

Branch coverage included in aggregate %.

69 of 73 new or added lines in 5 files covered. (94.52%)

2 existing lines in 1 file now uncovered.

5745 of 5974 relevant lines covered (96.17%)

177.41 hits per line

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

82.54

/src/datasources/balances-api/zerion-balances-api.service.ts

import { IConfigurationService } from '@/config/configuration.service.interface';
import { ChainAttributes } from '@/datasources/balances-api/entities/provider-chain-attributes.entity';
import {
  ZerionAttributes,
  ZerionBalance,
  ZerionBalances,
} from '@/datasources/balances-api/entities/zerion-balance.entity';
import { CacheFirstDataSource } from '@/datasources/cache/cache.first.data.source';
import { CacheRouter } from '@/datasources/cache/cache.router';
import {
  CacheService,
  ICacheService,
} from '@/datasources/cache/cache.service.interface';
import {
  Balance,
  Erc20Balance,
  NativeBalance,
} from '@/domain/balances/entities/balance.entity';
import { getNumberString } from '@/domain/common/utils/utils';
import { DataSourceError } from '@/domain/errors/data-source.error';
import { IBalancesApi } from '@/domain/interfaces/balances-api.interface';
import { asError } from '@/logging/utils';
import { Inject, Injectable } from '@nestjs/common';

export const IZerionBalancesApi = Symbol('IZerionBalancesApi');

@Injectable()
export class ZerionBalancesApi implements IBalancesApi {
  private readonly apiKey: string | undefined;
  private readonly baseUri: string;
  private readonly chainsConfiguration: Record<number, ChainAttributes>;
  private readonly defaultExpirationTimeInSeconds: number;
  private readonly defaultNotFoundExpirationTimeSeconds: number;

  constructor(
    @Inject(CacheService) private readonly cacheService: ICacheService,
    @Inject(IConfigurationService)
    private readonly configurationService: IConfigurationService,
    private readonly dataSource: CacheFirstDataSource,
  ) {
    this.apiKey = this.configurationService.get<string>(
      'balances.providers.zerion.apiKey',
    );
    this.baseUri = this.configurationService.getOrThrow<string>(
      'balances.providers.zerion.baseUri',
    );
    this.defaultExpirationTimeInSeconds =
      this.configurationService.getOrThrow<number>(
        'expirationTimeInSeconds.default',
      );
    this.defaultNotFoundExpirationTimeSeconds =
      this.configurationService.getOrThrow<number>(
        'expirationTimeInSeconds.notFound.default',
      );
    this.chainsConfiguration = this.configurationService.getOrThrow<
      Record<number, ChainAttributes>
    >('balances.providers.zerion.chains');
  }

  async getBalances(args: {
    chainId: string;
    safeAddress: string;
    fiatCode: string;
  }): Promise<Balance[]> {
    try {
      const cacheDir = CacheRouter.getZerionBalancesCacheDir(args);
      const chainName = this._getChainName(args.chainId);
      const currency = args.fiatCode.toLowerCase();
      const url = `${this.baseUri}/v1/wallets/${args.safeAddress}/positions?filter[chain_ids]=${chainName}&currency=${currency}&sort=value`;
      const { data } = await this.dataSource.get<ZerionBalances>({
        cacheDir,
        url,
        notFoundExpireTimeSeconds: this.defaultNotFoundExpirationTimeSeconds,
        networkRequest: { headers: { Authorization: `Basic ${this.apiKey}` } },
        expireTimeSeconds: this.defaultExpirationTimeInSeconds,
      });
      return this._mapBalances(chainName, data);
    } catch (error) {
      throw new DataSourceError(
        `Error getting ${args.safeAddress} balances from provider: ${asError(error).message}}`,
      );
    }
  }

  private _mapBalances(
    chainName: string,
    zerionBalances: ZerionBalance[],
  ): Balance[] {
    return zerionBalances
      .filter((zb) => zb.attributes.flags.displayable === true)
      .map((zb) => {
        const implementation = zb.attributes.fungible_info.implementations.find(
          (implementation) => implementation.chain_id === chainName,
        );
        if (!implementation)
          throw Error(
            `Zerion error: ${chainName} implementation not found for balance ${zb.id}`,
          );
        const fiatBalance = getNumberString(zb.attributes.value ?? 0);
        const fiatConversion = getNumberString(zb.attributes.price);

        return {
          ...(implementation.address === null
            ? this._mapNativeBalance(zb.attributes)
            : this._mapErc20Balance(zb.attributes, implementation.address)),
          fiatBalance,
          fiatConversion,
        };
      });
  }

  private _mapErc20Balance(
    zerionBalanceAttributes: ZerionAttributes,
    tokenAddress: string,
  ): Erc20Balance {
    const { fungible_info, quantity } = zerionBalanceAttributes;
    return {
      tokenAddress,
      token: {
        name: fungible_info.name!,
        symbol: fungible_info.symbol!,
        decimals: quantity.decimals,
        logoUri: fungible_info.icon.url ?? '',
      },
      balance: quantity.int,
    };
  }

  private _mapNativeBalance(
    zerionBalanceAttributes: ZerionAttributes,
  ): NativeBalance {
    return {
      tokenAddress: null,
      token: null,
      balance: zerionBalanceAttributes.quantity.int,
    };
  }

  async clearBalances(args: {
    chainId: string;
    safeAddress: string;
  }): Promise<void> {
    const key = CacheRouter.getZerionBalancesCacheKey(args);
    await this.cacheService.deleteByKey(key);
  }

  private _getChainName(chainId: string): string {
    const chainName = this.chainsConfiguration[Number(chainId)]?.chainName;
    if (!chainName)
      throw Error(
        `Chain ${chainId} balances retrieval via Zerion is not configured`,
      );
    return chainName;
  }
}

1	import { IConfigurationService } from '@/config/configuration.service.interface';	84✔
2	import { ChainAttributes } from '@/datasources/balances-api/entities/provider-chain-attributes.entity';
3	import {
4	ZerionAttributes,
5	ZerionBalance,
6	ZerionBalances,
7	} from '@/datasources/balances-api/entities/zerion-balance.entity';
8	import { CacheFirstDataSource } from '@/datasources/cache/cache.first.data.source';	84✔
9	import { CacheRouter } from '@/datasources/cache/cache.router';	84✔
10	import {	84✔
11	CacheService,
12	ICacheService,
13	} from '@/datasources/cache/cache.service.interface';
14	import {
15	Balance,
16	Erc20Balance,
17	NativeBalance,
18	} from '@/domain/balances/entities/balance.entity';
19	import { getNumberString } from '@/domain/common/utils/utils';	84✔
20	import { DataSourceError } from '@/domain/errors/data-source.error';	84✔
21	import { IBalancesApi } from '@/domain/interfaces/balances-api.interface';
22	import { asError } from '@/logging/utils';	84✔
23	import { Inject, Injectable } from '@nestjs/common';	84✔
24
25	export const IZerionBalancesApi = Symbol('IZerionBalancesApi');	84✔
26
27	@Injectable()
28	export class ZerionBalancesApi implements IBalancesApi {	84✔
29	private readonly apiKey: string \| undefined;
30	private readonly baseUri: string;
31	private readonly chainsConfiguration: Record<number, ChainAttributes>;
32	private readonly defaultExpirationTimeInSeconds: number;
33	private readonly defaultNotFoundExpirationTimeSeconds: number;
34
35	constructor(
36	@Inject(CacheService) private readonly cacheService: ICacheService,	614✔
37	@Inject(IConfigurationService)
38	private readonly configurationService: IConfigurationService,	614✔
39	private readonly dataSource: CacheFirstDataSource,	614✔
40	) {
41	this.apiKey = this.configurationService.get<string>(	614✔
42	'balances.providers.zerion.apiKey',
43	);
44	this.baseUri = this.configurationService.getOrThrow<string>(	614✔
45	'balances.providers.zerion.baseUri',
46	);
47	this.defaultExpirationTimeInSeconds =	614✔
48	this.configurationService.getOrThrow<number>(
49	'expirationTimeInSeconds.default',
50	);
51	this.defaultNotFoundExpirationTimeSeconds =	614✔
52	this.configurationService.getOrThrow<number>(
53	'expirationTimeInSeconds.notFound.default',
54	);
55	this.chainsConfiguration = this.configurationService.getOrThrow<	614✔
56	Record<number, ChainAttributes>
57	>('balances.providers.zerion.chains');
58	}
59
60	async getBalances(args: {
61	chainId: string;
62	safeAddress: string;
63	fiatCode: string;
64	}): Promise<Balance[]> {
65	try {	8✔
66	const cacheDir = CacheRouter.getZerionBalancesCacheDir(args);	8✔
67	const chainName = this._getChainName(args.chainId);	8✔
68	const currency = args.fiatCode.toLowerCase();	8✔
69	const url = `${this.baseUri}/v1/wallets/${args.safeAddress}/positions?filter[chain_ids]=${chainName}&currency=${currency}&sort=value`;	8✔
70	const { data } = await this.dataSource.get<ZerionBalances>({	8✔
71	cacheDir,
72	url,
73	notFoundExpireTimeSeconds: this.defaultNotFoundExpirationTimeSeconds,
74	networkRequest: { headers: { Authorization: `Basic ${this.apiKey}` } },
75	expireTimeSeconds: this.defaultExpirationTimeInSeconds,
76	});
77	return this._mapBalances(chainName, data);	6✔
78	} catch (error) {
79	throw new DataSourceError(	2✔
80	`Error getting ${args.safeAddress} balances from provider: ${asError(error).message}}`,
81	);
82	}
83	}
84
85	private _mapBalances(
86	chainName: string,
87	zerionBalances: ZerionBalance[],
88	): Balance[] {
89	return zerionBalances	6✔
90	.filter((zb) => zb.attributes.flags.displayable === true)	8✔
91	.map((zb) => {
92	const implementation = zb.attributes.fungible_info.implementations.find(	8✔
93	(implementation) => implementation.chain_id === chainName,	12✔
94	);
95	if (!implementation)	8!
NEW 96	throw Error(	×
97	`Zerion error: ${chainName} implementation not found for balance ${zb.id}`,
98	);
99	const fiatBalance = getNumberString(zb.attributes.value ?? 0);	8!
100	const fiatConversion = getNumberString(zb.attributes.price);	8✔
101
102	return {	8✔
103	...(implementation.address === null	4✔
104	? this._mapNativeBalance(zb.attributes)
105	: this._mapErc20Balance(zb.attributes, implementation.address)),
106	fiatBalance,
107	fiatConversion,
108	};
109	});
110	}
111
112	private _mapErc20Balance(
113	zerionBalanceAttributes: ZerionAttributes,
114	tokenAddress: string,
115	): Erc20Balance {
116	const { fungible_info, quantity } = zerionBalanceAttributes;	4✔
117	return {	4✔
118	tokenAddress,
119	token: {
120	name: fungible_info.name!,
121	symbol: fungible_info.symbol!,
122	decimals: quantity.decimals,
123	logoUri: fungible_info.icon.url ?? '',	6!
124	},
125	balance: quantity.int,
126	};
127	}
128
129	private _mapNativeBalance(
130	zerionBalanceAttributes: ZerionAttributes,
131	): NativeBalance {
132	return {	4✔
133	tokenAddress: null,
134	token: null,
135	balance: zerionBalanceAttributes.quantity.int,
136	};
137	}
138
139	async clearBalances(args: {
140	chainId: string;
141	safeAddress: string;
142	}): Promise<void> {
NEW 143	const key = CacheRouter.getZerionBalancesCacheKey(args);	×
NEW 144	await this.cacheService.deleteByKey(key);	×
145	}
146
147	private _getChainName(chainId: string): string {
148	const chainName = this.chainsConfiguration[Number(chainId)]?.chainName;	8!
149	if (!chainName)	8!
NEW 150	throw Error(	×
151	`Chain ${chainId} balances retrieval via Zerion is not configured`,
152	);
153	return chainName;	8✔
154	}
155	}

safe-global / safe-client-gateway / 7741747152

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