Add reward distributor check before setup

ionicprotocol · Jul 23, 2024 · 11d6d36 · 11d6d36
1 parent d22d2bc
commit 11d6d36
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tasks/flywheel/deploy.ts b/tasks/flywheel/deploy.ts
@@ -124,8 +124,10 @@ task("flywheel:add-to-pool", "Create pool if does not exist")
     }
 
     const comptroller = await viem.getContractAt("Comptroller", poolAddress);
-    const addTx = await comptroller.write._addRewardsDistributor([flywheelAddress]);
-
+    const rewardsDistributors = (await comptroller.read.getRewardsDistributors()) as Address[];
+    if (!rewardsDistributors.map((s) => s.toLowerCase()).includes(flywheelAddress.toLowerCase())) {
+      const addTx = await comptroller.write._addRewardsDistributor([flywheelAddress]);
+    }
     await publicClient.waitForTransactionReceipt({ hash: addTx });
     console.log({ addTx });
   });