on master help sign in
Update write_all
commit62e6232a329d8b0e5779236f0d3ca7b3b3591946 authorWojciech Kępka <wojtek.kepka@protonmail.com> dateSun 13 Jun 2021, 11:00 (5 years ago) parentd8b5d11
1 file changed+43−26
helix-term/src/commands.rs+43−26
helix-term/src/commands.rs+43−26
@@ -915,15 +915,7 @@ mod cmd {
915915 fn quit(editor: &mut Editor, args: &[&str], event: PromptEvent) {
916916 // last view and we have unsaved changes
917917 if editor.tree.views().count() == 1 {
918− let modified: Vec<_> = editor
919− .documents()
920− .filter(|doc| doc.is_modified())
921− .map(|doc| {
922− doc.relative_path()
923− .and_then(|path| path.to_str())
924− .unwrap_or("[scratch]")
925− })
926− .collect();
918+ let modified = _modified_left(editor);
927919
928920 if !modified.is_empty() {
929921 let err = format!(
@@ -1033,27 +1025,52 @@ mod cmd {
10331025 force_quit(editor, &[], event);
10341026 }
10351027
1028+ fn _modified_left(editor: &mut Editor) -> Vec<&str> {
1029+ editor
1030+ .documents()
1031+ .filter(|doc| doc.is_modified())
1032+ .map(|doc| {
1033+ doc.relative_path()
1034+ .and_then(|path| path.to_str())
1035+ .unwrap_or("[scratch]")
1036+ })
1037+ .collect()
1038+ }
1039+
10361040 fn _write_all(editor: &mut Editor, args: &[&str], event: PromptEvent, quit: bool, force: bool) {
1037− let ids = editor
1038− .tree
1039− .views()
1040− .map(|(view, _)| (view.id, view.doc))
1041− .collect::<Vec<_>>();
1042−
1043− for (view_id, doc_id) in ids {
1044− if let Some(doc) = editor.documents.get_mut(doc_id) {
1045− let view = editor.tree.get(view_id);
1046− if let Err(e) = _write(view, doc, None::<&str>) {
1047− editor.set_error(e.to_string());
1048− } else if quit {
1049− editor.close(view_id, false);
1050− continue;
1051− }
1041+ let mut errors = String::new();
10521042
1053− if force {
1054− editor.close(view_id, false);
1043+ // save all documents
1044+ for (id, mut doc) in &mut editor.documents {
1045+ if doc.path().is_none() {
1046+ errors.push_str("cannot write a buffer without a filename\n");
1047+ continue;
1048+ }
1049+ tokio::spawn(doc.save());
1050+ }
1051+ editor.set_error(errors);
1052+
1053+ if quit {
1054+ if !force {
1055+ // check if there are some unsaved buffers
1056+ let modified = _modified_left(editor);
1057+
1058+ if !modified.is_empty() {
1059+ let err = format!(
1060+ "{} unsaved buffer(s) remaining: {:?}",
1061+ modified.len(),
1062+ modified
1063+ );
1064+ editor.set_error(err);
1065+ return;
10551066 }
10561067 }
1068+
1069+ // close all views
1070+ let views: Vec<_> = editor.tree.views().map(|(view, _)| view.id).collect();
1071+ for view_id in views {
1072+ editor.close(view_id, false);
1073+ }
10571074 }
10581075 }
10591076