feat: added getWalletAssets function to tonapi service

Author: Ho3einWave

src/services/tonapi/tonapi.service.ts

@@ -1,5 +1,6 @@
+import { address } from '@ton/ton';
 import { Services } from '../../core/services';
-import { Balance, CustomPayload } from '../../types/swap';
+import { Balance, CustomPayload, IJettonsRate, WalletAssets, WalletInfo } from '../../types/swap';
 
 export class TonApi extends Services {
     public async getJettonData(walletAddr: string, jettonAddress: string) {
@@ -21,4 +22,59 @@ export class TonApi extends Services {
 
         return data;
     }
+
+    /**
+     * getWalletAssets
+     */
+    public async getWalletAssets(
+        walletAddress: string,
+        currencies: string[] = ['usd'],
+        custom_payload: boolean = true,
+    ) {
+        const { balances } = await this.client.request.send<WalletAssets>({
+            baseURL: 'https://tonapi.io/v2',
+            maxBodyLength: Infinity,
+            url: `/accounts/${walletAddress}/jettons?currencies=${currencies.join(',')}${
+                custom_payload ? '&supported_extensions=custom_payload' : ''
+            }`,
+        });
+
+        const addresses = balances.map((item) => item.jetton.address).join(',');
+
+        const { rates } = await this.client.request.send<IJettonsRate>({
+            baseURL: 'https://tonapi.io/v2',
+            url: `/rates?tokens=${
+                addresses.length > 0 ? addresses : ''
+            },EQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAM9c&currencies=usd`,
+        });
+
+        const { balance: tonBalance } = await this.client.request.send<WalletInfo>({
+            baseURL: 'https://tonapi.io/v2',
+            url: `/accounts/${walletAddress}`,
+        });
+
+        balances.push({
+            balance: String(tonBalance),
+            wallet_address: {
+                address: walletAddress,
+                is_scam: false,
+                is_wallet: true,
+            },
+            jetton: {
+                address: 'EQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAM9c',
+                symbol: 'TON',
+                name: 'TON',
+                image: 'https://asset.ston.fi/img/EQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAM9c/4ecd4687e0b5b8ff21a7fbe03f9d281c26a2dc13eac7b7d16048cc693fe0ec39',
+                decimals: 9,
+                verification: '',
+            },
+        });
+        const newBalances = balances.reduce((map, item) => {
+            const userFriendlyAddr = address(item.jetton.address).toString();
+            item.jetton.prices = rates[userFriendlyAddr].prices;
+            map.set(userFriendlyAddr, item);
+            return map;
+        }, new Map<string, Balance>());
+        return newBalances;
+    }
 }

src/services/tonapi/tonapi.test.ts

@@ -16,6 +16,20 @@ test('it should get fail on random address', async () => {
     expect(async () => await client.tonapi.getJettonData(userWallet, 'hmstrJetton')).toThrow();
 });
 
+test('it should get user assets in type of map', async () => {
+    const data = await client.tonapi.getWalletAssets(
+        'UQAaIQh7jVlOEylI8jM8OI1O-yYGZRqUfJRxuR-K57pskl9I',
+    );
+    expect(data instanceof Map).toBeTrue();
+    expect(
+        data.get('EQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAM9c')?.jetton.name,
+    ).not.toBeUndefined();
+
+    expect(data.get('EQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAM9c')?.jetton.name).toEqual(
+        'TON',
+    );
+});
+
 // test('it should get custom payload for Hamster Kombat Token', async () => {
 //     const customPayload = await client.tonapi.getCustomPayload(userWallet, hmstrJetton);
 //     expect(customPayload.custom_payload).not.toBeUndefined();

src/types/swap.ts

@@ -1,14 +1,10 @@
 export interface Balance {
     balance: string;
+    price?: Price;
     wallet_address: WalletAddress;
     jetton: Jetton;
     extensions?: string[];
-}
-
-export interface WalletAddress {
-    address: string;
-    is_scam: boolean;
-    is_wallet: boolean;
+    lock?: Lock;
 }
 
 export interface Jetton {
@@ -21,15 +17,64 @@ export interface Jetton {
     prices?: Prices;
     custom_payload_api_uri?: string;
 }
-export interface Prices {
-    USD: number;
-}
 
 export interface CustomPayload {
     custom_payload: string;
     state_init: string;
 }
-export interface CustomPayload {
-    custom_payload: string;
-    state_init: string;
+
+export interface WalletAssets {
+    balances: Balance[];
+}
+export interface IJettonsRate {
+    rates: Rates;
+}
+
+export interface Rates {
+    [key: string]: Price;
+}
+export interface Price {
+    prices: Prices;
+    diff_24h: Diff24h;
+    diff_7d: Diff7d;
+    diff_30d: Diff30d;
+}
+
+export interface Prices {
+    [key: string]: number;
+}
+
+export interface Diff24h {
+    [key: string]: number;
+}
+
+export interface Diff7d {
+    [key: string]: number;
+}
+
+export interface Diff30d {
+    TON: string;
+}
+
+export interface WalletAddress {
+    address: string;
+    name?: string;
+    is_scam: boolean;
+    icon?: string;
+    is_wallet: boolean;
+}
+
+export interface Lock {
+    amount: string;
+    till: number;
+}
+
+export interface WalletInfo {
+    address: string;
+    balance: number;
+    last_activity: number;
+    status: string;
+    interfaces: string[];
+    get_methods: string[];
+    is_wallet: boolean;
 }