Throw an unsupported error if 0 amount is expected from trade

EdgeApp · Sep 29, 2023 · 7e59943 · 7e59943
1 parent 9b72c1a
commit 7e59943
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 13 deletions.
diff --git a/src/swap/defi/uni-v2-based/plugins/spookySwap.ts b/src/swap/defi/uni-v2-based/plugins/spookySwap.ts
@@ -52,7 +52,7 @@ export function makeSpookySwapPlugin(
     request: EdgeSwapRequestPlugin,
     uid: string
   ): Promise<SwapOrder> => {
-    const { fromWallet, toWallet, fromTokenId, toTokenId, quoteFor } = request
+    const { fromWallet, toWallet, fromTokenId, toTokenId } = request
 
     // Sanity check: Both wallets should be of the same chain.
     if (
@@ -75,8 +75,8 @@ export function makeSpookySwapPlugin(
     const spookySwapRouter = makeSpookySwapRouterContract(provider)
     const { amountToSwap, expectedAmountOut } = await getSwapAmounts(
       spookySwapRouter,
-      quoteFor,
-      request.nativeAmount,
+      request,
+      swapInfo,
       [fromTokenAddress, toTokenAddress],
       isWrappingSwap
     )

diff --git a/src/swap/defi/uni-v2-based/plugins/tombSwap.ts b/src/swap/defi/uni-v2-based/plugins/tombSwap.ts
@@ -52,7 +52,7 @@ export function makeTombSwapPlugin(
     request: EdgeSwapRequestPlugin,
     uid: string
   ): Promise<SwapOrder> => {
-    const { fromWallet, toWallet, fromTokenId, toTokenId, quoteFor } = request
+    const { fromWallet, toWallet, fromTokenId, toTokenId } = request
 
     // Sanity check: Both wallets should be of the same chain.
     if (
@@ -75,8 +75,8 @@ export function makeTombSwapPlugin(
     const tombSwapRouter = makeTombSwapRouterContract(provider)
     const { amountToSwap, expectedAmountOut } = await getSwapAmounts(
       tombSwapRouter,
-      quoteFor,
-      request.nativeAmount,
+      request,
+      swapInfo,
       [fromTokenAddress, toTokenAddress],
       isWrappingSwap
     )

diff --git a/src/swap/defi/uni-v2-based/plugins/velodrome.ts b/src/swap/defi/uni-v2-based/plugins/velodrome.ts
@@ -43,7 +43,7 @@ export function makeVelodromePlugin(
     request: EdgeSwapRequestPlugin,
     uid: string
   ): Promise<SwapOrder> => {
-    const { fromWallet, toWallet, fromTokenId, toTokenId, quoteFor } = request
+    const { fromWallet, toWallet, fromTokenId, toTokenId } = request
 
     if (
       // Velodrome does not support reverse quotes
@@ -86,8 +86,8 @@ export function makeVelodromePlugin(
 
     const { amountToSwap, expectedAmountOut } = await getSwapAmounts(
       velodromeRouter,
-      quoteFor,
-      request.nativeAmount,
+      request,
+      swapInfo,
       path,
       isWrappingSwap
     )

diff --git a/src/swap/defi/uni-v2-based/uniV2Utils.ts b/src/swap/defi/uni-v2-based/uniV2Utils.ts
@@ -1,4 +1,5 @@
 import { mul, round, sub } from 'biggystring'
+import { EdgeSwapInfo, EdgeSwapRequest, SwapCurrencyError } from 'edge-core-js'
 import { BigNumber, Contract, ethers, PopulatedTransaction } from 'ethers'
 
 import { InOutTokenAddresses } from '../defiUtils'
@@ -10,11 +11,12 @@ import { makeErc20Contract } from './uniV2Contracts'
  */
 export const getSwapAmounts = async (
   router: Contract,
-  quoteFor: string,
-  nativeAmount: string,
+  request: EdgeSwapRequest,
+  swapInfo: EdgeSwapInfo,
   path: unknown,
   isWrappingSwap: boolean
 ): Promise<{ amountToSwap: string; expectedAmountOut: string }> => {
+  const { nativeAmount, quoteFor } = request
   const [amountToSwap, expectedAmountOut] = (isWrappingSwap
     ? [nativeAmount, nativeAmount]
     : quoteFor === 'to'
@@ -24,8 +26,14 @@ export const getSwapAmounts = async (
     : []
   ).map(String)
 
-  if (amountToSwap == null || expectedAmountOut == null)
-    throw new Error(`Failed to calculate amounts`)
+  if (
+    amountToSwap == null ||
+    expectedAmountOut == null ||
+    amountToSwap === '0' ||
+    expectedAmountOut === '0'
+  ) {
+    throw new SwapCurrencyError(swapInfo, request)
+  }
 
   return { amountToSwap, expectedAmountOut }
 }