feat(forge): prompt address and uint cheatcodes

crates/cheatcodes/spec/src/vm.rs

@@ -1427,6 +1427,14 @@ interface Vm {
     #[cheatcode(group = Filesystem)]
     function promptSecret(string calldata promptText) external returns (string memory input);
 
+    /// Prompts the user for an address in the terminal.
+    #[cheatcode(group = Filesystem)]
+    function promptAddress(string calldata promptText) external returns (address);
+
+    /// Prompts the user for uint256 in the terminal.
+    #[cheatcode(group = Filesystem)]
+    function promptUint(string calldata promptText) external returns (uint256);
+
     // ======== Environment Variables ========
 
     /// Sets environment variables.

crates/cheatcodes/src/fs.rs

@@ -1,6 +1,8 @@
 //! Implementations of [`Filesystem`](crate::Group::Filesystem) cheatcodes.
 
+use super::string::parse;
 use crate::{Cheatcode, Cheatcodes, Result, Vm::*};
+use alloy_dyn_abi::DynSolType;
 use alloy_json_abi::ContractObject;
 use alloy_primitives::U256;
 use alloy_sol_types::SolValue;
@@ -374,6 +376,20 @@ impl Cheatcode for promptSecretCall {
     }
 }
 
+impl Cheatcode for promptAddressCall {
+    fn apply(&self, state: &mut Cheatcodes) -> Result {
+        let Self { promptText: text } = self;
+        parse(&prompt(state, text, prompt_input)?, &DynSolType::Address)
+    }
+}
+
+impl Cheatcode for promptUintCall {
+    fn apply(&self, state: &mut Cheatcodes) -> Result {
+        let Self { promptText: text } = self;
+        parse(&prompt(state, text, prompt_input)?, &DynSolType::Uint(256))
+    }
+}
+
 pub(super) fn write_file(state: &Cheatcodes, path: &Path, contents: &[u8]) -> Result {
     let path = state.config.ensure_path_allowed(path, FsAccessKind::Write)?;
     // write access to foundry.toml is not allowed

testdata/default/cheats/Prompt.t.sol

@@ -3,6 +3,7 @@ pragma solidity 0.8.18;
 
 import "ds-test/test.sol";
 import "cheats/Vm.sol";
+import "../logs/console.sol";
 
 contract PromptTest is DSTest {
     Vm constant vm = Vm(HEVM_ADDRESS);
@@ -15,4 +16,14 @@ contract PromptTest is DSTest {
         vm._expectCheatcodeRevert();
         vm.promptSecret("test");
     }
+
+    function testPrompt_Address() public {
+        vm._expectCheatcodeRevert();
+        address test = vm.promptAddress("test");
+    }
+
+    function testPrompt_Uint() public {
+        vm._expectCheatcodeRevert();
+        uint256 test = vm.promptUint("test");
+    }
 }