@@ -6,14 +6,14 @@ import {
66 setupBankExtension as StargateSetupBankExtension ,
77 setupDistributionExtension as StargateDistributionExtension ,
88 setupStakingExtension as StargateStakingExtension ,
9- coinFromProto ,
109 AuthExtension ,
1110 BankExtension ,
1211 StakingExtension ,
1312 DistributionExtension ,
1413} from '@cosmjs/stargate' ;
1514
16- import { LumWallet , LumUtils , LumTypes } from '..' ;
15+ import { BaseAccount } from '../codec/cosmos/auth/v1beta1/auth' ;
16+ import { LumWallet , LumUtils , LumTypes , LumRegistry } from '..' ;
1717
1818export class LumClient {
1919 readonly tmClient : Tendermint34Client ;
@@ -110,10 +110,11 @@ export class LumClient {
110110 * @param address wallet address
111111 */
112112 getAccount = async ( address : string ) : Promise < LumTypes . Account | null > => {
113- const account = await this . queryClient . auth . account ( address ) ;
114- if ( ! account ) {
113+ const anyAccount = await this . queryClient . auth . verified . account ( address ) ;
114+ if ( ! anyAccount ) {
115115 return null ;
116116 }
117+ const account = LumRegistry . decode ( anyAccount ) as BaseAccount ;
117118 return {
118119 address : account . address ,
119120 accountNumber : Uint64 . fromString ( account . accountNumber . toString ( ) ) . toNumber ( ) ,
@@ -127,10 +128,11 @@ export class LumClient {
127128 * @param address wallet address
128129 */
129130 getAccountUnverified = async ( address : string ) : Promise < LumTypes . Account | null > => {
130- const account = await this . queryClient . auth . unverified . account ( address ) ;
131- if ( ! account ) {
131+ const anyAccount = await this . queryClient . auth . account ( address ) ;
132+ if ( ! anyAccount ) {
132133 return null ;
133134 }
135+ const account = LumRegistry . decode ( anyAccount ) as BaseAccount ;
134136 return {
135137 address : account . address ,
136138 accountNumber : Uint64 . fromString ( account . accountNumber . toString ( ) ) . toNumber ( ) ,
@@ -145,8 +147,8 @@ export class LumClient {
145147 * @param searchDenom Coin denomination (ex: lum)
146148 */
147149 getBalance = async ( address : string , searchDenom : string ) : Promise < LumTypes . Coin | null > => {
148- const balance = await this . queryClient . bank . balance ( address , searchDenom ) ;
149- return balance ? coinFromProto ( balance ) : null ;
150+ const balance = await this . queryClient . bank . verified . balance ( address , searchDenom ) ;
151+ return balance ? balance : null ;
150152 } ;
151153
152154 /**
@@ -156,8 +158,8 @@ export class LumClient {
156158 * @param searchDenom Coin denomination (ex: lum)
157159 */
158160 getBalanceUnverified = async ( address : string , searchDenom : string ) : Promise < LumTypes . Coin | null > => {
159- const balance = await this . queryClient . bank . unverified . balance ( address , searchDenom ) ;
160- return balance ? coinFromProto ( balance ) : null ;
161+ const balance = await this . queryClient . bank . balance ( address , searchDenom ) ;
162+ return balance ? balance : null ;
161163 } ;
162164
163165 /**
@@ -166,8 +168,8 @@ export class LumClient {
166168 * @param address wallet address
167169 */
168170 getAllBalancesUnverified = async ( address : string ) : Promise < LumTypes . Coin [ ] > => {
169- const balances = await this . queryClient . bank . unverified . allBalances ( address ) ;
170- return balances . map ( coinFromProto ) ;
171+ const balances = await this . queryClient . bank . allBalances ( address ) ;
172+ return balances ;
171173 } ;
172174
173175 /**
@@ -176,16 +178,16 @@ export class LumClient {
176178 * @param searchDenom Coin denomination (ex: lum)
177179 */
178180 getSupply = async ( searchDenom : string ) : Promise < LumTypes . Coin | null > => {
179- const supply = await this . queryClient . bank . unverified . supplyOf ( searchDenom ) ;
180- return supply ? coinFromProto ( supply ) : null ;
181+ const supply = await this . queryClient . bank . supplyOf ( searchDenom ) ;
182+ return supply ? supply : null ;
181183 } ;
182184
183185 /**
184186 * Get all coins supplies
185187 */
186188 getAllSupplies = async ( ) : Promise < LumTypes . Coin [ ] > => {
187- const supplies = await this . queryClient . bank . unverified . totalSupply ( ) ;
188- return supplies . map ( coinFromProto ) ;
189+ const supplies = await this . queryClient . bank . totalSupply ( ) ;
190+ return supplies ;
189191 } ;
190192
191193 /**
@@ -234,9 +236,9 @@ export class LumClient {
234236 *
235237 * @param params Search params
236238 */
237- private txsQuery = async ( params : LumTypes . TxSearchParams ) : Promise < LumTypes . TxResponse [ ] > => {
239+ private txsQuery = async ( params : LumTypes . TxSearchParams ) : Promise < readonly LumTypes . TxResponse [ ] > => {
238240 const results = await this . tmClient . txSearch ( params ) ;
239- return results . txs as LumTypes . TxResponse [ ] ;
241+ return results . txs ;
240242 } ;
241243
242244 /**
0 commit comments