Use CurrenciesApi in BalanceClient

galacticcouncil · Nov 5, 2024 · 0c3290f · 0c3290f
1 parent 9f2dce3
commit 0c3290f
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 21 deletions.
diff --git a/.changeset/fresh-chairs-carry.md b/.changeset/fresh-chairs-carry.md
@@ -0,0 +1,5 @@
+---
+'@galacticcouncil/sdk': minor
+---
+
+Use CurrenciesApi in BalanceClient, support for ERC20 tokens
diff --git a/.changeset/lazy-days-applaud.md b/.changeset/lazy-days-applaud.md
diff --git a/packages/sdk/src/client/BalanceClient.ts b/packages/sdk/src/client/BalanceClient.ts
@@ -1,4 +1,7 @@
 import { ApiPromise } from '@polkadot/api';
+import { u32, Vec } from '@polkadot/types';
+import { ITuple } from '@polkadot/types-codec/types';
+import { OrmlTokensAccountData } from '@polkadot/types/lookup';
 import { UnsubscribePromise } from '@polkadot/api-base/types';
 import { SYSTEM_ASSET_ID } from '../consts';
 import { BigNumber } from '../utils/bignumber';
@@ -10,26 +13,37 @@ export class BalanceClient extends PolkadotApiClient {
     super(api);
   }
 
-  async getBalance(accountId: string, tokenKey: string): Promise<BigNumber> {
-    return tokenKey === SYSTEM_ASSET_ID
-      ? await this.getSystemBalance(accountId)
-      : await this.getTokenBalance(accountId, tokenKey);
+  async getTokenBalanceData(accountId: string, tokenKey: string) {
+    const params = this.api.createType('(AssetId, AccountId)', [
+      tokenKey,
+      accountId,
+    ]);
+    const result = await this.api.rpc.state.call(
+      'CurrenciesApi_account',
+      params.toHex()
+    );
+    return this.api.createType<OrmlTokensAccountData>(
+      'OrmlTokensAccountData',
+      result
+    );
   }
 
-  async getSystemBalance(accountId: string): Promise<BigNumber> {
-    const { data } = await this.api.query.system.account(accountId);
-    return this.calculateFreeBalance(data);
-  }
+  async getAccountBalanceData(accountId: string) {
+    const params = this.api.createType('AccountId', accountId);
+    const result = await this.api.rpc.state.call(
+      'CurrenciesApi_accounts',
+      params.toHex()
+    );
 
-  async getTokenBalance(
-    accountId: string,
-    tokenKey: string
-  ): Promise<BigNumber> {
-    const { free, reserved, frozen } = await this.api.query.tokens.accounts(
-      accountId,
-      tokenKey
+    return this.api.createType<Vec<ITuple<[u32, OrmlTokensAccountData]>>>(
+      'Vec<(AssetId, OrmlTokensAccountData)>',
+      result
     );
-    return this.calculateFreeBalance({ free, feeFrozen: reserved, frozen });
+  }
+
+  async getBalance(accountId: string, tokenKey: string): Promise<BigNumber> {
+    const data = this.getTokenBalanceData(accountId, tokenKey);
+    return this.calculateFreeBalance(data);
   }
 
   async subscribeBalance(