feat(magicdrop 2): freeze or thaw 721 contract (#227)

Author: rovilay

cli-typescript/src/abis/index.ts

@@ -142,6 +142,20 @@ export const TRANSFER_OWNERSHIP_ABI = {
   type: 'function',
 } as const;
 
+export const SET_TRANSFERABLE_ABI = {
+  inputs: [
+    {
+      internalType: 'bool',
+      name: 'transferable',
+      type: 'bool',
+    },
+  ],
+  name: 'setTransferable',
+  outputs: [],
+  stateMutability: 'nonpayable',
+  type: 'function',
+} as const;
+
 export const ERC712M_ABIS = {
   setup: {
     type: 'function',
@@ -380,19 +394,6 @@ export const ERC712M_ABIS = {
 };
 
 export const ERC1155M_ABIS = {
-  setTransferable: {
-    inputs: [
-      {
-        internalType: 'bool',
-        name: 'transferable',
-        type: 'bool',
-      },
-    ],
-    name: 'setTransferable',
-    outputs: [],
-    stateMutability: 'nonpayable',
-    type: 'function',
-  },
   setup: {
     type: 'function',
     name: 'setup',

cli-typescript/src/cmds/general.ts

@@ -120,9 +120,7 @@ export const freezeThawContractCmd = () =>
     .command('freeze-thaw-contract <symbol>')
     .alias('ftc')
     .alias('freeze-thaw')
-    .description(
-      `Freeze/Thaw contract. Support for ${TOKEN_STANDARD.ERC1155} only`,
-    )
+    .description('Freeze or Thaw contract.')
     .addOption(getFreezeThawOption().makeOptionMandatory())
     .action(freezeThawContractAction);
 

cli-typescript/src/utils/ContractManager.ts

@@ -31,6 +31,7 @@ import {
   MagicDropTokenImplRegistryAbis,
   NEW_CONTRACT_INITIALIZED_EVENT_ABI,
   SET_TRANSFER_VALIDATOR_ABI,
+  SET_TRANSFERABLE_ABI,
   SUPPORTS_INTERFACE_ABI,
 } from '../abis';
 import { printTransactionHash, showText } from './display';
@@ -358,8 +359,8 @@ export class ContractManager {
     );
 
     const data = encodeFunctionData({
-      abi: [ERC1155M_ABIS.setTransferable],
-      functionName: ERC1155M_ABIS.setTransferable.name,
+      abi: [SET_TRANSFERABLE_ABI],
+      functionName: SET_TRANSFERABLE_ABI.name,
       args: [!freeze],
     });
 

cli-typescript/src/utils/cmdActions/freezeThawContractAction.ts

@@ -12,11 +12,6 @@ export const freezeThawContractAction = async (
   try {
     const { cm, config } = await actionPresets(symbol);
 
-    if (config.tokenStandard !== TOKEN_STANDARD.ERC1155)
-      throw new Error(
-        `this action is only supported for ${TOKEN_STANDARD.ERC1155} collections`,
-      );
-
     showText(
       `You're about to ${params.choice} the contract: ${config.deployment!.contract_address}`,
       '',
@@ -41,6 +36,9 @@ export const freezeThawContractAction = async (
       `Token transfers ${params.choice === 'freeze' ? 'frozen' : 'thawed'}.`,
     );
   } catch (error: any) {
-    showError({ text: `Failed to ${params.choice}: ${error.message}` });
+    showError({
+      text: `Failed to ${params.choice}: ${error.message}`,
+      subtext: `contract is probably already ${params.choice === 'freeze' ? 'frozen' : 'thawed'}. Try ${params.choice === 'freeze' ? 'thawing' : 'freezing'} it again.`,
+    });
   }
 };