refresh all views cursors

Author: mangas

helix-term/src/commands/typed.rs

@@ -1049,34 +1049,44 @@ fn reload_all(
         view.id
     };
 
-    let docs_views: Vec<(DocumentId, ViewId)> = cx
+    let docs_views: Vec<(DocumentId, Vec<ViewId>)> = cx
         .editor
         .documents_mut()
         .map(|doc| {
-            let target_view = if doc.selections().contains_key(&view_id) {
-                view_id
-            } else if let Some(view) = doc.selections().keys().next() {
-                *view
-            } else {
+            let mut views: Vec<_> = doc.selections().keys().cloned().collect();
+
+            if views.is_empty() {
                 doc.ensure_view_init(view_id);
-                view_id
+                views = vec![view_id];
             };
 
-            (doc.id(), target_view)
+            (doc.id(), views)
         })
         .collect();
 
-    for (doc_id, view_id) in docs_views {
-        let view = view_mut!(cx.editor, view_id);
+    for (doc_id, view_ids) in docs_views {
         let doc = doc_mut!(cx.editor, &doc_id);
 
+        let mut view_ids = view_ids.into_iter();
+
+        // Every doc is guaranteed to have at least 1 view at this point.
+        let view_id = view_ids.next().unwrap();
+        let view = view_mut!(cx.editor, view_id);
+
         doc.reload(view).map(|_| {
             view.ensure_cursor_in_view(doc, scrolloff);
         })?;
+
+        for view_id in view_ids.into_iter() {
+            let view = view_mut!(cx.editor, view_id);
+
+            view.ensure_cursor_in_view(doc, scrolloff);
+        }
     }
 
     Ok(())
 }
+
 /// Update the [`Document`] if it has been modified.
 fn update(
     cx: &mut compositor::Context,