Factory: PreTransfer: Add spender argument

Emmet-Finance · Jun 15, 2024 · 03ce84c · 03ce84c
1 parent 77e5070
commit 03ce84c
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/src/chains/index.ts b/src/chains/index.ts
@@ -129,6 +129,7 @@ export interface PreTransfer<Signer, GasArgs> {
   preTransfer: (
     signer: Signer,
     token: string,
+    spender: string,
     amount: bigint,
     gasArgs: GasArgs,
   ) => Promise<string>;

diff --git a/src/chains/web3.ts b/src/chains/web3.ts
@@ -137,10 +137,10 @@ export async function web3Helper({
     },
     nativeCoin: () => nativeCoin,
     chainName: () => chainName,
-    preTransfer: async (signer, tid, amt, gasArgs) => {
+    preTransfer: async (signer, tid, spender, amt, gasArgs) => {
       const erc = WrappedERC20__factory.connect(tid, signer);
       const preTransferGas = await erc.approve.estimateGas(bridge, amt);
-      const approved = await erc.approve(bridge, amt, {
+      const approved = await erc.approve(spender, amt, {
         ...gasArgs,
         gasLimit: preTransferGas,
       });

diff --git a/src/factory/factory.ts b/src/factory/factory.ts
@@ -73,8 +73,8 @@ export function ChainFactoryBuilder(
 
   return {
     inner,
-    preTransfer: async (chain, signer, tid, amt, ga) => {
-      const pt = await chain.preTransfer(signer, tid, amt, ga);
+    preTransfer: async (chain, signer, tid, spender, amt, ga) => {
+      const pt = await chain.preTransfer(signer, tid, spender, amt, ga);
       return pt;
     },
     getTxCount() {
@@ -141,6 +141,8 @@ export function ChainFactoryBuilder(
         uniqueUser: tx.uniqueUsers,
       };
     },
+    // async stakeTokenForPool(chain, signer, tokenSymbol, amount) {},
+    // async withdrawTokenForPool(chain, signer, tokenSymbol, amount) {},
     sendInstallment: async (
       chain,
       signer,

diff --git a/src/factory/types/index.ts b/src/factory/types/index.ts
@@ -87,6 +87,7 @@ export interface ChainFactory {
     chain: PreTransfer<Signer, GasArgs>,
     signer: Signer,
     tid: string,
+    spender: string,
     amount: bigint,
     gasArgs: GasArgs,
   ) => Promise<string>;