@openzeppelin/wizard 0.10.4 → 0.10.6

package/dist/utils/convert-strings.d.ts

@@ -1,11 +1,30 @@
 export declare const UINT256_MAX: bigint;
 /**
- * Checks that a string is a valid `uint256` value and converts it to bigint.
+ * Checks that a string is a valid number, and convert to bigint.
  *
  * @param value The string value to check and convert.
  * @param field The field name to use in the error message if the value is invalid.
- * @throws OptionsError if the value is not a valid number or is greater than the maximum value for `uint256`.
+ * @throws OptionsError if the value is not a valid number.
  * @returns The validated value as a bigint.
  */
+export declare function toBigInt(value: string, field: string): bigint;
+/**
+ * Checks that a bigint value fits within `uint256`.
+ *
+ * @param numValue The value to check.
+ * @param field The field name to use in the error if the value is invalid.
+ * @throws OptionsError if the value is greater than the maximum value for `uint256`.
+ * @returns The value as a bigint.
+ */
+export declare function validateUint256(numValue: bigint, field: string): bigint;
+/**
+ * Checks that a string is a valid number, and fits within `uint256`.
+ * Convenience function that calls `toBigInt` and `validateUint256`.
+ *
+ * @param value The value to check.
+ * @param field The field name to use in the error if the value is invalid.
+ * @throws OptionsError if the value is not a valid number or is greater than the maximum value for `uint256`.
+ * @returns The value as a bigint.
+ */
 export declare function toUint256(value: string, field: string): bigint;
 //# sourceMappingURL=convert-strings.d.ts.map

package/dist/utils/convert-strings.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convert-strings.d.ts","sourceRoot":"","sources":["../../src/utils/convert-strings.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,WAAW,QAAuC,CAAC;AAEhE;;;;;;;GAOG;AACH,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAc9D"}
+{"version":3,"file":"convert-strings.d.ts","sourceRoot":"","sources":["../../src/utils/convert-strings.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,WAAW,QAAuC,CAAC;AAEhE;;;;;;;GAOG;AACH,wBAAgB,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAQ7D;AAED;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAOvE;AAED;;;;;;;;GAQG;AACH,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAE9D"}

package/dist/utils/convert-strings.js

@@ -1,25 +1,37 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.UINT256_MAX = void 0;
+exports.toBigInt = toBigInt;
+exports.validateUint256 = validateUint256;
 exports.toUint256 = toUint256;
 const error_1 = require("../error");
 exports.UINT256_MAX = BigInt(2) ** BigInt(256) - BigInt(1);
 /**
- * Checks that a string is a valid `uint256` value and converts it to bigint.
+ * Checks that a string is a valid number, and convert to bigint.
  *
  * @param value The string value to check and convert.
  * @param field The field name to use in the error message if the value is invalid.
- * @throws OptionsError if the value is not a valid number or is greater than the maximum value for `uint256`.
+ * @throws OptionsError if the value is not a valid number.
  * @returns The validated value as a bigint.
  */
-function toUint256(value, field) {
+function toBigInt(value, field) {
     const isValidNumber = /^\d+$/.test(value);
     if (!isValidNumber) {
         throw new error_1.OptionsError({
             [field]: 'Not a valid number',
         });
     }
-    const numValue = BigInt(value);
+    return BigInt(value);
+}
+/**
+ * Checks that a bigint value fits within `uint256`.
+ *
+ * @param numValue The value to check.
+ * @param field The field name to use in the error if the value is invalid.
+ * @throws OptionsError if the value is greater than the maximum value for `uint256`.
+ * @returns The value as a bigint.
+ */
+function validateUint256(numValue, field) {
     if (numValue > exports.UINT256_MAX) {
         throw new error_1.OptionsError({
             [field]: 'Value is greater than uint256 max value',
@@ -27,4 +39,16 @@ function toUint256(value, field) {
     }
     return numValue;
 }
+/**
+ * Checks that a string is a valid number, and fits within `uint256`.
+ * Convenience function that calls `toBigInt` and `validateUint256`.
+ *
+ * @param value The value to check.
+ * @param field The field name to use in the error if the value is invalid.
+ * @throws OptionsError if the value is not a valid number or is greater than the maximum value for `uint256`.
+ * @returns The value as a bigint.
+ */
+function toUint256(value, field) {
+    return validateUint256(toBigInt(value, field), field);
+}
 //# sourceMappingURL=convert-strings.js.map

package/dist/utils/convert-strings.js.map

@@ -1 +1 @@
-{"version":3,"file":"convert-strings.js","sourceRoot":"","sources":["../../src/utils/convert-strings.ts"],"names":[],"mappings":";;;AAYA,8BAcC;AA1BD,oCAAwC;AAE3B,QAAA,WAAW,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;AAEhE;;;;;;;GAOG;AACH,SAAgB,SAAS,CAAC,KAAa,EAAE,KAAa;IACpD,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,oBAAY,CAAC;YACrB,CAAC,KAAK,CAAC,EAAE,oBAAoB;SAC9B,CAAC,CAAC;IACL,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,QAAQ,GAAG,mBAAW,EAAE,CAAC;QAC3B,MAAM,IAAI,oBAAY,CAAC;YACrB,CAAC,KAAK,CAAC,EAAE,yCAAyC;SACnD,CAAC,CAAC;IACL,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC"}
+{"version":3,"file":"convert-strings.js","sourceRoot":"","sources":["../../src/utils/convert-strings.ts"],"names":[],"mappings":";;;AAYA,4BAQC;AAUD,0CAOC;AAWD,8BAEC;AAlDD,oCAAwC;AAE3B,QAAA,WAAW,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;AAEhE;;;;;;;GAOG;AACH,SAAgB,QAAQ,CAAC,KAAa,EAAE,KAAa;IACnD,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,oBAAY,CAAC;YACrB,CAAC,KAAK,CAAC,EAAE,oBAAoB;SAC9B,CAAC,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,eAAe,CAAC,QAAgB,EAAE,KAAa;IAC7D,IAAI,QAAQ,GAAG,mBAAW,EAAE,CAAC;QAC3B,MAAM,IAAI,oBAAY,CAAC;YACrB,CAAC,KAAK,CAAC,EAAE,yCAAyC;SACnD,CAAC,CAAC;IACL,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;;;;;GAQG;AACH,SAAgB,SAAS,CAAC,KAAa,EAAE,KAAa;IACpD,OAAO,eAAe,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;AACxD,CAAC"}

package/dist/zip-hardhat-polkadot.d.ts

@@ -0,0 +1,5 @@
+import type JSZip from 'jszip';
+import type { Contract } from './contract';
+import type { GenericOptions } from './build-generic';
+export declare function zipHardhatPolkadot(c: Contract, opts?: GenericOptions): Promise<JSZip>;
+//# sourceMappingURL=zip-hardhat-polkadot.d.ts.map

package/dist/zip-hardhat-polkadot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"zip-hardhat-polkadot.d.ts","sourceRoot":"","sources":["../src/zip-hardhat-polkadot.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAC/B,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAE3C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAgFtD,wBAAsB,kBAAkB,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,CAE3F"}

package/dist/zip-hardhat-polkadot.js

@@ -0,0 +1,115 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.zipHardhatPolkadot = zipHardhatPolkadot;
+const zip_hardhat_1 = require("./zip-hardhat");
+const solidity_version_json_1 = __importDefault(require("./solidity-version.json"));
+class HardhatPolkadotZipGenerator extends zip_hardhat_1.HardhatZipGenerator {
+    getAdditionalHardhatImports() {
+        return ['@parity/hardhat-polkadot'];
+    }
+    getHardhatConfigJsonString() {
+        return `\
+{
+  solidity: {
+    version: "${solidity_version_json_1.default}",
+    settings: {
+      evmVersion: 'cancun',
+      optimizer: {
+        enabled: true,
+      },
+    },
+  },
+  resolc: {
+    compilerSource: 'npm',
+  },
+  networks: {
+    hardhat: {
+      polkavm: true,
+      nodeConfig: {
+        nodeBinaryPath: 'INSERT_PATH_TO_REVIVE_DEV_NODE',
+        rpcPort: 8000,
+        dev: true,
+      },
+      adapterConfig: {
+        adapterBinaryPath: 'INSERT_PATH_TO_ETH_RPC_ADAPTER',
+        dev: true,
+      },
+    },
+  },
+}`;
+    }
+    async getPackageJson(c) {
+        const { default: packageJson } = await Promise.resolve().then(() => __importStar(require('./environments/hardhat/polkadot/package.json')));
+        packageJson.license = c.license;
+        return packageJson;
+    }
+    async getPackageLock(c) {
+        const { default: packageLock } = await Promise.resolve().then(() => __importStar(require('./environments/hardhat/polkadot/package-lock.json')));
+        packageLock.packages[''].license = c.license;
+        return packageLock;
+    }
+    getReadmePrerequisitesSection() {
+        return `\
+## Prerequisites
+
+Ensure you have the following installed:
+- [Node.js 22.5+](https://nodejs.org/en/download/)
+- npm 10.9.0+
+
+`;
+    }
+    getReadmeTestingEnvironmentSetupSection() {
+        return `\
+## Setting up a testing environment
+
+Follow the steps in [Polkadot's documentation](https://docs.polkadot.com/smart-contracts/dev-environments/local-dev-node/) to set up a local development node and replace the placeholder values \`INSERT_PATH_TO_REVIVE_DEV_NODE\` and \`INSERT_PATH_TO_ETH_RPC_ADAPTER\` in \`hardhat.config.ts\`.
+
+`;
+    }
+    getGitIgnoreHardhatIgnition() {
+        return `
+# Hardhat Ignition default folder for deployments against a local Polkadot Revive Dev node
+ignition/deployments/chain-420420420
+    `;
+    }
+}
+async function zipHardhatPolkadot(c, opts) {
+    return new HardhatPolkadotZipGenerator().zipHardhat(c, opts);
+}
+//# sourceMappingURL=zip-hardhat-polkadot.js.map

package/dist/zip-hardhat-polkadot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zip-hardhat-polkadot.js","sourceRoot":"","sources":["../src/zip-hardhat-polkadot.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmFA,gDAEC;AAnFD,+CAAoD;AAEpD,oFAAuD;AAEvD,MAAM,2BAA4B,SAAQ,iCAAmB;IACjD,2BAA2B;QACnC,OAAO,CAAC,0BAA0B,CAAC,CAAC;IACtC,CAAC;IAES,0BAA0B;QAClC,OAAO;;;gBAGK,+BAAgB;;;;;;;;;;;;;;;;;;;;;;;;;EAyB9B,CAAC;IACD,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,CAAW;QACxC,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,wDAAa,8CAA8C,GAAC,CAAC;QAC9F,WAAW,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QAChC,OAAO,WAAW,CAAC;IACrB,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,CAAW;QACxC,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,wDAAa,mDAAmD,GAAC,CAAC;QACnG,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QAC7C,OAAO,WAAW,CAAC;IACrB,CAAC;IAES,6BAA6B;QACrC,OAAO;;;;;;;CAOV,CAAC;IACA,CAAC;IAES,uCAAuC;QAC/C,OAAO;;;;;CAKV,CAAC;IACA,CAAC;IAES,2BAA2B;QACnC,OAAO;;;KAGN,CAAC;IACJ,CAAC;CACF;AAEM,KAAK,UAAU,kBAAkB,CAAC,CAAW,EAAE,IAAqB;IACzE,OAAO,IAAI,2BAA2B,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;AAC/D,CAAC"}

package/dist/zip-hardhat.d.ts

@@ -1,5 +1,26 @@
 import JSZip from 'jszip';
 import type { GenericOptions } from './build-generic';
 import type { Contract } from './contract';
+export declare class HardhatZipGenerator {
+    protected getAdditionalHardhatImports(): string[];
+    protected getHardhatConfigJsonString(): string;
+    protected getHardhatConfig(upgradeable: boolean): string;
+    protected getTsConfig(): string;
+    protected getGitIgnoreHardhatIgnition(): string;
+    protected getGitIgnore(): string;
+    protected getTest(c: Contract, opts?: GenericOptions): string;
+    getDeploymentCall(c: Contract, args: string[]): string;
+    protected getScript(c: Contract): string;
+    private lowerFirstCharacter;
+    protected getIgnitionModule(c: Contract): string;
+    protected getReadmeTestingEnvironmentSetupSection(): string;
+    protected getReadmePrerequisitesSection(): string;
+    protected getReadme(c: Contract): string;
+    getHardhatPlugins(c: Contract): string[];
+    protected getPackageJson(c: Contract): Promise<unknown>;
+    protected getPackageLock(c: Contract): Promise<unknown>;
+    protected getPrintContract(c: Contract): string;
+    zipHardhat(c: Contract, opts?: GenericOptions): Promise<JSZip>;
+}
 export declare function zipHardhat(c: Contract, opts?: GenericOptions): Promise<JSZip>;
 //# sourceMappingURL=zip-hardhat.d.ts.map

package/dist/zip-hardhat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"zip-hardhat.d.ts","sourceRoot":"","sources":["../src/zip-hardhat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AA8M3C,wBAAsB,UAAU,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,cAAc,kBA8BlE"}
+{"version":3,"file":"zip-hardhat.d.ts","sourceRoot":"","sources":["../src/zip-hardhat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,QAAQ,EAAoB,MAAM,YAAY,CAAC;AAqF7D,qBAAa,mBAAmB;IAC9B,SAAS,CAAC,2BAA2B,IAAI,MAAM,EAAE;IAIjD,SAAS,CAAC,0BAA0B,IAAI,MAAM;IAe9C,SAAS,CAAC,gBAAgB,CAAC,WAAW,EAAE,OAAO,GAAG,MAAM;IAgBxD,SAAS,CAAC,WAAW,IAAI,MAAM;IAgB/B,SAAS,CAAC,2BAA2B,IAAI,MAAM;IAO/C,SAAS,CAAC,YAAY,IAAI,MAAM;IAehC,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,cAAc,GAAG,MAAM;IAItD,iBAAiB,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM;IAW7D,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,QAAQ,GAAG,MAAM;IA0BxC,OAAO,CAAC,mBAAmB;IAI3B,SAAS,CAAC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,GAAG,MAAM;IAchD,SAAS,CAAC,uCAAuC,IAAI,MAAM;IAI3D,SAAS,CAAC,6BAA6B,IAAI,MAAM;IAIjD,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,QAAQ,GAAG,MAAM;IA4BjC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,GAAG,MAAM,EAAE;cAQ/B,cAAc,CAAC,CAAC,EAAE,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC;cAQ7C,cAAc,CAAC,CAAC,EAAE,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC;IAQ7D,SAAS,CAAC,gBAAgB,CAAC,CAAC,EAAE,QAAQ,GAAG,MAAM;IAIzC,UAAU,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC;CAwBrE;AAED,wBAAsB,UAAU,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,CAEnF"}

package/dist/zip-hardhat.js

@@ -36,17 +36,87 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.HardhatZipGenerator = void 0;
 exports.zipHardhat = zipHardhat;
 const jszip_1 = __importDefault(require("jszip"));
 const print_1 = require("./print");
 const solidity_version_json_1 = __importDefault(require("./solidity-version.json"));
 const format_lines_1 = require("./utils/format-lines");
-const hardhatConfig = (upgradeable) => `\
-import { HardhatUserConfig } from "hardhat/config";
-import "@nomicfoundation/hardhat-toolbox";
-${upgradeable ? `import "@openzeppelin/hardhat-upgrades";` : ''}
-
-const config: HardhatUserConfig = {
+class TestGenerator {
+    constructor(parent) {
+        this.parent = parent;
+    }
+    getContent(c, opts) {
+        return (0, format_lines_1.formatLinesWithSpaces)(2, ...(0, format_lines_1.spaceBetween)(this.getImports(c), this.getTestCase(c, opts)));
+    }
+    getTestCase(c, opts) {
+        return [
+            `describe("${c.name}", function () {`,
+            [
+                'it("Test contract", async function () {',
+                (0, format_lines_1.spaceBetween)([`const ContractFactory = await ethers.getContractFactory("${c.name}");`], this.declareVariables(c.constructorArgs), this.getDeployLines(c, c.constructorArgs.map(a => a.name)), this.getExpects(opts)),
+                '});',
+            ],
+            '});',
+        ];
+    }
+    getImports(c) {
+        return [
+            'import { expect } from "chai";',
+            `import { ${this.parent.getHardhatPlugins(c).join(', ')} } from "hardhat";`,
+        ];
+    }
+    getExpects(opts) {
+        if (opts !== undefined) {
+            switch (opts.kind) {
+                case 'ERC20':
+                case 'ERC721':
+                    return [`expect(await instance.name()).to.equal("${opts.name}");`];
+                case 'ERC1155':
+                    return [`expect(await instance.uri(0)).to.equal("${opts.uri}");`];
+                case 'Account':
+                case 'Governor':
+                case 'Custom':
+                    break;
+                default:
+                    throw new Error('Unknown ERC');
+            }
+        }
+        return [];
+    }
+    declareVariables(args) {
+        return args.flatMap((arg, i) => {
+            if (arg.type === 'address') {
+                return [`const ${arg.name} = (await ethers.getSigners())[${i}].address;`];
+            }
+            else {
+                return [`// TODO: Set the following constructor argument`, `// const ${arg.name} = ...;`];
+            }
+        });
+    }
+    getDeployLines(c, argNames) {
+        if (c.constructorArgs.some(a => a.type !== 'address')) {
+            return [
+                `// TODO: Uncomment the below when the missing constructor arguments are set above`,
+                `// const instance = await ${this.parent.getDeploymentCall(c, argNames)};`,
+                `// await instance.waitForDeployment();`,
+            ];
+        }
+        else {
+            return [
+                `const instance = await ${this.parent.getDeploymentCall(c, argNames)};`,
+                'await instance.waitForDeployment();',
+            ];
+        }
+    }
+}
+class HardhatZipGenerator {
+    getAdditionalHardhatImports() {
+        return [];
+    }
+    getHardhatConfigJsonString() {
+        return `\
+{
   solidity: {
     version: "${solidity_version_json_1.default}",
     settings: {
@@ -56,11 +126,23 @@ const config: HardhatUserConfig = {
       },
     },
   },
-};
+}`;
+    }
+    getHardhatConfig(upgradeable) {
+        const additionalImports = this.getAdditionalHardhatImports();
+        const importsSection = additionalImports.length > 0 ? additionalImports.map(imp => `import "${imp}";`).join('\n') + '\n' : '';
+        return `\
+import { HardhatUserConfig } from "hardhat/config";
+import "@nomicfoundation/hardhat-toolbox";
+${importsSection}${upgradeable ? `import "@openzeppelin/hardhat-upgrades";` : ''}
+
+const config: HardhatUserConfig = ${this.getHardhatConfigJsonString()};
 
 export default config;
 `;
-const tsConfig = `\
+    }
+    getTsConfig() {
+        return `\
 {
   "compilerOptions": {
     "target": "es2020",
@@ -73,7 +155,15 @@ const tsConfig = `\
   }
 }
 `;
-const gitIgnore = `\
+    }
+    getGitIgnoreHardhatIgnition() {
+        return `
+# Hardhat Ignition default folder for deployments against a local node
+ignition/deployments/chain-31337
+`;
+    }
+    getGitIgnore() {
+        return `\
 node_modules
 .env
 coverage
@@ -84,78 +174,29 @@ typechain-types
 # Hardhat files
 cache
 artifacts
-`;
-const test = (c, opts) => {
-    return (0, format_lines_1.formatLinesWithSpaces)(2, ...(0, format_lines_1.spaceBetween)(getImports(c), getTestCase(c)));
-    function getTestCase(c) {
-        const args = getAddressArgs(c);
-        return [
-            `describe("${c.name}", function () {`,
-            [
-                'it("Test contract", async function () {',
-                (0, format_lines_1.spaceBetween)([`const ContractFactory = await ethers.getContractFactory("${c.name}");`], getAddressVariables(args), [`const instance = await ${getDeploymentCall(c, args)};`, 'await instance.waitForDeployment();'], getExpects()),
-                '});',
-            ],
-            '});',
-        ];
-    }
-    function getImports(c) {
-        return ['import { expect } from "chai";', `import { ${getHardhatPlugins(c).join(', ')} } from "hardhat";`];
-    }
-    function getExpects() {
-        if (opts !== undefined) {
-            switch (opts.kind) {
-                case 'ERC20':
-                case 'ERC721':
-                    return [`expect(await instance.name()).to.equal("${opts.name}");`];
-                case 'ERC1155':
-                    return [`expect(await instance.uri(0)).to.equal("${opts.uri}");`];
-                case 'Account':
-                case 'Governor':
-                case 'Custom':
-                    break;
-                default:
-                    throw new Error('Unknown ERC');
-            }
-        }
-        return [];
+${this.getGitIgnoreHardhatIgnition()}`;
     }
-    function getAddressVariables(args) {
-        const vars = [];
-        for (let i = 0; i < args.length; i++) {
-            vars.push(`const ${args[i]} = (await ethers.getSigners())[${i}].address;`);
-        }
-        return vars;
+    getTest(c, opts) {
+        return new TestGenerator(this).getContent(c, opts);
     }
-};
-function getAddressArgs(c) {
-    const args = [];
-    for (const constructorArg of c.constructorArgs) {
-        if (constructorArg.type === 'address') {
-            args.push(constructorArg.name);
-        }
+    getDeploymentCall(c, args) {
+        // TODO: remove that selector when the upgrades plugin supports @custom:oz-upgrades-unsafe-allow-reachable
+        const unsafeAllowConstructor = c.parents.find(p => ['EIP712'].includes(p.contract.name)) !== undefined;
+        return !c.upgradeable
+            ? `ContractFactory.deploy(${args.join(', ')})`
+            : unsafeAllowConstructor
+                ? `upgrades.deployProxy(ContractFactory, [${args.join(', ')}], { unsafeAllow: ['constructor'] })`
+                : `upgrades.deployProxy(ContractFactory, [${args.join(', ')}])`;
     }
-    return args;
-}
-function getDeploymentCall(c, args) {
-    // TODO: remove that selector when the upgrades plugin supports @custom:oz-upgrades-unsafe-allow-reachable
-    const unsafeAllowConstructor = c.parents.find(p => ['EIP712'].includes(p.contract.name)) !== undefined;
-    return !c.upgradeable
-        ? `ContractFactory.deploy(${args.join(', ')})`
-        : unsafeAllowConstructor
-            ? `upgrades.deployProxy(ContractFactory, [${args.join(', ')}], { unsafeAllow: 'constructor' })`
-            : `upgrades.deployProxy(ContractFactory, [${args.join(', ')}])`;
-}
-const script = (c) => {
-    const args = getAddressArgs(c);
-    return `\
-import { ${getHardhatPlugins(c).join(', ')} } from "hardhat";
+    getScript(c) {
+        return `\
+import { ${this.getHardhatPlugins(c).join(', ')} } from "hardhat";
 
 async function main() {
   const ContractFactory = await ethers.getContractFactory("${c.name}");
 
-  ${args.length > 0 ? '// TODO: Set addresses for the contract arguments below' : ''}
-  const instance = await ${getDeploymentCall(c, args)};
+  ${c.constructorArgs.length > 0 ? '// TODO: Set values for the constructor arguments below' : ''}
+  const instance = await ${this.getDeploymentCall(c, c.constructorArgs.map(a => a.name))};
   await instance.waitForDeployment();
 
   console.log(\`${c.upgradeable ? 'Proxy' : 'Contract'} deployed to \${await instance.getAddress()}\`);
@@ -168,34 +209,42 @@ main().catch((error) => {
   process.exitCode = 1;
 });
 `;
-};
-const lowerFirstCharacter = (str) => str.charAt(0).toLowerCase() + str.slice(1);
-const ignitionModule = (c) => {
-    const deployArguments = getAddressArgs(c);
-    const contractVariableName = lowerFirstCharacter(c.name);
-    return `import { buildModule } from "@nomicfoundation/hardhat-ignition/modules";
+    }
+    lowerFirstCharacter(str) {
+        return str.charAt(0).toLowerCase() + str.slice(1);
+    }
+    getIgnitionModule(c) {
+        const contractVariableName = this.lowerFirstCharacter(c.name);
+        return `import { buildModule } from "@nomicfoundation/hardhat-ignition/modules";
 
 export default buildModule("${c.name}Module", (m) => {
 
-  ${deployArguments.length > 0 ? '// TODO: Set addresses for the contract arguments below' : ''}
-  const ${contractVariableName} = m.contract("${c.name}", [${deployArguments.join(', ')}]);
+  ${c.constructorArgs.length > 0 ? '// TODO: Set values for the constructor arguments below' : ''}
+  const ${contractVariableName} = m.contract("${c.name}", [${c.constructorArgs.map(a => a.name).join(', ')}]);
 
   return { ${contractVariableName} };
 });
 `;
-};
-const readme = (c) => `\
+    }
+    getReadmeTestingEnvironmentSetupSection() {
+        return '';
+    }
+    getReadmePrerequisitesSection() {
+        return '';
+    }
+    getReadme(c) {
+        return `\
 # Sample Hardhat Project
 
 This project demonstrates a basic Hardhat use case. It comes with a contract generated by [OpenZeppelin Wizard](https://wizard.openzeppelin.com/), a test for that contract, ${c.upgradeable ? 'and a script that deploys that contract' : 'and a Hardhat Ignition module that deploys that contract'}.
 
-## Installing dependencies
+${this.getReadmePrerequisitesSection()}## Installing dependencies
 
 \`\`\`
 npm install
 \`\`\`
 
-## Testing the contract
+${this.getReadmeTestingEnvironmentSetupSection()}## Testing the contract
 
 \`\`\`
 npm test
@@ -209,37 +258,54 @@ You can target any network from your Hardhat config using:
 ${c.upgradeable ? 'npx hardhat run --network <network-name> scripts/deploy.ts' : `npx hardhat ignition deploy ignition/modules/${c.name}.ts --network <network-name>`}
 \`\`\`
 `;
-function getHardhatPlugins(c) {
-    const plugins = ['ethers'];
-    if (c.upgradeable) {
-        plugins.push('upgrades');
     }
-    return plugins;
+    getHardhatPlugins(c) {
+        const plugins = ['ethers'];
+        if (c.upgradeable) {
+            plugins.push('upgrades');
+        }
+        return plugins;
+    }
+    async getPackageJson(c) {
+        const { default: packageJson } = c.upgradeable
+            ? await Promise.resolve().then(() => __importStar(require('./environments/hardhat/upgradeable/package.json')))
+            : await Promise.resolve().then(() => __importStar(require('./environments/hardhat/package.json')));
+        packageJson.license = c.license;
+        return packageJson;
+    }
+    async getPackageLock(c) {
+        const { default: packageLock } = c.upgradeable
+            ? await Promise.resolve().then(() => __importStar(require('./environments/hardhat/upgradeable/package-lock.json')))
+            : await Promise.resolve().then(() => __importStar(require('./environments/hardhat/package-lock.json')));
+        packageLock.packages[''].license = c.license;
+        return packageLock;
+    }
+    getPrintContract(c) {
+        return (0, print_1.printContract)(c);
+    }
+    async zipHardhat(c, opts) {
+        const zip = new jszip_1.default();
+        const packageJson = await this.getPackageJson(c);
+        const packageLock = await this.getPackageLock(c);
+        zip.file(`contracts/${c.name}.sol`, this.getPrintContract(c));
+        zip.file('test/test.ts', this.getTest(c, opts));
+        if (c.upgradeable) {
+            zip.file('scripts/deploy.ts', this.getScript(c));
+        }
+        else {
+            zip.file(`ignition/modules/${c.name}.ts`, this.getIgnitionModule(c));
+        }
+        zip.file('.gitignore', this.getGitIgnore());
+        zip.file('hardhat.config.ts', this.getHardhatConfig(c.upgradeable));
+        zip.file('package.json', JSON.stringify(packageJson, null, 2));
+        zip.file(`package-lock.json`, JSON.stringify(packageLock, null, 2));
+        zip.file('README.md', this.getReadme(c));
+        zip.file('tsconfig.json', this.getTsConfig());
+        return zip;
+    }
 }
+exports.HardhatZipGenerator = HardhatZipGenerator;
 async function zipHardhat(c, opts) {
-    const zip = new jszip_1.default();
-    const { default: packageJson } = c.upgradeable
-        ? await Promise.resolve().then(() => __importStar(require('./environments/hardhat/upgradeable/package.json')))
-        : await Promise.resolve().then(() => __importStar(require('./environments/hardhat/package.json')));
-    packageJson.license = c.license;
-    const { default: packageLock } = c.upgradeable
-        ? await Promise.resolve().then(() => __importStar(require('./environments/hardhat/upgradeable/package-lock.json')))
-        : await Promise.resolve().then(() => __importStar(require('./environments/hardhat/package-lock.json')));
-    packageLock.packages[''].license = c.license;
-    zip.file(`contracts/${c.name}.sol`, (0, print_1.printContract)(c));
-    zip.file('test/test.ts', test(c, opts));
-    if (c.upgradeable) {
-        zip.file('scripts/deploy.ts', script(c));
-    }
-    else {
-        zip.file(`ignition/modules/${c.name}.ts`, ignitionModule(c));
-    }
-    zip.file('.gitignore', gitIgnore);
-    zip.file('hardhat.config.ts', hardhatConfig(c.upgradeable));
-    zip.file('package.json', JSON.stringify(packageJson, null, 2));
-    zip.file(`package-lock.json`, JSON.stringify(packageLock, null, 2));
-    zip.file('README.md', readme(c));
-    zip.file('tsconfig.json', tsConfig);
-    return zip;
+    return new HardhatZipGenerator().zipHardhat(c, opts);
 }
 //# sourceMappingURL=zip-hardhat.js.map