diff --git a/helix-term/src/commands.rs b/helix-term/src/commands.rs index 2fe841628..3f6495728 100644 --- a/helix-term/src/commands.rs +++ b/helix-term/src/commands.rs @@ -954,17 +954,22 @@ mod cmd { }; } - fn write(editor: &mut Editor, args: &[&str], event: PromptEvent) { - let (view, doc) = editor.current(); - if let Some(path) = args.get(0) { - if let Err(err) = doc.set_path(Path::new(path)) { - editor.set_error(format!("invalid filepath: {}", err)); - return; + fn _write>( + view: &View, + doc: &mut Document, + path: Option

, + ) -> Result<(), anyhow::Error> { + use anyhow::anyhow; + + if let Some(path) = path { + if let Err(err) = doc.set_path(path.as_ref()) { + return Err(anyhow!("invalid filepath: {}", err)); }; } if doc.path().is_none() { - editor.set_error("cannot write a buffer without a filename".to_string()); - return; + return Err(anyhow!( + "cannot write a buffer without a filename".to_string() + )); } let autofmt = doc .language_config() @@ -974,6 +979,14 @@ mod cmd { doc.format(view.id); // TODO: merge into save } tokio::spawn(doc.save()); + Ok(()) + } + + fn write(editor: &mut Editor, args: &[&str], event: PromptEvent) { + let (view, doc) = editor.current(); + if let Err(e) = _write(view, doc, args.first()) { + editor.set_error(e.to_string()); + }; } fn new_file(editor: &mut Editor, args: &[&str], event: PromptEvent) { @@ -1010,6 +1023,52 @@ mod cmd { doc.later(view.id, uk) } + fn write_quit(editor: &mut Editor, args: &[&str], event: PromptEvent) { + write(editor, args, event.clone()); + quit(editor, &[], event); + } + + fn force_write_quit(editor: &mut Editor, args: &[&str], event: PromptEvent) { + write(editor, args, event.clone()); + force_quit(editor, &[], event); + } + + fn _write_all(editor: &mut Editor, args: &[&str], event: PromptEvent, quit: bool, force: bool) { + let ids = editor + .tree + .views() + .map(|(view, _)| (view.id, view.doc)) + .collect::>(); + + for (view_id, doc_id) in ids { + if let Some(doc) = editor.documents.get_mut(doc_id) { + let view = editor.tree.get(view_id); + if let Err(e) = _write(view, doc, None::<&str>) { + editor.set_error(e.to_string()); + } else if quit { + editor.close(view_id, false); + continue; + } + + if force { + editor.close(view_id, false); + } + } + } + } + + fn write_all(editor: &mut Editor, args: &[&str], event: PromptEvent) { + _write_all(editor, args, event, false, false) + } + + fn write_all_quit(editor: &mut Editor, args: &[&str], event: PromptEvent) { + _write_all(editor, args, event, true, false) + } + + fn force_write_all_quit(editor: &mut Editor, args: &[&str], event: PromptEvent) { + _write_all(editor, args, event, true, true) + } + pub const COMMAND_LIST: &[Command] = &[ Command { name: "quit", @@ -1067,6 +1126,41 @@ mod cmd { fun: later, completer: None, }, + Command { + name: "write-quit", + alias: Some("wq"), + doc: "Writes changes to disk and closes the current view. Accepts an optional path (:wq some/path.txt)", + fun: write_quit, + completer: Some(completers::filename), + }, + Command { + name: "write-quit!", + alias: Some("wq!"), + doc: "Writes changes to disk and closes the current view forcefully. Accepts an optional path (:wq! some/path.txt)", + fun: force_write_quit, + completer: Some(completers::filename), + }, + Command { + name: "write-all", + alias: Some("wa"), + doc: "Writes changes from all views to disk.", + fun: write_all, + completer: None, + }, + Command { + name: "write-all-quit", + alias: Some("waq"), + doc: "Writes changes from all views to disk and close all views.", + fun: write_all_quit, + completer: None, + }, + Command { + name: "write-all-quit!", + alias: Some("waq!"), + doc: "Writes changes from all views to disk and close all views forcefully (ignoring errors).", + fun: force_write_all_quit, + completer: None, + }, ]; pub static COMMANDS: Lazy> = Lazy::new(|| { diff --git a/helix-term/src/ui/prompt.rs b/helix-term/src/ui/prompt.rs index 8367b12c1..c6429736c 100644 --- a/helix-term/src/ui/prompt.rs +++ b/helix-term/src/ui/prompt.rs @@ -18,7 +18,7 @@ pub struct Prompt { pub doc_fn: Box Option<&'static str>>, } -#[derive(PartialEq)] +#[derive(Clone, PartialEq)] pub enum PromptEvent { /// The prompt input has been updated. Update,