Allow message splitting with multiple cursors (#2627)

Antonio Scandurra created 2 years ago

Closes
https://linear.app/zed-industries/issue/Z-2408/make-block-splitting-work-with-multiple-cursors

Change summary

crates/ai/src/assistant.rs | 12 +++++++++---
1 file changed, 9 insertions(+), 3 deletions(-)

Detailed changes

crates/ai/src/assistant.rs

@@ -8,7 +8,7 @@ use collections::{HashMap, HashSet};
 use editor::{
     display_map::{BlockDisposition, BlockId, BlockProperties, BlockStyle, ToDisplayPoint},
     scroll::autoscroll::{Autoscroll, AutoscrollStrategy},
-    Anchor, Editor,
+    Anchor, Editor, ToOffset,
 };
 use fs::Fs;
 use futures::{io::BufReader, AsyncBufReadExt, AsyncReadExt, Stream, StreamExt};
@@ -1294,8 +1294,14 @@ impl AssistantEditor {
 
     fn split(&mut self, _: &Split, cx: &mut ViewContext<Self>) {
         self.assistant.update(cx, |assistant, cx| {
-            let range = self.editor.read(cx).selections.newest::<usize>(cx).range();
-            assistant.split_message(range, cx);
+            let selections = self.editor.read(cx).selections.disjoint_anchors();
+            for selection in selections.into_iter() {
+                let buffer = self.editor.read(cx).buffer().read(cx).snapshot(cx);
+                let range = selection
+                    .map(|endpoint| endpoint.to_offset(&buffer))
+                    .range();
+                assistant.split_message(range, cx);
+            }
         });
     }