76 lines
1.8 KiB
Rust
76 lines
1.8 KiB
Rust
use std::{collections::HashMap, io::Write};
|
|
|
|
|
|
|
|
pub fn simple_prompt(p: &str) -> String {
|
|
|
|
print!("{c}prompt{r}: {p}",
|
|
c=anstyle::AnsiColor::Magenta.render_fg(),
|
|
r=anstyle::Style::new().render_reset()
|
|
);
|
|
|
|
// I dont care if it fails
|
|
let _ = std::io::stdout().flush();
|
|
|
|
let mut buf = String::new();
|
|
let _ = std::io::stdin().read_line(&mut buf);
|
|
|
|
buf
|
|
}
|
|
|
|
pub fn prompt_with_options(p: &str, options: &[&str]) -> usize {
|
|
println!("{c}prompt{r}: {p}",
|
|
c=anstyle::AnsiColor::Magenta.render_fg(),
|
|
r=anstyle::Style::new().render_reset()
|
|
);
|
|
|
|
for (i, op) in options.iter().enumerate() {
|
|
println!(" - {}: {}", i, op);
|
|
}
|
|
|
|
print!("> ");
|
|
// I dont care if it fails
|
|
let _ = std::io::stdout().flush();
|
|
|
|
let mut buf = String::new();
|
|
let _ = std::io::stdin().read_line(&mut buf);
|
|
|
|
if let Ok(num) = buf.parse::<usize>() {
|
|
if num <= options.len() {
|
|
return num;
|
|
} else {
|
|
log::error!("Number not in range");
|
|
return prompt_with_options(p, options);
|
|
}
|
|
} else {
|
|
log::error!("Not a number");
|
|
return prompt_with_options(p, options);
|
|
}
|
|
}
|
|
|
|
pub fn prompt_with_named_options(p: &str, options: HashMap<&str, &str>) -> String {
|
|
println!("{c}prompt{r}: {p}",
|
|
c=anstyle::AnsiColor::Magenta.render_fg(),
|
|
r=anstyle::Style::new().render_reset()
|
|
);
|
|
|
|
let mut keys = Vec::new();
|
|
|
|
for (k, v) in &options {
|
|
println!(" - {}: {}", k, v);
|
|
keys.push(k.trim().to_lowercase())
|
|
}
|
|
|
|
print!("> ");
|
|
|
|
// I dont care if it fails
|
|
let _ = std::io::stdout().flush();
|
|
|
|
let mut buf = String::new();
|
|
let _ = std::io::stdin().read_line(&mut buf);
|
|
if !keys.contains(&buf.trim().to_lowercase()) {
|
|
return prompt_with_named_options(p, options);
|
|
}
|
|
buf
|
|
}
|