fix: deployment files

Author: rhlsthrm

chainDeploy/helpers/irms.ts

@@ -1,4 +1,4 @@
-import { parseEther } from "viem";
+import { Hash, parseEther } from "viem";
 import { IrmDeployFnParams } from "../../chains/types";
 
 export const deployIRMs = async ({
@@ -21,6 +21,6 @@ export const deployIRMs = async ({
     ],
     log: true
   });
-  if (jrm.transactionHash) await publicClient.waitForTransactionReceipt({ hash: jrm.transactionHash });
+  if (jrm.transactionHash) await publicClient.waitForTransactionReceipt({ hash: jrm.transactionHash as Hash });
   console.log("JumpRateModel: ", jrm.address);
 };

deploy/02-deploy-comptroller-set-extensions.ts

@@ -0,0 +1,116 @@
+import { DeployFunction } from "hardhat-deploy/types";
+import { Address, encodeFunctionData, Hash, zeroAddress } from "viem";
+
+import { logTransaction } from "../chainDeploy/helpers/logging";
+
+const func: DeployFunction = async ({ viem, getNamedAccounts, deployments }) => {
+  const { deployer, multisig } = await getNamedAccounts();
+  const publicClient = await viem.getPublicClient();
+
+  const oldComptroller = await deployments.getOrNull("Comptroller");
+  const fuseFeeDistributor = await viem.getContractAt(
+    "FeeDistributor",
+    (await deployments.get("FeeDistributor")).address as Address
+  );
+  let tx: Hash;
+
+  const comp = await deployments.deploy("Comptroller", {
+    contract: "Comptroller.sol:Comptroller",
+    from: deployer,
+    args: [],
+    log: true
+  });
+  if (comp.transactionHash) await publicClient.waitForTransactionReceipt({ hash: comp.transactionHash as Hash });
+  console.log("Comptroller ", comp.address);
+
+  const compFirstExtension = await deployments.deploy("ComptrollerFirstExtension", {
+    contract: "ComptrollerFirstExtension",
+    from: deployer,
+    args: [],
+    log: true
+  });
+  if (compFirstExtension.transactionHash)
+    await publicClient.waitForTransactionReceipt({ hash: compFirstExtension.transactionHash as Hash });
+  console.log("ComptrollerFirstExtension", compFirstExtension.address);
+
+  const comptroller = await viem.getContractAt(
+    "Comptroller",
+    (await deployments.get("Comptroller")).address as Address
+  );
+
+  /// LATEST IMPLEMENTATIONS
+  // Comptroller
+  if (oldComptroller) {
+    const latestComptrollerImplementation = await fuseFeeDistributor.read.latestComptrollerImplementation([
+      oldComptroller.address as Address
+    ]);
+    if (latestComptrollerImplementation === zeroAddress || latestComptrollerImplementation !== comptroller.address) {
+      if ((await fuseFeeDistributor.read.owner()).toLowerCase() !== deployer.toLowerCase()) {
+        logTransaction(
+          "Set Latest Comptroller Implementation",
+          encodeFunctionData({
+            abi: fuseFeeDistributor.abi,
+            functionName: "_setLatestComptrollerImplementation",
+            args: [oldComptroller.address as Address, comptroller.address]
+          })
+        );
+      } else {
+        tx = await fuseFeeDistributor.write._setLatestComptrollerImplementation([
+          oldComptroller.address as Address,
+          comptroller.address
+        ]);
+        await publicClient.waitForTransactionReceipt({ hash: tx });
+        console.log(
+          `Set the latest Comptroller implementation for ${oldComptroller.address} to ${comptroller.address}`
+        );
+      }
+    } else {
+      console.log(
+        `No change in the latest Comptroller implementation ${latestComptrollerImplementation} for ${comptroller.address}`
+      );
+    }
+  } else {
+    // on the first deploy to a chain
+    if (multisig && (await fuseFeeDistributor.read.owner()).toLowerCase() !== deployer.toLowerCase()) {
+      logTransaction(
+        "Set Latest Comptroller Implementation",
+        encodeFunctionData({
+          abi: fuseFeeDistributor.abi,
+          functionName: "_setLatestComptrollerImplementation",
+          args: [zeroAddress, comptroller.address]
+        })
+      );
+    } else {
+      tx = await fuseFeeDistributor.write._setLatestComptrollerImplementation([zeroAddress, comptroller.address]);
+      await publicClient.waitForTransactionReceipt({ hash: tx });
+      console.log(`Set the latest Comptroller implementation for ${zeroAddress} to ${comptroller.address}`);
+    }
+  }
+
+  const comptrollerExtensions = await fuseFeeDistributor.read.getComptrollerExtensions([comptroller.address]);
+  if (comptrollerExtensions.length == 0 || comptrollerExtensions[1] != compFirstExtension.address) {
+    if (multisig && (await fuseFeeDistributor.read.owner()).toLowerCase() !== deployer.toLowerCase()) {
+      logTransaction(
+        "Set Comptroller Extensions",
+        encodeFunctionData({
+          abi: fuseFeeDistributor.abi,
+          functionName: "_setComptrollerExtensions",
+          args: [comptroller.address, [comptroller.address, compFirstExtension.address as Address]]
+        })
+      );
+    } else {
+      tx = await fuseFeeDistributor.write._setComptrollerExtensions([
+        comptroller.address,
+        [comptroller.address, compFirstExtension.address as Address]
+      ]);
+      await publicClient.waitForTransactionReceipt({ hash: tx });
+      console.log(`configured the extensions for comptroller ${comptroller.address}`);
+    }
+  } else {
+    console.log(`comptroller extensions already configured`);
+  }
+};
+
+func.tags = ["prod", "comptroller-setup"];
+
+export default func;