TimeMachine.t.sol

// SPDX-License-Identifier: BUSL-1.1
pragma solidity ^0.8.27;

import "forge-std/Test.sol";
import "src/test/utils/Logger.t.sol";

contract TimeMachine is Test, Logger {
    uint[] public snapshots;

    function NAME() public view virtual override returns (string memory) {
        return "TimeMachine";
    }

    /// -----------------------------------------------------------------------
    /// Setters
    /// -----------------------------------------------------------------------

    function createSnapshot() public returns (uint snapshot) {
        snapshots.push(snapshot = cheats.snapshotState());
        print.method("createSnapshot", cheats.toString(snapshot));
    }

    function travelToLast() public returns (uint currentSnapshot) {
        // Safety check to make sure createSnapshot is called before attempting
        // to warp so we don't accidentally prevent our own births.
        assertTrue(pastExists(), "Global.warpToPast: invalid usage, past does not exist");
        uint last = lastSnapshot();
        // print.method("travelToLast", cheats.toString(last));
        currentSnapshot = createSnapshot();
        cheats.revertToState(last);
    }

    function travel(uint snapshot) public {
        // print.method("travel", cheats.toString(snapshot));
        cheats.revertToState(snapshot);
    }

    /// -----------------------------------------------------------------------
    /// Getters
    /// -----------------------------------------------------------------------

    function lastSnapshot() public view returns (uint) {
        return snapshots[snapshots.length - 1];
    }

    function pastExists() public view returns (bool) {
        return snapshots.length != 0;
    }
}