From f52200a340649eac2e65895f01f83891e32891a4 Mon Sep 17 00:00:00 2001
From: Antonio Scandurra <me@as-cii.com>
Date: Mon, 2 Oct 2023 15:21:58 +0200
Subject: [PATCH] Prevent deploying the inline assistant when selection spans
 multiple excerpts

---
 crates/assistant/src/assistant_panel.rs | 40 ++++++++++++++-----------
 crates/assistant/src/prompts.rs         |  4 +--
 2 files changed, 24 insertions(+), 20 deletions(-)

diff --git a/crates/assistant/src/assistant_panel.rs b/crates/assistant/src/assistant_panel.rs
index 4a4dc08790..0d9f69011e 100644
--- a/crates/assistant/src/assistant_panel.rs
+++ b/crates/assistant/src/assistant_panel.rs
@@ -274,13 +274,17 @@ impl AssistantPanel {
             return;
         };
 
+        let selection = editor.read(cx).selections.newest_anchor().clone();
+        if selection.start.excerpt_id() != selection.end.excerpt_id() {
+            return;
+        }
+
         let inline_assist_id = post_inc(&mut self.next_inline_assist_id);
         let snapshot = editor.read(cx).buffer().read(cx).snapshot(cx);
         let provider = Arc::new(OpenAICompletionProvider::new(
             api_key,
             cx.background().clone(),
         ));
-        let selection = editor.read(cx).selections.newest_anchor().clone();
         let codegen_kind = if editor.read(cx).selections.newest::<usize>(cx).is_empty() {
             CodegenKind::Generate {
                 position: selection.start,
@@ -542,25 +546,25 @@ impl AssistantPanel {
             self.inline_prompt_history.pop_front();
         }
 
-        let multi_buffer = editor.read(cx).buffer().read(cx);
-        let multi_buffer_snapshot = multi_buffer.snapshot(cx);
-        let snapshot = if multi_buffer.is_singleton() {
-            multi_buffer.as_singleton().unwrap().read(cx).snapshot()
+        let snapshot = editor.read(cx).buffer().read(cx).snapshot(cx);
+        let range = pending_assist.codegen.read(cx).range();
+        let start = snapshot.point_to_buffer_offset(range.start);
+        let end = snapshot.point_to_buffer_offset(range.end);
+        let (buffer, range) = if let Some((start, end)) = start.zip(end) {
+            let (start_buffer, start_buffer_offset) = start;
+            let (end_buffer, end_buffer_offset) = end;
+            if start_buffer.remote_id() == end_buffer.remote_id() {
+                (start_buffer, start_buffer_offset..end_buffer_offset)
+            } else {
+                self.finish_inline_assist(inline_assist_id, false, cx);
+                return;
+            }
         } else {
+            self.finish_inline_assist(inline_assist_id, false, cx);
             return;
         };
 
-        let range = pending_assist.codegen.read(cx).range();
-        let language_range = snapshot.anchor_at(
-            range.start.to_offset(&multi_buffer_snapshot),
-            language::Bias::Left,
-        )
-            ..snapshot.anchor_at(
-                range.end.to_offset(&multi_buffer_snapshot),
-                language::Bias::Right,
-            );
-
-        let language = snapshot.language_at(language_range.start);
+        let language = buffer.language_at(range.start);
         let language_name = if let Some(language) = language.as_ref() {
             if Arc::ptr_eq(language, &language::PLAIN_TEXT) {
                 None
@@ -576,8 +580,8 @@ impl AssistantPanel {
         let prompt = generate_content_prompt(
             user_prompt.to_string(),
             language_name,
-            &snapshot,
-            language_range,
+            &buffer,
+            range,
             codegen_kind,
         );
 
diff --git a/crates/assistant/src/prompts.rs b/crates/assistant/src/prompts.rs
index 0646534e01..2451369a18 100644
--- a/crates/assistant/src/prompts.rs
+++ b/crates/assistant/src/prompts.rs
@@ -119,7 +119,7 @@ pub fn generate_content_prompt(
     user_prompt: String,
     language_name: Option<&str>,
     buffer: &BufferSnapshot,
-    range: Range<language::Anchor>,
+    range: Range<impl ToOffset>,
     kind: CodegenKind,
 ) -> String {
     let mut prompt = String::new();
@@ -131,7 +131,7 @@ pub fn generate_content_prompt(
         writeln!(prompt, "You're an expert engineer.\n").unwrap();
     }
 
-    let outline = summarize(buffer, range.clone());
+    let outline = summarize(buffer, range);
     writeln!(
         prompt,
         "The file you are currently working on has the following outline:"