feat: add allowance for curve

OriginProtocol · Sep 13, 2023 · eed0b54 · eed0b54
1 parent e0a0a18
commit eed0b54
Showing 1 changed file with 35 additions and 1 deletion.
diff --git a/libs/oeth/swap/src/actions/swapCurve/index.ts b/libs/oeth/swap/src/actions/swapCurve/index.ts
@@ -1,5 +1,6 @@
 import { ETH_ADDRESS_CURVE, isNilOrEmpty } from '@origin/shared/utils';
 import {
+  erc20ABI,
   getAccount,
   getPublicClient,
   prepareWriteContract,
@@ -149,10 +150,43 @@ const swap: Swap = async ({
   slippage,
   curve,
 }) => {
-  if (amountIn === 0n) {
+  const { address } = getAccount();
+
+  if (amountIn === 0n || isNilOrEmpty(address)) {
     return;
   }
 
+  if (!isNilOrEmpty(tokenIn.address) && !isNilOrEmpty(tokenOut.address)) {
+    const allowance = await readContract({
+      address: tokenIn.address,
+      abi: erc20ABI,
+      functionName: 'allowance',
+      args: [address, curve.CurveRegistryExchange.address],
+    });
+
+    if (allowance < amountIn) {
+      try {
+        const { request } = await prepareWriteContract({
+          address: tokenIn.address,
+          abi: erc20ABI,
+          functionName: 'approve',
+          args: [curve.CurveRegistryExchange.address, amountIn],
+        });
+        const { hash } = await writeContract(request);
+        await waitForTransaction({ hash });
+
+        // TODO trigger notification
+        console.log(`swap curve exchange multiple approval done!`);
+      } catch (e) {
+        // TODO trigger notification
+        console.error(
+          `swap curve exchange multiple approval error!\n${e.message}`,
+        );
+        return;
+      }
+    }
+  }
+
   const minAmountOut = parseUnits(
     (
       +formatUnits(amountOut, tokenOut.decimals) -