Add getLiquidityPoolAssets, handle liquidity pool asset in getBalanceByAccountID

Author: zaelgohary

packages/stellar_client/bin/stellar_client.dart

@@ -1,14 +1,13 @@
 import 'package:stellar_client/stellar_client.dart';
 
 void main() async {
-  final stellarClient = Client.create(NetworkType.PUBLIC);
-  await stellarClient.activateThroughThreefoldService();
+  final stellarClient = Client(NetworkType.PUBLIC,
+      'SCVWI4TCVUPB7EN4I5X5DQ56RV5E3JNFWDY2NHJCBXOMH5WKVCJ4RSGG');
+  // print({stellarClient.accountId, stellarClient.secretSeed});
+  // print(await stellarClient.getBalance());
+  final balance = await getBalanceByAccountID(
+      network: NetworkType.PUBLIC,
+      accountId: 'GBE366DFQTLJNQSAWAE2YSGLAPXS4MZMJPCIBKWCHUUKMQLIJ2PNUIFC');
 
-  await stellarClient.transfer(
-      destinationAddress: "destination-public-key",
-      amount: "20",
-      currency: "TFT",
-      memoText: "Memo Text");
-
-  await stellarClient.getTransactions();
+  print('balanceeeee: $balance');
 }

packages/stellar_client/lib/src/helpers.dart

@@ -1,9 +1,41 @@
 part of '../stellar_client.dart';
 
+Future<BalanceData> getLiquidityPoolAssets(
+    NetworkType network, String liquidityPoolId) async {
+  late StellarSDK _sdk;
+  late Uri liquidityPoolUri;
+  late BalanceData balance;
+  switch (network) {
+    case NetworkType.TESTNET:
+      _sdk = StellarSDK.TESTNET;
+      break;
+    default:
+      _sdk = StellarSDK.PUBLIC;
+  }
+  try {
+    if (network == Network.TESTNET) {
+      liquidityPoolUri = Uri.parse(
+          'https://horizon-testnet.stellar.org/liquidity_pools/$liquidityPoolId');
+    } else {
+      liquidityPoolUri = Uri.parse(
+          'https://horizon.stellar.org/liquidity_pools/$liquidityPoolId');
+    }
+    LiquidityPoolResponse liquidityPool =
+        await _sdk.liquidityPools.liquidityPool(liquidityPoolUri);
+
+    for (var reserve in liquidityPool.reserves) {
+      balance =
+          BalanceData(assetCode: reserve.asset.type, balance: reserve.amount);
+    }
+    return balance;
+  } catch (e) {
+    throw Exception(e);
+  }
+}
+
 Future<List<BalanceInfo>> getBalanceByAccountID(
     {required NetworkType network, required String accountId}) async {
   late StellarSDK _sdk;
-
   switch (network) {
     case NetworkType.TESTNET:
       _sdk = StellarSDK.TESTNET;
@@ -12,22 +44,28 @@ Future<List<BalanceInfo>> getBalanceByAccountID(
       _sdk = StellarSDK.PUBLIC;
       break;
   }
-
-  List<BalanceInfo> balancesList = [];
-  AccountResponse account = await _sdk.accounts.account(accountId);
-
-  for (Balance balance in account.balances) {
-    BalanceData balanceData;
-    switch (balance.assetType) {
-      case Asset.TYPE_NATIVE:
-        balanceData = BalanceData(assetCode: 'XLM', balance: balance.balance);
-        break;
-      default:
-        balanceData = BalanceData(
-            assetCode: balance.assetCode!, balance: balance.balance);
+  try {
+    List<BalanceInfo> balancesList = [];
+    AccountResponse account = await _sdk.accounts.account(accountId);
+    for (Balance balance in account.balances) {
+      BalanceData balanceData;
+      switch (balance.assetType) {
+        case Asset.TYPE_NATIVE:
+          balanceData = BalanceData(assetCode: 'XLM', balance: balance.balance);
+          break;
+        case Asset.TYPE_POOL_SHARE:
+          final b =
+              await getLiquidityPoolAssets(network, balance.liquidityPoolId!);
+          balanceData =
+              BalanceData(assetCode: b.assetCode, balance: balance.balance);
+        default:
+          balanceData = BalanceData(
+              assetCode: balance.assetCode!, balance: balance.balance);
+      }
+      balancesList.add(balanceData);
     }
-    balancesList.add(balanceData);
+    return balancesList;
+  } catch (e) {
+    throw Exception(e);
   }
-
-  return balancesList;
 }