status-im
/
safe-react
mirror of https://github.com/status-im/safe-react.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix token price function to be more generic

This commit is contained in:
Mati Dastugue 2020-12-28 14:09:30 -03:00
parent 6ffb7368cd
commit a00539e8c2
1 changed files with 9 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/routes/safe/components/Balances/dataFetcher.ts
View File

@ -13,7 +13,7 @@ export const BALANCE_TABLE_VALUE_ID = 'value'
const { nativeCoin } = getNetworkInfo() const { nativeCoin } = getNetworkInfo()
const getTokenPriceValue = (token: Token, currencyValues?: BalanceCurrencyList, currencyRate?: number): string => { const getTokenValue = (token: Token, currencyValues?: BalanceCurrencyList, currencyRate?: number): string => {
const currencyValue = currencyValues?.find(({ tokenAddress }) => { const currencyValue = currencyValues?.find(({ tokenAddress }) => {
if (token.address === nativeCoin.address && !tokenAddress) { if (token.address === nativeCoin.address && !tokenAddress) {
return true return true
@ -27,36 +27,16 @@ const getTokenPriceValue = (token: Token, currencyValues?: BalanceCurrencyList,
} }
const { balanceInBaseCurrency } = currencyValue const { balanceInBaseCurrency } = currencyValue
const balance = new BigNumber(balanceInBaseCurrency).times(currencyRate).toFixed(2) const balance = new BigNumber(balanceInBaseCurrency).times(currencyRate).toString()
return formatAmountInUsFormat(balance) return balance
} }
const getTokenPriceInCurrency = ( const getTokenPriceInCurrency = (balance: string, currencySelected?: string): string => {
token: Token,
currencySelected?: string,
currencyValues?: BalanceCurrencyList,
currencyRate?: number,
): string => {
if (!currencySelected) { if (!currencySelected) {
return '' return Number('').toFixed(2)
} }
const currencyValue = currencyValues?.find(({ tokenAddress }) => { return `${formatAmountInUsFormat(Number(balance).toFixed(2))} ${currencySelected}`
if (token.address === nativeCoin.address && !tokenAddress) {
return true
}
return token.address === tokenAddress
})
if (!currencyValue || !currencyRate) {
return ''
}
const { balanceInBaseCurrency } = currencyValue
const balance = new BigNumber(balanceInBaseCurrency).times(currencyRate).toFixed(2)
return `${formatAmountInUsFormat(balance)} ${currencySelected}`
} }
export interface BalanceData { export interface BalanceData {
@ -77,6 +57,7 @@ export const getBalanceData = (
): List<BalanceData> => { ): List<BalanceData> => {
const { nativeCoin } = getNetworkInfo() const { nativeCoin } = getNetworkInfo()
return activeTokens.map((token) => { return activeTokens.map((token) => {
const balance = getTokenValue(token, currencyValues, currencyRate)
return { return {
[BALANCE_TABLE_ASSET_ID]: { [BALANCE_TABLE_ASSET_ID]: {
name: token.name, name: token.name,
@ -88,8 +69,8 @@ export const getBalanceData = (
[BALANCE_TABLE_BALANCE_ID]: `${formatAmountInUsFormat(token.balance?.toString() || '0')} ${token.symbol}`, [BALANCE_TABLE_BALANCE_ID]: `${formatAmountInUsFormat(token.balance?.toString() || '0')} ${token.symbol}`,
balanceOrder: Number(token.balance), balanceOrder: Number(token.balance),
[FIXED]: token.symbol === nativeCoin.symbol, [FIXED]: token.symbol === nativeCoin.symbol,
[BALANCE_TABLE_VALUE_ID]: getTokenPriceInCurrency(token, currencySelected, currencyValues, currencyRate), [BALANCE_TABLE_VALUE_ID]: getTokenPriceInCurrency(balance, currencySelected),
valueOrder: Number(getTokenPriceValue(token, currencyValues, currencyRate)), valueOrder: Number(balance),
} }
}) })
} }