rigg 0.17.0

Configuration-as-code CLI for Azure AI Search and Microsoft Foundry
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

//! Shared confirmation prompt utilities

use std::io::{self, BufRead, Write};

use anyhow::Result;

/// Prompt the user with a yes/no question. Default is "no".
/// Returns true only if the user types "y" or "yes" (case-insensitive).
pub fn prompt_yes_no(message: &str) -> Result<bool> {
    print!("{} [y/N] ", message);
    io::stdout().flush()?;

    let mut input = String::new();
    io::stdin().lock().read_line(&mut input)?;

    Ok(matches!(
        input.trim().to_ascii_lowercase().as_str(),
        "y" | "yes"
    ))
}

/// Prompt the user with a yes/no question. Default is "yes".
/// Returns false only if the user types "n" or "no" (case-insensitive).
pub fn prompt_yes_default(message: &str) -> Result<bool> {
    print!("{} [Y/n] ", message);
    io::stdout().flush()?;

    let mut input = String::new();
    io::stdin().lock().read_line(&mut input)?;

    Ok(!matches!(
        input.trim().to_ascii_lowercase().as_str(),
        "n" | "no"
    ))
}

#[cfg(test)]
mod tests {
    #[test]
    fn test_yes_default_parsing_logic() {
        // Test the parsing logic for default-yes prompts
        let check = |input: &str| -> bool {
            !matches!(input.trim().to_ascii_lowercase().as_str(), "n" | "no")
        };

        assert!(check("y"));
        assert!(check("Y"));
        assert!(check("yes"));
        assert!(check("")); // empty = yes (default)
        assert!(check("  ")); // whitespace = yes (default)
        assert!(!check("n"));
        assert!(!check("N"));
        assert!(!check("no"));
        assert!(!check("NO"));
        assert!(!check("No"));
    }

    #[test]
    fn test_yes_no_parsing_logic() {
        // Test the parsing logic without stdin
        let check = |input: &str| -> bool {
            matches!(input.trim().to_ascii_lowercase().as_str(), "y" | "yes")
        };

        assert!(check("y"));
        assert!(check("Y"));
        assert!(check("yes"));
        assert!(check("YES"));
        assert!(check("Yes"));
        assert!(check("  y  "));
        assert!(!check("n"));
        assert!(!check("no"));
        assert!(!check(""));
        assert!(!check("yep"));
        assert!(!check("nope"));
    }
}