Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve background highlighting of inline assists

Browse source
This commit is contained in:
Antonio Scandurra 2023-08-25 12:35:36 +02:00
parent 144f5c5d41
commit 971c833e80
1 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

30
crates/ai/src/assistant.rs
View file

@ -26,7 +26,7 @@ use gpui::{
Subscription, Task, View, ViewContext, ViewHandle, WeakViewHandle, WindowContext,
};
use language::{
language_settings::SoftWrap, Buffer, LanguageRegistry, Point, Rope, Selection, ToOffset as _,
language_settings::SoftWrap, Buffer, LanguageRegistry, Point, Rope, ToOffset as _,
TransactionId,
};
use search::BufferSearchBar;
@ -244,7 +244,9 @@ impl AssistantPanel {
fn new_inline_assist(&mut self, editor: &ViewHandle<Editor>, cx: &mut ViewContext<Self>) {
let id = post_inc(&mut self.next_inline_assist_id);
let snapshot = editor.read(cx).buffer().read(cx).snapshot(cx);
let selection = editor.read(cx).selections.newest_anchor().clone();
let range = selection.start.bias_left(&snapshot)..selection.end.bias_right(&snapshot);
let assist_kind = if editor.read(cx).selections.newest::<usize>(cx).is_empty() {
InlineAssistKind::Insert
} else {
@ -269,16 +271,14 @@ impl AssistantPanel {
});
let block_id = editor.update(cx, |editor, cx| {
editor.highlight_background::<Self>(
vec![selection.start..selection.end],
vec![range.clone()],
|theme| theme.assistant.inline.pending_edit_background,
cx,
);
editor.insert_blocks(
[BlockProperties {
style: BlockStyle::Flex,
position: selection
.head()
.bias_left(&editor.buffer().read(cx).snapshot(cx)),
position: selection.head().bias_left(&snapshot),
height: 2,
render: Arc::new({
let inline_assistant = inline_assistant.clone();
@ -308,7 +308,7 @@ impl AssistantPanel {
PendingInlineAssist {
kind: assist_kind,
editor: editor.downgrade(),
selection,
range,
inline_assistant_block_id: Some(block_id),
code_generation: Task::ready(None),
transaction_id: None,
@ -418,15 +418,15 @@ impl AssistantPanel {
return;
};
let selection = pending_assist.selection.clone();
let range = pending_assist.range.clone();
let snapshot = editor.read(cx).buffer().read(cx).snapshot(cx);
let selected_text = snapshot
.text_for_range(selection.start..selection.end)
.text_for_range(range.start..range.end)
.collect::<Rope>();
let mut base_indentation: Option<language::IndentSize> = None;
let selection_start = selection.start.to_point(&snapshot);
let selection_end = selection.end.to_point(&snapshot);
let selection_start = range.start.to_point(&snapshot);
let selection_end = range.end.to_point(&snapshot);
let mut start_row = selection_start.row;
if snapshot.is_line_blank(start_row) {
if let Some(prev_non_blank_row) = snapshot.prev_non_blank_row(start_row) {
@ -470,7 +470,7 @@ impl AssistantPanel {
}
let language_name = snapshot
.language_at(selection.start)
.language_at(range.start)
.map(|language| language.name());
let language_name = language_name.as_deref().unwrap_or("");
@ -494,11 +494,11 @@ impl AssistantPanel {
}
InlineAssistKind::Insert => {
writeln!(prompt, "```{language_name}").unwrap();
for chunk in snapshot.text_for_range(Anchor::min()..selection.head()) {
for chunk in snapshot.text_for_range(Anchor::min()..range.start) {
write!(prompt, "{chunk}").unwrap();
}
write!(prompt, "<|>").unwrap();
for chunk in snapshot.text_for_range(selection.head()..Anchor::max()) {
for chunk in snapshot.text_for_range(range.start..Anchor::max()) {
write!(prompt, "{chunk}").unwrap();
}
writeln!(prompt).unwrap();
@ -543,7 +543,7 @@ impl AssistantPanel {
}
});
let mut edit_start = selection.start.to_offset(&snapshot);
let mut edit_start = range.start.to_offset(&snapshot);
let (mut hunks_tx, mut hunks_rx) = mpsc::channel(1);
let diff = cx.background().spawn(async move {
@ -2697,7 +2697,7 @@ impl InlineAssistant {
struct PendingInlineAssist {
kind: InlineAssistKind,
editor: WeakViewHandle<Editor>,
selection: Selection<Anchor>,
range: Range<Anchor>,
inline_assistant_block_id: Option<BlockId>,
code_generation: Task<Option<()>>,
transaction_id: Option<TransactionId>,