From 51211b34f543f8fddd31cd4afc6511db4a32e132 Mon Sep 17 00:00:00 2001
From: julian <julian@cypherstack.com>
Date: Fri, 7 Apr 2023 16:02:28 -0600
Subject: [PATCH] eth token fee parsing fix

---
 .../ethereum/cached_eth_token_balance.dart     | 10 ++--------
 lib/services/ethereum/ethereum_api.dart        | 18 +++++-------------
 2 files changed, 7 insertions(+), 21 deletions(-)

diff --git a/lib/services/ethereum/cached_eth_token_balance.dart b/lib/services/ethereum/cached_eth_token_balance.dart
index c08b40fd5..f36cf2ea3 100644
--- a/lib/services/ethereum/cached_eth_token_balance.dart
+++ b/lib/services/ethereum/cached_eth_token_balance.dart
@@ -23,14 +23,8 @@ class CachedEthTokenBalance with EthTokenCache {
       await updateCachedBalance(
         TokenBalance(
           contractAddress: token.address,
-          total: Amount(
-            rawValue: BigInt.from(response.value!),
-            fractionDigits: token.decimals,
-          ),
-          spendable: Amount(
-            rawValue: BigInt.from(response.value!),
-            fractionDigits: token.decimals,
-          ),
+          total: response.value!,
+          spendable: response.value!,
           blockedTotal: Amount(
             rawValue: BigInt.zero,
             fractionDigits: token.decimals,
diff --git a/lib/services/ethereum/ethereum_api.dart b/lib/services/ethereum/ethereum_api.dart
index 3bcb81b71..d31bcfd40 100644
--- a/lib/services/ethereum/ethereum_api.dart
+++ b/lib/services/ethereum/ethereum_api.dart
@@ -1,5 +1,4 @@
 import 'dart:convert';
-import 'dart:math';
 
 import 'package:decimal/decimal.dart';
 import 'package:http/http.dart';
@@ -9,6 +8,7 @@ import 'package:stackwallet/dto/ethereum/eth_tx_dto.dart';
 import 'package:stackwallet/dto/ethereum/pending_eth_tx_dto.dart';
 import 'package:stackwallet/models/isar/models/ethereum/eth_contract.dart';
 import 'package:stackwallet/models/paymint/fee_object_model.dart';
+import 'package:stackwallet/utilities/amount/amount.dart';
 import 'package:stackwallet/utilities/default_nodes.dart';
 import 'package:stackwallet/utilities/eth_commons.dart';
 import 'package:stackwallet/utilities/extensions/extensions.dart';
@@ -396,7 +396,7 @@ abstract class EthereumAPI {
   //   }
   // }
 
-  static Future<EthereumResponse<int>> getWalletTokenBalance({
+  static Future<EthereumResponse<Amount>> getWalletTokenBalance({
     required String address,
     required String contractAddress,
   }) async {
@@ -411,19 +411,11 @@ abstract class EthereumAPI {
         if (json["data"] is List) {
           final map = json["data"].first as Map;
 
-          final bal = Decimal.tryParse(map["balance"].toString());
-          final int balance;
-          if (bal == null) {
-            balance = 0;
-          } else {
-            final int decimals = map["decimals"] as int;
-            balance = (bal * Decimal.fromInt(pow(10, decimals).truncate()))
-                .toBigInt()
-                .toInt();
-          }
+          final balance =
+              Decimal.tryParse(map["balance"].toString()) ?? Decimal.zero;
 
           return EthereumResponse(
-            balance,
+            Amount.fromDecimal(balance, fractionDigits: map["decimals"] as int),
             null,
           );
         } else {