add script for adding implementation for released contracts (#225)

* add script for adding implementation for released contracts

* comments

* fix package.json

* correct abi

Co-authored-by: Yash <kumaryashcse@gmail.com>

Author: joaquim-verges

package.json

@@ -15,6 +15,7 @@
     "@openzeppelin/contracts": "4.5.0",
     "@openzeppelin/contracts-upgradeable": "4.5.1",
     "@primitivefi/hardhat-dodoc": "^0.2.0",
+    "@thirdweb-dev/sdk": "^2.3.39",
     "@typechain/ethers-v5": "^10.0.0",
     "@typechain/hardhat": "^4.0.0",
     "@types/fs-extra": "^9.0.13",

scripts/release/add_implementations_from_release.ts

@@ -0,0 +1,75 @@
+import "dotenv/config";
+import { SUPPORTED_CHAIN_ID, ThirdwebSDK } from "@thirdweb-dev/sdk";
+import { readFileSync } from "fs";
+import { chainIdToName } from "./constants";
+
+////// To run this script: `npx ts-node scripts/release/add_implementations_from_release.ts` //////
+///// MAKE SURE TO PUT IN THE RIGHT CONTRACT NAME HERE AFTER CREATING A RELEASE FOR IT /////
+//// THE RELEASE SHOULD HAVE THE IMPLEMENTATIONS ALREADY DEPLOYED AND RECORDED (via dashboard) ////
+const releasedContractName = "SignatureDrop";
+const privateKey: string = process.env.DEPLOYER_KEY as string; // should be the correct deployer key
+
+const polygonSDK = ThirdwebSDK.fromPrivateKey(privateKey, "polygon");
+
+async function main() {
+  const releaser = await polygonSDK.wallet.getAddress();
+  console.log("Releasing as", releaser);
+
+  const latest = await polygonSDK.getPublisher().getLatest(releaser, releasedContractName);
+
+  if (latest && latest.metadataUri) {
+    console.log(latest);
+    const prev = await polygonSDK.getPublisher().fetchPublishedContractInfo(latest);
+
+    console.log("Fetched latest version", prev);
+    const prevReleaseMetadata = prev.publishedMetadata;
+
+    const implementations = prev.publishedMetadata.factoryDeploymentData?.implementationAddresses;
+    console.log("Implementations", implementations);
+
+    if (!implementations) {
+      console.log("No implementations to approve");
+      return;
+    }
+
+    // Adding implementations
+    console.log("Adding implementations...");
+    for (const [chainId, implementation] of Object.entries(implementations)) {
+      const chainName = chainIdToName[parseInt(chainId) as SUPPORTED_CHAIN_ID];
+      const chainSDK = ThirdwebSDK.fromPrivateKey(privateKey, chainName);
+      const factoryAddr = prevReleaseMetadata?.factoryDeploymentData?.factoryAddresses?.[chainId];
+      if (implementation && factoryAddr) {
+        const factory = chainSDK.getContractFromAbi(
+          factoryAddr,
+          JSON.parse(readFileSync("artifacts_forge/TWFactory.sol/TWFactory.json", "utf-8")).abi,
+        );
+        const approved = await factory.call("approval", implementation);
+        if (!approved) {
+          try {
+            console.log("Adding implementation", implementation, "on", chainName, "to", factoryAddr);
+            await factory.call("addImplementation", implementation);
+          } catch (e) {
+            console.log("Failed to add implementation on", chainName, e);
+          }
+        } else {
+          console.log("Implementation", implementation, "already approved on", chainName);
+        }
+      } else {
+        console.log("No implementation or factory address for", chainName);
+      }
+    }
+  } else {
+    console.log("No previous release found");
+    return;
+  }
+
+  console.log("All done.");
+  console.log("Release page:", `https://thirdweb.com/${releaser}/${releasedContractName}`);
+}
+
+main()
+  .then(() => process.exit(0))
+  .catch(e => {
+    console.error(e);
+    process.exit(1);
+  });

scripts/release/constants.ts

@@ -0,0 +1,51 @@
+import { ChainId, CONTRACT_ADDRESSES } from "@thirdweb-dev/sdk";
+
+export const nativeTokenWrapper: Record<number, string> = {
+  1: "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2", // mainnet
+  4: "0xc778417E063141139Fce010982780140Aa0cD5Ab", // rinkeby
+  5: "0xB4FBF271143F4FBf7B91A5ded31805e42b2208d6", // goerli
+  137: "0x0d500B1d8E8eF31E21C99d1Db9A6444d3ADf1270", // polygon
+  80001: "0x9c3C9283D3e44854697Cd22D3Faa240Cfb032889", // mumbai
+  43114: "0xB31f66AA3C1e785363F0875A1B74E27b85FD66c7", // avalanche
+  43113: "0xd00ae08403B9bbb9124bB305C09058E32C39A48c", // avalanche fuji testnet
+  250: "0x21be370D5312f44cB42ce377BC9b8a0cEF1A4C83", // fantom
+  4002: "0xf1277d1Ed8AD466beddF92ef448A132661956621", // fantom testnet
+  10: "0x4200000000000000000000000000000000000006", // optimism
+  69: "0xbC6F6b680bc61e30dB47721c6D1c5cde19C1300d", // optimism testnet
+  42161: "0x82af49447d8a07e3bd95bd0d56f35241523fbab1", // arbitrum
+  421611: "0xEBbc3452Cc911591e4F18f3b36727Df45d6bd1f9", // arbitrum testnet
+};
+
+export const chainIdToName: Record<number, string> = {
+  [ChainId.Mumbai]: "mumbai",
+  [ChainId.Rinkeby]: "rinkeby",
+  [ChainId.Goerli]: "goerli",
+  [ChainId.Polygon]: "polygon",
+  [ChainId.Mainnet]: "mainnet",
+  [ChainId.Optimism]: "optimism",
+  [ChainId.OptimismTestnet]: "optimism-testnet",
+  [ChainId.Arbitrum]: "arbitrum",
+  [ChainId.ArbitrumTestnet]: "arbitrum-testnet",
+  [ChainId.Fantom]: "fantom",
+  [ChainId.FantomTestnet]: "fantom-testnet",
+  [ChainId.Avalanche]: "avalanche",
+  [ChainId.AvalancheFujiTestnet]: "avalanche-testnet",
+  // [ChainId.BinanceSmartChainMainnet]: "binance",
+  // [ChainId.BinanceSmartChainTestnet]: "binance-testnet",
+};
+
+export const defaultFactories: Record<number, string> = {
+  [ChainId.Mainnet]: CONTRACT_ADDRESSES[ChainId.Mainnet].twFactory,
+  [ChainId.Goerli]: CONTRACT_ADDRESSES[ChainId.Goerli].twFactory,
+  [ChainId.Rinkeby]: CONTRACT_ADDRESSES[ChainId.Rinkeby].twFactory,
+  [ChainId.Polygon]: CONTRACT_ADDRESSES[ChainId.Polygon].twFactory,
+  [ChainId.Mumbai]: CONTRACT_ADDRESSES[ChainId.Mumbai].twFactory,
+  [ChainId.Fantom]: CONTRACT_ADDRESSES[ChainId.Fantom].twFactory,
+  [ChainId.FantomTestnet]: CONTRACT_ADDRESSES[ChainId.FantomTestnet].twFactory,
+  [ChainId.Optimism]: CONTRACT_ADDRESSES[ChainId.Optimism].twFactory,
+  [ChainId.OptimismTestnet]: CONTRACT_ADDRESSES[ChainId.OptimismTestnet].twFactory,
+  [ChainId.Arbitrum]: CONTRACT_ADDRESSES[ChainId.Arbitrum].twFactory,
+  [ChainId.ArbitrumTestnet]: CONTRACT_ADDRESSES[ChainId.ArbitrumTestnet].twFactory,
+  [ChainId.Avalanche]: CONTRACT_ADDRESSES[ChainId.Avalanche].twFactory,
+  [ChainId.AvalancheFujiTestnet]: CONTRACT_ADDRESSES[ChainId.AvalancheFujiTestnet].twFactory,
+};