Feat: Index cheatcode for Strings

crates/cheatcodes/spec/src/vm.rs

@@ -1678,6 +1678,10 @@ interface Vm {
     /// Splits the given `string` into an array of strings divided by the `delimiter`.
     #[cheatcode(group = String)]
     function split(string calldata input, string calldata delimiter) external pure returns (string[] memory outputs);
+    /// Returns the index of the first occurrence of a `key` in an `input` string.
+    /// Returns `NOT_FOUND` (i.e. `type(uint256).max`) if the `key` is not found.
+    #[cheatcode(group = String)]
+    function indexOf(string memory input, string memory key) external pure returns (uint256);
 
     // ======== JSON Parsing and Manipulation ========
 

crates/cheatcodes/src/string.rs

@@ -135,6 +135,20 @@ impl Cheatcode for splitCall {
     }
 }
 
+// indexOf
+impl Cheatcode for indexOfCall {
+    fn apply(&self, _state: &mut Cheatcodes) -> Result {
+        let Self { input, key } = self;
+        let Some(index) = input.find(key) else {
+            return parse(
+                "0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff",
+                &DynSolType::Uint(256),
+            );
+        };
+        parse(index.to_string().as_str(), &DynSolType::Uint(256))
+    }
+}
+
 pub(super) fn parse(s: &str, ty: &DynSolType) -> Result {
     parse_value(s, ty).map(|v| v.abi_encode())
 }

testdata/default/cheats/StringUtils.t.sol

@@ -39,4 +39,16 @@ contract StringManipulationTest is DSTest {
         assertEq("World", splitResult[1]);
         assertEq("Reth", splitResult[2]);
     }
+
+    function testIndexOf() public {
+        string memory input = "Hello, World!";
+        string memory key1 = "Hello";
+        string memory key2 = "World";
+        string memory key3 = "!";
+        string memory key4 = "foundry";
+        assertEq(vm.indexOf(input, key1), 0);
+        assertEq(vm.indexOf(input, key2), 7);
+        assertEq(vm.indexOf(input, key3), 12);
+        assertEq(vm.indexOf(input, key4), type(uint256).max);
+    }
 }