fixup! Update getInOutTokenAddresses to use EdgeTokens

EdgeApp · Sep 22, 2023 · eb14fb9 · eb14fb9
1 parent 29dfd52
commit eb14fb9
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/swap/defi/defiUtils.ts b/src/swap/defi/defiUtils.ts
@@ -1,3 +1,4 @@
+import { asMaybe, asObject, asString } from 'cleaners'
 import { EdgeCurrencyConfig, EdgeToken } from 'edge-core-js/types'
 import { ethers } from 'ethers'
 
@@ -14,6 +15,10 @@ export interface InOutTokenAddresses {
   isToWrappedCurrency: boolean
 }
 
+const asContractLocation = asObject({
+  contractAddress: asString
+})
+
 /**
  * Determine if the tokens are wrapped and return the appropriate wrapped
  * contract addresses, if different.
@@ -36,9 +41,11 @@ export const getInOutTokenAddresses = (
     ? undefined
     : allTokens[toTokenId]
 
-  const fromTokenAddress: string =
-    fromToken?.networkLocation?.contractAddress ?? ''
-  const toTokenAddress: string = toToken?.networkLocation?.contractAddress ?? ''
+  const fromTokenAddress =
+    asMaybe(asContractLocation)(fromToken?.networkLocation)?.contractAddress ??
+    ''
+  const toTokenAddress =
+    asMaybe(asContractLocation)(toToken?.networkLocation)?.contractAddress ?? ''
 
   const isFromWrappedCurrency =
     fromTokenAddress.toLowerCase() === wrappedMainnetAddress.toLowerCase()