feat: token delegate

aave · Dec 21, 2023 · 7ec04b0 · 7ec04b0
1 parent 7ec0959
commit 7ec04b0
Showing 1 changed file with 28 additions and 1 deletion.
diff --git a/packages/contract-helpers/src/governance-v3/aave-token-v3/index.ts b/packages/contract-helpers/src/governance-v3/aave-token-v3/index.ts
@@ -1,14 +1,16 @@
-import { providers } from 'ethers';
+import { BigNumber, PopulatedTransaction, providers } from 'ethers';
 import { AaveTokenV3 } from '../typechain/AaveTokenV3';
 import { AaveTokenV3__factory } from '../typechain/factories/AaveTokenV3__factory';
 
 export enum GovernancePowerType {
   VOTING,
   PROPOSITION,
+  ALL,
 }
 
 export class AaveTokenV3Service {
   readonly _contract: AaveTokenV3;
+  readonly _contractInterface = AaveTokenV3__factory.createInterface();
 
   constructor(tokenAddress: string, provider: providers.Provider) {
     this._contract = AaveTokenV3__factory.connect(tokenAddress, provider);
@@ -27,4 +29,29 @@ export class AaveTokenV3Service {
       blockTag: blockNumber,
     });
   }
+
+  public getDelegateTxData(
+    user: string,
+    delegateTo: string,
+    type: GovernancePowerType,
+  ): PopulatedTransaction {
+    const tx: PopulatedTransaction = {};
+    if (type === GovernancePowerType.ALL) {
+      tx.data = this._contractInterface.encodeFunctionData('delegate', [
+        delegateTo,
+      ]);
+    } else {
+      tx.data = this._contractInterface.encodeFunctionData('delegateByType', [
+        delegateTo,
+        type,
+      ]);
+    }
+
+    return {
+      ...tx,
+      to: this._contract.address,
+      from: user,
+      gasLimit: BigNumber.from('1000000'),
+    };
+  }
 }