Custom Hooks

interface IHook {
    function run(
        AuthCaptureEscrow.PaymentInfo calldata paymentInfo,
        uint256 amount,
        address caller,
        bytes calldata data
    ) external;
}

contract MyHook is BaseHook {
    constructor(address escrow, bytes32 authorizedCodehash)
        BaseHook(escrow, authorizedCodehash)
    {}

    function run(
        AuthCaptureEscrow.PaymentInfo calldata paymentInfo,
        uint256 amount,
        address /* caller */,
        bytes calldata /* data */
    ) external override {
        bytes32 hash = _verifyAndHash(paymentInfo);
        // Your recording logic here, keyed by `hash`
    }
}

contract CaptureCountHook is BaseHook {
    mapping(bytes32 => uint256) public captureCount;
    mapping(bytes32 => uint256) public totalCaptured;

    constructor(address escrow, bytes32 authorizedCodehash)
        BaseHook(escrow, authorizedCodehash)
    {}

    function run(
        AuthCaptureEscrow.PaymentInfo calldata paymentInfo,
        uint256 amount,
        address /* caller */,
        bytes calldata /* data */
    ) external override {
        bytes32 hash = _verifyAndHash(paymentInfo);
        captureCount[hash]++;
        totalCaptured[hash] += amount;
    }

    function getStats(bytes32 paymentHash)
        external view
        returns (uint256 count, uint256 total)
    {
        return (captureCount[paymentHash], totalCaptured[paymentHash]);
    }
}

contract CaptureCountHookTest is Test {
    CaptureCountHook hook;

    function setUp() public {
        hook = new CaptureCountHook(address(escrow), bytes32(0));
    }

    function test_incrementsOnRun() public {
        vm.prank(paymentInfo.operator);
        hook.run(paymentInfo, 100e6, address(0), "");
        bytes32 hash = escrow.getHash(paymentInfo);
        (uint256 count, uint256 total) = hook.getStats(hash);
        assertEq(count, 1);
        assertEq(total, 100e6);
    }

    function test_tracksMultipleCaptures() public {
        vm.startPrank(paymentInfo.operator);
        hook.run(paymentInfo, 50e6, address(0), "");
        hook.run(paymentInfo, 30e6, address(0), "");
        vm.stopPrank();
        bytes32 hash = escrow.getHash(paymentInfo);
        (uint256 count, uint256 total) = hook.getStats(hash);
        assertEq(count, 2);
        assertEq(total, 80e6);
    }
}