Merge pull request #450 from soroswap/fix/nativeTokenInBalances

✨ Improve native token support to BalancesTable
soroswap · May 15, 2024 · 4eb2454 · 4eb2454
2 parents 7cfc63c + d1682fd
commit 4eb2454
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 5 deletions.
diff --git a/src/components/Balances.tsx b/src/components/Balances.tsx
@@ -9,6 +9,7 @@ import BalancesTable from './BalancesTable/BalancesTable';
 import { ButtonPrimary } from './Buttons/Button';
 import { WalletButton } from './Buttons/WalletButton';
 import { MintCustomToken } from './MintCustomToken';
+import { xlmTokenList } from 'constants/xlmToken';
 
 const PageWrapper = styled(Paper)`
   background: ${({ theme }) => `linear-gradient(${theme.palette.customBackground.bg2}, ${
@@ -37,6 +38,26 @@ export function Balances() {
 
   const [currentMintingToken, setCurrentMintingToken] = useState<TokenType | null>(null);
   const [isMinting, setIsMinting] = useState(false);
+  const getNetwork = () => {
+    switch (sorobanContext.activeChain?.networkPassphrase) {
+      case Networks.TESTNET:
+        return 'testnet';
+      case Networks.PUBLIC:
+        return 'mainnet';
+      case Networks.STANDALONE:
+        return 'standalone';
+      case Networks.FUTURENET:
+        return 'futurenet';
+      default:
+        return 'Unknown';
+    }
+  }
+
+  const getNativeToken = () => {
+    const network = getNetwork();
+    const nativeToken = xlmTokenList.find((token) => token.network === network);
+    return nativeToken?.assets[0];
+  }
 
   const handleMint = () => {
     setIsMinting(true);
@@ -104,7 +125,7 @@ export function Balances() {
             )}
           </Box>
           <Box>
-            <BalancesTable />
+            <BalancesTable nativeToken={getNativeToken()} />
           </Box>
         </>
       )}

diff --git a/src/components/BalancesTable/BalancesTable.tsx b/src/components/BalancesTable/BalancesTable.tsx
@@ -119,15 +119,15 @@ function BalancesTableHead(props: BalancesTableProps) {
   );
 }
 
-export default function BalancesTable() {
+export default function BalancesTable(props: any) {
+  const { nativeToken } = props;
   const { tokens, tokenBalancesResponse } = useGetMyBalances();
-
   const rows =
-    tokenBalancesResponse?.balances?.map((x) => ({
+  tokenBalancesResponse?.balances?.map((x) => ({
       ...x,
       type: x.issuer
         ? 'Stellar Classic Asset'
-        : x.name == 'Stellar Lumens'
+        : (x.contract === nativeToken.contract && x.code === nativeToken.code)
         ? 'Native'
         : 'Soroban Token',
     })) ?? [];