From ed027187568c91a14c545c1962a219552e4654e7 Mon Sep 17 00:00:00 2001
From: 魏曹先生 <1992414357@qq.com>
Date: Wed, 29 Oct 2025 15:27:11 +0800
Subject: Add input utility functions for user confirmation

---
 src/utils/input.rs | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 src/utils/input.rs

(limited to 'src/utils/input.rs')

diff --git a/src/utils/input.rs b/src/utils/input.rs
new file mode 100644
index 0000000..217cede
--- /dev/null
+++ b/src/utils/input.rs
@@ -0,0 +1,52 @@
+/// Confirm the current operation
+/// Waits for user input of 'y' or 'n'
+pub async fn confirm_hint(text: impl Into<String>) -> bool {
+    use tokio::io::{self, AsyncBufReadExt, AsyncWriteExt, BufReader};
+
+    let prompt = text.into().trim().to_string();
+
+    let mut stdout = io::stdout();
+    let mut stdin = BufReader::new(io::stdin());
+
+    stdout
+        .write_all(prompt.as_bytes())
+        .await
+        .expect("Failed to write prompt");
+    stdout.flush().await.expect("Failed to flush stdout");
+
+    let mut input = String::new();
+    stdin
+        .read_line(&mut input)
+        .await
+        .expect("Failed to read input");
+
+    input.trim().eq_ignore_ascii_case("y")
+}
+
+/// Confirm the current operation, or execute a closure if rejected
+/// Waits for user input of 'y' or 'n'
+/// If 'n' is entered, executes the provided closure and returns false
+pub async fn confirm_hint_or<F>(text: impl Into<String>, on_reject: F) -> bool
+where
+    F: FnOnce(),
+{
+    let confirmed = confirm_hint(text).await;
+    if !confirmed {
+        on_reject();
+    }
+    confirmed
+}
+
+/// Confirm the current operation, and execute a closure if confirmed
+/// Waits for user input of 'y' or 'n'
+/// If 'y' is entered, executes the provided closure and returns true
+pub async fn confirm_hint_then<F>(text: impl Into<String>, on_confirm: F) -> bool
+where
+    F: FnOnce(),
+{
+    let confirmed = confirm_hint(text).await;
+    if confirmed {
+        on_confirm();
+    }
+    confirmed
+}
-- 
cgit 


From 251218ed09d640d7af44f26c6917d8fdb90fc263 Mon Sep 17 00:00:00 2001
From: 魏曹先生 <1992414357@qq.com>
Date: Wed, 29 Oct 2025 16:34:43 +0800
Subject: Add input_with_editor function for text editing

This function opens the system editor with default text in a cache file,
reads back the modified content after editing, and removes comment
lines.
---
 src/utils/input.rs | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

(limited to 'src/utils/input.rs')

diff --git a/src/utils/input.rs b/src/utils/input.rs
index 217cede..a728c77 100644
--- a/src/utils/input.rs
+++ b/src/utils/input.rs
@@ -1,3 +1,5 @@
+use tokio::{fs, process::Command};
+
 /// Confirm the current operation
 /// Waits for user input of 'y' or 'n'
 pub async fn confirm_hint(text: impl Into<String>) -> bool {
@@ -50,3 +52,54 @@ where
     }
     confirmed
 }
+
+/// Input text using the system editor
+/// Opens the system editor (from EDITOR environment variable) with default text in a cache file,
+/// then reads back the modified content after the editor closes, removing comment lines
+pub async fn input_with_editor(
+    default_text: impl AsRef<str>,
+    cache_file: impl AsRef<std::path::Path>,
+    comment_char: impl AsRef<str>,
+) -> Result<String, std::io::Error> {
+    let cache_path = cache_file.as_ref();
+    let default_content = default_text.as_ref();
+    let comment_prefix = comment_char.as_ref();
+
+    // Write default text to cache file
+    fs::write(cache_path, default_content).await?;
+
+    // Get editor from environment variable
+    let editor = std::env::var("EDITOR").unwrap_or_else(|_| "vi".to_string());
+
+    // Open editor with cache file
+    let status = Command::new(editor).arg(cache_path).status().await?;
+
+    if !status.success() {
+        return Err(std::io::Error::new(
+            std::io::ErrorKind::Other,
+            "Editor exited with non-zero status",
+        ));
+    }
+
+    // Read the modified content
+    let content = fs::read_to_string(cache_path).await?;
+
+    // Remove comment lines and trim
+    let processed_content: String = content
+        .lines()
+        .filter_map(|line| {
+            let trimmed = line.trim();
+            if trimmed.starts_with(comment_prefix) {
+                None
+            } else {
+                Some(line)
+            }
+        })
+        .collect::<Vec<&str>>()
+        .join("\n");
+
+    // Delete the cache file
+    let _ = fs::remove_file(cache_path).await;
+
+    Ok(processed_content)
+}
-- 
cgit