summaryrefslogtreecommitdiff
path: root/vendor/dialoguer/src/edit.rs
diff options
context:
space:
mode:
authorValentin Popov <valentin@popov.link>2024-01-08 00:21:28 +0300
committerValentin Popov <valentin@popov.link>2024-01-08 00:21:28 +0300
commit1b6a04ca5504955c571d1c97504fb45ea0befee4 (patch)
tree7579f518b23313e8a9748a88ab6173d5e030b227 /vendor/dialoguer/src/edit.rs
parent5ecd8cf2cba827454317368b68571df0d13d7842 (diff)
downloadfparkan-1b6a04ca5504955c571d1c97504fb45ea0befee4.tar.xz
fparkan-1b6a04ca5504955c571d1c97504fb45ea0befee4.zip
Initial vendor packages
Signed-off-by: Valentin Popov <valentin@popov.link>
Diffstat (limited to 'vendor/dialoguer/src/edit.rs')
-rw-r--r--vendor/dialoguer/src/edit.rs131
1 files changed, 131 insertions, 0 deletions
diff --git a/vendor/dialoguer/src/edit.rs b/vendor/dialoguer/src/edit.rs
new file mode 100644
index 0000000..f69844c
--- /dev/null
+++ b/vendor/dialoguer/src/edit.rs
@@ -0,0 +1,131 @@
+use std::{
+ env,
+ ffi::{OsStr, OsString},
+ fs, io,
+ io::{Read, Write},
+ process,
+};
+
+/// Launches the default editor to edit a string.
+///
+/// ## Example
+///
+/// ```rust,no_run
+/// use dialoguer::Editor;
+///
+/// if let Some(rv) = Editor::new().edit("Enter a commit message").unwrap() {
+/// println!("Your message:");
+/// println!("{}", rv);
+/// } else {
+/// println!("Abort!");
+/// }
+/// ```
+pub struct Editor {
+ editor: OsString,
+ extension: String,
+ require_save: bool,
+ trim_newlines: bool,
+}
+
+fn get_default_editor() -> OsString {
+ if let Some(prog) = env::var_os("VISUAL") {
+ return prog;
+ }
+ if let Some(prog) = env::var_os("EDITOR") {
+ return prog;
+ }
+ if cfg!(windows) {
+ "notepad.exe".into()
+ } else {
+ "vi".into()
+ }
+}
+
+impl Default for Editor {
+ fn default() -> Self {
+ Self::new()
+ }
+}
+
+impl Editor {
+ /// Creates a new editor.
+ pub fn new() -> Self {
+ Self {
+ editor: get_default_editor(),
+ extension: ".txt".into(),
+ require_save: true,
+ trim_newlines: true,
+ }
+ }
+
+ /// Sets a specific editor executable.
+ pub fn executable<S: AsRef<OsStr>>(&mut self, val: S) -> &mut Self {
+ self.editor = val.as_ref().into();
+ self
+ }
+
+ /// Sets a specific extension
+ pub fn extension(&mut self, val: &str) -> &mut Self {
+ self.extension = val.into();
+ self
+ }
+
+ /// Enables or disables the save requirement.
+ pub fn require_save(&mut self, val: bool) -> &mut Self {
+ self.require_save = val;
+ self
+ }
+
+ /// Enables or disables trailing newline stripping.
+ ///
+ /// This is on by default.
+ pub fn trim_newlines(&mut self, val: bool) -> &mut Self {
+ self.trim_newlines = val;
+ self
+ }
+
+ /// Launches the editor to edit a string.
+ ///
+ /// Returns `None` if the file was not saved or otherwise the
+ /// entered text.
+ pub fn edit(&self, s: &str) -> io::Result<Option<String>> {
+ let mut f = tempfile::Builder::new()
+ .prefix("edit-")
+ .suffix(&self.extension)
+ .rand_bytes(12)
+ .tempfile()?;
+ f.write_all(s.as_bytes())?;
+ f.flush()?;
+ let ts = fs::metadata(f.path())?.modified()?;
+
+ let s: String = self.editor.clone().into_string().unwrap();
+ let (cmd, args) = match shell_words::split(&s) {
+ Ok(mut parts) => {
+ let cmd = parts.remove(0);
+ (cmd, parts)
+ }
+ Err(_) => (s, vec![]),
+ };
+
+ let rv = process::Command::new(cmd)
+ .args(args)
+ .arg(f.path())
+ .spawn()?
+ .wait()?;
+
+ if rv.success() && self.require_save && ts >= fs::metadata(f.path())?.modified()? {
+ return Ok(None);
+ }
+
+ let mut new_f = fs::File::open(f.path())?;
+ let mut rv = String::new();
+ new_f.read_to_string(&mut rv)?;
+
+ if self.trim_newlines {
+ let len = rv.trim_end_matches(&['\n', '\r'][..]).len();
+ rv.truncate(len);
+ }
+
+ Ok(Some(rv))
+ }
+}