feat: compare and upgrade all contracts in batch transaction

scripts/safe-scripts/helpers/utils.ts

@@ -0,0 +1,18 @@
+const { ethers, upgrades } = require("hardhat");
+import { artifacts } from "hardhat";
+
+export async function getDeployedBytecode(address: string, provider: any) {
+  const contractImpl = await upgrades.erc1967.getImplementationAddress(address);
+  const response = await provider.getCode(contractImpl);
+  return response;
+}
+
+export async function getArtifact(name: string) {
+  const artifact = await artifacts.readArtifact(name);
+  return artifact.deployedBytecode;
+}
+
+export async function forceImportDeployedProxies(contractAddress: string, contractName: string) {
+  const contractArtifact = await ethers.getContractFactory(contractName);
+  await upgrades.forceImport(contractAddress, contractArtifact, { kind: "transparent" });
+}

scripts/safe-scripts/upgradeAll.ts

@@ -1,8 +1,8 @@
-const { ethers, upgrades } = require("hardhat");
+const { ethers } = require("hardhat");
 import upgradeHelper from "./helpers/upgrade";
 import networkAddresses from "./address.json";
 import proposeTransaction from "./helpers/proposeTransaction";
-import { artifacts } from "hardhat";
+import {getArtifact, getDeployedBytecode, forceImportDeployedProxies} from "./helpers/utils";
 
 async function main(networks: { [networkName: string]: { contracts: { name: string; address: string }[] } }) {
   const provider = ethers.provider;
@@ -38,22 +38,6 @@ async function main(networks: { [networkName: string]: { contracts: { name: stri
   }
 }
 
-async function getDeployedBytecode(address: string, provider: any) {
-  const contractImpl = await upgrades.erc1967.getImplementationAddress(address);
-  const response = await provider.getCode(contractImpl);
-  return response;
-}
-
-async function getArtifact(name: string) {
-  const artifact = await artifacts.readArtifact(name);
-  return artifact.deployedBytecode;
-}
-
-async function forceImportDeployedProxies(contractAddress: string, contractName: string) {
-  const contractArtifact = await ethers.getContractFactory(contractName);
-  await upgrades.forceImport(contractAddress, contractArtifact, { kind: "transparent" });
-}
-
 main(networkAddresses)
   .then(() => process.exit(0))
   .catch((error) => {

scripts/safe-scripts/upgradeAllBatch.ts

@@ -1,21 +1,46 @@
 import upgradeHelper from "./helpers/upgrade";
 import networkAddresses from "./address.json";
 import proposeTransactions from "./helpers/proposeTransactions";
-import {  MetaTransactionData } from "@safe-global/safe-core-sdk-types";
+const { ethers } = require("hardhat");
+import { MetaTransactionData } from "@safe-global/safe-core-sdk-types";
+import { getArtifact, getDeployedBytecode, forceImportDeployedProxies } from "./helpers/utils";
 
 async function main(networks: { [networkName: string]: { contracts: { name: string; address: string }[] } }) {
+  const provider = ethers.provider;
   const networkName = await hre.network.name;
   const networkContracts = networks[networkName].contracts;
+
+  console.log(`Checking contracts on network "${networkName}":`, "\n");
   const targets = [];
   const values = [];
   const params = [];
-  console.log(`Checking contracts on network "${networkName}":`, "\n");
 
   for (let { name, address } of networkContracts) {
-    const { to, value, data } = await upgradeHelper(address, name);
-    targets.push(to);
-    values.push(value);
-    params.push(data);
+    console.log(`Checking contract "${name}" at address ${address}`);
+
+    const deployedBytecode = await getDeployedBytecode(address, provider);
+    if (!deployedBytecode) {
+      console.error(`Failed to retrieve deployed bytecode for "${name}". Skipping.`);
+      continue;
+    }
+    try {
+      // Uncomment below line if network files are lost and need to be force import.
+      // await forceImportDeployedProxies(address, name);
+      const compiledBytecode = await getArtifact(name);
+
+      if (deployedBytecode !== compiledBytecode) {
+        console.warn(`Contract "${name}" is out of date!`);
+        console.log(`Upgrading to latest version...`);
+        const { to, value, data } = await upgradeHelper(address, name);
+        targets.push(to);
+        values.push(value);
+        params.push(data);
+      } else {
+        console.log(`"${name}" is already up to date on network "${networkName}".`);
+      }
+    } catch (error) {
+      console.error(`Error checking or upgrading "${name}" on network "${networkName}":`, error);
+    }
   }
   const multisigData = await buildMultiSigTx(targets, values, params);
   await proposeTransactions(multisigData);