feat: prudentia scripts

Author: rhlsthrm

tasks/chain-specific/mode.ts

@@ -1,5 +1,6 @@
 import { task } from "hardhat/config";
 import { assets as modeAssets } from "../../chains/mode/assets";
+import { Address } from "viem";
 
 task("market:set-cf:mode:main", "Sets caps on a market").setAction(async (_, { viem, run }) => {
   const COMPTROLLER = "0xfb3323e24743caf4add0fdccfb268565c0685556";
@@ -16,3 +17,68 @@ task("market:set-cf:mode:main", "Sets caps on a market").setAction(async (_, { v
     }
   }
 });
+
+task("prudentia:upgrade:pool", "Upgrades a pool to the latest comptroller implementation").setAction(
+  async (_, { viem, deployments, getNamedAccounts }) => {
+    const { deployer } = await getNamedAccounts();
+    console.log("deployer: ", deployer);
+    const publicClient = await viem.getPublicClient();
+    const fuseFeeDistributor = await viem.getContractAt(
+      "FeeDistributor",
+      (await deployments.get("FeeDistributor")).address as Address
+    );
+    const COMPTROLLER = "0xfb3323e24743caf4add0fdccfb268565c0685556";
+    const unitroller = await viem.getContractAt("Unitroller", COMPTROLLER);
+    const admin = await unitroller.read.admin();
+    console.log("pool admin", admin);
+
+    const implBefore = await unitroller.read.comptrollerImplementation();
+
+    const latestImpl = await fuseFeeDistributor.read.latestComptrollerImplementation([implBefore]);
+    console.log(`current impl ${implBefore} latest ${latestImpl}`);
+
+    const shouldUpgrade = implBefore !== latestImpl;
+
+    if (shouldUpgrade) {
+      const tx = await unitroller.write._upgrade();
+      await publicClient.waitForTransactionReceipt({ hash: tx });
+      console.log(`Upgraded pool ${COMPTROLLER} with tx ${tx}`);
+    }
+  }
+);
+
+task("prudentia:config", "Sets prudentia config").setAction(async (_, { viem, getNamedAccounts }) => {
+  const { deployer } = await getNamedAccounts();
+  console.log("deployer: ", deployer);
+  const publicClient = await viem.getPublicClient();
+  const COMPTROLLER = "0xfb3323e24743caf4add0fdccfb268565c0685556";
+  const pool = await viem.getContractAt("ComptrollerPrudentiaCapsExt", COMPTROLLER);
+  const admin = await pool.read.admin();
+  console.log("admin: ", admin);
+  let tx = await pool.write._setSupplyCapConfig([
+    { controller: "0x425Ed58c3B836B1c5a073ab5dae3ee6c94336B21", offset: 0, decimalShift: -4 }
+  ]);
+  console.log("set supply cap config tx: ", tx);
+  await publicClient.waitForTransactionReceipt({ hash: tx });
+  // set borrow cap config
+  tx = await pool.write._setBorrowCapConfig([
+    { controller: "0x3060759F0D0BF60c373f9057BB1269c28Bd5Bb66", offset: 0, decimalShift: -4 }
+  ]);
+
+  // set ionusdc IRM
+  const ionUSDC = "0x2BE717340023C9e14C1Bb12cb3ecBcfd3c3fB038";
+  const ionUSDCirm = "0x6a40D802080a37E210Ec87735ABF995b5BC636A6";
+
+  const cToken = await viem.getContractAt("ICErc20", ionUSDC);
+  tx = await cToken.write._setInterestRateModel([ionUSDCirm]);
+  await publicClient.waitForTransactionReceipt({ hash: tx });
+  console.log(`Set IRM of ${await cToken.read.symbol()} to ${ionUSDCirm}`);
+
+  // set ionusdt IRM
+  const ionUSDT = "0x94812F2eEa03A49869f95e1b5868C6f3206ee3D3";
+  const ionUSDTirm = "0xC58DCC0cbc02355cF1aD6b5398dE49152ae72E2E";
+  const cToken2 = await viem.getContractAt("ICErc20", ionUSDT);
+  tx = await cToken2.write._setInterestRateModel([ionUSDTirm]);
+  await publicClient.waitForTransactionReceipt({ hash: tx });
+  console.log(`Set IRM of ${await cToken2.read.symbol()} to ${ionUSDTirm}`);
+});