feat: add ability to replace wallets in DV migration (#4456)

Author: mrice32

packages/scripts/src/migration/dvm2-designated-voting/2_Delegate.ts

@@ -4,8 +4,14 @@
 
 const hre = require("hardhat");
 
-import { VotingTokenEthers, DesignatedVotingV2FactoryEthers, VotingV2Ethers } from "@uma/contracts-node";
+import {
+  VotingTokenEthers,
+  DesignatedVotingV2FactoryEthers,
+  VotingV2Ethers,
+  DesignatedVotingV2Ethers,
+} from "@uma/contracts-node";
 import { getContractInstance } from "../../utils/contracts";
+import { utils } from "ethers";
 
 import { baseSafePayload, appendTxToSafePayload } from "./common";
 
@@ -23,6 +29,20 @@ async function main() {
   if (!process.env.GNOSIS_SAFE) throw new Error("No GNOSIS_SAFE set");
   const safe = process.env.GNOSIS_SAFE || "";
 
+  // Replace hot wallets based on provided env variable.
+  // Note: the env variable should be formatted as follows:
+  // HOT_WALLET_REPLACEMENTS="0x1234:0x5678,0x9abc:0xdef0"
+  const replacementPairs = process.env.HOT_WALLET_REPLACEMENTS?.split(",") || [];
+  const oldToNewHotWallet = Object.fromEntries(
+    replacementPairs.map((replacementPair) => {
+      // Split by ":".
+      const [originalWallet, replacementWallet] = replacementPair.split(":");
+      if (!originalWallet || !replacementWallet) throw new Error("Invalid HOT_WALLET_REPLACEMENTS provided");
+      // Ensure that the addresses are formatted consistently.
+      return [utils.getAddress(originalWallet), utils.getAddress(replacementWallet)];
+    })
+  );
+
   // Step 1: fetch all the deployed DesignatedVotingV2 deployed from the factory.
   const factoryV2 = await getContractInstance<DesignatedVotingV2FactoryEthers>("DesignatedVotingV2Factory");
   const votingV2 = await getContractInstance<VotingV2Ethers>("VotingV2");
@@ -44,6 +64,14 @@ async function main() {
   const votingToken = await getContractInstance<VotingTokenEthers>("VotingToken");
   for (const designatedVoting of designatedVotingContracts) {
     const amount = await votingToken.balanceOf(designatedVoting);
+    const dvContract = await getContractInstance<DesignatedVotingV2Ethers>("DesignatedVotingV2", designatedVoting);
+    const currentVoter = utils.getAddress(await dvContract.getMember(1));
+    const updatedHotWallet = oldToNewHotWallet[currentVoter];
+    if (updatedHotWallet)
+      payload = appendTxToSafePayload(payload, designatedVoting, updateVoter, {
+        roleId: 1,
+        newMember: updatedHotWallet,
+      });
     payload = appendTxToSafePayload(payload, designatedVoting, stakeInput, { amount, address: votingV2.address });
     payload = appendTxToSafePayload(payload, designatedVoting, delegateToVoter, {});
   }
@@ -63,6 +91,15 @@ main().then(
   }
 );
 
+const updateVoter = {
+  inputs: [
+    { internalType: "uint256", name: "roleId", type: "uint256" },
+    { internalType: "address", name: "newMember", type: "address" },
+  ],
+  name: "resetMember",
+  payable: false,
+};
+
 const stakeInput = {
   inputs: [
     { internalType: "uint256", name: "amount", type: "uint256" },

yarn.lock

@@ -1316,11 +1316,6 @@
     exec-sh "^0.3.2"
     minimist "^1.2.0"
 
-"@colors/colors@1.5.0":
-  version "1.5.0"
-  resolved "https://registry.yarnpkg.com/@colors/colors/-/colors-1.5.0.tgz#bb504579c1cae923e6576a4f5da43d25f97bdbd9"
-  integrity sha512-ooWCrlZP11i8GImSjTHYHLkvFDP48nS4+204nGb1RiX/WXYHmJA2III9/e2DWVabCESdW7hBAEzHRqUn9OUVvQ==
-
 "@dabh/diagnostics@^2.0.2":
   version "2.0.2"
   resolved "https://registry.yarnpkg.com/@dabh/diagnostics/-/diagnostics-2.0.2.tgz#290d08f7b381b8f94607dc8f471a12c675f9db31"
@@ -7535,11 +7530,6 @@ async@^3.1.0:
   resolved "https://registry.yarnpkg.com/async/-/async-3.2.0.tgz#b3a2685c5ebb641d3de02d161002c60fc9f85720"
   integrity sha512-TR2mEZFVOj2pLStYxLht7TyfuRzaydfpxr3k9RpHIzMgw7A64dzsdqCxH1WJyQdoe8T10nDXd9wnEigmiuHIZw==
 
-async@^3.2.3:
-  version "3.2.4"
-  resolved "https://registry.yarnpkg.com/async/-/async-3.2.4.tgz#2d22e00f8cddeb5fde5dd33522b56d1cf569a81c"
-  integrity sha512-iAB+JbDEGXhyIUavoDl9WP/Jj106Kz9DEn1DPgYw5ruDn0e3Wgi3sKFm55sASdGBNOQB8F59d9qQ7deqrHA8wQ==
-
 asynckit@^0.4.0:
   version "0.4.0"
   resolved "https://registry.yarnpkg.com/asynckit/-/asynckit-0.4.0.tgz#c79ed97f7f34cb8f2ba1bc9790bcc366474b4b79"
@@ -17848,17 +17838,6 @@ logform@^2.2.0:
     ms "^2.1.1"
     triple-beam "^1.3.0"
 
-logform@^2.3.2, logform@^2.4.0:
-  version "2.4.2"
-  resolved "https://registry.yarnpkg.com/logform/-/logform-2.4.2.tgz#a617983ac0334d0c3b942c34945380062795b47c"
-  integrity sha512-W4c9himeAwXEdZ05dQNerhFz2XG80P9Oj0loPUMV23VC2it0orMHQhJm4hdnnor3rd1HsGf6a2lPwBM1zeXHGw==
-  dependencies:
-    "@colors/colors" "1.5.0"
-    fecha "^4.2.0"
-    ms "^2.1.1"
-    safe-stable-stringify "^2.3.1"
-    triple-beam "^1.3.0"
-
 lolex@^5.0.0:
   version "5.1.2"
   resolved "https://registry.yarnpkg.com/lolex/-/lolex-5.1.2.tgz#953694d098ce7c07bc5ed6d0e42bc6c0c6d5a367"
@@ -18512,7 +18491,7 @@ minimist@^1.1.1, minimist@^1.1.3, minimist@^1.2.3, minimist@^1.2.5, minimist@~1.
   resolved "https://registry.yarnpkg.com/minimist/-/minimist-1.2.6.tgz#8637a5b759ea0d6e98702cfb3a9283323c93af44"
   integrity sha512-Jsjnk4bw3YJqYzbdyBiNsPWHPfO++UGG749Cxs6peCu5Xg4nrena6OVxOYxrQTqww0Jmwt+Ref8rggumkTLz9Q==
 
-minimist@^1.2.0, minimist@^1.2.6, minimist@^1.2.7:
+minimist@^1.2.0, minimist@^1.2.7:
   version "1.2.7"
   resolved "https://registry.yarnpkg.com/minimist/-/minimist-1.2.7.tgz#daa1c4d91f507390437c6a8bc01078e7000c4d18"
   integrity sha512-bzfL1YUZsP41gmu/qjrEk0Q6i2ix/cVeAhbCbqH9u3zYutS1cLg00qhrD0M2MVdCcx4Sc0UpP2eBWo9rotpq6g==
@@ -22569,11 +22548,6 @@ safe-regex@^1.1.0:
   dependencies:
     ret "~0.1.10"
 
-safe-stable-stringify@^2.3.1:
-  version "2.4.1"
-  resolved "https://registry.yarnpkg.com/safe-stable-stringify/-/safe-stable-stringify-2.4.1.tgz#34694bd8a30575b7f94792aa51527551bd733d61"
-  integrity sha512-dVHE6bMtS/bnL2mwualjc6IxEv1F+OCUpA46pKUj6F8uDbUM0jCCulPqRNPSnWwGNKx5etqMjZYdXtrm5KJZGA==
-
 "safer-buffer@>= 2.1.2 < 3", "safer-buffer@>= 2.1.2 < 3.0.0", safer-buffer@^2.0.2, safer-buffer@^2.1.0, safer-buffer@~2.1.0:
   version "2.1.2"
   resolved "https://registry.yarnpkg.com/safer-buffer/-/safer-buffer-2.1.2.tgz#44fa161b0187b9549dd84bb91802f9bd8385cd6a"
@@ -26859,15 +26833,6 @@ winston-transport@^4.3.0, winston-transport@^4.4.0:
     readable-stream "^2.3.7"
     triple-beam "^1.2.0"
 
-winston-transport@^4.5.0:
-  version "4.5.0"
-  resolved "https://registry.yarnpkg.com/winston-transport/-/winston-transport-4.5.0.tgz#6e7b0dd04d393171ed5e4e4905db265f7ab384fa"
-  integrity sha512-YpZzcUzBedhlTAfJg6vJDlyEai/IFMIVcaEZZyl3UXIl4gmqRpU7AE89AHLkbzLUsv0NVmw7ts+iztqKxxPW1Q==
-  dependencies:
-    logform "^2.3.2"
-    readable-stream "^3.6.0"
-    triple-beam "^1.3.0"
-
 winston@^3.2.1:
   version "3.3.3"
   resolved "https://registry.yarnpkg.com/winston/-/winston-3.3.3.tgz#ae6172042cafb29786afa3d09c8ff833ab7c9170"
@@ -26883,23 +26848,6 @@ winston@^3.2.1:
     triple-beam "^1.3.0"
     winston-transport "^4.4.0"
 
-winston@^3.8.1:
-  version "3.8.2"
-  resolved "https://registry.yarnpkg.com/winston/-/winston-3.8.2.tgz#56e16b34022eb4cff2638196d9646d7430fdad50"
-  integrity sha512-MsE1gRx1m5jdTTO9Ld/vND4krP2To+lgDoMEHGGa4HIlAUyXJtfc7CxQcGXVyz2IBpw5hbFkj2b/AtUdQwyRew==
-  dependencies:
-    "@colors/colors" "1.5.0"
-    "@dabh/diagnostics" "^2.0.2"
-    async "^3.2.3"
-    is-stream "^2.0.0"
-    logform "^2.4.0"
-    one-time "^1.0.0"
-    readable-stream "^3.4.0"
-    safe-stable-stringify "^2.3.1"
-    stack-trace "0.0.x"
-    triple-beam "^1.3.0"
-    winston-transport "^4.5.0"
-
 word-wrap@^1.2.3, word-wrap@~1.2.3:
   version "1.2.3"
   resolved "https://registry.yarnpkg.com/word-wrap/-/word-wrap-1.2.3.tgz#610636f6b1f703891bd34771ccb17fb93b47079c"