feat: approve methods for cross-asset exchange

curvefi · Oct 27, 2021 · 337a741 · 337a741
1 parent 85c25e8
commit 337a741
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -12,6 +12,9 @@ import {
     crossAssetExchangeAvailable,
     crossAssetExchangeOutputAndSlippage,
     crossAssetExchangeExpected,
+    crossAssetExchangeIsApproved,
+    crossAssetExchangeApproveEstimateGas,
+    crossAssetExchangeApprove,
     crossAssetExchangeEstimateGas,
     crossAssetExchange,
 } from "./pools";
@@ -59,18 +62,21 @@ const curve = {
     hasAllowance,
     ensureAllowance,
     getBestPoolAndOutput,
+    exchangeExpected,
     exchangeIsApproved,
     exchangeApprove,
-    exchangeExpected,
     exchange,
     crossAssetExchangeAvailable,
     crossAssetExchangeOutputAndSlippage,
     crossAssetExchangeExpected,
+    crossAssetExchangeIsApproved,
+    crossAssetExchangeApprove,
     crossAssetExchange,
     estimateGas: {
         ensureAllowance: ensureAllowanceEstimateGas,
         exchangeApprove: exchangeApproveEstimateGas,
         exchange: exchangeEstimateGas,
+        crossAssetExchangeApprove: crossAssetExchangeApproveEstimateGas,
         crossAssetExchange: crossAssetExchangeEstimateGas,
     },
     boosting: {

diff --git a/src/pools.ts b/src/pools.ts
@@ -1841,6 +1841,18 @@ export const crossAssetExchangeExpected = async (inputCoin: string, outputCoin:
     return ethers.utils.formatUnits(_expected, outputCoinDecimals)
 }
 
+export const crossAssetExchangeIsApproved = async (inputCoin: string, amount: string): Promise<boolean> => {
+    return await hasAllowance([inputCoin], [amount], curve.signerAddress, ALIASES.router);
+}
+
+export const crossAssetExchangeApproveEstimateGas = async (inputCoin: string, amount: string): Promise<number> => {
+    return await ensureAllowanceEstimateGas([inputCoin], [amount], ALIASES.router);
+}
+
+export const crossAssetExchangeApprove = async (inputCoin: string, amount: string): Promise<string[]> => {
+    return await ensureAllowance([inputCoin], [amount], ALIASES.router);
+}
+
 export const crossAssetExchangeEstimateGas = async (inputCoin: string, outputCoin: string, amount: string, maxSlippage = 0.02): Promise<number> => {
     if (!(await crossAssetExchangeAvailable(inputCoin, outputCoin))) throw Error("Such exchange is not available");