INNER CODE UNIT · Rust

yn_prompt

Drew-Alleman/DataSurgeon · src/utils/mod.rs:11

fn yn_prompt(message: &str) -> bool {
    println!("[+] {}", message);

    let mut input = String::new();

    match io::stdin().read_line(&mut input) {
        Ok(_) => {
            match input.trim().to_lowercase().as_str() {
                "y" => true,
                "n" => false,
                _ => {
                    println!("[-] Invalid input. Please enter 'y' or 'n'.");
                    yn_prompt(message) // Ask the prompt again for invalid input
                }
            }
        }
        Err(error) => {
            println!("[-] Failed to read input: {}", error);

View source record →