[pbs-devel] [PATCH proxmox v3] router: cli: added `ask_for_confirmation` helper

Gabriel Goller g.goller at proxmox.com
Fri Sep 8 13:44:09 CEST 2023


Added `ask_for_confirmation` helper that outputs a prompt and
lets the user confirm or deny it. Also added localization using
`libc::nl_langinfo()` to match/show a `yes` or `no` in the correct
language.
Implemented to close #4763.

Signed-off-by: Gabriel Goller <g.goller at proxmox.com>
---

update v3:
 - added localization using `libc::nl_langinfo`

note: skipped to version 3, so that the whole patch series has 
      the same version number.

 proxmox-router/Cargo.toml     |  1 +
 proxmox-router/src/cli/mod.rs | 34 +++++++++++++++++++++++++++++++++-
 2 files changed, 34 insertions(+), 1 deletion(-)

diff --git a/proxmox-router/Cargo.toml b/proxmox-router/Cargo.toml
index 1c08ce2..e0d67d0 100644
--- a/proxmox-router/Cargo.toml
+++ b/proxmox-router/Cargo.toml
@@ -19,6 +19,7 @@ percent-encoding.workspace = true
 serde_json.workspace = true
 serde.workspace = true
 unicode-width ="0.1.8"
+regex.workspace = true
 
 # cli:
 tokio = { workspace = true, features = [], optional = true }
diff --git a/proxmox-router/src/cli/mod.rs b/proxmox-router/src/cli/mod.rs
index 208df4a..1dcfaf0 100644
--- a/proxmox-router/src/cli/mod.rs
+++ b/proxmox-router/src/cli/mod.rs
@@ -12,7 +12,10 @@
 //! - Ability to create interactive commands (using ``rustyline``)
 //! - Supports complex/nested commands
 
-use std::collections::HashMap;
+use std::{
+    collections::HashMap,
+    io::{self, Write},
+};
 
 use crate::ApiMethod;
 
@@ -62,6 +65,35 @@ pub fn init_cli_logger(env_var_name: &str, default_log_level: &str) {
     .init();
 }
 
+pub fn ask_for_confirmation(query: String) -> Result<(), io::Error> {
+    use std::ffi::CStr;
+    let yes_expr = unsafe { CStr::from_ptr(libc::nl_langinfo(libc::YESEXPR)) }
+        .to_str()
+        .unwrap_or("^[yY]$");
+    let no_expr = unsafe { CStr::from_ptr(libc::nl_langinfo(libc::NOEXPR)) }
+        .to_str()
+        .unwrap_or("^[nN]$");
+
+    let n = no_expr.chars().find(|c| c.is_uppercase()).unwrap_or('N');
+    let y = yes_expr.chars().find(|c| c.is_uppercase()).unwrap_or('Y');
+    print!("{query} [{y}/{n}]: ");
+
+    use regex::Regex;
+    // .unwrap() is okay, because we should always get a correct regex (see `.unwrap_or()` calls above)
+    let yes_regex: Regex = Regex::new(yes_expr).unwrap();
+
+    io::stdout().flush()?;
+    let stdin = io::stdin();
+    let mut line = String::new();
+    stdin.read_line(&mut line)?;
+
+    if yes_regex.is_match(line.trim()) {
+        Ok(())
+    } else {
+        Err(io::Error::new(io::ErrorKind::Other, "Aborted"))
+    }
+}
+
 /// Define a simple CLI command.
 pub struct CliCommand {
     /// The Schema definition.
-- 
2.39.2






More information about the pbs-devel mailing list