diff --git a/src/services/BalanceUtilities.tsx b/src/services/BalanceUtilities.tsx index 2e3b4114..d1ba0cb4 100644 --- a/src/services/BalanceUtilities.tsx +++ b/src/services/BalanceUtilities.tsx @@ -4,6 +4,7 @@ import { PublicIdentity, BlockchainUtils, } from '@kiltprotocol/sdk-js' +import { Balances as KiltBalances } from '@kiltprotocol/types' import BN from 'bn.js' import React from 'react' import { Store } from 'redux' @@ -46,8 +47,10 @@ class BalanceUtilities { } public static async getMyBalance(identity: IMyIdentity): Promise { - const balance: BN = await Balance.getBalance(identity.identity.address) - return balance + const balance: KiltBalances = await Balance.getBalances( + identity.identity.address + ) + return balance.free } public static connectMyIdentities( @@ -117,21 +120,21 @@ class BalanceUtilities { private static listener( account: PublicIdentity['address'], - balance: BN, - change: BN + balance: KiltBalances, + change: KiltBalances ): void { - if (!change.isZero()) { - const inDeCreased = `${change.gtn(0) ? 'in' : 'de'}creased` + if (!change.free.isZero()) { + const inDeCreased = `${change.free.gtn(0) ? 'in' : 'de'}creased` notify(
Balance of {inDeCreased} by{' '} - . + .
) } persistentStoreInstance.store.dispatch( - Balances.Store.updateBalance(account, balance) + Balances.Store.updateBalance(account, balance.free) ) } }