@diamondslab/diamonds-hardhat-foundry 2.2.2 → 2.3.0

package/README.md

@@ -14,7 +14,8 @@ Hardhat plugin that seamlessly integrates Foundry testing with [ERC-2535 Diamond
 - 🚀 **Automated Diamond Deployment** - Deploy Diamond contracts with a single command
 - 📝 **Helper Generation** - Automatically generate Solidity helpers with deployment data
 - 🧪 **Test Scaffolding** - Create unit, integration, and fuzz test templates
-- 🔧 **Hardhat Tasks** - CLI tasks for init, deploy, generate, and test workflows
+- � **Coverage Testing** - Run forge coverage with full Diamond integration and multiple report formats
+- 🔧 **Hardhat Tasks** - CLI tasks for init, deploy, generate, test, and coverage workflows
 - 🎯 **Programmatic API** - Use framework classes directly in scripts
 - 📚 **Base Contracts** - Reusable Solidity utilities and test base classes
 - ⚡ **Foundry Integration** - Leverage Forge's speed and fuzzing capabilities
@@ -342,6 +343,80 @@ npx hardhat diamonds-forge:test --match-test "testOwnership" --verbosity 4
 npx hardhat diamonds-forge:test --skip-deployment --match-contract "MyTest"
 ```
 
+### `diamonds-forge:coverage`
+
+Run forge coverage for your Diamond contracts with full integration support.
+
+**Important:** Coverage requires a deployed Diamond on a persistent network (like localhost). See workflow below.
+
+**Options:**
+
+- `--diamond-name <name>` - Diamond to analyze (default: from config)
+- `--network <name>` - Network for deployment (default: from config)
+- `--report <format>` - Report format: summary, lcov, debug, bytecode (default: summary)
+- `--report-file <path>` - Output file for coverage report
+- `--lcov-version <version>` - LCOV version for LCOV reports (default: v1)
+- `--match-test <pattern>` - Run tests matching pattern
+- `--match-contract <pattern>` - Run tests in matching contracts
+- `--match-path <glob>` - Run tests in files matching glob
+- `--no-match-test <pattern>` - Skip tests matching pattern
+- `--no-match-contract <pattern>` - Skip tests in matching contracts
+- `--no-match-path <glob>` - Skip tests in files matching glob
+- `--no-match-coverage <pattern>` - Exclude contracts from coverage
+- `--verbosity <level>` - Verbosity level: 0-5 (default: 0)
+- `--color <mode>` - Color output: auto, always, never (default: auto)
+- And many more options for filtering, optimization, and EVM configuration
+
+**Workflow (Required):**
+
+```bash
+# Step 1: Start Hardhat node (persistent network)
+npx hardhat node
+
+# Step 2: Deploy Diamond to localhost network (in another terminal)
+npx hardhat diamonds-forge:deploy --diamond-name MyDiamond --network localhost
+
+# Step 3: Run coverage against deployed Diamond
+npx hardhat diamonds-forge:coverage --diamond-name MyDiamond --network localhost
+```
+
+**Examples:**
+
+```bash
+# Basic coverage with default summary
+npx hardhat diamonds-forge:coverage --diamond-name MyDiamond --network localhost
+
+# Generate LCOV report for CI/CD
+npx hardhat diamonds-forge:coverage \
+  --diamond-name MyDiamond \
+  --network localhost \
+  --report lcov \
+  --report-file coverage/lcov.info
+
+# Coverage for specific test patterns
+npx hardhat diamonds-forge:coverage \
+  --diamond-name MyDiamond \
+  --network localhost \
+  --match-contract "Unit" \
+  --verbosity 2
+
+# Multiple report formats
+npx hardhat diamonds-forge:coverage \
+  --diamond-name MyDiamond \
+  --network localhost \
+  --report summary \
+  --report lcov \
+  --report debug
+```
+
+**Important Notes:**
+- Always specify `--network localhost` (coverage needs deployed contracts)
+- Cannot use `--network hardhat` (ephemeral in-memory network)
+- Diamond must be deployed before running coverage
+- Same workflow as `diamonds-forge:test` - both require persistent network
+
+**See the [Coverage Guide](../../docs/FOUNDRY_FORGE_DIAMONDS_COVERAGE.md) for complete documentation, CI/CD integration examples, and best practices.**
+
 ## Dynamic Helper Generation
 
 Version 2.0.0+ introduces **dynamic helper generation** that creates deployment-specific Solidity helpers without hardcoded addresses.

package/dist/framework/ForgeCoverageFramework.d.ts

@@ -0,0 +1,61 @@
+import { HardhatRuntimeEnvironment } from "hardhat/types";
+import { CoverageOptions } from "../types/config";
+/**
+ * ForgeCoverageFramework - Orchestrates forge coverage execution for Diamond contracts
+ *
+ * Responsibilities:
+ * 1. Build forge coverage command with all options
+ * 2. Execute forge coverage with proper fork URL
+ * 3. Stream output to terminal
+ * 4. Handle errors appropriately
+ */
+export declare class ForgeCoverageFramework {
+    private hre;
+    constructor(hre: HardhatRuntimeEnvironment);
+    /**
+     * Run forge coverage with provided options
+     *
+     * @param options - Coverage execution options
+     * @returns Promise<boolean> - true if coverage succeeds, false otherwise
+     */
+    runCoverage(options?: CoverageOptions): Promise<boolean>;
+    /**
+     * Build complete forge coverage command arguments
+     *
+     * @param options - Coverage options
+     * @returns Array of command arguments
+     */
+    private buildCoverageCommand;
+    /**
+     * Build report-related options
+     */
+    private buildReportOptions;
+    /**
+     * Build test filtering options
+     */
+    private buildFilterOptions;
+    /**
+     * Build display options
+     */
+    private buildDisplayOptions;
+    /**
+     * Build test execution options
+     */
+    private buildTestOptions;
+    /**
+     * Build EVM options
+     */
+    private buildEvmOptions;
+    /**
+     * Build build options
+     */
+    private buildBuildOptions;
+    /**
+     * Execute forge coverage command and stream output
+     *
+     * @param args - Command arguments
+     * @returns Promise<boolean> - true if command succeeds
+     */
+    private executeForge;
+}
+//# sourceMappingURL=ForgeCoverageFramework.d.ts.map

package/dist/framework/ForgeCoverageFramework.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ForgeCoverageFramework.d.ts","sourceRoot":"","sources":["../../src/framework/ForgeCoverageFramework.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,yBAAyB,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAGlD;;;;;;;;GAQG;AACH,qBAAa,sBAAsB;IACrB,OAAO,CAAC,GAAG;gBAAH,GAAG,EAAE,yBAAyB;IAElD;;;;;OAKG;IACG,WAAW,CAAC,OAAO,GAAE,eAAoB,GAAG,OAAO,CAAC,OAAO,CAAC;IA0BlE;;;;;OAKG;IACH,OAAO,CAAC,oBAAoB;IAmB5B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAiC1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAkC1B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IA2B3B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA0BxB;;OAEG;IACH,OAAO,CAAC,eAAe;IAsBvB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA0BzB;;;;;OAKG;IACH,OAAO,CAAC,YAAY;CAoBrB"}

package/dist/framework/ForgeCoverageFramework.js

@@ -0,0 +1,238 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ForgeCoverageFramework = void 0;
+const child_process_1 = require("child_process");
+const logger_1 = require("../utils/logger");
+/**
+ * ForgeCoverageFramework - Orchestrates forge coverage execution for Diamond contracts
+ *
+ * Responsibilities:
+ * 1. Build forge coverage command with all options
+ * 2. Execute forge coverage with proper fork URL
+ * 3. Stream output to terminal
+ * 4. Handle errors appropriately
+ */
+class ForgeCoverageFramework {
+    hre;
+    constructor(hre) {
+        this.hre = hre;
+    }
+    /**
+     * Run forge coverage with provided options
+     *
+     * @param options - Coverage execution options
+     * @returns Promise<boolean> - true if coverage succeeds, false otherwise
+     */
+    async runCoverage(options = {}) {
+        logger_1.Logger.section("Forge Coverage - Execution");
+        try {
+            // Build command arguments
+            const args = this.buildCoverageCommand(options);
+            logger_1.Logger.info(`Executing: forge coverage ${args.join(" ")}`);
+            logger_1.Logger.info("⏳ Running coverage analysis (this may take a while)...\n");
+            // Execute forge coverage
+            const success = await this.executeForge(args);
+            if (success) {
+                logger_1.Logger.success("✅ Coverage analysis completed successfully");
+            }
+            else {
+                logger_1.Logger.error("❌ Coverage analysis failed");
+            }
+            return success;
+        }
+        catch (error) {
+            logger_1.Logger.error(`Coverage execution error: ${error.message}`);
+            return false;
+        }
+    }
+    /**
+     * Build complete forge coverage command arguments
+     *
+     * @param options - Coverage options
+     * @returns Array of command arguments
+     */
+    buildCoverageCommand(options) {
+        const args = [];
+        // Add fork URL if provided
+        if (options.forkUrl) {
+            args.push("--fork-url", options.forkUrl);
+        }
+        // Add all option groups
+        args.push(...this.buildReportOptions(options));
+        args.push(...this.buildFilterOptions(options));
+        args.push(...this.buildDisplayOptions(options));
+        args.push(...this.buildTestOptions(options));
+        args.push(...this.buildEvmOptions(options));
+        args.push(...this.buildBuildOptions(options));
+        return args.filter((arg) => arg !== "");
+    }
+    /**
+     * Build report-related options
+     */
+    buildReportOptions(options) {
+        const args = [];
+        // Multiple --report flags
+        if (options.report && options.report.length > 0) {
+            options.report.forEach((reportType) => {
+                args.push("--report", reportType);
+            });
+        }
+        if (options.reportFile) {
+            args.push("--report-file", options.reportFile);
+        }
+        if (options.lcovVersion) {
+            args.push("--lcov-version", options.lcovVersion);
+        }
+        if (options.includeLibs) {
+            args.push("--include-libs");
+        }
+        if (options.excludeTests) {
+            args.push("--exclude-tests");
+        }
+        if (options.irMinimum) {
+            args.push("--ir-minimum");
+        }
+        return args;
+    }
+    /**
+     * Build test filtering options
+     */
+    buildFilterOptions(options) {
+        const args = [];
+        if (options.matchTest) {
+            args.push("--match-test", options.matchTest);
+        }
+        if (options.noMatchTest) {
+            args.push("--no-match-test", options.noMatchTest);
+        }
+        if (options.matchContract) {
+            args.push("--match-contract", options.matchContract);
+        }
+        if (options.noMatchContract) {
+            args.push("--no-match-contract", options.noMatchContract);
+        }
+        if (options.matchPath) {
+            args.push("--match-path", options.matchPath);
+        }
+        if (options.noMatchPath) {
+            args.push("--no-match-path", options.noMatchPath);
+        }
+        if (options.noMatchCoverage) {
+            args.push("--no-match-coverage", options.noMatchCoverage);
+        }
+        return args;
+    }
+    /**
+     * Build display options
+     */
+    buildDisplayOptions(options) {
+        const args = [];
+        // Verbosity (-v, -vv, -vvv, etc.)
+        if (options.verbosity && options.verbosity > 0) {
+            args.push("-" + "v".repeat(options.verbosity));
+        }
+        if (options.quiet) {
+            args.push("--quiet");
+        }
+        if (options.json) {
+            args.push("--json");
+        }
+        if (options.md) {
+            args.push("--md");
+        }
+        if (options.color) {
+            args.push("--color", options.color);
+        }
+        return args;
+    }
+    /**
+     * Build test execution options
+     */
+    buildTestOptions(options) {
+        const args = [];
+        if (options.threads !== undefined) {
+            args.push("--threads", options.threads.toString());
+        }
+        if (options.fuzzRuns !== undefined) {
+            args.push("--fuzz-runs", options.fuzzRuns.toString());
+        }
+        if (options.fuzzSeed) {
+            args.push("--fuzz-seed", options.fuzzSeed);
+        }
+        if (options.failFast) {
+            args.push("--fail-fast");
+        }
+        if (options.allowFailure) {
+            args.push("--allow-failure");
+        }
+        return args;
+    }
+    /**
+     * Build EVM options
+     */
+    buildEvmOptions(options) {
+        const args = [];
+        if (options.forkBlockNumber !== undefined) {
+            args.push("--fork-block-number", options.forkBlockNumber.toString());
+        }
+        if (options.initialBalance) {
+            args.push("--initial-balance", options.initialBalance);
+        }
+        if (options.sender) {
+            args.push("--sender", options.sender);
+        }
+        if (options.ffi) {
+            args.push("--ffi");
+        }
+        return args;
+    }
+    /**
+     * Build build options
+     */
+    buildBuildOptions(options) {
+        const args = [];
+        if (options.force) {
+            args.push("--force");
+        }
+        if (options.noCache) {
+            args.push("--no-cache");
+        }
+        if (options.optimize) {
+            args.push("--optimize");
+        }
+        if (options.optimizerRuns !== undefined) {
+            args.push("--optimizer-runs", options.optimizerRuns.toString());
+        }
+        if (options.viaIr) {
+            args.push("--via-ir");
+        }
+        return args;
+    }
+    /**
+     * Execute forge coverage command and stream output
+     *
+     * @param args - Command arguments
+     * @returns Promise<boolean> - true if command succeeds
+     */
+    executeForge(args) {
+        return new Promise((resolve, reject) => {
+            const forgeProcess = (0, child_process_1.spawn)("forge", ["coverage", ...args], {
+                cwd: this.hre.config.paths.root,
+                stdio: "inherit", // Stream output directly to terminal
+            });
+            forgeProcess.on("close", (code) => {
+                if (code === 0) {
+                    resolve(true);
+                }
+                else {
+                    resolve(false);
+                }
+            });
+            forgeProcess.on("error", (error) => {
+                reject(new Error(`Failed to execute forge coverage: ${error.message}`));
+            });
+        });
+    }
+}
+exports.ForgeCoverageFramework = ForgeCoverageFramework;
+//# sourceMappingURL=ForgeCoverageFramework.js.map

package/dist/framework/ForgeCoverageFramework.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ForgeCoverageFramework.js","sourceRoot":"","sources":["../../src/framework/ForgeCoverageFramework.ts"],"names":[],"mappings":";;;AAAA,iDAAsC;AAGtC,4CAAyC;AAEzC;;;;;;;;GAQG;AACH,MAAa,sBAAsB;IACb;IAApB,YAAoB,GAA8B;QAA9B,QAAG,GAAH,GAAG,CAA2B;IAAG,CAAC;IAEtD;;;;;OAKG;IACH,KAAK,CAAC,WAAW,CAAC,UAA2B,EAAE;QAC7C,eAAM,CAAC,OAAO,CAAC,4BAA4B,CAAC,CAAC;QAE7C,IAAI;YACF,0BAA0B;YAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAEhD,eAAM,CAAC,IAAI,CAAC,6BAA6B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC3D,eAAM,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;YAExE,yBAAyB;YACzB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAE9C,IAAI,OAAO,EAAE;gBACX,eAAM,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAC;aAC9D;iBAAM;gBACL,eAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;aAC5C;YAED,OAAO,OAAO,CAAC;SAChB;QAAC,OAAO,KAAU,EAAE;YACnB,eAAM,CAAC,KAAK,CAAC,6BAA6B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3D,OAAO,KAAK,CAAC;SACd;IACH,CAAC;IAED;;;;;OAKG;IACK,oBAAoB,CAAC,OAAwB;QACnD,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,2BAA2B;QAC3B,IAAI,OAAO,CAAC,OAAO,EAAE;YACnB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;SAC1C;QAED,wBAAwB;QACxB,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;QAChD,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7C,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC;QAE9C,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,OAAwB;QACjD,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,0BAA0B;QAC1B,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;YAC/C,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;gBACpC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;SACJ;QAED,IAAI,OAAO,CAAC,UAAU,EAAE;YACtB,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;SAChD;QAED,IAAI,OAAO,CAAC,WAAW,EAAE;YACvB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;SAClD;QAED,IAAI,OAAO,CAAC,WAAW,EAAE;YACvB,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;SAC7B;QAED,IAAI,OAAO,CAAC,YAAY,EAAE;YACxB,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;SAC9B;QAED,IAAI,OAAO,CAAC,SAAS,EAAE;YACrB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;SAC3B;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,OAAwB;QACjD,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,IAAI,OAAO,CAAC,SAAS,EAAE;YACrB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;SAC9C;QAED,IAAI,OAAO,CAAC,WAAW,EAAE;YACvB,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;SACnD;QAED,IAAI,OAAO,CAAC,aAAa,EAAE;YACzB,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;SACtD;QAED,IAAI,OAAO,CAAC,eAAe,EAAE;YAC3B,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;SAC3D;QAED,IAAI,OAAO,CAAC,SAAS,EAAE;YACrB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;SAC9C;QAED,IAAI,OAAO,CAAC,WAAW,EAAE;YACvB,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;SACnD;QAED,IAAI,OAAO,CAAC,eAAe,EAAE;YAC3B,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;SAC3D;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,OAAwB;QAClD,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,kCAAkC;QAClC,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,GAAG,CAAC,EAAE;YAC9C,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;SAChD;QAED,IAAI,OAAO,CAAC,KAAK,EAAE;YACjB,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;SACtB;QAED,IAAI,OAAO,CAAC,IAAI,EAAE;YAChB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;SACrB;QAED,IAAI,OAAO,CAAC,EAAE,EAAE;YACd,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;SACnB;QAED,IAAI,OAAO,CAAC,KAAK,EAAE;YACjB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;SACrC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,OAAwB;QAC/C,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE;YACjC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;SACpD;QAED,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE;YAClC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;SACvD;QAED,IAAI,OAAO,CAAC,QAAQ,EAAE;YACpB,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;SAC5C;QAED,IAAI,OAAO,CAAC,QAAQ,EAAE;YACpB,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;SAC1B;QAED,IAAI,OAAO,CAAC,YAAY,EAAE;YACxB,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;SAC9B;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,OAAwB;QAC9C,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,EAAE;YACzC,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC,CAAC;SACtE;QAED,IAAI,OAAO,CAAC,cAAc,EAAE;YAC1B,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC;SACxD;QAED,IAAI,OAAO,CAAC,MAAM,EAAE;YAClB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;SACvC;QAED,IAAI,OAAO,CAAC,GAAG,EAAE;YACf,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;SACpB;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,OAAwB;QAChD,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,IAAI,OAAO,CAAC,KAAK,EAAE;YACjB,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;SACtB;QAED,IAAI,OAAO,CAAC,OAAO,EAAE;YACnB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;SACzB;QAED,IAAI,OAAO,CAAC,QAAQ,EAAE;YACpB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;SACzB;QAED,IAAI,OAAO,CAAC,aAAa,KAAK,SAAS,EAAE;YACvC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC,CAAC;SACjE;QAED,IAAI,OAAO,CAAC,KAAK,EAAE;YACjB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;SACvB;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACK,YAAY,CAAC,IAAc;QACjC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,YAAY,GAAG,IAAA,qBAAK,EAAC,OAAO,EAAE,CAAC,UAAU,EAAE,GAAG,IAAI,CAAC,EAAE;gBACzD,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI;gBAC/B,KAAK,EAAE,SAAS,EAAE,qCAAqC;aACxD,CAAC,CAAC;YAEH,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;gBAChC,IAAI,IAAI,KAAK,CAAC,EAAE;oBACd,OAAO,CAAC,IAAI,CAAC,CAAC;iBACf;qBAAM;oBACL,OAAO,CAAC,KAAK,CAAC,CAAC;iBAChB;YACH,CAAC,CAAC,CAAC;YAEH,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjC,MAAM,CAAC,IAAI,KAAK,CAAC,qCAAqC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC1E,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAhRD,wDAgRC","sourcesContent":["import { spawn } from \"child_process\";\nimport { HardhatRuntimeEnvironment } from \"hardhat/types\";\nimport { CoverageOptions } from \"../types/config\";\nimport { Logger } from \"../utils/logger\";\n\n/**\n * ForgeCoverageFramework - Orchestrates forge coverage execution for Diamond contracts\n *\n * Responsibilities:\n * 1. Build forge coverage command with all options\n * 2. Execute forge coverage with proper fork URL\n * 3. Stream output to terminal\n * 4. Handle errors appropriately\n */\nexport class ForgeCoverageFramework {\n  constructor(private hre: HardhatRuntimeEnvironment) {}\n\n  /**\n   * Run forge coverage with provided options\n   *\n   * @param options - Coverage execution options\n   * @returns Promise<boolean> - true if coverage succeeds, false otherwise\n   */\n  async runCoverage(options: CoverageOptions = {}): Promise<boolean> {\n    Logger.section(\"Forge Coverage - Execution\");\n\n    try {\n      // Build command arguments\n      const args = this.buildCoverageCommand(options);\n\n      Logger.info(`Executing: forge coverage ${args.join(\" \")}`);\n      Logger.info(\"⏳ Running coverage analysis (this may take a while)...\\n\");\n\n      // Execute forge coverage\n      const success = await this.executeForge(args);\n\n      if (success) {\n        Logger.success(\"✅ Coverage analysis completed successfully\");\n      } else {\n        Logger.error(\"❌ Coverage analysis failed\");\n      }\n\n      return success;\n    } catch (error: any) {\n      Logger.error(`Coverage execution error: ${error.message}`);\n      return false;\n    }\n  }\n\n  /**\n   * Build complete forge coverage command arguments\n   *\n   * @param options - Coverage options\n   * @returns Array of command arguments\n   */\n  private buildCoverageCommand(options: CoverageOptions): string[] {\n    const args: string[] = [];\n\n    // Add fork URL if provided\n    if (options.forkUrl) {\n      args.push(\"--fork-url\", options.forkUrl);\n    }\n\n    // Add all option groups\n    args.push(...this.buildReportOptions(options));\n    args.push(...this.buildFilterOptions(options));\n    args.push(...this.buildDisplayOptions(options));\n    args.push(...this.buildTestOptions(options));\n    args.push(...this.buildEvmOptions(options));\n    args.push(...this.buildBuildOptions(options));\n\n    return args.filter((arg) => arg !== \"\");\n  }\n\n  /**\n   * Build report-related options\n   */\n  private buildReportOptions(options: CoverageOptions): string[] {\n    const args: string[] = [];\n\n    // Multiple --report flags\n    if (options.report && options.report.length > 0) {\n      options.report.forEach((reportType) => {\n        args.push(\"--report\", reportType);\n      });\n    }\n\n    if (options.reportFile) {\n      args.push(\"--report-file\", options.reportFile);\n    }\n\n    if (options.lcovVersion) {\n      args.push(\"--lcov-version\", options.lcovVersion);\n    }\n\n    if (options.includeLibs) {\n      args.push(\"--include-libs\");\n    }\n\n    if (options.excludeTests) {\n      args.push(\"--exclude-tests\");\n    }\n\n    if (options.irMinimum) {\n      args.push(\"--ir-minimum\");\n    }\n\n    return args;\n  }\n\n  /**\n   * Build test filtering options\n   */\n  private buildFilterOptions(options: CoverageOptions): string[] {\n    const args: string[] = [];\n\n    if (options.matchTest) {\n      args.push(\"--match-test\", options.matchTest);\n    }\n\n    if (options.noMatchTest) {\n      args.push(\"--no-match-test\", options.noMatchTest);\n    }\n\n    if (options.matchContract) {\n      args.push(\"--match-contract\", options.matchContract);\n    }\n\n    if (options.noMatchContract) {\n      args.push(\"--no-match-contract\", options.noMatchContract);\n    }\n\n    if (options.matchPath) {\n      args.push(\"--match-path\", options.matchPath);\n    }\n\n    if (options.noMatchPath) {\n      args.push(\"--no-match-path\", options.noMatchPath);\n    }\n\n    if (options.noMatchCoverage) {\n      args.push(\"--no-match-coverage\", options.noMatchCoverage);\n    }\n\n    return args;\n  }\n\n  /**\n   * Build display options\n   */\n  private buildDisplayOptions(options: CoverageOptions): string[] {\n    const args: string[] = [];\n\n    // Verbosity (-v, -vv, -vvv, etc.)\n    if (options.verbosity && options.verbosity > 0) {\n      args.push(\"-\" + \"v\".repeat(options.verbosity));\n    }\n\n    if (options.quiet) {\n      args.push(\"--quiet\");\n    }\n\n    if (options.json) {\n      args.push(\"--json\");\n    }\n\n    if (options.md) {\n      args.push(\"--md\");\n    }\n\n    if (options.color) {\n      args.push(\"--color\", options.color);\n    }\n\n    return args;\n  }\n\n  /**\n   * Build test execution options\n   */\n  private buildTestOptions(options: CoverageOptions): string[] {\n    const args: string[] = [];\n\n    if (options.threads !== undefined) {\n      args.push(\"--threads\", options.threads.toString());\n    }\n\n    if (options.fuzzRuns !== undefined) {\n      args.push(\"--fuzz-runs\", options.fuzzRuns.toString());\n    }\n\n    if (options.fuzzSeed) {\n      args.push(\"--fuzz-seed\", options.fuzzSeed);\n    }\n\n    if (options.failFast) {\n      args.push(\"--fail-fast\");\n    }\n\n    if (options.allowFailure) {\n      args.push(\"--allow-failure\");\n    }\n\n    return args;\n  }\n\n  /**\n   * Build EVM options\n   */\n  private buildEvmOptions(options: CoverageOptions): string[] {\n    const args: string[] = [];\n\n    if (options.forkBlockNumber !== undefined) {\n      args.push(\"--fork-block-number\", options.forkBlockNumber.toString());\n    }\n\n    if (options.initialBalance) {\n      args.push(\"--initial-balance\", options.initialBalance);\n    }\n\n    if (options.sender) {\n      args.push(\"--sender\", options.sender);\n    }\n\n    if (options.ffi) {\n      args.push(\"--ffi\");\n    }\n\n    return args;\n  }\n\n  /**\n   * Build build options\n   */\n  private buildBuildOptions(options: CoverageOptions): string[] {\n    const args: string[] = [];\n\n    if (options.force) {\n      args.push(\"--force\");\n    }\n\n    if (options.noCache) {\n      args.push(\"--no-cache\");\n    }\n\n    if (options.optimize) {\n      args.push(\"--optimize\");\n    }\n\n    if (options.optimizerRuns !== undefined) {\n      args.push(\"--optimizer-runs\", options.optimizerRuns.toString());\n    }\n\n    if (options.viaIr) {\n      args.push(\"--via-ir\");\n    }\n\n    return args;\n  }\n\n  /**\n   * Execute forge coverage command and stream output\n   *\n   * @param args - Command arguments\n   * @returns Promise<boolean> - true if command succeeds\n   */\n  private executeForge(args: string[]): Promise<boolean> {\n    return new Promise((resolve, reject) => {\n      const forgeProcess = spawn(\"forge\", [\"coverage\", ...args], {\n        cwd: this.hre.config.paths.root,\n        stdio: \"inherit\", // Stream output directly to terminal\n      });\n\n      forgeProcess.on(\"close\", (code) => {\n        if (code === 0) {\n          resolve(true);\n        } else {\n          resolve(false);\n        }\n      });\n\n      forgeProcess.on(\"error\", (error) => {\n        reject(new Error(`Failed to execute forge coverage: ${error.message}`));\n      });\n    });\n  }\n}\n"]}

package/dist/framework/HelperGenerator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"HelperGenerator.d.ts","sourceRoot":"","sources":["../../src/framework/HelperGenerator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAE5D,OAAO,EAAE,yBAAyB,EAAE,MAAM,eAAe,CAAC;AAI1D;;GAEG;AACH,qBAAa,eAAe;IACd,OAAO,CAAC,GAAG;gBAAH,GAAG,EAAE,yBAAyB;IAElD;;OAEG;IACG,eAAe,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBxD;;;OAGG;IACG,yBAAyB,CAC7B,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,mBAAmB,EACnC,OAAO,CAAC,EAAE,GAAG,GACZ,OAAO,CAAC,MAAM,CAAC;IAmClB;;;OAGG;IACH,OAAO,CAAC,4BAA4B;IAuBpC;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAuE/C;;;OAGG;IACH,OAAO,CAAC,qBAAqB;CA0I9B"}
+{"version":3,"file":"HelperGenerator.d.ts","sourceRoot":"","sources":["../../src/framework/HelperGenerator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAW,MAAM,uBAAuB,CAAC;AAErE,OAAO,EAAE,yBAAyB,EAAE,MAAM,eAAe,CAAC;AAI1D;;GAEG;AACH,qBAAa,eAAe;IACd,OAAO,CAAC,GAAG;gBAAH,GAAG,EAAE,yBAAyB;IAElD;;OAEG;IACG,eAAe,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBxD;;;OAGG;IACG,yBAAyB,CAC7B,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,mBAAmB,EACnC,OAAO,CAAC,EAAE,GAAG,GACZ,OAAO,CAAC,MAAM,CAAC;IAmClB;;;OAGG;IACH,OAAO,CAAC,4BAA4B;IA2BpC;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAuE/C;;;OAGG;IACH,OAAO,CAAC,qBAAqB;CA0I9B"}

package/dist/framework/HelperGenerator.js

@@ -56,8 +56,12 @@ class HelperGenerator {
      */
     setForgeEnvironmentVariables(diamond, deploymentData) {
         try {
-            // Get Diamond ABI path
-            const abiPath = diamond.getDiamondAbiFilePath?.() || `./diamond-abi/${deploymentData.DiamondAddress}.json`;
+            // Get Diamond ABI path - remove leading "./" for Foundry compatibility
+            let abiPath = diamond.getDiamondAbiFilePath?.() || `diamond-abi/${diamond.getDiamondConfig().diamondName}.json` || `diamond-abi/Diamond.json`;
+            // Normalize path - remove leading "./" if present
+            if (abiPath.startsWith('./')) {
+                abiPath = abiPath.substring(2);
+            }
             // Get Diamond address
             const diamondAddress = deploymentData.DiamondAddress;
             // Set environment variables
@@ -170,7 +174,7 @@ class HelperGenerator {
         // Diamond ABI path
         source += `    /// @notice Path to the Diamond ABI file\n`;
         source += `    /// @dev Used by DiamondFuzzBase to load ABI for testing\n`;
-        source += `    string constant DIAMOND_ABI_PATH = "./diamond-abi/${diamondName}.json";\n\n`;
+        source += `    string constant DIAMOND_ABI_PATH = "diamond-abi/${diamondName}.json";\n\n`;
         // Diamond address
         source += `    /// @notice Address of the deployed ${diamondName} contract\n`;
         source += `    /// @dev This is the main Diamond proxy address\n`;

package/dist/framework/HelperGenerator.js.map

@@ -1 +1 @@
-{"version":3,"file":"HelperGenerator.js","sourceRoot":"","sources":["../../src/framework/HelperGenerator.ts"],"names":[],"mappings":";;;AACA,2BAAwE;AAExE,+BAA4B;AAC5B,4CAAyC;AAEzC;;GAEG;AACH,MAAa,eAAe;IACN;IAApB,YAAoB,GAA8B;QAA9B,QAAG,GAAH,GAAG,CAA2B;IAAG,CAAC;IAEtD;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,SAAkB;QACtC,MAAM,UAAU,GAAG,SAAS,IAAI,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC;QACpE,MAAM,QAAQ,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAE9D,eAAM,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;QAEnD,qBAAqB;QACrB,eAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvC,IAAA,cAAS,EAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzC,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,gBAAgB,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE1D,eAAM,CAAC,OAAO,CAAC,6BAA6B,QAAQ,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,yBAAyB,CAC7B,WAAmB,EACnB,WAAmB,EACnB,OAAe,EACf,cAAmC,EACnC,OAAa,CAAC,sDAAsD;;QAEpE,eAAM,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QAEvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC;QACvD,MAAM,UAAU,GAAG,IAAA,WAAI,EACrB,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAC1B,UAAU,EACV,uBAAuB,CACxB,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,qBAAqB,CACxC,WAAW,EACX,WAAW,EACX,OAAO,EACP,cAAc,CACf,CAAC;QAEF,0BAA0B;QAC1B,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE;YACtD,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAEH,aAAa;QACb,IAAA,kBAAa,EAAC,UAAU,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAE3C,eAAM,CAAC,OAAO,CAAC,cAAc,UAAU,EAAE,CAAC,CAAC;QAE3C,2EAA2E;QAC3E,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,4BAA4B,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;SAC5D;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACK,4BAA4B,CAClC,OAAY,EACZ,cAAmC;QAEnC,IAAI;YACF,uBAAuB;YACvB,MAAM,OAAO,GAAG,OAAO,CAAC,qBAAqB,EAAE,EAAE,IAAI,iBAAiB,cAAc,CAAC,cAAc,OAAO,CAAC;YAE3G,sBAAsB;YACtB,MAAM,cAAc,GAAG,cAAc,CAAC,cAAc,CAAC;YAErD,4BAA4B;YAC5B,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,OAAO,CAAC;YACvC,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,cAAc,CAAC;YAE7C,eAAM,CAAC,IAAI,CAAC,yBAAyB,OAAO,EAAE,CAAC,CAAC;YAChD,eAAM,CAAC,IAAI,CAAC,wBAAwB,cAAc,EAAE,CAAC,CAAC;SACvD;QAAC,OAAO,KAAU,EAAE;YACnB,eAAM,CAAC,IAAI,CAAC,wCAAwC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACrE,eAAM,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC;SACpE;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB;QACxB,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,YAAY,CAAC;QAEvD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,gBAAgB,EAAE;YAC9C,eAAM,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;YACrD,OAAO,SAAS,CAAC;SAClB;QAED,eAAM,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QACrE,MAAM,aAAa,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,cAAc,CAAC,CAAC;QAEtD,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE;YAC3B,IAAI,YAAY,GAAG,EAAE,CAAC;YACtB,IAAI,UAAU,GAAG,EAAE,CAAC;YAEpB,QAAQ,IAAI,EAAE;gBACZ,KAAK,MAAM;oBACT,YAAY,GAAG,IAAA,WAAI,EAAC,aAAa,EAAE,gCAAgC,CAAC,CAAC;oBACrE,UAAU,GAAG,IAAA,WAAI,EAAC,QAAQ,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;oBACzD,MAAM;gBACR,KAAK,aAAa;oBAChB,YAAY,GAAG,IAAA,WAAI,EAAC,aAAa,EAAE,uCAAuC,CAAC,CAAC;oBAC5E,UAAU,GAAG,IAAA,WAAI,EAAC,QAAQ,EAAE,aAAa,EAAE,0BAA0B,CAAC,CAAC;oBACvE,MAAM;gBACR,KAAK,MAAM;oBACT,YAAY,GAAG,IAAA,WAAI,EAAC,aAAa,EAAE,gCAAgC,CAAC,CAAC;oBACrE,UAAU,GAAG,IAAA,WAAI,EAAC,QAAQ,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;oBACzD,MAAM;gBACR;oBACE,eAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;oBAC7C,SAAS;aACZ;YAED,IAAI;gBACF,2BAA2B;gBAC3B,IAAI,CAAC,IAAA,eAAU,EAAC,YAAY,CAAC,EAAE;oBAC7B,eAAM,CAAC,IAAI,CAAC,uBAAuB,YAAY,EAAE,CAAC,CAAC;oBACnD,SAAS;iBACV;gBAED,wBAAwB;gBACxB,MAAM,eAAe,GAAG,IAAA,iBAAY,EAAC,YAAY,EAAE,MAAM,CAAC,CAAC;gBAE3D,iCAAiC;gBACjC,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAErD,+BAA+B;gBAC/B,IAAI,IAAA,eAAU,EAAC,UAAU,CAAC,EAAE;oBAC1B,eAAM,CAAC,IAAI,CAAC,YAAY,IAAI,8BAA8B,UAAU,EAAE,CAAC,CAAC;oBACxE,SAAS;iBACV;gBAED,0BAA0B;gBAC1B,IAAA,kBAAa,EAAC,UAAU,EAAE,eAAe,EAAE,MAAM,CAAC,CAAC;gBACnD,eAAM,CAAC,OAAO,CAAC,aAAa,IAAI,aAAa,UAAU,EAAE,CAAC,CAAC;gBAC3D,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;aAC5B;YAAC,OAAO,KAAU,EAAE;gBACnB,eAAM,CAAC,KAAK,CAAC,sBAAsB,IAAI,aAAa,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;aACtE;SACF;QAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;YAC1B,eAAM,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;SACnE;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;OAGG;IACK,qBAAqB,CAC3B,WAAmB,EACnB,WAAmB,EACnB,OAAe,EACf,cAAmC;QAEnC,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,MAAM,WAAW,GAAG,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;QAChD,MAAM,kBAAkB,GAAG,GAAG,WAAW,CAAC,WAAW,EAAE,IAAI,WAAW,OAAO,CAAC;QAC9E,MAAM,kBAAkB,GAAG,YAAY,WAAW,gBAAgB,kBAAkB,EAAE,CAAC;QAEvF,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,kBAAkB;QAClB,MAAM,IAAI,mCAAmC,CAAC;QAC9C,MAAM,IAAI,8BAA8B,CAAC;QAEzC,kBAAkB;QAClB,MAAM,IAAI,OAAO,CAAC;QAClB,MAAM,IAAI,+BAA+B,CAAC;QAC1C,MAAM,IAAI,iDAAiD,WAAW,IAAI,CAAC;QAC3E,MAAM,IAAI,8EAA8E,CAAC;QACzF,MAAM,IAAI,oFAAoF,CAAC;QAC/F,MAAM,IAAI,qDAAqD,CAAC;QAChE,MAAM,IAAI,MAAM,CAAC;QACjB,MAAM,IAAI,sBAAsB,kBAAkB,IAAI,CAAC;QACvD,MAAM,IAAI,oBAAoB,SAAS,IAAI,CAAC;QAC5C,MAAM,IAAI,MAAM,CAAC;QACjB,MAAM,IAAI,+BAA+B,CAAC;QAC1C,MAAM,IAAI,8DAA8D,WAAW,IAAI,CAAC;QACxF,MAAM,IAAI,MAAM,CAAC;QACjB,MAAM,IAAI,gFAAgF,CAAC;QAC3F,MAAM,IAAI,OAAO,CAAC;QAClB,MAAM,IAAI,+BAA+B,CAAC;QAE1C,eAAe;QACf,MAAM,IAAI,gDAAgD,CAAC;QAC3D,MAAM,IAAI,0DAA0D,CAAC;QACrE,MAAM,IAAI,uCAAuC,WAAW,QAAQ,CAAC;QAErE,mBAAmB;QACnB,MAAM,IAAI,gDAAgD,CAAC;QAC3D,MAAM,IAAI,gEAAgE,CAAC;QAC3E,MAAM,IAAI,yDAAyD,WAAW,aAAa,CAAC;QAE5F,kBAAkB;QAClB,MAAM,IAAI,2CAA2C,WAAW,aAAa,CAAC;QAC9E,MAAM,IAAI,uDAAuD,CAAC;QAClE,MAAM,IAAI,0CAA0C,cAAc,CAAC,cAAc,OAAO,CAAC;QAEzF,mBAAmB;QACnB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,kDAAkD,CAAC;QAC7D,MAAM,IAAI,2CAA2C,cAAc,CAAC,eAAe,OAAO,CAAC;QAE3F,kBAAkB;QAClB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,0BAA0B,CAAC;QACrC,MAAM,IAAI,qDAAqD,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,cAAc,IAAI,EAAE,CAAC;QACnD,KAAK,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YAC3D,MAAM,YAAY,GAAG,SAAS;iBAC3B,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;iBACrB,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;iBAC1B,WAAW,EAAE;iBACb,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC;YAEhC,MAAM,IAAI,8BAA8B,SAAS,mBAAmB,CAAC;YACrE,MAAM,IAAI,wBAAwB,YAAY,MAAM,SAAS,CAAC,OAAO,KAAK,CAAC;SAC5E;QACD,MAAM,IAAI,IAAI,CAAC;QAEf,mBAAmB;QACnB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,2BAA2B,CAAC;QACtC,MAAM,IAAI,qDAAqD,CAAC;QAEhE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,uCAAuC,CAAC;QAClD,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,yEAAyE,CAAC;QACpF,MAAM,IAAI,gCAAgC,CAAC;QAC3C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,uDAAuD,CAAC;QAClE,MAAM,IAAI,wDAAwD,CAAC;QACnE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,4EAA4E,CAAC;QACvF,MAAM,IAAI,oCAAoC,CAAC;QAC/C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,4DAA4D,CAAC;QACvE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,sEAAsE,CAAC;QACjF,MAAM,IAAI,mCAAmC,CAAC;QAC9C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,2CAA2C,CAAC;QACtD,MAAM,IAAI,sDAAsD,CAAC;QACjE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,uEAAuE,CAAC;QAClF,MAAM,IAAI,oCAAoC,CAAC;QAC/C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,2DAA2D,CAAC;QACtE,MAAM,IAAI,iDAAiD,CAAC;QAC5D,MAAM,IAAI,0DAA0D,CAAC;QACrE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,2FAA2F,CAAC;QAEtG,IAAI,UAAU,GAAG,IAAI,CAAC;QACtB,KAAK,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YAC3D,MAAM,YAAY,GAAG,SAAS;iBAC3B,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;iBACrB,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;iBAC1B,WAAW,EAAE;iBACb,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC;YAEhC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAChD,MAAM,IAAI,WAAW,SAAS,qDAAqD,SAAS,UAAU,CAAC;YACvG,MAAM,IAAI,sBAAsB,YAAY,KAAK,CAAC;YAClD,MAAM,IAAI,aAAa,CAAC;YACxB,UAAU,GAAG,KAAK,CAAC;SACpB;QACD,MAAM,IAAI,8BAA8B,CAAC;QACzC,MAAM,IAAI,SAAS,CAAC;QAEpB,MAAM,IAAI,KAAK,CAAC;QAEhB,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAtTD,0CAsTC","sourcesContent":["import { DeployedDiamondData } from \"@diamondslab/diamonds\";\nimport { existsSync, mkdirSync, readFileSync, writeFileSync } from \"fs\";\nimport { HardhatRuntimeEnvironment } from \"hardhat/types\";\nimport { join } from \"path\";\nimport { Logger } from \"../utils/logger\";\n\n/**\n * HelperGenerator - Generates Solidity helper files for testing\n */\nexport class HelperGenerator {\n  constructor(private hre: HardhatRuntimeEnvironment) {}\n\n  /**\n   * Scaffold project with initial test structure\n   */\n  async scaffoldProject(outputDir?: string): Promise<void> {\n    const helpersDir = outputDir || this.hre.diamondsFoundry.helpersDir;\n    const basePath = join(this.hre.config.paths.root, helpersDir);\n\n    Logger.section(\"Scaffolding Forge Test Structure\");\n    \n    // Create directories\n    Logger.step(\"Creating directories...\");\n    mkdirSync(basePath, { recursive: true });\n    mkdirSync(join(basePath, \"../unit\"), { recursive: true });\n    mkdirSync(join(basePath, \"../integration\"), { recursive: true });\n    mkdirSync(join(basePath, \"../fuzz\"), { recursive: true });\n\n    Logger.success(`Test structure created at ${basePath}`);\n  }\n\n  /**\n   * Generate DiamondDeployment.sol from deployment record\n   * Also sets DIAMOND_ADDRESS and DIAMOND_ABI_PATH environment variables for Forge tests\n   */\n  async generateDeploymentHelpers(\n    diamondName: string,\n    networkName: string,\n    chainId: number,\n    deploymentData: DeployedDiamondData,\n    diamond?: any // Diamond instance from @diamondslab/hardhat-diamonds\n  ): Promise<string> {\n    Logger.section(\"Generating Diamond Deployment Helper\");\n\n    const helpersDir = this.hre.diamondsFoundry.helpersDir;\n    const outputPath = join(\n      this.hre.config.paths.root,\n      helpersDir,\n      \"DiamondDeployment.sol\"\n    );\n\n    const content = this.generateLibrarySource(\n      diamondName,\n      networkName,\n      chainId,\n      deploymentData\n    );\n\n    // Ensure directory exists\n    mkdirSync(join(this.hre.config.paths.root, helpersDir), {\n      recursive: true,\n    });\n\n    // Write file\n    writeFileSync(outputPath, content, \"utf8\");\n\n    Logger.success(`Generated: ${outputPath}`);\n\n    // Set environment variables for Forge tests (if Diamond instance provided)\n    if (diamond) {\n      this.setForgeEnvironmentVariables(diamond, deploymentData);\n    }\n\n    return outputPath;\n  }\n\n  /**\n   * Set environment variables for Forge tests\n   * @private\n   */\n  private setForgeEnvironmentVariables(\n    diamond: any,\n    deploymentData: DeployedDiamondData\n  ): void {\n    try {\n      // Get Diamond ABI path\n      const abiPath = diamond.getDiamondAbiFilePath?.() || `./diamond-abi/${deploymentData.DiamondAddress}.json`;\n      \n      // Get Diamond address\n      const diamondAddress = deploymentData.DiamondAddress;\n\n      // Set environment variables\n      process.env.DIAMOND_ABI_PATH = abiPath;\n      process.env.DIAMOND_ADDRESS = diamondAddress;\n\n      Logger.info(`Set DIAMOND_ABI_PATH: ${abiPath}`);\n      Logger.info(`Set DIAMOND_ADDRESS: ${diamondAddress}`);\n    } catch (error: any) {\n      Logger.warn(`Failed to set environment variables: ${error.message}`);\n      Logger.warn(\"Tests can still use DiamondDeployment.sol overrides\");\n    }\n  }\n\n  /**\n   * Generate example test files\n   */\n  async generateExampleTests(): Promise<string[]> {\n    const generated: string[] = [];\n    const examples = this.hre.diamondsFoundry.exampleTests;\n\n    if (!this.hre.diamondsFoundry.generateExamples) {\n      Logger.info(\"Example generation disabled in config\");\n      return generated;\n    }\n\n    Logger.section(\"Generating Example Tests\");\n\n    const basePath = join(this.hre.config.paths.root, \"test\", \"foundry\");\n    const templatesPath = join(__dirname, \"../templates\");\n\n    for (const type of examples) {\n      let templateFile = \"\";\n      let outputPath = \"\";\n\n      switch (type) {\n        case \"unit\":\n          templateFile = join(templatesPath, \"ExampleUnitTest.t.sol.template\");\n          outputPath = join(basePath, \"unit\", \"ExampleUnit.t.sol\");\n          break;\n        case \"integration\":\n          templateFile = join(templatesPath, \"ExampleIntegrationTest.t.sol.template\");\n          outputPath = join(basePath, \"integration\", \"ExampleIntegration.t.sol\");\n          break;\n        case \"fuzz\":\n          templateFile = join(templatesPath, \"ExampleFuzzTest.t.sol.template\");\n          outputPath = join(basePath, \"fuzz\", \"ExampleFuzz.t.sol\");\n          break;\n        default:\n          Logger.warn(`Unknown example type: ${type}`);\n          continue;\n      }\n\n      try {\n        // Check if template exists\n        if (!existsSync(templateFile)) {\n          Logger.warn(`Template not found: ${templateFile}`);\n          continue;\n        }\n\n        // Read template content\n        const templateContent = readFileSync(templateFile, \"utf8\");\n\n        // Ensure output directory exists\n        mkdirSync(join(basePath, type), { recursive: true });\n\n        // Check if file already exists\n        if (existsSync(outputPath)) {\n          Logger.info(`Skipping ${type} example (already exists): ${outputPath}`);\n          continue;\n        }\n\n        // Write example test file\n        writeFileSync(outputPath, templateContent, \"utf8\");\n        Logger.success(`Generated ${type} example: ${outputPath}`);\n        generated.push(outputPath);\n      } catch (error: any) {\n        Logger.error(`Failed to generate ${type} example: ${error.message}`);\n      }\n    }\n\n    if (generated.length === 0) {\n      Logger.info(\"No new example tests generated (may already exist)\");\n    }\n\n    return generated;\n  }\n\n  /**\n   * Generate Solidity library source from deployment data\n   * @private\n   */\n  private generateLibrarySource(\n    diamondName: string,\n    networkName: string,\n    chainId: number,\n    deploymentData: DeployedDiamondData\n  ): string {\n    const timestamp = new Date().toISOString();\n    const networkInfo = `${networkName}-${chainId}`;\n    const deploymentFileName = `${diamondName.toLowerCase()}-${networkInfo}.json`;\n    const deploymentFilePath = `diamonds/${diamondName}/deployments/${deploymentFileName}`;\n\n    let source = \"\";\n\n    // SPDX and pragma\n    source += \"// SPDX-License-Identifier: MIT\\n\";\n    source += \"pragma solidity ^0.8.19;\\n\\n\";\n\n    // Header comments\n    source += \"/**\\n\";\n    source += ` * @title DiamondDeployment\\n`;\n    source += ` * @notice Auto-generated deployment data for ${diamondName}\\n`;\n    source += ` * @dev This library provides constants and helper functions for accessing\\n`;\n    source += ` *      deployment data in Forge tests. It is auto-generated from the deployment\\n`;\n    source += ` *      record and should not be edited manually.\\n`;\n    source += ` *\\n`;\n    source += ` * Generated from: ${deploymentFilePath}\\n`;\n    source += ` * Generated at: ${timestamp}\\n`;\n    source += ` *\\n`;\n    source += ` * To regenerate this file:\\n`;\n    source += ` *   npx hardhat diamonds-forge:generate-helpers --diamond ${diamondName}\\n`;\n    source += ` *\\n`;\n    source += ` * ⚠️  DO NOT EDIT MANUALLY - Changes will be overwritten on next generation\\n`;\n    source += \" */\\n\";\n    source += \"library DiamondDeployment {\\n\";\n\n    // Diamond name\n    source += `    /// @notice Name of the Diamond contract\\n`;\n    source += `    /// @dev Used for identifying the Diamond in tests\\n`;\n    source += `    string constant DIAMOND_NAME = \"${diamondName}\";\\n\\n`;\n\n    // Diamond ABI path\n    source += `    /// @notice Path to the Diamond ABI file\\n`;\n    source += `    /// @dev Used by DiamondFuzzBase to load ABI for testing\\n`;\n    source += `    string constant DIAMOND_ABI_PATH = \"./diamond-abi/${diamondName}.json\";\\n\\n`;\n\n    // Diamond address\n    source += `    /// @notice Address of the deployed ${diamondName} contract\\n`;\n    source += `    /// @dev This is the main Diamond proxy address\\n`;\n    source += `    address constant DIAMOND_ADDRESS = ${deploymentData.DiamondAddress};\\n\\n`;\n\n    // Deployer address\n    source += `    /// @notice Address of the deployer account\\n`;\n    source += `    /// @dev Account that deployed the Diamond\\n`;\n    source += `    address constant DEPLOYER_ADDRESS = ${deploymentData.DeployerAddress};\\n\\n`;\n\n    // Facet addresses\n    source += \"    // ========================================\\n\";\n    source += \"    // Facet Addresses\\n\";\n    source += \"    // ========================================\\n\\n\";\n\n    const facets = deploymentData.DeployedFacets ?? {};\n    for (const [facetName, facetData] of Object.entries(facets)) {\n      const constantName = facetName\n        .replace(/Facet$/, \"\")\n        .replace(/([A-Z])/g, \"_$1\")\n        .toUpperCase()\n        .replace(/^_/, \"\") + \"_FACET\";\n\n      source += `    /// @notice Address of ${facetName} implementation\\n`;\n      source += `    address constant ${constantName} = ${facetData.address};\\n`;\n    }\n    source += \"\\n\";\n\n    // Helper functions\n    source += \"    // ========================================\\n\";\n    source += \"    // Helper Functions\\n\";\n    source += \"    // ========================================\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the Diamond name\\n\";\n    source += \"     * @return The name of the Diamond contract\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDiamondName() internal pure returns (string memory) {\\n\";\n    source += \"        return DIAMOND_NAME;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the path to the Diamond ABI file\\n\";\n    source += \"     * @return The path to the Diamond ABI JSON file\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDiamondABIPath() internal pure returns (string memory) {\\n\";\n    source += \"        return DIAMOND_ABI_PATH;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the Diamond contract address\\n\";\n    source += \"     * @return The address of the deployed Diamond proxy\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDiamondAddress() internal pure returns (address) {\\n\";\n    source += \"        return DIAMOND_ADDRESS;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the deployer address\\n\";\n    source += \"     * @return The address of the deployer account\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDeployerAddress() internal pure returns (address) {\\n\";\n    source += \"        return DEPLOYER_ADDRESS;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get facet implementation address by name\\n\";\n    source += \"     * @param facetName The name of the facet\\n\";\n    source += \"     * @return The address of the facet implementation\\n\";\n    source += \"     */\\n\";\n    source += \"    function getFacetAddress(string memory facetName) internal pure returns (address) {\\n\";\n\n    let firstFacet = true;\n    for (const [facetName, facetData] of Object.entries(facets)) {\n      const constantName = facetName\n        .replace(/Facet$/, \"\")\n        .replace(/([A-Z])/g, \"_$1\")\n        .toUpperCase()\n        .replace(/^_/, \"\") + \"_FACET\";\n\n      const condition = firstFacet ? \"if\" : \"else if\";\n      source += `        ${condition} (keccak256(bytes(facetName)) == keccak256(bytes(\"${facetName}\"))) {\\n`;\n      source += `            return ${constantName};\\n`;\n      source += \"        }\\n\";\n      firstFacet = false;\n    }\n    source += \"        return address(0);\\n\";\n    source += \"    }\\n\";\n\n    source += \"}\\n\";\n\n    return source;\n  }\n}\n"]}
+{"version":3,"file":"HelperGenerator.js","sourceRoot":"","sources":["../../src/framework/HelperGenerator.ts"],"names":[],"mappings":";;;AACA,2BAAwE;AAExE,+BAA4B;AAC5B,4CAAyC;AAEzC;;GAEG;AACH,MAAa,eAAe;IACN;IAApB,YAAoB,GAA8B;QAA9B,QAAG,GAAH,GAAG,CAA2B;IAAG,CAAC;IAEtD;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,SAAkB;QACtC,MAAM,UAAU,GAAG,SAAS,IAAI,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC;QACpE,MAAM,QAAQ,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAE9D,eAAM,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;QAEnD,qBAAqB;QACrB,eAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvC,IAAA,cAAS,EAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzC,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,gBAAgB,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE1D,eAAM,CAAC,OAAO,CAAC,6BAA6B,QAAQ,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,yBAAyB,CAC7B,WAAmB,EACnB,WAAmB,EACnB,OAAe,EACf,cAAmC,EACnC,OAAa,CAAC,sDAAsD;;QAEpE,eAAM,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QAEvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC;QACvD,MAAM,UAAU,GAAG,IAAA,WAAI,EACrB,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAC1B,UAAU,EACV,uBAAuB,CACxB,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,qBAAqB,CACxC,WAAW,EACX,WAAW,EACX,OAAO,EACP,cAAc,CACf,CAAC;QAEF,0BAA0B;QAC1B,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE;YACtD,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAEH,aAAa;QACb,IAAA,kBAAa,EAAC,UAAU,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAE3C,eAAM,CAAC,OAAO,CAAC,cAAc,UAAU,EAAE,CAAC,CAAC;QAE3C,2EAA2E;QAC3E,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,4BAA4B,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;SAC5D;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACK,4BAA4B,CAClC,OAAgB,EAChB,cAAmC;QAEnC,IAAI;YACF,uEAAuE;YACvE,IAAI,OAAO,GAAG,OAAO,CAAC,qBAAqB,EAAE,EAAE,IAAI,eAAe,OAAO,CAAC,gBAAgB,EAAE,CAAC,WAAW,OAAO,IAAI,0BAA0B,CAAC;YAC9I,kDAAkD;YAClD,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE;gBAC5B,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;aAChC;YAED,sBAAsB;YACtB,MAAM,cAAc,GAAG,cAAc,CAAC,cAAc,CAAC;YAErD,4BAA4B;YAC5B,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,OAAO,CAAC;YACvC,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,cAAc,CAAC;YAE7C,eAAM,CAAC,IAAI,CAAC,yBAAyB,OAAO,EAAE,CAAC,CAAC;YAChD,eAAM,CAAC,IAAI,CAAC,wBAAwB,cAAc,EAAE,CAAC,CAAC;SACvD;QAAC,OAAO,KAAU,EAAE;YACnB,eAAM,CAAC,IAAI,CAAC,wCAAwC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACrE,eAAM,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC;SACpE;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB;QACxB,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,YAAY,CAAC;QAEvD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,gBAAgB,EAAE;YAC9C,eAAM,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;YACrD,OAAO,SAAS,CAAC;SAClB;QAED,eAAM,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QACrE,MAAM,aAAa,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,cAAc,CAAC,CAAC;QAEtD,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE;YAC3B,IAAI,YAAY,GAAG,EAAE,CAAC;YACtB,IAAI,UAAU,GAAG,EAAE,CAAC;YAEpB,QAAQ,IAAI,EAAE;gBACZ,KAAK,MAAM;oBACT,YAAY,GAAG,IAAA,WAAI,EAAC,aAAa,EAAE,gCAAgC,CAAC,CAAC;oBACrE,UAAU,GAAG,IAAA,WAAI,EAAC,QAAQ,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;oBACzD,MAAM;gBACR,KAAK,aAAa;oBAChB,YAAY,GAAG,IAAA,WAAI,EAAC,aAAa,EAAE,uCAAuC,CAAC,CAAC;oBAC5E,UAAU,GAAG,IAAA,WAAI,EAAC,QAAQ,EAAE,aAAa,EAAE,0BAA0B,CAAC,CAAC;oBACvE,MAAM;gBACR,KAAK,MAAM;oBACT,YAAY,GAAG,IAAA,WAAI,EAAC,aAAa,EAAE,gCAAgC,CAAC,CAAC;oBACrE,UAAU,GAAG,IAAA,WAAI,EAAC,QAAQ,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;oBACzD,MAAM;gBACR;oBACE,eAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;oBAC7C,SAAS;aACZ;YAED,IAAI;gBACF,2BAA2B;gBAC3B,IAAI,CAAC,IAAA,eAAU,EAAC,YAAY,CAAC,EAAE;oBAC7B,eAAM,CAAC,IAAI,CAAC,uBAAuB,YAAY,EAAE,CAAC,CAAC;oBACnD,SAAS;iBACV;gBAED,wBAAwB;gBACxB,MAAM,eAAe,GAAG,IAAA,iBAAY,EAAC,YAAY,EAAE,MAAM,CAAC,CAAC;gBAE3D,iCAAiC;gBACjC,IAAA,cAAS,EAAC,IAAA,WAAI,EAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAErD,+BAA+B;gBAC/B,IAAI,IAAA,eAAU,EAAC,UAAU,CAAC,EAAE;oBAC1B,eAAM,CAAC,IAAI,CAAC,YAAY,IAAI,8BAA8B,UAAU,EAAE,CAAC,CAAC;oBACxE,SAAS;iBACV;gBAED,0BAA0B;gBAC1B,IAAA,kBAAa,EAAC,UAAU,EAAE,eAAe,EAAE,MAAM,CAAC,CAAC;gBACnD,eAAM,CAAC,OAAO,CAAC,aAAa,IAAI,aAAa,UAAU,EAAE,CAAC,CAAC;gBAC3D,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;aAC5B;YAAC,OAAO,KAAU,EAAE;gBACnB,eAAM,CAAC,KAAK,CAAC,sBAAsB,IAAI,aAAa,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;aACtE;SACF;QAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;YAC1B,eAAM,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;SACnE;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;OAGG;IACK,qBAAqB,CAC3B,WAAmB,EACnB,WAAmB,EACnB,OAAe,EACf,cAAmC;QAEnC,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,MAAM,WAAW,GAAG,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;QAChD,MAAM,kBAAkB,GAAG,GAAG,WAAW,CAAC,WAAW,EAAE,IAAI,WAAW,OAAO,CAAC;QAC9E,MAAM,kBAAkB,GAAG,YAAY,WAAW,gBAAgB,kBAAkB,EAAE,CAAC;QAEvF,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,kBAAkB;QAClB,MAAM,IAAI,mCAAmC,CAAC;QAC9C,MAAM,IAAI,8BAA8B,CAAC;QAEzC,kBAAkB;QAClB,MAAM,IAAI,OAAO,CAAC;QAClB,MAAM,IAAI,+BAA+B,CAAC;QAC1C,MAAM,IAAI,iDAAiD,WAAW,IAAI,CAAC;QAC3E,MAAM,IAAI,8EAA8E,CAAC;QACzF,MAAM,IAAI,oFAAoF,CAAC;QAC/F,MAAM,IAAI,qDAAqD,CAAC;QAChE,MAAM,IAAI,MAAM,CAAC;QACjB,MAAM,IAAI,sBAAsB,kBAAkB,IAAI,CAAC;QACvD,MAAM,IAAI,oBAAoB,SAAS,IAAI,CAAC;QAC5C,MAAM,IAAI,MAAM,CAAC;QACjB,MAAM,IAAI,+BAA+B,CAAC;QAC1C,MAAM,IAAI,8DAA8D,WAAW,IAAI,CAAC;QACxF,MAAM,IAAI,MAAM,CAAC;QACjB,MAAM,IAAI,gFAAgF,CAAC;QAC3F,MAAM,IAAI,OAAO,CAAC;QAClB,MAAM,IAAI,+BAA+B,CAAC;QAE1C,eAAe;QACf,MAAM,IAAI,gDAAgD,CAAC;QAC3D,MAAM,IAAI,0DAA0D,CAAC;QACrE,MAAM,IAAI,uCAAuC,WAAW,QAAQ,CAAC;QAErE,mBAAmB;QACnB,MAAM,IAAI,gDAAgD,CAAC;QAC3D,MAAM,IAAI,gEAAgE,CAAC;QAC3E,MAAM,IAAI,uDAAuD,WAAW,aAAa,CAAC;QAE1F,kBAAkB;QAClB,MAAM,IAAI,2CAA2C,WAAW,aAAa,CAAC;QAC9E,MAAM,IAAI,uDAAuD,CAAC;QAClE,MAAM,IAAI,0CAA0C,cAAc,CAAC,cAAc,OAAO,CAAC;QAEzF,mBAAmB;QACnB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,kDAAkD,CAAC;QAC7D,MAAM,IAAI,2CAA2C,cAAc,CAAC,eAAe,OAAO,CAAC;QAE3F,kBAAkB;QAClB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,0BAA0B,CAAC;QACrC,MAAM,IAAI,qDAAqD,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,cAAc,IAAI,EAAE,CAAC;QACnD,KAAK,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YAC3D,MAAM,YAAY,GAAG,SAAS;iBAC3B,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;iBACrB,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;iBAC1B,WAAW,EAAE;iBACb,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC;YAEhC,MAAM,IAAI,8BAA8B,SAAS,mBAAmB,CAAC;YACrE,MAAM,IAAI,wBAAwB,YAAY,MAAM,SAAS,CAAC,OAAO,KAAK,CAAC;SAC5E;QACD,MAAM,IAAI,IAAI,CAAC;QAEf,mBAAmB;QACnB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,2BAA2B,CAAC;QACtC,MAAM,IAAI,qDAAqD,CAAC;QAEhE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,uCAAuC,CAAC;QAClD,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,yEAAyE,CAAC;QACpF,MAAM,IAAI,gCAAgC,CAAC;QAC3C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,uDAAuD,CAAC;QAClE,MAAM,IAAI,wDAAwD,CAAC;QACnE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,4EAA4E,CAAC;QACvF,MAAM,IAAI,oCAAoC,CAAC;QAC/C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,mDAAmD,CAAC;QAC9D,MAAM,IAAI,4DAA4D,CAAC;QACvE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,sEAAsE,CAAC;QACjF,MAAM,IAAI,mCAAmC,CAAC;QAC9C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,2CAA2C,CAAC;QACtD,MAAM,IAAI,sDAAsD,CAAC;QACjE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,uEAAuE,CAAC;QAClF,MAAM,IAAI,oCAAoC,CAAC;QAC/C,MAAM,IAAI,WAAW,CAAC;QAEtB,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,2DAA2D,CAAC;QACtE,MAAM,IAAI,iDAAiD,CAAC;QAC5D,MAAM,IAAI,0DAA0D,CAAC;QACrE,MAAM,IAAI,WAAW,CAAC;QACtB,MAAM,IAAI,2FAA2F,CAAC;QAEtG,IAAI,UAAU,GAAG,IAAI,CAAC;QACtB,KAAK,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YAC3D,MAAM,YAAY,GAAG,SAAS;iBAC3B,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;iBACrB,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;iBAC1B,WAAW,EAAE;iBACb,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC;YAEhC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAChD,MAAM,IAAI,WAAW,SAAS,qDAAqD,SAAS,UAAU,CAAC;YACvG,MAAM,IAAI,sBAAsB,YAAY,KAAK,CAAC;YAClD,MAAM,IAAI,aAAa,CAAC;YACxB,UAAU,GAAG,KAAK,CAAC;SACpB;QACD,MAAM,IAAI,8BAA8B,CAAC;QACzC,MAAM,IAAI,SAAS,CAAC;QAEpB,MAAM,IAAI,KAAK,CAAC;QAEhB,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AA1TD,0CA0TC","sourcesContent":["import { DeployedDiamondData, Diamond } from \"@diamondslab/diamonds\";\nimport { existsSync, mkdirSync, readFileSync, writeFileSync } from \"fs\";\nimport { HardhatRuntimeEnvironment } from \"hardhat/types\";\nimport { join } from \"path\";\nimport { Logger } from \"../utils/logger\";\n\n/**\n * HelperGenerator - Generates Solidity helper files for testing\n */\nexport class HelperGenerator {\n  constructor(private hre: HardhatRuntimeEnvironment) {}\n\n  /**\n   * Scaffold project with initial test structure\n   */\n  async scaffoldProject(outputDir?: string): Promise<void> {\n    const helpersDir = outputDir || this.hre.diamondsFoundry.helpersDir;\n    const basePath = join(this.hre.config.paths.root, helpersDir);\n\n    Logger.section(\"Scaffolding Forge Test Structure\");\n    \n    // Create directories\n    Logger.step(\"Creating directories...\");\n    mkdirSync(basePath, { recursive: true });\n    mkdirSync(join(basePath, \"../unit\"), { recursive: true });\n    mkdirSync(join(basePath, \"../integration\"), { recursive: true });\n    mkdirSync(join(basePath, \"../fuzz\"), { recursive: true });\n\n    Logger.success(`Test structure created at ${basePath}`);\n  }\n\n  /**\n   * Generate DiamondDeployment.sol from deployment record\n   * Also sets DIAMOND_ADDRESS and DIAMOND_ABI_PATH environment variables for Forge tests\n   */\n  async generateDeploymentHelpers(\n    diamondName: string,\n    networkName: string,\n    chainId: number,\n    deploymentData: DeployedDiamondData,\n    diamond?: any // Diamond instance from @diamondslab/hardhat-diamonds\n  ): Promise<string> {\n    Logger.section(\"Generating Diamond Deployment Helper\");\n\n    const helpersDir = this.hre.diamondsFoundry.helpersDir;\n    const outputPath = join(\n      this.hre.config.paths.root,\n      helpersDir,\n      \"DiamondDeployment.sol\"\n    );\n\n    const content = this.generateLibrarySource(\n      diamondName,\n      networkName,\n      chainId,\n      deploymentData\n    );\n\n    // Ensure directory exists\n    mkdirSync(join(this.hre.config.paths.root, helpersDir), {\n      recursive: true,\n    });\n\n    // Write file\n    writeFileSync(outputPath, content, \"utf8\");\n\n    Logger.success(`Generated: ${outputPath}`);\n\n    // Set environment variables for Forge tests (if Diamond instance provided)\n    if (diamond) {\n      this.setForgeEnvironmentVariables(diamond, deploymentData);\n    }\n\n    return outputPath;\n  }\n\n  /**\n   * Set environment variables for Forge tests\n   * @private\n   */\n  private setForgeEnvironmentVariables(\n    diamond: Diamond,\n    deploymentData: DeployedDiamondData\n  ): void {\n    try {\n      // Get Diamond ABI path - remove leading \"./\" for Foundry compatibility\n      let abiPath = diamond.getDiamondAbiFilePath?.() || `diamond-abi/${diamond.getDiamondConfig().diamondName}.json` || `diamond-abi/Diamond.json`;\n      // Normalize path - remove leading \"./\" if present\n      if (abiPath.startsWith('./')) {\n        abiPath = abiPath.substring(2);\n      }\n      \n      // Get Diamond address\n      const diamondAddress = deploymentData.DiamondAddress;\n\n      // Set environment variables\n      process.env.DIAMOND_ABI_PATH = abiPath;\n      process.env.DIAMOND_ADDRESS = diamondAddress;\n\n      Logger.info(`Set DIAMOND_ABI_PATH: ${abiPath}`);\n      Logger.info(`Set DIAMOND_ADDRESS: ${diamondAddress}`);\n    } catch (error: any) {\n      Logger.warn(`Failed to set environment variables: ${error.message}`);\n      Logger.warn(\"Tests can still use DiamondDeployment.sol overrides\");\n    }\n  }\n\n  /**\n   * Generate example test files\n   */\n  async generateExampleTests(): Promise<string[]> {\n    const generated: string[] = [];\n    const examples = this.hre.diamondsFoundry.exampleTests;\n\n    if (!this.hre.diamondsFoundry.generateExamples) {\n      Logger.info(\"Example generation disabled in config\");\n      return generated;\n    }\n\n    Logger.section(\"Generating Example Tests\");\n\n    const basePath = join(this.hre.config.paths.root, \"test\", \"foundry\");\n    const templatesPath = join(__dirname, \"../templates\");\n\n    for (const type of examples) {\n      let templateFile = \"\";\n      let outputPath = \"\";\n\n      switch (type) {\n        case \"unit\":\n          templateFile = join(templatesPath, \"ExampleUnitTest.t.sol.template\");\n          outputPath = join(basePath, \"unit\", \"ExampleUnit.t.sol\");\n          break;\n        case \"integration\":\n          templateFile = join(templatesPath, \"ExampleIntegrationTest.t.sol.template\");\n          outputPath = join(basePath, \"integration\", \"ExampleIntegration.t.sol\");\n          break;\n        case \"fuzz\":\n          templateFile = join(templatesPath, \"ExampleFuzzTest.t.sol.template\");\n          outputPath = join(basePath, \"fuzz\", \"ExampleFuzz.t.sol\");\n          break;\n        default:\n          Logger.warn(`Unknown example type: ${type}`);\n          continue;\n      }\n\n      try {\n        // Check if template exists\n        if (!existsSync(templateFile)) {\n          Logger.warn(`Template not found: ${templateFile}`);\n          continue;\n        }\n\n        // Read template content\n        const templateContent = readFileSync(templateFile, \"utf8\");\n\n        // Ensure output directory exists\n        mkdirSync(join(basePath, type), { recursive: true });\n\n        // Check if file already exists\n        if (existsSync(outputPath)) {\n          Logger.info(`Skipping ${type} example (already exists): ${outputPath}`);\n          continue;\n        }\n\n        // Write example test file\n        writeFileSync(outputPath, templateContent, \"utf8\");\n        Logger.success(`Generated ${type} example: ${outputPath}`);\n        generated.push(outputPath);\n      } catch (error: any) {\n        Logger.error(`Failed to generate ${type} example: ${error.message}`);\n      }\n    }\n\n    if (generated.length === 0) {\n      Logger.info(\"No new example tests generated (may already exist)\");\n    }\n\n    return generated;\n  }\n\n  /**\n   * Generate Solidity library source from deployment data\n   * @private\n   */\n  private generateLibrarySource(\n    diamondName: string,\n    networkName: string,\n    chainId: number,\n    deploymentData: DeployedDiamondData\n  ): string {\n    const timestamp = new Date().toISOString();\n    const networkInfo = `${networkName}-${chainId}`;\n    const deploymentFileName = `${diamondName.toLowerCase()}-${networkInfo}.json`;\n    const deploymentFilePath = `diamonds/${diamondName}/deployments/${deploymentFileName}`;\n\n    let source = \"\";\n\n    // SPDX and pragma\n    source += \"// SPDX-License-Identifier: MIT\\n\";\n    source += \"pragma solidity ^0.8.19;\\n\\n\";\n\n    // Header comments\n    source += \"/**\\n\";\n    source += ` * @title DiamondDeployment\\n`;\n    source += ` * @notice Auto-generated deployment data for ${diamondName}\\n`;\n    source += ` * @dev This library provides constants and helper functions for accessing\\n`;\n    source += ` *      deployment data in Forge tests. It is auto-generated from the deployment\\n`;\n    source += ` *      record and should not be edited manually.\\n`;\n    source += ` *\\n`;\n    source += ` * Generated from: ${deploymentFilePath}\\n`;\n    source += ` * Generated at: ${timestamp}\\n`;\n    source += ` *\\n`;\n    source += ` * To regenerate this file:\\n`;\n    source += ` *   npx hardhat diamonds-forge:generate-helpers --diamond ${diamondName}\\n`;\n    source += ` *\\n`;\n    source += ` * ⚠️  DO NOT EDIT MANUALLY - Changes will be overwritten on next generation\\n`;\n    source += \" */\\n\";\n    source += \"library DiamondDeployment {\\n\";\n\n    // Diamond name\n    source += `    /// @notice Name of the Diamond contract\\n`;\n    source += `    /// @dev Used for identifying the Diamond in tests\\n`;\n    source += `    string constant DIAMOND_NAME = \"${diamondName}\";\\n\\n`;\n\n    // Diamond ABI path\n    source += `    /// @notice Path to the Diamond ABI file\\n`;\n    source += `    /// @dev Used by DiamondFuzzBase to load ABI for testing\\n`;\n    source += `    string constant DIAMOND_ABI_PATH = \"diamond-abi/${diamondName}.json\";\\n\\n`;\n\n    // Diamond address\n    source += `    /// @notice Address of the deployed ${diamondName} contract\\n`;\n    source += `    /// @dev This is the main Diamond proxy address\\n`;\n    source += `    address constant DIAMOND_ADDRESS = ${deploymentData.DiamondAddress};\\n\\n`;\n\n    // Deployer address\n    source += `    /// @notice Address of the deployer account\\n`;\n    source += `    /// @dev Account that deployed the Diamond\\n`;\n    source += `    address constant DEPLOYER_ADDRESS = ${deploymentData.DeployerAddress};\\n\\n`;\n\n    // Facet addresses\n    source += \"    // ========================================\\n\";\n    source += \"    // Facet Addresses\\n\";\n    source += \"    // ========================================\\n\\n\";\n\n    const facets = deploymentData.DeployedFacets ?? {};\n    for (const [facetName, facetData] of Object.entries(facets)) {\n      const constantName = facetName\n        .replace(/Facet$/, \"\")\n        .replace(/([A-Z])/g, \"_$1\")\n        .toUpperCase()\n        .replace(/^_/, \"\") + \"_FACET\";\n\n      source += `    /// @notice Address of ${facetName} implementation\\n`;\n      source += `    address constant ${constantName} = ${facetData.address};\\n`;\n    }\n    source += \"\\n\";\n\n    // Helper functions\n    source += \"    // ========================================\\n\";\n    source += \"    // Helper Functions\\n\";\n    source += \"    // ========================================\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the Diamond name\\n\";\n    source += \"     * @return The name of the Diamond contract\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDiamondName() internal pure returns (string memory) {\\n\";\n    source += \"        return DIAMOND_NAME;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the path to the Diamond ABI file\\n\";\n    source += \"     * @return The path to the Diamond ABI JSON file\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDiamondABIPath() internal pure returns (string memory) {\\n\";\n    source += \"        return DIAMOND_ABI_PATH;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the Diamond contract address\\n\";\n    source += \"     * @return The address of the deployed Diamond proxy\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDiamondAddress() internal pure returns (address) {\\n\";\n    source += \"        return DIAMOND_ADDRESS;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get the deployer address\\n\";\n    source += \"     * @return The address of the deployer account\\n\";\n    source += \"     */\\n\";\n    source += \"    function getDeployerAddress() internal pure returns (address) {\\n\";\n    source += \"        return DEPLOYER_ADDRESS;\\n\";\n    source += \"    }\\n\\n\";\n\n    source += \"    /**\\n\";\n    source += \"     * @notice Get facet implementation address by name\\n\";\n    source += \"     * @param facetName The name of the facet\\n\";\n    source += \"     * @return The address of the facet implementation\\n\";\n    source += \"     */\\n\";\n    source += \"    function getFacetAddress(string memory facetName) internal pure returns (address) {\\n\";\n\n    let firstFacet = true;\n    for (const [facetName, facetData] of Object.entries(facets)) {\n      const constantName = facetName\n        .replace(/Facet$/, \"\")\n        .replace(/([A-Z])/g, \"_$1\")\n        .toUpperCase()\n        .replace(/^_/, \"\") + \"_FACET\";\n\n      const condition = firstFacet ? \"if\" : \"else if\";\n      source += `        ${condition} (keccak256(bytes(facetName)) == keccak256(bytes(\"${facetName}\"))) {\\n`;\n      source += `            return ${constantName};\\n`;\n      source += \"        }\\n\";\n      firstFacet = false;\n    }\n    source += \"        return address(0);\\n\";\n    source += \"    }\\n\";\n\n    source += \"}\\n\";\n\n    return source;\n  }\n}\n"]}

package/dist/index.d.ts

@@ -1,3 +1,4 @@
+import "./tasks/coverage";
 import "./tasks/deploy";
 import "./tasks/generate-helpers";
 import "./tasks/init";

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,gBAAgB,CAAC;AACxB,OAAO,0BAA0B,CAAC;AAClC,OAAO,cAAc,CAAC;AACtB,OAAO,cAAc,CAAC;AACtB,OAAO,iBAAiB,CAAC;AAmBzB,cAAc,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,kBAAkB,CAAC;AAC1B,OAAO,gBAAgB,CAAC;AACxB,OAAO,0BAA0B,CAAC;AAClC,OAAO,cAAc,CAAC;AACtB,OAAO,cAAc,CAAC;AACtB,OAAO,iBAAiB,CAAC;AAmBzB,cAAc,gBAAgB,CAAC"}

package/dist/index.js

@@ -18,6 +18,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const config_1 = require("hardhat/config");
+require("./tasks/coverage");
 require("./tasks/deploy");
 require("./tasks/generate-helpers");
 require("./tasks/init");

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AAAA,2CAAqF;AACrF,0BAAwB;AACxB,oCAAkC;AAClC,wBAAsB;AACtB,wBAAsB;AACtB,2BAAyB;AAEzB,2BAA+C;AAC/C,iEAG0C;AAE1C,gDAAwB;AACxB,4DAAoC;AACpC,uCAKmB;AACnB,mDAAoD;AAEpD,eAAe;AACf,iDAA+B;AAE/B,MAAM,iBAAiB,GAAG,cAAc,CAAC;AAEzC,IAAI,eAAe,GAAG,KAAK,CAAC;AAE5B,8CAA8C;AAC9C,IAAA,qBAAY,EAAC,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE;IAClC,0CAA0C;IAC1C,MAAM,CAAC,eAAe,GAAG,IAAA,2BAAc,EAAC,UAAU,CAAC,eAAe,CAAC,CAAC;IAEpE,mFAAmF;IACnF,IAAI,CAAC,IAAA,eAAU,EAAC,cAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC,EAAE;QAC7D,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE;YAC7C,OAAO,CAAC,GAAG,CACT,oBAAU,CAAC,MAAM,CACf,oIAAoI,iBAAiB,kBAAkB,CACxK,CACF,CAAC;SACH;QACD,OAAO;KACR;IAED,sBAAsB;IACtB,MAAM,aAAa,GAAG,IAAA,wBAAc,GAAE,CAAC;IAEvC,6BAA6B;IAC7B,IACE,aAAa,EAAE,GAAG,KAAK,SAAS;QAChC,aAAa,EAAE,UAAU,KAAK,SAAS,EACvC;QACA,MAAM,IAAI,6BAAmB,CAC3B,qFAAqF,CACtF,CAAC;KACH;IAED,gEAAgE;IAChE,MAAM,eAAe,GAAG,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC;IAClD,MAAM,kBAAkB,GAAG,aAAa,CAAC,GAAG,CAAC;IAE7C,IACE,eAAe,KAAK,SAAS;QAC7B,cAAI,CAAC,OAAO,CAAC,eAAe,CAAC,KAAK,cAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAClE;QACA,MAAM,IAAI,6BAAmB,CAC3B,iCAAiC,eAAe,+CAA+C,kBAAkB,GAAG,CACrH,CAAC;KACH;IAED,mBAAmB;IACnB,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,cAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;IAE3E,2DAA2D;IAC3D,MAAM,gBAAgB,GAAG,cAAI,CAAC,OAAO,CACnC,MAAM,CAAC,KAAK,CAAC,IAAI,EACjB,aAAa,CAAC,UAAU,CACzB,CAAC;IACF,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,KAAK,gBAAgB,EAAE;QAC3C,MAAM,CAAC,KAAK,CAAC,KAAK,GAAG,eAAe,CAAC;KACtC;IAED,eAAe,GAAG,IAAI,CAAC;AACzB,CAAC,CAAC,CAAC;AAEH,0DAA0D;AAC1D,IAAA,0BAAiB,EAAC,CAAC,GAA8B,EAAE,EAAE;IACnD,GAAG,CAAC,eAAe,GAAG,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC;AACnD,CAAC,CAAC,CAAC;AAEH,4DAA4D;AAC5D,IAAA,qBAAY,EAAC,+CAAkC,CAAC,CAAC,SAAS,CACxD,KAAK,EACH,EACE,UAAU,EACV,gBAAgB,GACkC,EACpD,IAAI,EACa,EAAE;IACnB,8FAA8F;IAC9F,IAAI,gBAAgB,EAAE;QACpB,OAAO,UAAU,CAAC;KACnB;IACD,MAAM,IAAI,6BAAmB,CAC3B,+EAA+E,CAChF,CAAC;AACJ,CAAC,CACF,CAAC;AAEF,IAAA,qBAAY,EAAC,wCAA2B,CAAC,CAAC,SAAS,CACjD,KAAK,IAAqC,EAAE;IAC1C,IAAI,CAAC,eAAe,EAAE;QACpB,OAAO,EAAE,CAAC;KACX;IAED,OAAO,IAAA,uBAAa,GAAE,CAAC;AACzB,CAAC,CACF,CAAC;AAEF,IAAA,aAAI,EACF,iBAAiB,EACjB,qDAAqD,EACrD,KAAK,EAAE,CAAC,EAAE,GAA8B,EAAE,EAAE;IAC1C,MAAM,iBAAiB,GAAG,cAAI,CAAC,OAAO,CACpC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EACrB,cAAc,CACf,CAAC;IAEF,IAAI,IAAA,eAAU,EAAC,iBAAiB,CAAC,EAAE;QACjC,OAAO,CAAC,IAAI,CACV,oBAAU,CAAC,MAAM,CAAC,6CAA6C,CAAC,CACjE,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;KACjB;IAED,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAE7C,IAAA,kBAAa,EACX,iBAAiB,EACjB;QACE,mBAAmB;QACnB,UAAU,cAAI,CAAC,QAAQ,CACrB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EACrB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CACzB,GAAG;QACJ,aAAa;QACb,gCAAgC;QAChC,WAAW,cAAI,CAAC,QAAQ,CACtB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EACrB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CACvB,GAAG;QACJ,6BAA6B;KAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;IAEF,MAAM,IAAA,2BAAiB,EAAC,sBAAsB,CAAC,CAAC;AAClD,CAAC,CACF,CAAC","sourcesContent":["import { extendConfig, extendEnvironment, internalTask, task } from \"hardhat/config\";\nimport \"./tasks/deploy\";\nimport \"./tasks/generate-helpers\";\nimport \"./tasks/init\";\nimport \"./tasks/test\";\nimport \"./types/hardhat\";\n\nimport { existsSync, writeFileSync } from \"fs\";\nimport {\n  TASK_COMPILE_GET_REMAPPINGS,\n  TASK_COMPILE_TRANSFORM_IMPORT_NAME,\n} from \"hardhat/builtin-tasks/task-names\";\nimport { HardhatRuntimeEnvironment } from \"hardhat/types\";\nimport path from \"path\";\nimport picocolors from \"picocolors\";\nimport {\n  getForgeConfig,\n  getRemappings,\n  HardhatFoundryError,\n  installDependency,\n} from \"./foundry\";\nimport { validateConfig } from \"./utils/validation\";\n\n// Export types\nexport * from \"./types/config\";\n\nconst TASK_INIT_FOUNDRY = \"init-foundry\";\n\nlet pluginActivated = false;\n\n// Extend config with diamondsFoundry settings\nextendConfig((config, userConfig) => {\n  // Validate and set diamondsFoundry config\n  config.diamondsFoundry = validateConfig(userConfig.diamondsFoundry);\n\n  // Check foundry.toml presence. Don't warn when running foundry initialization task\n  if (!existsSync(path.join(config.paths.root, \"foundry.toml\"))) {\n    if (!process.argv.includes(TASK_INIT_FOUNDRY)) {\n      console.log(\n        picocolors.yellow(\n          `Warning: You are using the diamonds-hardhat-foundry plugin but there isn't a foundry.toml file in your project. Run 'npx hardhat ${TASK_INIT_FOUNDRY}' to create one.`\n        )\n      );\n    }\n    return;\n  }\n\n  // Load foundry config\n  const foundryConfig = getForgeConfig();\n\n  // Ensure required keys exist\n  if (\n    foundryConfig?.src === undefined ||\n    foundryConfig?.cache_path === undefined\n  ) {\n    throw new HardhatFoundryError(\n      \"Couldn't find `src` or `cache_path` config keys after running `forge config --json`\"\n    );\n  }\n\n  // Ensure foundry src path doesn't mismatch user-configured path\n  const userSourcesPath = userConfig.paths?.sources;\n  const foundrySourcesPath = foundryConfig.src;\n\n  if (\n    userSourcesPath !== undefined &&\n    path.resolve(userSourcesPath) !== path.resolve(foundrySourcesPath)\n  ) {\n    throw new HardhatFoundryError(\n      `User-configured sources path (${userSourcesPath}) doesn't match path configured in foundry (${foundrySourcesPath})`\n    );\n  }\n\n  // Set sources path\n  config.paths.sources = path.resolve(config.paths.root, foundrySourcesPath);\n\n  // Change hardhat's cache path if it clashes with foundry's\n  const foundryCachePath = path.resolve(\n    config.paths.root,\n    foundryConfig.cache_path\n  );\n  if (config.paths.cache === foundryCachePath) {\n    config.paths.cache = \"cache_hardhat\";\n  }\n\n  pluginActivated = true;\n});\n\n// Extend environment to add diamondsFoundry config to HRE\nextendEnvironment((hre: HardhatRuntimeEnvironment) => {\n  hre.diamondsFoundry = hre.config.diamondsFoundry;\n});\n\n// This task is in place to detect old hardhat-core versions\ninternalTask(TASK_COMPILE_TRANSFORM_IMPORT_NAME).setAction(\n  async (\n    {\n      importName,\n      deprecationCheck,\n    }: { importName: string; deprecationCheck: boolean },\n    _hre\n  ): Promise<string> => {\n    // When the deprecationCheck param is passed, it means a new enough hardhat-core is being used\n    if (deprecationCheck) {\n      return importName;\n    }\n    throw new HardhatFoundryError(\n      \"This version of diamonds-hardhat-foundry depends on hardhat version >= 2.17.2\"\n    );\n  }\n);\n\ninternalTask(TASK_COMPILE_GET_REMAPPINGS).setAction(\n  async (): Promise<Record<string, string>> => {\n    if (!pluginActivated) {\n      return {};\n    }\n\n    return getRemappings();\n  }\n);\n\ntask(\n  TASK_INIT_FOUNDRY,\n  \"Initialize foundry setup in current hardhat project\",\n  async (_, hre: HardhatRuntimeEnvironment) => {\n    const foundryConfigPath = path.resolve(\n      hre.config.paths.root,\n      \"foundry.toml\"\n    );\n\n    if (existsSync(foundryConfigPath)) {\n      console.warn(\n        picocolors.yellow(`File foundry.toml already exists. Aborting.`)\n      );\n      process.exit(1);\n    }\n\n    console.log(`Creating foundry.toml file...`);\n\n    writeFileSync(\n      foundryConfigPath,\n      [\n        `[profile.default]`,\n        `src = '${path.relative(\n          hre.config.paths.root,\n          hre.config.paths.sources\n        )}'`,\n        `out = 'out'`,\n        `libs = ['node_modules', 'lib']`,\n        `test = '${path.relative(\n          hre.config.paths.root,\n          hre.config.paths.tests\n        )}'`,\n        `cache_path  = 'cache_forge'`,\n      ].join(\"\\n\")\n    );\n\n    await installDependency(\"foundry-rs/forge-std\");\n  }\n);\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AAAA,2CAAqF;AACrF,4BAA0B;AAC1B,0BAAwB;AACxB,oCAAkC;AAClC,wBAAsB;AACtB,wBAAsB;AACtB,2BAAyB;AAEzB,2BAA+C;AAC/C,iEAG0C;AAE1C,gDAAwB;AACxB,4DAAoC;AACpC,uCAKmB;AACnB,mDAAoD;AAEpD,eAAe;AACf,iDAA+B;AAE/B,MAAM,iBAAiB,GAAG,cAAc,CAAC;AAEzC,IAAI,eAAe,GAAG,KAAK,CAAC;AAE5B,8CAA8C;AAC9C,IAAA,qBAAY,EAAC,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE;IAClC,0CAA0C;IAC1C,MAAM,CAAC,eAAe,GAAG,IAAA,2BAAc,EAAC,UAAU,CAAC,eAAe,CAAC,CAAC;IAEpE,mFAAmF;IACnF,IAAI,CAAC,IAAA,eAAU,EAAC,cAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC,EAAE;QAC7D,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE;YAC7C,OAAO,CAAC,GAAG,CACT,oBAAU,CAAC,MAAM,CACf,oIAAoI,iBAAiB,kBAAkB,CACxK,CACF,CAAC;SACH;QACD,OAAO;KACR;IAED,sBAAsB;IACtB,MAAM,aAAa,GAAG,IAAA,wBAAc,GAAE,CAAC;IAEvC,6BAA6B;IAC7B,IACE,aAAa,EAAE,GAAG,KAAK,SAAS;QAChC,aAAa,EAAE,UAAU,KAAK,SAAS,EACvC;QACA,MAAM,IAAI,6BAAmB,CAC3B,qFAAqF,CACtF,CAAC;KACH;IAED,gEAAgE;IAChE,MAAM,eAAe,GAAG,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC;IAClD,MAAM,kBAAkB,GAAG,aAAa,CAAC,GAAG,CAAC;IAE7C,IACE,eAAe,KAAK,SAAS;QAC7B,cAAI,CAAC,OAAO,CAAC,eAAe,CAAC,KAAK,cAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAClE;QACA,MAAM,IAAI,6BAAmB,CAC3B,iCAAiC,eAAe,+CAA+C,kBAAkB,GAAG,CACrH,CAAC;KACH;IAED,mBAAmB;IACnB,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,cAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;IAE3E,2DAA2D;IAC3D,MAAM,gBAAgB,GAAG,cAAI,CAAC,OAAO,CACnC,MAAM,CAAC,KAAK,CAAC,IAAI,EACjB,aAAa,CAAC,UAAU,CACzB,CAAC;IACF,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,KAAK,gBAAgB,EAAE;QAC3C,MAAM,CAAC,KAAK,CAAC,KAAK,GAAG,eAAe,CAAC;KACtC;IAED,eAAe,GAAG,IAAI,CAAC;AACzB,CAAC,CAAC,CAAC;AAEH,0DAA0D;AAC1D,IAAA,0BAAiB,EAAC,CAAC,GAA8B,EAAE,EAAE;IACnD,GAAG,CAAC,eAAe,GAAG,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC;AACnD,CAAC,CAAC,CAAC;AAEH,4DAA4D;AAC5D,IAAA,qBAAY,EAAC,+CAAkC,CAAC,CAAC,SAAS,CACxD,KAAK,EACH,EACE,UAAU,EACV,gBAAgB,GACkC,EACpD,IAAI,EACa,EAAE;IACnB,8FAA8F;IAC9F,IAAI,gBAAgB,EAAE;QACpB,OAAO,UAAU,CAAC;KACnB;IACD,MAAM,IAAI,6BAAmB,CAC3B,+EAA+E,CAChF,CAAC;AACJ,CAAC,CACF,CAAC;AAEF,IAAA,qBAAY,EAAC,wCAA2B,CAAC,CAAC,SAAS,CACjD,KAAK,IAAqC,EAAE;IAC1C,IAAI,CAAC,eAAe,EAAE;QACpB,OAAO,EAAE,CAAC;KACX;IAED,OAAO,IAAA,uBAAa,GAAE,CAAC;AACzB,CAAC,CACF,CAAC;AAEF,IAAA,aAAI,EACF,iBAAiB,EACjB,qDAAqD,EACrD,KAAK,EAAE,CAAC,EAAE,GAA8B,EAAE,EAAE;IAC1C,MAAM,iBAAiB,GAAG,cAAI,CAAC,OAAO,CACpC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EACrB,cAAc,CACf,CAAC;IAEF,IAAI,IAAA,eAAU,EAAC,iBAAiB,CAAC,EAAE;QACjC,OAAO,CAAC,IAAI,CACV,oBAAU,CAAC,MAAM,CAAC,6CAA6C,CAAC,CACjE,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;KACjB;IAED,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAE7C,IAAA,kBAAa,EACX,iBAAiB,EACjB;QACE,mBAAmB;QACnB,UAAU,cAAI,CAAC,QAAQ,CACrB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EACrB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CACzB,GAAG;QACJ,aAAa;QACb,gCAAgC;QAChC,WAAW,cAAI,CAAC,QAAQ,CACtB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EACrB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CACvB,GAAG;QACJ,6BAA6B;KAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;IAEF,MAAM,IAAA,2BAAiB,EAAC,sBAAsB,CAAC,CAAC;AAClD,CAAC,CACF,CAAC","sourcesContent":["import { extendConfig, extendEnvironment, internalTask, task } from \"hardhat/config\";\nimport \"./tasks/coverage\";\nimport \"./tasks/deploy\";\nimport \"./tasks/generate-helpers\";\nimport \"./tasks/init\";\nimport \"./tasks/test\";\nimport \"./types/hardhat\";\n\nimport { existsSync, writeFileSync } from \"fs\";\nimport {\n    TASK_COMPILE_GET_REMAPPINGS,\n    TASK_COMPILE_TRANSFORM_IMPORT_NAME,\n} from \"hardhat/builtin-tasks/task-names\";\nimport { HardhatRuntimeEnvironment } from \"hardhat/types\";\nimport path from \"path\";\nimport picocolors from \"picocolors\";\nimport {\n    getForgeConfig,\n    getRemappings,\n    HardhatFoundryError,\n    installDependency,\n} from \"./foundry\";\nimport { validateConfig } from \"./utils/validation\";\n\n// Export types\nexport * from \"./types/config\";\n\nconst TASK_INIT_FOUNDRY = \"init-foundry\";\n\nlet pluginActivated = false;\n\n// Extend config with diamondsFoundry settings\nextendConfig((config, userConfig) => {\n  // Validate and set diamondsFoundry config\n  config.diamondsFoundry = validateConfig(userConfig.diamondsFoundry);\n\n  // Check foundry.toml presence. Don't warn when running foundry initialization task\n  if (!existsSync(path.join(config.paths.root, \"foundry.toml\"))) {\n    if (!process.argv.includes(TASK_INIT_FOUNDRY)) {\n      console.log(\n        picocolors.yellow(\n          `Warning: You are using the diamonds-hardhat-foundry plugin but there isn't a foundry.toml file in your project. Run 'npx hardhat ${TASK_INIT_FOUNDRY}' to create one.`\n        )\n      );\n    }\n    return;\n  }\n\n  // Load foundry config\n  const foundryConfig = getForgeConfig();\n\n  // Ensure required keys exist\n  if (\n    foundryConfig?.src === undefined ||\n    foundryConfig?.cache_path === undefined\n  ) {\n    throw new HardhatFoundryError(\n      \"Couldn't find `src` or `cache_path` config keys after running `forge config --json`\"\n    );\n  }\n\n  // Ensure foundry src path doesn't mismatch user-configured path\n  const userSourcesPath = userConfig.paths?.sources;\n  const foundrySourcesPath = foundryConfig.src;\n\n  if (\n    userSourcesPath !== undefined &&\n    path.resolve(userSourcesPath) !== path.resolve(foundrySourcesPath)\n  ) {\n    throw new HardhatFoundryError(\n      `User-configured sources path (${userSourcesPath}) doesn't match path configured in foundry (${foundrySourcesPath})`\n    );\n  }\n\n  // Set sources path\n  config.paths.sources = path.resolve(config.paths.root, foundrySourcesPath);\n\n  // Change hardhat's cache path if it clashes with foundry's\n  const foundryCachePath = path.resolve(\n    config.paths.root,\n    foundryConfig.cache_path\n  );\n  if (config.paths.cache === foundryCachePath) {\n    config.paths.cache = \"cache_hardhat\";\n  }\n\n  pluginActivated = true;\n});\n\n// Extend environment to add diamondsFoundry config to HRE\nextendEnvironment((hre: HardhatRuntimeEnvironment) => {\n  hre.diamondsFoundry = hre.config.diamondsFoundry;\n});\n\n// This task is in place to detect old hardhat-core versions\ninternalTask(TASK_COMPILE_TRANSFORM_IMPORT_NAME).setAction(\n  async (\n    {\n      importName,\n      deprecationCheck,\n    }: { importName: string; deprecationCheck: boolean },\n    _hre\n  ): Promise<string> => {\n    // When the deprecationCheck param is passed, it means a new enough hardhat-core is being used\n    if (deprecationCheck) {\n      return importName;\n    }\n    throw new HardhatFoundryError(\n      \"This version of diamonds-hardhat-foundry depends on hardhat version >= 2.17.2\"\n    );\n  }\n);\n\ninternalTask(TASK_COMPILE_GET_REMAPPINGS).setAction(\n  async (): Promise<Record<string, string>> => {\n    if (!pluginActivated) {\n      return {};\n    }\n\n    return getRemappings();\n  }\n);\n\ntask(\n  TASK_INIT_FOUNDRY,\n  \"Initialize foundry setup in current hardhat project\",\n  async (_, hre: HardhatRuntimeEnvironment) => {\n    const foundryConfigPath = path.resolve(\n      hre.config.paths.root,\n      \"foundry.toml\"\n    );\n\n    if (existsSync(foundryConfigPath)) {\n      console.warn(\n        picocolors.yellow(`File foundry.toml already exists. Aborting.`)\n      );\n      process.exit(1);\n    }\n\n    console.log(`Creating foundry.toml file...`);\n\n    writeFileSync(\n      foundryConfigPath,\n      [\n        `[profile.default]`,\n        `src = '${path.relative(\n          hre.config.paths.root,\n          hre.config.paths.sources\n        )}'`,\n        `out = 'out'`,\n        `libs = ['node_modules', 'lib']`,\n        `test = '${path.relative(\n          hre.config.paths.root,\n          hre.config.paths.tests\n        )}'`,\n        `cache_path  = 'cache_forge'`,\n      ].join(\"\\n\")\n    );\n\n    await installDependency(\"foundry-rs/forge-std\");\n  }\n);\n"]}

package/dist/tasks/coverage.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=coverage.d.ts.map

package/dist/tasks/coverage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"coverage.d.ts","sourceRoot":"","sources":["../../src/tasks/coverage.ts"],"names":[],"mappings":""}