@diamondslab/diamonds-hardhat-foundry 1.0.3 → 2.2.0

package/contracts/DiamondFuzzBase.sol

@@ -2,127 +2,335 @@
 pragma solidity ^0.8.0;
 
 import "forge-std/Test.sol";
-import "forge-std/console.sol";
-import "./DiamondForgeHelpers.sol";
-
-/**
- * @title DiamondFuzzBase
- * @notice Base contract for Diamond fuzz testing with Forge
- * @dev Provides setup methods and utilities for fuzz testing Diamond contracts
- */
+import "./DiamondABILoader.sol";
+
+/// @title DiamondFuzzBase
+/// @author DiamondsLab
+/// @notice Abstract base contract for Diamond fuzzing and integration tests
+/// @dev Inherit from this contract to create Diamond-based tests with built-in helpers for calling functions, measuring gas, and verifying routing
+/// @custom:security This contract is intended for testing purposes only
+/// @custom:usage
+/// ```solidity
+/// contract MyDiamondTest is DiamondFuzzBase {
+///     function setUp() public override {
+///         super.setUp();
+///         // Additional setup
+///     }
+/// }
+/// ```
 abstract contract DiamondFuzzBase is Test {
-    using DiamondForgeHelpers for address;
+    using DiamondABILoader for string;
 
-    // Diamond contract address (set by child contracts)
-    address public diamond;
+    /// @notice Deployed Diamond contract address
+    /// @dev Loaded in setUp(), can be overridden by implementing _loadDiamondAddress()
+    address internal diamond;
 
-    // Facet addresses (set by child contracts)
-    mapping(string => address) public facets;
+    /// @notice Diamond ABI JSON content
+    /// @dev Loaded from diamond-abi file in setUp()
+    string internal diamondABI;
 
-    // Test accounts
-    address public deployer;
-    address public user1;
-    address public user2;
-    address public user3;
+    /// @notice Array of all function selectors in the Diamond
+    bytes4[] internal diamondSelectors;
+
+    /// @notice Array of all function signatures in the Diamond
+    string[] internal diamondSignatures;
+
+    /// @notice Mapping of selector to function name for debugging
+    mapping(bytes4 => string) internal selectorToName;
+
+    /// @notice Error for failed Diamond calls
+    /// @param selector The function selector that was called
+    /// @param data The calldata that was sent
+    /// @param returnData The raw error data returned
+    error DiamondCallFailed(bytes4 selector, bytes data, bytes returnData);
+
+    /// @notice Error for unexpected Diamond success when expecting revert
+    /// @param selector The function selector that was called
+    /// @param data The calldata that was sent
+    /// @param returnData The return data from the unexpected success
+    error DiamondCallUnexpectedSuccess(bytes4 selector, bytes data, bytes returnData);
+
+    /// @notice Event emitted when Diamond is loaded in setUp
+    /// @param diamondAddress The address of the loaded Diamond
+    /// @param functionCount The number of functions found in the ABI
+    event DiamondLoaded(address indexed diamondAddress, uint256 functionCount);
+
+    /// @notice Event emitted when gas is measured for a function call
+    /// @param selector The function selector
+    /// @param functionName The function signature
+    /// @param gasUsed The gas consumed by the call
+    event GasMeasurement(bytes4 indexed selector, string functionName, uint256 gasUsed);
+
+    /// @notice Load deployed Diamond address
+    /// @dev Override this function to provide custom Diamond address loading logic
+    /// @dev Default implementation uses vm.envAddress("DIAMOND_ADDRESS") if available, otherwise uses a test deployment
+    /// @return diamondAddress The deployed Diamond contract address
+    function _loadDiamondAddress() internal view virtual returns (address diamondAddress) {
+        // Try to load from environment variable
+        try vm.envAddress("DIAMOND_ADDRESS") returns (address addr) {
+            return addr;
+        } catch {
+            // Fall back to expecting a deployment file or generated library
+            revert(
+                "DiamondFuzzBase: Override _loadDiamondAddress() or set DIAMOND_ADDRESS env var"
+            );
+        }
+    }
+
+    /// @notice Get the path to the Diamond ABI file
+    /// @dev Override this to use a different ABI file path
+    /// @return abiPath The path to the Diamond ABI JSON file
+    function _getDiamondABIPath() internal view virtual returns (string memory abiPath) {
+        return "./diamond-abi/ExampleDiamond.json";
+    }
+
+    /// @notice Load and parse Diamond ABI file
+    /// @dev Uses DiamondABILoader to extract selectors and signatures
+    /// @dev Can be overridden to customize ABI loading behavior
+    function _loadDiamondABI() internal virtual {
+        string memory abiPath = _getDiamondABIPath();
+
+        // Load ABI JSON
+        diamondABI = DiamondABILoader.loadDiamondABI(abiPath);
+
+        // Extract selectors and signatures
+        diamondSelectors = DiamondABILoader.extractSelectors(diamondABI);
+        diamondSignatures = DiamondABILoader.extractSignatures(diamondABI);
+
+        require(diamondSelectors.length > 0, "DiamondFuzzBase: No selectors found");
+        require(
+            diamondSelectors.length == diamondSignatures.length,
+            "DiamondFuzzBase: Selector/signature count mismatch"
+        );
+
+        // Build selector to name mapping for debugging
+        for (uint256 i = 0; i < diamondSelectors.length; i++) {
+            selectorToName[diamondSelectors[i]] = diamondSignatures[i];
+        }
+
+        emit DiamondLoaded(diamond, diamondSelectors.length);
+    }
+
+    /// @notice Call a Diamond function with selector and encoded arguments
+    /// @dev Low-level call helper for non-payable functions
+    /// @param selector The function selector (bytes4)
+    /// @param data The encoded function arguments (without selector)
+    /// @return success Whether the call succeeded
+    /// @return returnData The raw return data from the call
+    function _callDiamond(
+        bytes4 selector,
+        bytes memory data
+    ) internal virtual returns (bool success, bytes memory returnData) {
+        bytes memory callData = abi.encodePacked(selector, data);
+        (success, returnData) = diamond.call(callData);
+    }
+
+    /// @notice Call a Diamond function with value (for payable functions)
+    /// @dev Low-level call helper for payable functions
+    /// @param selector The function selector (bytes4)
+    /// @param data The encoded function arguments (without selector)
+    /// @param value The ETH value to send with the call
+    /// @return success Whether the call succeeded
+    /// @return returnData The raw return data from the call
+    function _callDiamondWithValue(
+        bytes4 selector,
+        bytes memory data,
+        uint256 value
+    ) internal virtual returns (bool success, bytes memory returnData) {
+        bytes memory callData = abi.encodePacked(selector, data);
+        (success, returnData) = diamond.call{value: value}(callData);
+    }
+
+    /// @notice Expect a Diamond call to revert with specific error
+    /// @dev Helper for testing expected reverts
+    /// @param selector The function selector to call
+    /// @param data The encoded function arguments
+    /// @param expectedError The expected error data (can be empty to expect any revert)
+    function _expectDiamondRevert(
+        bytes4 selector,
+        bytes memory data,
+        bytes memory expectedError
+    ) internal virtual {
+        (bool success, bytes memory returnData) = _callDiamond(selector, data);
+
+        if (success) {
+            revert DiamondCallUnexpectedSuccess(selector, data, returnData);
+        }
+
+        if (expectedError.length > 0) {
+            // Verify exact error match
+            assertEq(returnData, expectedError, "Unexpected error data");
+        }
+        // If expectedError is empty, just verify it reverted (which we already did)
+    }
+
+    /// @notice Verify function selector routes to expected facet address
+    /// @dev Uses DiamondLoupe to check facet routing
+    /// @param selector The function selector to check
+    /// @param expectedFacet The expected facet address (address(0) to just verify it exists)
+    /// @return facetAddress The actual facet address for this selector
+    function _verifyFacetRouting(
+        bytes4 selector,
+        address expectedFacet
+    ) internal view virtual returns (address facetAddress) {
+        // Call facetAddress(bytes4) from DiamondLoupe
+        bytes memory callData = abi.encodeWithSignature("facetAddress(bytes4)", selector);
+        (bool success, bytes memory returnData) = diamond.staticcall(callData);
+
+        require(success, "DiamondFuzzBase: facetAddress call failed");
+        facetAddress = abi.decode(returnData, (address));
+
+        if (expectedFacet != address(0)) {
+            assertEq(facetAddress, expectedFacet, "Selector routes to unexpected facet");
+        } else {
+            assertTrue(facetAddress != address(0), "Selector has no facet");
+        }
 
-    /**
-     * @notice Setup function called before each test
-     * @dev Override this in child contracts to set up Diamond and facets
-     */
-    function setUp() public virtual {
-        // Set up test accounts
-        deployer = address(this);
-        user1 = makeAddr("user1");
-        user2 = makeAddr("user2");
-        user3 = makeAddr("user3");
-
-        // Fund test accounts
-        vm.deal(user1, 100 ether);
-        vm.deal(user2, 100 ether);
-        vm.deal(user3, 100 ether);
-
-        // Child contracts should override and call setDiamondAddress()
-    }
-
-    /**
-     * @notice Set the Diamond contract address
-     * @param _diamond Address of the deployed Diamond
-     */
-    function setDiamondAddress(address _diamond) internal {
-        require(_diamond != address(0), "Diamond address cannot be zero");
-        diamond = _diamond;
-        DiamondForgeHelpers.assertValidDiamond(_diamond);
-    }
-
-    /**
-     * @notice Register a facet address
-     * @param name Name of the facet
-     * @param facetAddress Address of the facet
-     */
-    function registerFacet(string memory name, address facetAddress) internal {
-        require(facetAddress != address(0), "Facet address cannot be zero");
-        facets[name] = facetAddress;
-        DiamondForgeHelpers.assertValidFacet(facetAddress, name);
-    }
-
-    /**
-     * @notice Get a facet address by name
-     * @param name Name of the facet
-     * @return Address of the facet
-     */
-    function getFacet(string memory name) internal view returns (address) {
-        address facetAddress = facets[name];
-        require(facetAddress != address(0), string(abi.encodePacked("Facet not found: ", name)));
         return facetAddress;
     }
 
-    /**
-     * @notice Assume valid Ethereum address for fuzzing
-     * @param addr Address to validate
-     */
-    function assumeValidAddress(address addr) internal pure {
-        vm.assume(addr != address(0));
-        vm.assume(addr != address(0xdead));
-        vm.assume(uint160(addr) > 0xFF); // Avoid precompiles
-    }
-
-    /**
-     * @notice Assume valid amount for fuzzing (not zero, not unreasonably large)
-     * @param amount Amount to validate
-     */
-    function assumeValidAmount(uint256 amount) internal pure {
-        vm.assume(amount > 0);
-        vm.assume(amount < type(uint128).max); // Reasonable upper bound
-    }
-
-    /**
-     * @notice Bound a fuzzed value to a specific range
-     * @param value The fuzzed value
-     * @param min Minimum value (inclusive)
-     * @param max Maximum value (inclusive)
-     * @return Bounded value
-     */
-    function boundValue(uint256 value, uint256 min, uint256 max) internal pure returns (uint256) {
-        return bound(value, min, max);
-    }
-
-    /**
-     * @notice Log test context for debugging
-     * @param testName Name of the test
-     */
-    function logTestContext(string memory testName) internal view {
-        console.log("=== Test:", testName, "===");
-        console.log("Diamond:", diamond);
-        console.log("Deployer:", deployer);
-        console.log("Block number:", block.number);
-        console.log("Block timestamp:", block.timestamp);
-    }
-
-    /**
-     * @notice Expect a revert with specific error message
-     * @param errorMessage Expected error message
-     */
-    function expectRevertWithMessage(string memory errorMessage) internal {
-        vm.expectRevert(bytes(errorMessage));
+    /// @notice Measure gas consumption of a Diamond function call
+    /// @dev Gas profiling helper for optimization testing
+    /// @param selector The function selector to call
+    /// @param data The encoded function arguments
+    /// @return gasUsed The gas consumed by the call
+    function _measureDiamondGas(
+        bytes4 selector,
+        bytes memory data
+    ) internal virtual returns (uint256 gasUsed) {
+        uint256 gasBefore = gasleft();
+        (bool success, bytes memory returnData) = _callDiamond(selector, data);
+        uint256 gasAfter = gasleft();
+
+        if (!success) {
+            revert DiamondCallFailed(selector, data, returnData);
+        }
+
+        gasUsed = gasBefore - gasAfter;
+        emit GasMeasurement(selector, selectorToName[selector], gasUsed);
+
+        return gasUsed;
+    }
+
+    /// @notice Get the Diamond owner address
+    /// @dev Helper to get owner for ownership checks (assumes owner() function exists)
+    /// @return owner The current Diamond owner address
+    function _getDiamondOwner() internal view virtual returns (address owner) {
+        // Call owner() function - standard in most ownership facets
+        bytes memory callData = abi.encodeWithSignature("owner()");
+        (bool success, bytes memory returnData) = diamond.staticcall(callData);
+
+        require(success, "DiamondFuzzBase: owner() call failed");
+        owner = abi.decode(returnData, (address));
+
+        return owner;
+    }
+
+    /// @notice Check if an address has a specific role
+    /// @dev Helper for access control testing (assumes role-based access control)
+    /// @param role The role identifier (bytes32)
+    /// @param account The address to check
+    /// @return hasRole True if the account has the role
+    function _hasRole(bytes32 role, address account) internal view virtual returns (bool hasRole) {
+        bytes memory callData = abi.encodeWithSignature("hasRole(bytes32,address)", role, account);
+        (bool success, bytes memory returnData) = diamond.staticcall(callData);
+
+        if (!success) {
+            return false;
+        }
+
+        hasRole = abi.decode(returnData, (bool));
+        return hasRole;
+    }
+
+    /// @notice Grant a role to an address (requires appropriate permissions)
+    /// @dev Helper for access control testing
+    /// @dev Note: The caller must have permission to grant the role (e.g., have DEFAULT_ADMIN_ROLE)
+    /// @dev For tests, use vm.prank() to call from an address with the appropriate permissions
+    /// @param role The role identifier
+    /// @param account The address to grant the role to
+    function _grantRole(bytes32 role, address account) internal virtual {
+        bytes4 selector = bytes4(keccak256("grantRole(bytes32,address)"));
+        bytes memory data = abi.encode(role, account);
+        (bool success, bytes memory returnData) = _callDiamond(selector, data);
+
+        if (!success) {
+            revert DiamondCallFailed(selector, data, returnData);
+        }
+    }
+
+    /// @notice Grant a role to the test contract itself
+    /// @dev Convenience helper that grants a role to address(this)
+    /// @dev The caller must have permission to grant the role - use vm.prank() as needed
+    /// @dev Common pattern in setUp(): vm.prank(owner); _grantRoleToSelf(DEFAULT_ADMIN_ROLE);
+    /// @param role The role identifier to grant to the test contract
+    function _grantRoleToSelf(bytes32 role) internal virtual {
+        _grantRole(role, address(this));
+    }
+
+    /// @notice Revoke a role from an address (requires appropriate permissions)
+    /// @dev Helper for access control testing
+    /// @param role The role identifier
+    /// @param account The address to revoke the role from
+    function _revokeRole(bytes32 role, address account) internal virtual {
+        bytes4 selector = bytes4(keccak256("revokeRole(bytes32,address)"));
+        bytes memory data = abi.encode(role, account);
+        (bool success, bytes memory returnData) = _callDiamond(selector, data);
+
+        if (!success) {
+            revert DiamondCallFailed(selector, data, returnData);
+        }
+    }
+
+    /// @notice Get all function selectors in the Diamond
+    /// @dev Helper to access loaded selectors
+    /// @return selectors Array of all function selectors
+    function _getDiamondSelectors() internal view virtual returns (bytes4[] memory selectors) {
+        return diamondSelectors;
+    }
+
+    /// @notice Get all function signatures in the Diamond
+    /// @dev Helper to access loaded signatures
+    /// @return signatures Array of all function signatures
+    function _getDiamondSignatures() internal view virtual returns (string[] memory signatures) {
+        return diamondSignatures;
+    }
+
+    /// @notice Get function name/signature for a selector
+    /// @dev Helper for debugging and logging
+    /// @param selector The function selector
+    /// @return name The function signature/name
+    function _getFunctionName(bytes4 selector) internal view virtual returns (string memory name) {
+        return selectorToName[selector];
+    }
+
+    /// @notice Setup function that loads Diamond address and ABI
+    /// @dev Override this to customize setup behavior, but call super.setUp() to load Diamond
+    /// @dev For access control tests, grant necessary roles in setUp() after calling super.setUp()
+    /// @custom:example
+    /// ```solidity
+    /// function setUp() public override {
+    ///     super.setUp(); // Load Diamond and ABI
+    ///
+    ///     // Grant DEFAULT_ADMIN_ROLE to test contract for access control tests
+    ///     vm.prank(_getDiamondOwner());
+    ///     _grantRoleToSelf(DEFAULT_ADMIN_ROLE);
+    ///
+    ///     // Additional custom setup
+    /// }
+    /// ```
+    function setUp() public virtual {
+        // Load Diamond address
+        diamond = _loadDiamondAddress();
+
+        // Verify Diamond has code deployed
+        require(diamond.code.length > 0, "DiamondFuzzBase: Diamond has no code");
+
+        // Load and parse Diamond ABI
+        _loadDiamondABI();
+
+        // Log for debugging
+        console.log("Diamond loaded at:", diamond);
+        console.log("Functions found:", diamondSelectors.length);
     }
 }

package/dist/foundry.d.ts

@@ -7,5 +7,33 @@ export declare function getForgeConfig(): any;
 export declare function parseRemappings(remappingsTxt: string): Remappings;
 export declare function getRemappings(): Promise<Remappings>;
 export declare function installDependency(dependency: string): Promise<void>;
+/**
+ * Check if Foundry is installed
+ */
+export declare function isFoundryInstalled(): boolean;
+/**
+ * Compile Forge contracts
+ */
+export declare function compileForge(options: {
+    cwd?: string;
+    verbose?: boolean;
+}): Promise<{
+    success: boolean;
+    output?: string;
+}>;
+/**
+ * Run Forge tests with optional forking
+ */
+export declare function runForgeTest(options: {
+    matchTest?: string;
+    matchContract?: string;
+    verbosity?: number;
+    gasReport?: boolean;
+    forkUrl?: string;
+    cwd?: string;
+}): Promise<{
+    success: boolean;
+    output?: string;
+}>;
 export {};
 //# sourceMappingURL=foundry.d.ts.map

package/dist/foundry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"foundry.d.ts","sourceRoot":"","sources":["../src/foundry.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,2BAA2B,EAAE,MAAM,8BAA8B,CAAC;AAM3E,KAAK,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AAIzC,qBAAa,mBAAoB,SAAQ,2BAA2B;gBACtD,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK;CAG5C;AAcD,wBAAgB,cAAc,QAE7B;AAED,wBAAgB,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,UAAU,CA+BjE;AAED,wBAAsB,aAAa,wBAOlC;AAED,wBAAsB,iBAAiB,CAAC,UAAU,EAAE,MAAM,iBAgBzD"}
+{"version":3,"file":"foundry.d.ts","sourceRoot":"","sources":["../src/foundry.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,2BAA2B,EAAE,MAAM,8BAA8B,CAAC;AAM3E,KAAK,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AAIzC,qBAAa,mBAAoB,SAAQ,2BAA2B;gBACtD,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK;CAG5C;AAcD,wBAAgB,cAAc,QAE7B;AAED,wBAAgB,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,UAAU,CA+BjE;AAED,wBAAsB,aAAa,wBAOlC;AAED,wBAAsB,iBAAiB,CAAC,UAAU,EAAE,MAAM,iBAgBzD;AA4CD;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,OAAO,CAO5C;AAED;;GAEG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE;IAC1C,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAejD;AAED;;GAEG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE;IAC1C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAyDjD"}

package/dist/foundry.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.installDependency = exports.getRemappings = exports.parseRemappings = exports.getForgeConfig = exports.HardhatFoundryError = void 0;
+exports.runForgeTest = exports.compileForge = exports.isFoundryInstalled = exports.installDependency = exports.getRemappings = exports.parseRemappings = exports.getForgeConfig = exports.HardhatFoundryError = void 0;
 const child_process_1 = require("child_process");
 const errors_1 = require("hardhat/internal/core/errors");
 const picocolors_1 = __importDefault(require("picocolors"));
@@ -101,4 +101,85 @@ function buildForgeExecutionError(exitCode, message) {
             return new HardhatFoundryError(`Unexpected error while running \`forge\`: ${message}`);
     }
 }
+/**
+ * Check if Foundry is installed
+ */
+function isFoundryInstalled() {
+    try {
+        (0, child_process_1.execSync)("forge --version", { stdio: "pipe" });
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+exports.isFoundryInstalled = isFoundryInstalled;
+/**
+ * Compile Forge contracts
+ */
+async function compileForge(options) {
+    const { cwd = process.cwd(), verbose = false } = options;
+    try {
+        const args = ["build"];
+        if (!verbose) {
+            args.push("--quiet");
+        }
+        const { stdout, stderr } = await exec(`forge ${args.join(" ")}`, { cwd });
+        return { success: true, output: stdout || stderr };
+    }
+    catch (error) {
+        console.error("Forge build failed:", error.message);
+        return { success: false, output: error.message };
+    }
+}
+exports.compileForge = compileForge;
+/**
+ * Run Forge tests with optional forking
+ */
+async function runForgeTest(options) {
+    const { matchTest, matchContract, verbosity = 2, gasReport = false, forkUrl, cwd = process.cwd(), } = options;
+    try {
+        const args = ["test"];
+        // Add verbosity
+        if (verbosity > 0 && verbosity <= 5) {
+            args.push(`-${"v".repeat(verbosity)}`);
+        }
+        // Add test filters
+        if (matchTest) {
+            args.push("--match-test", matchTest);
+        }
+        if (matchContract) {
+            args.push("--match-contract", matchContract);
+        }
+        // Add gas reporting
+        if (gasReport) {
+            args.push("--gas-report");
+        }
+        // CRITICAL: Add fork URL to connect to deployed Diamond
+        // This allows Forge tests to access the Hardhat-deployed Diamond contract
+        if (forkUrl) {
+            args.push("--fork-url", forkUrl);
+        }
+        const cmd = `forge ${args.join(" ")}`;
+        console.log(`\nRunning: ${picocolors_1.default.blue(cmd)}\n`);
+        const { stdout, stderr } = await exec(cmd, { cwd, maxBuffer: 10 * 1024 * 1024 });
+        console.log(stdout);
+        if (stderr)
+            console.error(stderr);
+        return { success: true, output: stdout };
+    }
+    catch (error) {
+        // Forge returns non-zero exit code when tests fail
+        // We still want to show the output
+        if (error.stdout)
+            console.log(error.stdout);
+        if (error.stderr)
+            console.error(error.stderr);
+        return {
+            success: false,
+            output: error.stdout || error.stderr || error.message,
+        };
+    }
+}
+exports.runForgeTest = runForgeTest;
 //# sourceMappingURL=foundry.js.map

package/dist/foundry.js.map

@@ -1 +1 @@
-{"version":3,"file":"foundry.js","sourceRoot":"","sources":["../src/foundry.ts"],"names":[],"mappings":";;;;;;AAAA,iDAA+D;AAC/D,yDAA2E;AAC3E,4DAAoC;AACpC,+BAAiC;AAEjC,MAAM,IAAI,GAAG,IAAA,gBAAS,EAAC,oBAAY,CAAC,CAAC;AAIrC,IAAI,gBAAiD,CAAC;AAEtD,MAAa,mBAAoB,SAAQ,oCAA2B;IAClE,YAAY,OAAe,EAAE,MAAc;QACzC,KAAK,CAAC,0BAA0B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACrD,CAAC;CACF;AAJD,kDAIC;AAED,MAAM,iBAAkB,SAAQ,mBAAmB;IACjD,YAAY,UAAkB,EAAE,MAAa;QAC3C,KAAK,CACH,qBAAqB,UAAU;;EAEnC,MAAM,CAAC,OAAO;CACf,EACK,MAAM,CACP,CAAC;IACJ,CAAC;CACF;AAED,SAAgB,cAAc;IAC5B,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC,CAAC;AACvD,CAAC;AAFD,wCAEC;AAED,SAAgB,eAAe,CAAC,aAAqB;IACnD,MAAM,UAAU,GAAe,EAAE,CAAC;IAClC,MAAM,cAAc,GAAG,aAAa,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACzD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;QAC1C,IAAI,aAAa,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC/B,SAAS;SACV;QAED,IAAI,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;YAC/B,MAAM,IAAI,mBAAmB,CAC3B,sBAAsB,aAAa,uCAAuC,CAC3E,CAAC;SACH;QAED,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,mBAAmB,CAC3B,sBAAsB,aAAa,gDAAgD,CACpF,CAAC;SACH;QAED,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAExC,2EAA2E;QAC3E,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE;YACvC,SAAS;SACV;QAED,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;KACnC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AA/BD,0CA+BC;AAEM,KAAK,UAAU,aAAa;IACjC,2BAA2B;IAC3B,IAAI,gBAAgB,KAAK,SAAS,EAAE;QAClC,gBAAgB,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;KACrE;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAPD,sCAOC;AAEM,KAAK,UAAU,iBAAiB,CAAC,UAAkB;IACxD,6EAA6E;IAC7E,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;IACtD,MAAM,eAAe,GAAG,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAEzD,MAAM,GAAG,GAAG,iBACV,eAAe,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,EACpC,IAAI,UAAU,EAAE,CAAC;IAEjB,OAAO,CAAC,GAAG,CAAC,YAAY,oBAAU,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEjD,IAAI;QACF,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;KACjB;IAAC,OAAO,KAAU,EAAE;QACnB,MAAM,IAAI,iBAAiB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;KAChD;AACH,CAAC;AAhBD,8CAgBC;AAED,SAAS,UAAU,CAAC,GAAW;IAC7B,IAAI;QACF,OAAO,IAAA,wBAAQ,EAAC,GAAG,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;KACpD;IAAC,OAAO,KAAU,EAAE;QACnB,MAAM,WAAW,GAAG,wBAAwB,CAC1C,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CACxB,CAAC;QAEF,MAAM,WAAW,CAAC;KACnB;AACH,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,GAAW;IAC/B,IAAI;QACF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;QACnC,OAAO,MAAM,CAAC;KACf;IAAC,OAAO,KAAU,EAAE;QACnB,MAAM,wBAAwB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;KAC3D;AACH,CAAC;AAED,SAAS,wBAAwB,CAC/B,QAA4B,EAC5B,OAAe;IAEf,QAAQ,QAAQ,EAAE;QAChB,KAAK,GAAG;YACN,OAAO,IAAI,mBAAmB,CAC5B,+EAA+E,CAChF,CAAC;QACJ,KAAK,GAAG;YACN,OAAO,IAAI,mBAAmB,CAC5B,8EAA8E,CAC/E,CAAC;QACJ;YACE,OAAO,IAAI,mBAAmB,CAC5B,6CAA6C,OAAO,EAAE,CACvD,CAAC;KACL;AACH,CAAC","sourcesContent":["import { exec as execCallback, execSync } from \"child_process\";\nimport { NomicLabsHardhatPluginError } from \"hardhat/internal/core/errors\";\nimport picocolors from \"picocolors\";\nimport { promisify } from \"util\";\n\nconst exec = promisify(execCallback);\n\ntype Remappings = Record<string, string>;\n\nlet cachedRemappings: Promise<Remappings> | undefined;\n\nexport class HardhatFoundryError extends NomicLabsHardhatPluginError {\n  constructor(message: string, parent?: Error) {\n    super(\"diamonds-hardhat-foundry\", message, parent);\n  }\n}\n\nclass ForgeInstallError extends HardhatFoundryError {\n  constructor(dependency: string, parent: Error) {\n    super(\n      `Couldn't install '${dependency}', please install it manually.\n\n${parent.message}\n`,\n      parent\n    );\n  }\n}\n\nexport function getForgeConfig() {\n  return JSON.parse(runCmdSync(\"forge config --json\"));\n}\n\nexport function parseRemappings(remappingsTxt: string): Remappings {\n  const remappings: Remappings = {};\n  const remappingLines = remappingsTxt.split(/\\r\\n|\\r|\\n/);\n  for (const remappingLine of remappingLines) {\n    if (remappingLine.trim() === \"\") {\n      continue;\n    }\n\n    if (remappingLine.includes(\":\")) {\n      throw new HardhatFoundryError(\n        `Invalid remapping '${remappingLine}', remapping contexts are not allowed`\n      );\n    }\n\n    if (!remappingLine.includes(\"=\")) {\n      throw new HardhatFoundryError(\n        `Invalid remapping '${remappingLine}', remappings without a target are not allowed`\n      );\n    }\n\n    const fromTo = remappingLine.split(\"=\");\n\n    // if the remapping already exists, we ignore it because the first one wins\n    if (remappings[fromTo[0]] !== undefined) {\n      continue;\n    }\n\n    remappings[fromTo[0]] = fromTo[1];\n  }\n\n  return remappings;\n}\n\nexport async function getRemappings() {\n  // Get remappings only once\n  if (cachedRemappings === undefined) {\n    cachedRemappings = runCmd(\"forge remappings\").then(parseRemappings);\n  }\n\n  return cachedRemappings;\n}\n\nexport async function installDependency(dependency: string) {\n  // Check if --no-commit flag is supported. Best way is checking the help text\n  const helpText = await runCmd(\"forge install --help\");\n  const useNoCommitFlag = helpText.includes(\"--no-commit\");\n\n  const cmd = `forge install ${\n    useNoCommitFlag ? \"--no-commit\" : \"\"\n  } ${dependency}`;\n\n  console.log(`Running '${picocolors.blue(cmd)}'`);\n\n  try {\n    await exec(cmd);\n  } catch (error: any) {\n    throw new ForgeInstallError(dependency, error);\n  }\n}\n\nfunction runCmdSync(cmd: string): string {\n  try {\n    return execSync(cmd, { stdio: \"pipe\" }).toString();\n  } catch (error: any) {\n    const pluginError = buildForgeExecutionError(\n      error.status,\n      error.stderr.toString()\n    );\n\n    throw pluginError;\n  }\n}\n\nasync function runCmd(cmd: string): Promise<string> {\n  try {\n    const { stdout } = await exec(cmd);\n    return stdout;\n  } catch (error: any) {\n    throw buildForgeExecutionError(error.code, error.message);\n  }\n}\n\nfunction buildForgeExecutionError(\n  exitCode: number | undefined,\n  message: string\n) {\n  switch (exitCode) {\n    case 127:\n      return new HardhatFoundryError(\n        \"Couldn't run `forge`. Please check that your foundry installation is correct.\"\n      );\n    case 134:\n      return new HardhatFoundryError(\n        \"Running `forge` failed. Please check that your foundry.toml file is correct.\"\n      );\n    default:\n      return new HardhatFoundryError(\n        `Unexpected error while running \\`forge\\`: ${message}`\n      );\n  }\n}\n"]}
+{"version":3,"file":"foundry.js","sourceRoot":"","sources":["../src/foundry.ts"],"names":[],"mappings":";;;;;;AAAA,iDAA+D;AAC/D,yDAA2E;AAC3E,4DAAoC;AACpC,+BAAiC;AAEjC,MAAM,IAAI,GAAG,IAAA,gBAAS,EAAC,oBAAY,CAAC,CAAC;AAIrC,IAAI,gBAAiD,CAAC;AAEtD,MAAa,mBAAoB,SAAQ,oCAA2B;IAClE,YAAY,OAAe,EAAE,MAAc;QACzC,KAAK,CAAC,0BAA0B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACrD,CAAC;CACF;AAJD,kDAIC;AAED,MAAM,iBAAkB,SAAQ,mBAAmB;IACjD,YAAY,UAAkB,EAAE,MAAa;QAC3C,KAAK,CACH,qBAAqB,UAAU;;EAEnC,MAAM,CAAC,OAAO;CACf,EACK,MAAM,CACP,CAAC;IACJ,CAAC;CACF;AAED,SAAgB,cAAc;IAC5B,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC,CAAC;AACvD,CAAC;AAFD,wCAEC;AAED,SAAgB,eAAe,CAAC,aAAqB;IACnD,MAAM,UAAU,GAAe,EAAE,CAAC;IAClC,MAAM,cAAc,GAAG,aAAa,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACzD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;QAC1C,IAAI,aAAa,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC/B,SAAS;SACV;QAED,IAAI,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;YAC/B,MAAM,IAAI,mBAAmB,CAC3B,sBAAsB,aAAa,uCAAuC,CAC3E,CAAC;SACH;QAED,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;YAChC,MAAM,IAAI,mBAAmB,CAC3B,sBAAsB,aAAa,gDAAgD,CACpF,CAAC;SACH;QAED,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAExC,2EAA2E;QAC3E,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE;YACvC,SAAS;SACV;QAED,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;KACnC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AA/BD,0CA+BC;AAEM,KAAK,UAAU,aAAa;IACjC,2BAA2B;IAC3B,IAAI,gBAAgB,KAAK,SAAS,EAAE;QAClC,gBAAgB,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;KACrE;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAPD,sCAOC;AAEM,KAAK,UAAU,iBAAiB,CAAC,UAAkB;IACxD,6EAA6E;IAC7E,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;IACtD,MAAM,eAAe,GAAG,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAEzD,MAAM,GAAG,GAAG,iBACV,eAAe,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,EACpC,IAAI,UAAU,EAAE,CAAC;IAEjB,OAAO,CAAC,GAAG,CAAC,YAAY,oBAAU,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEjD,IAAI;QACF,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;KACjB;IAAC,OAAO,KAAU,EAAE;QACnB,MAAM,IAAI,iBAAiB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;KAChD;AACH,CAAC;AAhBD,8CAgBC;AAED,SAAS,UAAU,CAAC,GAAW;IAC7B,IAAI;QACF,OAAO,IAAA,wBAAQ,EAAC,GAAG,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;KACpD;IAAC,OAAO,KAAU,EAAE;QACnB,MAAM,WAAW,GAAG,wBAAwB,CAC1C,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CACxB,CAAC;QAEF,MAAM,WAAW,CAAC;KACnB;AACH,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,GAAW;IAC/B,IAAI;QACF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;QACnC,OAAO,MAAM,CAAC;KACf;IAAC,OAAO,KAAU,EAAE;QACnB,MAAM,wBAAwB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;KAC3D;AACH,CAAC;AAED,SAAS,wBAAwB,CAC/B,QAA4B,EAC5B,OAAe;IAEf,QAAQ,QAAQ,EAAE;QAChB,KAAK,GAAG;YACN,OAAO,IAAI,mBAAmB,CAC5B,+EAA+E,CAChF,CAAC;QACJ,KAAK,GAAG;YACN,OAAO,IAAI,mBAAmB,CAC5B,8EAA8E,CAC/E,CAAC;QACJ;YACE,OAAO,IAAI,mBAAmB,CAC5B,6CAA6C,OAAO,EAAE,CACvD,CAAC;KACL;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,kBAAkB;IAChC,IAAI;QACF,IAAA,wBAAQ,EAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC;KACb;IAAC,MAAM;QACN,OAAO,KAAK,CAAC;KACd;AACH,CAAC;AAPD,gDAOC;AAED;;GAEG;AACI,KAAK,UAAU,YAAY,CAAC,OAGlC;IACC,MAAM,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,EAAE,OAAO,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;IAEzD,IAAI;QACF,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,IAAI,CAAC,OAAO,EAAE;YACZ,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;SACtB;QAED,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,IAAI,MAAM,EAAE,CAAC;KACpD;IAAC,OAAO,KAAU,EAAE;QACnB,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACpD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;KAClD;AACH,CAAC;AAlBD,oCAkBC;AAED;;GAEG;AACI,KAAK,UAAU,YAAY,CAAC,OAOlC;IACC,MAAM,EACJ,SAAS,EACT,aAAa,EACb,SAAS,GAAG,CAAC,EACb,SAAS,GAAG,KAAK,EACjB,OAAO,EACP,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,GACpB,GAAG,OAAO,CAAC;IAEZ,IAAI;QACF,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QAEtB,gBAAgB;QAChB,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,IAAI,CAAC,EAAE;YACnC,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;SACxC;QAED,mBAAmB;QACnB,IAAI,SAAS,EAAE;YACb,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;SACtC;QAED,IAAI,aAAa,EAAE;YACjB,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;SAC9C;QAED,oBAAoB;QACpB,IAAI,SAAS,EAAE;YACb,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;SAC3B;QAED,wDAAwD;QACxD,0EAA0E;QAC1E,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;SAClC;QAED,MAAM,GAAG,GAAG,SAAS,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,cAAc,oBAAU,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEpD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC,CAAC;QACjF,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpB,IAAI,MAAM;YAAE,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAElC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;KAC1C;IAAC,OAAO,KAAU,EAAE;QACnB,mDAAmD;QACnD,mCAAmC;QACnC,IAAI,KAAK,CAAC,MAAM;YAAE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5C,IAAI,KAAK,CAAC,MAAM;YAAE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE9C,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO;SACtD,CAAC;KACH;AACH,CAAC;AAhED,oCAgEC","sourcesContent":["import { exec as execCallback, execSync } from \"child_process\";\nimport { NomicLabsHardhatPluginError } from \"hardhat/internal/core/errors\";\nimport picocolors from \"picocolors\";\nimport { promisify } from \"util\";\n\nconst exec = promisify(execCallback);\n\ntype Remappings = Record<string, string>;\n\nlet cachedRemappings: Promise<Remappings> | undefined;\n\nexport class HardhatFoundryError extends NomicLabsHardhatPluginError {\n  constructor(message: string, parent?: Error) {\n    super(\"diamonds-hardhat-foundry\", message, parent);\n  }\n}\n\nclass ForgeInstallError extends HardhatFoundryError {\n  constructor(dependency: string, parent: Error) {\n    super(\n      `Couldn't install '${dependency}', please install it manually.\n\n${parent.message}\n`,\n      parent\n    );\n  }\n}\n\nexport function getForgeConfig() {\n  return JSON.parse(runCmdSync(\"forge config --json\"));\n}\n\nexport function parseRemappings(remappingsTxt: string): Remappings {\n  const remappings: Remappings = {};\n  const remappingLines = remappingsTxt.split(/\\r\\n|\\r|\\n/);\n  for (const remappingLine of remappingLines) {\n    if (remappingLine.trim() === \"\") {\n      continue;\n    }\n\n    if (remappingLine.includes(\":\")) {\n      throw new HardhatFoundryError(\n        `Invalid remapping '${remappingLine}', remapping contexts are not allowed`\n      );\n    }\n\n    if (!remappingLine.includes(\"=\")) {\n      throw new HardhatFoundryError(\n        `Invalid remapping '${remappingLine}', remappings without a target are not allowed`\n      );\n    }\n\n    const fromTo = remappingLine.split(\"=\");\n\n    // if the remapping already exists, we ignore it because the first one wins\n    if (remappings[fromTo[0]] !== undefined) {\n      continue;\n    }\n\n    remappings[fromTo[0]] = fromTo[1];\n  }\n\n  return remappings;\n}\n\nexport async function getRemappings() {\n  // Get remappings only once\n  if (cachedRemappings === undefined) {\n    cachedRemappings = runCmd(\"forge remappings\").then(parseRemappings);\n  }\n\n  return cachedRemappings;\n}\n\nexport async function installDependency(dependency: string) {\n  // Check if --no-commit flag is supported. Best way is checking the help text\n  const helpText = await runCmd(\"forge install --help\");\n  const useNoCommitFlag = helpText.includes(\"--no-commit\");\n\n  const cmd = `forge install ${\n    useNoCommitFlag ? \"--no-commit\" : \"\"\n  } ${dependency}`;\n\n  console.log(`Running '${picocolors.blue(cmd)}'`);\n\n  try {\n    await exec(cmd);\n  } catch (error: any) {\n    throw new ForgeInstallError(dependency, error);\n  }\n}\n\nfunction runCmdSync(cmd: string): string {\n  try {\n    return execSync(cmd, { stdio: \"pipe\" }).toString();\n  } catch (error: any) {\n    const pluginError = buildForgeExecutionError(\n      error.status,\n      error.stderr.toString()\n    );\n\n    throw pluginError;\n  }\n}\n\nasync function runCmd(cmd: string): Promise<string> {\n  try {\n    const { stdout } = await exec(cmd);\n    return stdout;\n  } catch (error: any) {\n    throw buildForgeExecutionError(error.code, error.message);\n  }\n}\n\nfunction buildForgeExecutionError(\n  exitCode: number | undefined,\n  message: string\n) {\n  switch (exitCode) {\n    case 127:\n      return new HardhatFoundryError(\n        \"Couldn't run `forge`. Please check that your foundry installation is correct.\"\n      );\n    case 134:\n      return new HardhatFoundryError(\n        \"Running `forge` failed. Please check that your foundry.toml file is correct.\"\n      );\n    default:\n      return new HardhatFoundryError(\n        `Unexpected error while running \\`forge\\`: ${message}`\n      );\n  }\n}\n\n/**\n * Check if Foundry is installed\n */\nexport function isFoundryInstalled(): boolean {\n  try {\n    execSync(\"forge --version\", { stdio: \"pipe\" });\n    return true;\n  } catch {\n    return false;\n  }\n}\n\n/**\n * Compile Forge contracts\n */\nexport async function compileForge(options: {\n  cwd?: string;\n  verbose?: boolean;\n}): Promise<{ success: boolean; output?: string }> {\n  const { cwd = process.cwd(), verbose = false } = options;\n  \n  try {\n    const args = [\"build\"];\n    if (!verbose) {\n      args.push(\"--quiet\");\n    }\n    \n    const { stdout, stderr } = await exec(`forge ${args.join(\" \")}`, { cwd });\n    return { success: true, output: stdout || stderr };\n  } catch (error: any) {\n    console.error(\"Forge build failed:\", error.message);\n    return { success: false, output: error.message };\n  }\n}\n\n/**\n * Run Forge tests with optional forking\n */\nexport async function runForgeTest(options: {\n  matchTest?: string;\n  matchContract?: string;\n  verbosity?: number;\n  gasReport?: boolean;\n  forkUrl?: string;\n  cwd?: string;\n}): Promise<{ success: boolean; output?: string }> {\n  const {\n    matchTest,\n    matchContract,\n    verbosity = 2,\n    gasReport = false,\n    forkUrl,\n    cwd = process.cwd(),\n  } = options;\n\n  try {\n    const args = [\"test\"];\n\n    // Add verbosity\n    if (verbosity > 0 && verbosity <= 5) {\n      args.push(`-${\"v\".repeat(verbosity)}`);\n    }\n\n    // Add test filters\n    if (matchTest) {\n      args.push(\"--match-test\", matchTest);\n    }\n\n    if (matchContract) {\n      args.push(\"--match-contract\", matchContract);\n    }\n\n    // Add gas reporting\n    if (gasReport) {\n      args.push(\"--gas-report\");\n    }\n\n    // CRITICAL: Add fork URL to connect to deployed Diamond\n    // This allows Forge tests to access the Hardhat-deployed Diamond contract\n    if (forkUrl) {\n      args.push(\"--fork-url\", forkUrl);\n    }\n\n    const cmd = `forge ${args.join(\" \")}`;\n    console.log(`\\nRunning: ${picocolors.blue(cmd)}\\n`);\n\n    const { stdout, stderr } = await exec(cmd, { cwd, maxBuffer: 10 * 1024 * 1024 });\n    console.log(stdout);\n    if (stderr) console.error(stderr);\n\n    return { success: true, output: stdout };\n  } catch (error: any) {\n    // Forge returns non-zero exit code when tests fail\n    // We still want to show the output\n    if (error.stdout) console.log(error.stdout);\n    if (error.stderr) console.error(error.stderr);\n\n    return {\n      success: false,\n      output: error.stdout || error.stderr || error.message,\n    };\n  }\n}\n"]}