terminal: Extend Ask default actions, prompts

epilys · epilys · commit 49dcbc5e58be · 2024-09-23T22:19:40.000+03:00
Signed-off-by: Manos Pitsidianakis &lt;manos@pitsidianak.is&gt;
diff --git a/meli/src/conf.rs b/meli/src/conf.rs
@@ -325,12 +325,10 @@ impl FileSettings {
                     .set_kind(ErrorKind::Configuration));
             }
             #[cfg(not(test))]
-            let ask = crate::terminal::Ask {
-                message: format!(
-                    "No configuration found. Would you like to generate one in {}?",
-                    path_string
-                ),
-            };
+            let ask = crate::terminal::Ask::new(format!(
+                "No configuration found. Would you like to generate one in {}?",
+                path_string
+            ));
             #[cfg(not(test))]
             if ask.run() {
                 create_config_file(&config_path)?;
diff --git a/meli/src/terminal.rs b/meli/src/terminal.rs
@@ -422,33 +422,66 @@ derive_csi_sequence!(
     (QuerySynchronizedOutputSupport, "?2026$p")
 );
 
-pub struct Ask {
-    pub message: String,
+pub struct Ask<'m> {
+    message: Cow<'m, str>,
+    default: Option<bool>,
 }
 
-impl Ask {
+impl<'m> Ask<'m> {
+    pub fn new<M>(m: M) -> Self
+    where
+        M: Into<Cow<'m, str>>,
+    {
+        let message = m.into();
+        Self {
+            message,
+            default: Some(true),
+        }
+    }
+
+    pub fn yes_by_default(self, default: bool) -> Self {
+        Self {
+            default: Some(default),
+            ..self
+        }
+    }
+
+    pub fn without_default(self) -> Self {
+        Self {
+            default: None,
+            ..self
+        }
+    }
+
     pub fn run(self) -> bool {
         let mut buffer = String::new();
         let stdin = std::io::stdin();
         let mut handle = stdin.lock();
 
-        print!("{} [Y/n] ", &self.message);
+        let default = match self.default {
+            None => "y/n",
+            Some(true) => "Y/n",
+            Some(false) => "y/N",
+        };
+
+        print!("{} [{default}] ", self.message.as_ref());
         let _ = std::io::stdout().flush();
         loop {
             buffer.clear();
             handle
                 .read_line(&mut buffer)
                 .expect("Could not read from stdin.");
 
-            match buffer.trim() {
-                "" | "Y" | "y" | "yes" | "YES" | "Yes" => {
+            match (buffer.trim(), self.default) {
+                ("", Some(val)) => return val,
+                ("Y" | "y" | "yes" | "YES" | "Yes", _) => {
                     return true;
                 }
-                "n" | "N" | "no" | "No" | "NO" => {
+                ("n" | "N" | "no" | "No" | "NO", _) => {
                     return false;
                 }
                 _ => {
-                    print!("\n{} [Y/n] ", &self.message);
+                    print!("\n{} [{default}] ", self.message.as_ref());
                     let _ = std::io::stdout().flush();
                 }
             }
