1972 lines
78 KiB
Rust
1972 lines
78 KiB
Rust
use std::path::Path;
|
||
use std::rc::Rc;
|
||
use std::sync::Arc;
|
||
use std::time::Duration;
|
||
|
||
use agentic_coding_protocol::{self as acp};
|
||
use collections::{HashMap, HashSet};
|
||
use editor::{
|
||
AnchorRangeExt, ContextMenuOptions, ContextMenuPlacement, Editor, EditorElement, EditorMode,
|
||
EditorStyle, MinimapVisibility, MultiBuffer,
|
||
};
|
||
use file_icons::FileIcons;
|
||
use futures::channel::oneshot;
|
||
use gpui::{
|
||
Animation, AnimationExt, App, BorderStyle, EdgesRefinement, Empty, Entity, EntityId, Focusable,
|
||
Hsla, Length, ListOffset, ListState, SharedString, StyleRefinement, Subscription, TextStyle,
|
||
TextStyleRefinement, Transformation, UnderlineStyle, WeakEntity, Window, div, list, percentage,
|
||
prelude::*, pulsating_between,
|
||
};
|
||
use gpui::{FocusHandle, Task};
|
||
use language::language_settings::SoftWrap;
|
||
use language::{Buffer, Language};
|
||
use markdown::{HeadingLevelStyles, Markdown, MarkdownElement, MarkdownStyle};
|
||
use parking_lot::Mutex;
|
||
use project::Project;
|
||
use settings::Settings as _;
|
||
use theme::ThemeSettings;
|
||
use ui::{Disclosure, Tooltip, prelude::*};
|
||
use util::ResultExt;
|
||
use workspace::Workspace;
|
||
use zed_actions::agent::{Chat, NextHistoryMessage, PreviousHistoryMessage};
|
||
|
||
use ::acp::{
|
||
AcpThread, AcpThreadEvent, AgentThreadEntry, AssistantMessage, AssistantMessageChunk, Diff,
|
||
LoadError, MentionPath, ThreadStatus, ToolCall, ToolCallConfirmation, ToolCallContent,
|
||
ToolCallId, ToolCallStatus,
|
||
};
|
||
|
||
use crate::acp::completion_provider::{ContextPickerCompletionProvider, MentionSet};
|
||
use crate::acp::message_history::MessageHistory;
|
||
|
||
const RESPONSE_PADDING_X: Pixels = px(19.);
|
||
|
||
pub struct AcpThreadView {
|
||
workspace: WeakEntity<Workspace>,
|
||
project: Entity<Project>,
|
||
thread_state: ThreadState,
|
||
diff_editors: HashMap<EntityId, Entity<Editor>>,
|
||
message_editor: Entity<Editor>,
|
||
mention_set: Arc<Mutex<MentionSet>>,
|
||
last_error: Option<Entity<Markdown>>,
|
||
list_state: ListState,
|
||
auth_task: Option<Task<()>>,
|
||
expanded_tool_calls: HashSet<ToolCallId>,
|
||
expanded_thinking_blocks: HashSet<(usize, usize)>,
|
||
message_history: MessageHistory<acp::UserMessage>,
|
||
}
|
||
|
||
enum ThreadState {
|
||
Loading {
|
||
_task: Task<()>,
|
||
},
|
||
Ready {
|
||
thread: Entity<AcpThread>,
|
||
_subscription: Subscription,
|
||
},
|
||
LoadError(LoadError),
|
||
Unauthenticated {
|
||
thread: Entity<AcpThread>,
|
||
},
|
||
}
|
||
|
||
impl AcpThreadView {
|
||
pub fn new(
|
||
workspace: WeakEntity<Workspace>,
|
||
project: Entity<Project>,
|
||
window: &mut Window,
|
||
cx: &mut Context<Self>,
|
||
) -> Self {
|
||
let language = Language::new(
|
||
language::LanguageConfig {
|
||
completion_query_characters: HashSet::from_iter(['.', '-', '_', '@']),
|
||
..Default::default()
|
||
},
|
||
None,
|
||
);
|
||
|
||
let mention_set = Arc::new(Mutex::new(MentionSet::default()));
|
||
|
||
let message_editor = cx.new(|cx| {
|
||
let buffer = cx.new(|cx| Buffer::local("", cx).with_language(Arc::new(language), cx));
|
||
let buffer = cx.new(|cx| MultiBuffer::singleton(buffer, cx));
|
||
|
||
let mut editor = Editor::new(
|
||
editor::EditorMode::AutoHeight {
|
||
min_lines: 4,
|
||
max_lines: None,
|
||
},
|
||
buffer,
|
||
None,
|
||
window,
|
||
cx,
|
||
);
|
||
editor.set_placeholder_text("Message the agent － @ to include files", cx);
|
||
editor.set_show_indent_guides(false, cx);
|
||
editor.set_soft_wrap();
|
||
editor.set_use_modal_editing(true);
|
||
editor.set_completion_provider(Some(Rc::new(ContextPickerCompletionProvider::new(
|
||
mention_set.clone(),
|
||
workspace.clone(),
|
||
cx.weak_entity(),
|
||
))));
|
||
editor.set_context_menu_options(ContextMenuOptions {
|
||
min_entries_visible: 12,
|
||
max_entries_visible: 12,
|
||
placement: Some(ContextMenuPlacement::Above),
|
||
});
|
||
editor
|
||
});
|
||
|
||
let list_state = ListState::new(
|
||
0,
|
||
gpui::ListAlignment::Bottom,
|
||
px(2048.0),
|
||
cx.processor({
|
||
move |this: &mut Self, index: usize, window, cx| {
|
||
let Some((entry, len)) = this.thread().and_then(|thread| {
|
||
let entries = &thread.read(cx).entries();
|
||
Some((entries.get(index)?, entries.len()))
|
||
}) else {
|
||
return Empty.into_any();
|
||
};
|
||
this.render_entry(index, len, entry, window, cx)
|
||
}
|
||
}),
|
||
);
|
||
|
||
Self {
|
||
workspace,
|
||
project: project.clone(),
|
||
thread_state: Self::initial_state(project, window, cx),
|
||
message_editor,
|
||
mention_set,
|
||
diff_editors: Default::default(),
|
||
list_state: list_state,
|
||
last_error: None,
|
||
auth_task: None,
|
||
expanded_tool_calls: HashSet::default(),
|
||
expanded_thinking_blocks: HashSet::default(),
|
||
message_history: MessageHistory::new(),
|
||
}
|
||
}
|
||
|
||
fn initial_state(
|
||
project: Entity<Project>,
|
||
window: &mut Window,
|
||
cx: &mut Context<Self>,
|
||
) -> ThreadState {
|
||
let root_dir = project
|
||
.read(cx)
|
||
.visible_worktrees(cx)
|
||
.next()
|
||
.map(|worktree| worktree.read(cx).abs_path())
|
||
.unwrap_or_else(|| paths::home_dir().as_path().into());
|
||
|
||
let load_task = cx.spawn_in(window, async move |this, cx| {
|
||
let thread = match AcpThread::spawn(agent_servers::Gemini, &root_dir, project, cx).await
|
||
{
|
||
Ok(thread) => thread,
|
||
Err(err) => {
|
||
this.update(cx, |this, cx| {
|
||
this.handle_load_error(err, cx);
|
||
cx.notify();
|
||
})
|
||
.log_err();
|
||
return;
|
||
}
|
||
};
|
||
|
||
let init_response = async {
|
||
let resp = thread
|
||
.read_with(cx, |thread, _cx| thread.initialize())?
|
||
.await?;
|
||
anyhow::Ok(resp)
|
||
};
|
||
|
||
let result = match init_response.await {
|
||
Err(e) => {
|
||
let mut cx = cx.clone();
|
||
if e.downcast_ref::<oneshot::Canceled>().is_some() {
|
||
let child_status = thread
|
||
.update(&mut cx, |thread, _| thread.child_status())
|
||
.ok()
|
||
.flatten();
|
||
if let Some(child_status) = child_status {
|
||
match child_status.await {
|
||
Ok(_) => Err(e),
|
||
Err(e) => Err(e),
|
||
}
|
||
} else {
|
||
Err(e)
|
||
}
|
||
} else {
|
||
Err(e)
|
||
}
|
||
}
|
||
Ok(response) => {
|
||
if !response.is_authenticated {
|
||
this.update(cx, |this, _| {
|
||
this.thread_state = ThreadState::Unauthenticated { thread };
|
||
})
|
||
.ok();
|
||
return;
|
||
};
|
||
Ok(())
|
||
}
|
||
};
|
||
|
||
this.update_in(cx, |this, window, cx| {
|
||
match result {
|
||
Ok(()) => {
|
||
let subscription =
|
||
cx.subscribe_in(&thread, window, Self::handle_thread_event);
|
||
this.list_state
|
||
.splice(0..0, thread.read(cx).entries().len());
|
||
|
||
this.thread_state = ThreadState::Ready {
|
||
thread,
|
||
_subscription: subscription,
|
||
};
|
||
cx.notify();
|
||
}
|
||
Err(err) => {
|
||
this.handle_load_error(err, cx);
|
||
}
|
||
};
|
||
})
|
||
.log_err();
|
||
});
|
||
|
||
ThreadState::Loading { _task: load_task }
|
||
}
|
||
|
||
fn handle_load_error(&mut self, err: anyhow::Error, cx: &mut Context<Self>) {
|
||
if let Some(load_err) = err.downcast_ref::<LoadError>() {
|
||
self.thread_state = ThreadState::LoadError(load_err.clone());
|
||
} else {
|
||
self.thread_state = ThreadState::LoadError(LoadError::Other(err.to_string().into()))
|
||
}
|
||
cx.notify();
|
||
}
|
||
|
||
fn thread(&self) -> Option<&Entity<AcpThread>> {
|
||
match &self.thread_state {
|
||
ThreadState::Ready { thread, .. } | ThreadState::Unauthenticated { thread } => {
|
||
Some(thread)
|
||
}
|
||
ThreadState::Loading { .. } | ThreadState::LoadError(..) => None,
|
||
}
|
||
}
|
||
|
||
pub fn title(&self, cx: &App) -> SharedString {
|
||
match &self.thread_state {
|
||
ThreadState::Ready { thread, .. } => thread.read(cx).title(),
|
||
ThreadState::Loading { .. } => "Loading…".into(),
|
||
ThreadState::LoadError(_) => "Failed to load".into(),
|
||
ThreadState::Unauthenticated { .. } => "Not authenticated".into(),
|
||
}
|
||
}
|
||
|
||
pub fn cancel(&mut self, cx: &mut Context<Self>) {
|
||
self.last_error.take();
|
||
|
||
if let Some(thread) = self.thread() {
|
||
thread.update(cx, |thread, cx| thread.cancel(cx)).detach();
|
||
}
|
||
}
|
||
|
||
fn chat(&mut self, _: &Chat, window: &mut Window, cx: &mut Context<Self>) {
|
||
self.last_error.take();
|
||
|
||
let mut ix = 0;
|
||
let mut chunks: Vec<acp::UserMessageChunk> = Vec::new();
|
||
|
||
let project = self.project.clone();
|
||
self.message_editor.update(cx, |editor, cx| {
|
||
let text = editor.text(cx);
|
||
editor.display_map.update(cx, |map, cx| {
|
||
let snapshot = map.snapshot(cx);
|
||
for (crease_id, crease) in snapshot.crease_snapshot.creases() {
|
||
if let Some(project_path) =
|
||
self.mention_set.lock().path_for_crease_id(crease_id)
|
||
{
|
||
let crease_range = crease.range().to_offset(&snapshot.buffer_snapshot);
|
||
if crease_range.start > ix {
|
||
chunks.push(acp::UserMessageChunk::Text {
|
||
chunk: text[ix..crease_range.start].to_string(),
|
||
});
|
||
}
|
||
if let Some(abs_path) = project.read(cx).absolute_path(&project_path, cx) {
|
||
chunks.push(acp::UserMessageChunk::Path { path: abs_path });
|
||
}
|
||
ix = crease_range.end;
|
||
}
|
||
}
|
||
|
||
if ix < text.len() {
|
||
let last_chunk = text[ix..].trim();
|
||
if !last_chunk.is_empty() {
|
||
chunks.push(last_chunk.into());
|
||
}
|
||
}
|
||
})
|
||
});
|
||
|
||
if chunks.is_empty() {
|
||
return;
|
||
}
|
||
|
||
let Some(thread) = self.thread() else { return };
|
||
let message = acp::UserMessage { chunks };
|
||
let task = thread.update(cx, |thread, cx| thread.send(message.clone(), cx));
|
||
|
||
cx.spawn(async move |this, cx| {
|
||
let result = task.await;
|
||
|
||
this.update(cx, |this, cx| {
|
||
if let Err(err) = result {
|
||
this.last_error =
|
||
Some(cx.new(|cx| Markdown::new(err.to_string().into(), None, None, cx)))
|
||
}
|
||
})
|
||
})
|
||
.detach();
|
||
|
||
let mention_set = self.mention_set.clone();
|
||
|
||
self.message_editor.update(cx, |editor, cx| {
|
||
editor.clear(window, cx);
|
||
editor.remove_creases(mention_set.lock().drain(), cx)
|
||
});
|
||
|
||
self.message_history.push(message);
|
||
}
|
||
|
||
fn previous_history_message(
|
||
&mut self,
|
||
_: &PreviousHistoryMessage,
|
||
window: &mut Window,
|
||
cx: &mut Context<Self>,
|
||
) {
|
||
Self::set_draft_message(
|
||
self.message_editor.clone(),
|
||
self.mention_set.clone(),
|
||
self.project.clone(),
|
||
self.message_history.prev(),
|
||
window,
|
||
cx,
|
||
);
|
||
}
|
||
|
||
fn next_history_message(
|
||
&mut self,
|
||
_: &NextHistoryMessage,
|
||
window: &mut Window,
|
||
cx: &mut Context<Self>,
|
||
) {
|
||
Self::set_draft_message(
|
||
self.message_editor.clone(),
|
||
self.mention_set.clone(),
|
||
self.project.clone(),
|
||
self.message_history.next(),
|
||
window,
|
||
cx,
|
||
);
|
||
}
|
||
|
||
fn set_draft_message(
|
||
message_editor: Entity<Editor>,
|
||
mention_set: Arc<Mutex<MentionSet>>,
|
||
project: Entity<Project>,
|
||
message: Option<&acp::UserMessage>,
|
||
window: &mut Window,
|
||
cx: &mut Context<Self>,
|
||
) {
|
||
cx.notify();
|
||
|
||
let Some(message) = message else {
|
||
message_editor.update(cx, |editor, cx| {
|
||
editor.clear(window, cx);
|
||
editor.remove_creases(mention_set.lock().drain(), cx)
|
||
});
|
||
return;
|
||
};
|
||
|
||
let mut text = String::new();
|
||
let mut mentions = Vec::new();
|
||
|
||
for chunk in &message.chunks {
|
||
match chunk {
|
||
acp::UserMessageChunk::Text { chunk } => {
|
||
text.push_str(&chunk);
|
||
}
|
||
acp::UserMessageChunk::Path { path } => {
|
||
let start = text.len();
|
||
let content = MentionPath::new(path).to_string();
|
||
text.push_str(&content);
|
||
let end = text.len();
|
||
if let Some(project_path) =
|
||
project.read(cx).project_path_for_absolute_path(path, cx)
|
||
{
|
||
let filename: SharedString = path
|
||
.file_name()
|
||
.unwrap_or_default()
|
||
.to_string_lossy()
|
||
.to_string()
|
||
.into();
|
||
mentions.push((start..end, project_path, filename));
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
let snapshot = message_editor.update(cx, |editor, cx| {
|
||
editor.set_text(text, window, cx);
|
||
editor.buffer().read(cx).snapshot(cx)
|
||
});
|
||
|
||
for (range, project_path, filename) in mentions {
|
||
let crease_icon_path = if project_path.path.is_dir() {
|
||
FileIcons::get_folder_icon(false, cx)
|
||
.unwrap_or_else(|| IconName::Folder.path().into())
|
||
} else {
|
||
FileIcons::get_icon(Path::new(project_path.path.as_ref()), cx)
|
||
.unwrap_or_else(|| IconName::File.path().into())
|
||
};
|
||
|
||
let anchor = snapshot.anchor_before(range.start);
|
||
let crease_id = crate::context_picker::insert_crease_for_mention(
|
||
anchor.excerpt_id,
|
||
anchor.text_anchor,
|
||
range.end - range.start,
|
||
filename,
|
||
crease_icon_path,
|
||
message_editor.clone(),
|
||
window,
|
||
cx,
|
||
);
|
||
if let Some(crease_id) = crease_id {
|
||
mention_set.lock().insert(crease_id, project_path);
|
||
}
|
||
}
|
||
}
|
||
|
||
fn handle_thread_event(
|
||
&mut self,
|
||
thread: &Entity<AcpThread>,
|
||
event: &AcpThreadEvent,
|
||
window: &mut Window,
|
||
cx: &mut Context<Self>,
|
||
) {
|
||
let count = self.list_state.item_count();
|
||
match event {
|
||
AcpThreadEvent::NewEntry => {
|
||
self.sync_thread_entry_view(thread.read(cx).entries().len() - 1, window, cx);
|
||
self.list_state.splice(count..count, 1);
|
||
}
|
||
AcpThreadEvent::EntryUpdated(index) => {
|
||
let index = *index;
|
||
self.sync_thread_entry_view(index, window, cx);
|
||
self.list_state.splice(index..index + 1, 1);
|
||
}
|
||
}
|
||
cx.notify();
|
||
}
|
||
|
||
fn sync_thread_entry_view(
|
||
&mut self,
|
||
entry_ix: usize,
|
||
window: &mut Window,
|
||
cx: &mut Context<Self>,
|
||
) {
|
||
let Some(multibuffer) = self.entry_diff_multibuffer(entry_ix, cx) else {
|
||
return;
|
||
};
|
||
|
||
if self.diff_editors.contains_key(&multibuffer.entity_id()) {
|
||
return;
|
||
}
|
||
|
||
let editor = cx.new(|cx| {
|
||
let mut editor = Editor::new(
|
||
EditorMode::Full {
|
||
scale_ui_elements_with_buffer_font_size: false,
|
||
show_active_line_background: false,
|
||
sized_by_content: true,
|
||
},
|
||
multibuffer.clone(),
|
||
None,
|
||
window,
|
||
cx,
|
||
);
|
||
editor.set_show_gutter(false, cx);
|
||
editor.disable_inline_diagnostics();
|
||
editor.disable_expand_excerpt_buttons(cx);
|
||
editor.set_show_vertical_scrollbar(false, cx);
|
||
editor.set_minimap_visibility(MinimapVisibility::Disabled, window, cx);
|
||
editor.set_soft_wrap_mode(SoftWrap::None, cx);
|
||
editor.scroll_manager.set_forbid_vertical_scroll(true);
|
||
editor.set_show_indent_guides(false, cx);
|
||
editor.set_read_only(true);
|
||
editor.set_show_breakpoints(false, cx);
|
||
editor.set_show_code_actions(false, cx);
|
||
editor.set_show_git_diff_gutter(false, cx);
|
||
editor.set_expand_all_diff_hunks(cx);
|
||
editor.set_text_style_refinement(TextStyleRefinement {
|
||
font_size: Some(
|
||
TextSize::Small
|
||
.rems(cx)
|
||
.to_pixels(ThemeSettings::get_global(cx).agent_font_size(cx))
|
||
.into(),
|
||
),
|
||
..Default::default()
|
||
});
|
||
editor
|
||
});
|
||
let entity_id = multibuffer.entity_id();
|
||
cx.observe_release(&multibuffer, move |this, _, _| {
|
||
this.diff_editors.remove(&entity_id);
|
||
})
|
||
.detach();
|
||
|
||
self.diff_editors.insert(entity_id, editor);
|
||
}
|
||
|
||
fn entry_diff_multibuffer(&self, entry_ix: usize, cx: &App) -> Option<Entity<MultiBuffer>> {
|
||
let entry = self.thread()?.read(cx).entries().get(entry_ix)?;
|
||
if let AgentThreadEntry::ToolCall(ToolCall {
|
||
content: Some(ToolCallContent::Diff { diff }),
|
||
..
|
||
}) = &entry
|
||
{
|
||
Some(diff.multibuffer.clone())
|
||
} else {
|
||
None
|
||
}
|
||
}
|
||
|
||
fn authenticate(&mut self, window: &mut Window, cx: &mut Context<Self>) {
|
||
let Some(thread) = self.thread().cloned() else {
|
||
return;
|
||
};
|
||
|
||
self.last_error.take();
|
||
let authenticate = thread.read(cx).authenticate();
|
||
self.auth_task = Some(cx.spawn_in(window, {
|
||
let project = self.project.clone();
|
||
async move |this, cx| {
|
||
let result = authenticate.await;
|
||
|
||
this.update_in(cx, |this, window, cx| {
|
||
if let Err(err) = result {
|
||
this.last_error = Some(cx.new(|cx| {
|
||
Markdown::new(format!("Error: {err}").into(), None, None, cx)
|
||
}))
|
||
} else {
|
||
this.thread_state = Self::initial_state(project.clone(), window, cx)
|
||
}
|
||
this.auth_task.take()
|
||
})
|
||
.ok();
|
||
}
|
||
}));
|
||
}
|
||
|
||
fn authorize_tool_call(
|
||
&mut self,
|
||
id: ToolCallId,
|
||
outcome: acp::ToolCallConfirmationOutcome,
|
||
cx: &mut Context<Self>,
|
||
) {
|
||
let Some(thread) = self.thread() else {
|
||
return;
|
||
};
|
||
thread.update(cx, |thread, cx| {
|
||
thread.authorize_tool_call(id, outcome, cx);
|
||
});
|
||
cx.notify();
|
||
}
|
||
|
||
fn render_entry(
|
||
&self,
|
||
index: usize,
|
||
total_entries: usize,
|
||
entry: &AgentThreadEntry,
|
||
window: &mut Window,
|
||
cx: &Context<Self>,
|
||
) -> AnyElement {
|
||
match &entry {
|
||
AgentThreadEntry::UserMessage(message) => div()
|
||
.py_4()
|
||
.px_2()
|
||
.child(
|
||
v_flex()
|
||
.p_3()
|
||
.gap_1p5()
|
||
.rounded_lg()
|
||
.shadow_md()
|
||
.bg(cx.theme().colors().editor_background)
|
||
.border_1()
|
||
.border_color(cx.theme().colors().border)
|
||
.text_xs()
|
||
.child(self.render_markdown(
|
||
message.content.clone(),
|
||
user_message_markdown_style(window, cx),
|
||
)),
|
||
)
|
||
.into_any(),
|
||
AgentThreadEntry::AssistantMessage(AssistantMessage { chunks }) => {
|
||
let style = default_markdown_style(false, window, cx);
|
||
let message_body = v_flex()
|
||
.w_full()
|
||
.gap_2p5()
|
||
.children(chunks.iter().enumerate().map(|(chunk_ix, chunk)| {
|
||
match chunk {
|
||
AssistantMessageChunk::Text { chunk } => self
|
||
.render_markdown(chunk.clone(), style.clone())
|
||
.into_any_element(),
|
||
AssistantMessageChunk::Thought { chunk } => self.render_thinking_block(
|
||
index,
|
||
chunk_ix,
|
||
chunk.clone(),
|
||
window,
|
||
cx,
|
||
),
|
||
}
|
||
}))
|
||
.into_any();
|
||
|
||
v_flex()
|
||
.px_5()
|
||
.py_1()
|
||
.when(index + 1 == total_entries, |this| this.pb_4())
|
||
.w_full()
|
||
.text_ui(cx)
|
||
.child(message_body)
|
||
.into_any()
|
||
}
|
||
AgentThreadEntry::ToolCall(tool_call) => div()
|
||
.py_1p5()
|
||
.px_5()
|
||
.child(self.render_tool_call(index, tool_call, window, cx))
|
||
.into_any(),
|
||
}
|
||
}
|
||
|
||
fn tool_card_header_bg(&self, cx: &Context<Self>) -> Hsla {
|
||
cx.theme()
|
||
.colors()
|
||
.element_background
|
||
.blend(cx.theme().colors().editor_foreground.opacity(0.025))
|
||
}
|
||
|
||
fn tool_card_border_color(&self, cx: &Context<Self>) -> Hsla {
|
||
cx.theme().colors().border.opacity(0.6)
|
||
}
|
||
|
||
fn tool_name_font_size(&self) -> Rems {
|
||
rems_from_px(13.)
|
||
}
|
||
|
||
fn render_thinking_block(
|
||
&self,
|
||
entry_ix: usize,
|
||
chunk_ix: usize,
|
||
chunk: Entity<Markdown>,
|
||
window: &Window,
|
||
cx: &Context<Self>,
|
||
) -> AnyElement {
|
||
let header_id = SharedString::from(format!("thinking-block-header-{}", entry_ix));
|
||
let key = (entry_ix, chunk_ix);
|
||
let is_open = self.expanded_thinking_blocks.contains(&key);
|
||
|
||
v_flex()
|
||
.child(
|
||
h_flex()
|
||
.id(header_id)
|
||
.group("disclosure-header")
|
||
.w_full()
|
||
.justify_between()
|
||
.opacity(0.8)
|
||
.hover(|style| style.opacity(1.))
|
||
.child(
|
||
h_flex()
|
||
.gap_1p5()
|
||
.child(
|
||
Icon::new(IconName::ToolBulb)
|
||
.size(IconSize::Small)
|
||
.color(Color::Muted),
|
||
)
|
||
.child(
|
||
div()
|
||
.text_size(self.tool_name_font_size())
|
||
.child("Thinking"),
|
||
),
|
||
)
|
||
.child(
|
||
div().visible_on_hover("disclosure-header").child(
|
||
Disclosure::new("thinking-disclosure", is_open)
|
||
.opened_icon(IconName::ChevronUp)
|
||
.closed_icon(IconName::ChevronDown)
|
||
.on_click(cx.listener({
|
||
move |this, _event, _window, cx| {
|
||
if is_open {
|
||
this.expanded_thinking_blocks.remove(&key);
|
||
} else {
|
||
this.expanded_thinking_blocks.insert(key);
|
||
}
|
||
cx.notify();
|
||
}
|
||
})),
|
||
),
|
||
)
|
||
.on_click(cx.listener({
|
||
move |this, _event, _window, cx| {
|
||
if is_open {
|
||
this.expanded_thinking_blocks.remove(&key);
|
||
} else {
|
||
this.expanded_thinking_blocks.insert(key);
|
||
}
|
||
cx.notify();
|
||
}
|
||
})),
|
||
)
|
||
.when(is_open, |this| {
|
||
this.child(
|
||
div()
|
||
.relative()
|
||
.mt_1p5()
|
||
.ml(px(7.))
|
||
.pl_4()
|
||
.border_l_1()
|
||
.border_color(self.tool_card_border_color(cx))
|
||
.text_ui_sm(cx)
|
||
.child(
|
||
self.render_markdown(chunk, default_markdown_style(false, window, cx)),
|
||
),
|
||
)
|
||
})
|
||
.into_any_element()
|
||
}
|
||
|
||
fn render_tool_call(
|
||
&self,
|
||
entry_ix: usize,
|
||
tool_call: &ToolCall,
|
||
window: &Window,
|
||
cx: &Context<Self>,
|
||
) -> Div {
|
||
let header_id = SharedString::from(format!("tool-call-header-{}", entry_ix));
|
||
|
||
let status_icon = match &tool_call.status {
|
||
ToolCallStatus::WaitingForConfirmation { .. } => None,
|
||
ToolCallStatus::Allowed {
|
||
status: acp::ToolCallStatus::Running,
|
||
..
|
||
} => Some(
|
||
Icon::new(IconName::ArrowCircle)
|
||
.color(Color::Accent)
|
||
.size(IconSize::Small)
|
||
.with_animation(
|
||
"running",
|
||
Animation::new(Duration::from_secs(2)).repeat(),
|
||
|icon, delta| icon.transform(Transformation::rotate(percentage(delta))),
|
||
)
|
||
.into_any(),
|
||
),
|
||
ToolCallStatus::Allowed {
|
||
status: acp::ToolCallStatus::Finished,
|
||
..
|
||
} => None,
|
||
ToolCallStatus::Rejected
|
||
| ToolCallStatus::Canceled
|
||
| ToolCallStatus::Allowed {
|
||
status: acp::ToolCallStatus::Error,
|
||
..
|
||
} => Some(
|
||
Icon::new(IconName::X)
|
||
.color(Color::Error)
|
||
.size(IconSize::Small)
|
||
.into_any_element(),
|
||
),
|
||
};
|
||
|
||
let needs_confirmation = match &tool_call.status {
|
||
ToolCallStatus::WaitingForConfirmation { .. } => true,
|
||
_ => tool_call
|
||
.content
|
||
.iter()
|
||
.any(|content| matches!(content, ToolCallContent::Diff { .. })),
|
||
};
|
||
|
||
let is_collapsible = tool_call.content.is_some() && !needs_confirmation;
|
||
let is_open = !is_collapsible || self.expanded_tool_calls.contains(&tool_call.id);
|
||
|
||
let content = if is_open {
|
||
match &tool_call.status {
|
||
ToolCallStatus::WaitingForConfirmation { confirmation, .. } => {
|
||
Some(self.render_tool_call_confirmation(
|
||
tool_call.id,
|
||
confirmation,
|
||
tool_call.content.as_ref(),
|
||
window,
|
||
cx,
|
||
))
|
||
}
|
||
ToolCallStatus::Allowed { .. } | ToolCallStatus::Canceled => {
|
||
tool_call.content.as_ref().map(|content| {
|
||
div()
|
||
.py_1p5()
|
||
.child(self.render_tool_call_content(content, window, cx))
|
||
.into_any_element()
|
||
})
|
||
}
|
||
ToolCallStatus::Rejected => None,
|
||
}
|
||
} else {
|
||
None
|
||
};
|
||
|
||
v_flex()
|
||
.when(needs_confirmation, |this| {
|
||
this.rounded_lg()
|
||
.border_1()
|
||
.border_color(self.tool_card_border_color(cx))
|
||
.bg(cx.theme().colors().editor_background)
|
||
.overflow_hidden()
|
||
})
|
||
.child(
|
||
h_flex()
|
||
.id(header_id)
|
||
.w_full()
|
||
.gap_1()
|
||
.justify_between()
|
||
.map(|this| {
|
||
if needs_confirmation {
|
||
this.px_2()
|
||
.py_1()
|
||
.rounded_t_md()
|
||
.bg(self.tool_card_header_bg(cx))
|
||
.border_b_1()
|
||
.border_color(self.tool_card_border_color(cx))
|
||
} else {
|
||
this.opacity(0.8).hover(|style| style.opacity(1.))
|
||
}
|
||
})
|
||
.child(
|
||
h_flex()
|
||
.id("tool-call-header")
|
||
.overflow_x_scroll()
|
||
.map(|this| {
|
||
if needs_confirmation {
|
||
this.text_xs()
|
||
} else {
|
||
this.text_size(self.tool_name_font_size())
|
||
}
|
||
})
|
||
.gap_1p5()
|
||
.child(
|
||
Icon::new(tool_call.icon)
|
||
.size(IconSize::Small)
|
||
.color(Color::Muted),
|
||
)
|
||
.child(self.render_markdown(
|
||
tool_call.label.clone(),
|
||
default_markdown_style(needs_confirmation, window, cx),
|
||
)),
|
||
)
|
||
.child(
|
||
h_flex()
|
||
.gap_0p5()
|
||
.when(is_collapsible, |this| {
|
||
this.child(
|
||
Disclosure::new(("expand", tool_call.id.0), is_open)
|
||
.opened_icon(IconName::ChevronUp)
|
||
.closed_icon(IconName::ChevronDown)
|
||
.on_click(cx.listener({
|
||
let id = tool_call.id;
|
||
move |this: &mut Self, _, _, cx: &mut Context<Self>| {
|
||
if is_open {
|
||
this.expanded_tool_calls.remove(&id);
|
||
} else {
|
||
this.expanded_tool_calls.insert(id);
|
||
}
|
||
cx.notify();
|
||
}
|
||
})),
|
||
)
|
||
})
|
||
.children(status_icon),
|
||
)
|
||
.on_click(cx.listener({
|
||
let id = tool_call.id;
|
||
move |this: &mut Self, _, _, cx: &mut Context<Self>| {
|
||
if is_open {
|
||
this.expanded_tool_calls.remove(&id);
|
||
} else {
|
||
this.expanded_tool_calls.insert(id);
|
||
}
|
||
cx.notify();
|
||
}
|
||
})),
|
||
)
|
||
.when(is_open, |this| {
|
||
this.child(
|
||
div()
|
||
.text_xs()
|
||
.when(is_collapsible, |this| {
|
||
this.mt_1()
|
||
.border_1()
|
||
.border_color(self.tool_card_border_color(cx))
|
||
.bg(cx.theme().colors().editor_background)
|
||
.rounded_lg()
|
||
})
|
||
.children(content),
|
||
)
|
||
})
|
||
}
|
||
|
||
fn render_tool_call_content(
|
||
&self,
|
||
content: &ToolCallContent,
|
||
window: &Window,
|
||
cx: &Context<Self>,
|
||
) -> AnyElement {
|
||
match content {
|
||
ToolCallContent::Markdown { markdown } => self
|
||
.render_markdown(markdown.clone(), default_markdown_style(false, window, cx))
|
||
.into_any_element(),
|
||
ToolCallContent::Diff {
|
||
diff: Diff {
|
||
path, multibuffer, ..
|
||
},
|
||
..
|
||
} => self.render_diff_editor(multibuffer, path),
|
||
}
|
||
}
|
||
|
||
fn render_tool_call_confirmation(
|
||
&self,
|
||
tool_call_id: ToolCallId,
|
||
confirmation: &ToolCallConfirmation,
|
||
content: Option<&ToolCallContent>,
|
||
window: &Window,
|
||
cx: &Context<Self>,
|
||
) -> AnyElement {
|
||
let confirmation_container = v_flex().mt_1().py_1p5();
|
||
|
||
let button_container = h_flex()
|
||
.pt_1p5()
|
||
.px_1p5()
|
||
.gap_1()
|
||
.justify_end()
|
||
.border_t_1()
|
||
.border_color(self.tool_card_border_color(cx));
|
||
|
||
match confirmation {
|
||
ToolCallConfirmation::Edit { description } => confirmation_container
|
||
.child(
|
||
div()
|
||
.px_2()
|
||
.children(description.clone().map(|description| {
|
||
self.render_markdown(
|
||
description,
|
||
default_markdown_style(false, window, cx),
|
||
)
|
||
})),
|
||
)
|
||
.children(content.map(|content| self.render_tool_call_content(content, window, cx)))
|
||
.child(
|
||
button_container
|
||
.child(
|
||
Button::new(("always_allow", tool_call_id.0), "Always Allow Edits")
|
||
.icon(IconName::CheckDouble)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::AlwaysAllow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("allow", tool_call_id.0), "Allow")
|
||
.icon(IconName::Check)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Allow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("reject", tool_call_id.0), "Reject")
|
||
.icon(IconName::X)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Error)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Reject,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
),
|
||
)
|
||
.into_any(),
|
||
ToolCallConfirmation::Execute {
|
||
command,
|
||
root_command,
|
||
description,
|
||
} => confirmation_container
|
||
.child(v_flex().px_2().pb_1p5().child(command.clone()).children(
|
||
description.clone().map(|description| {
|
||
self.render_markdown(description, default_markdown_style(false, window, cx))
|
||
.on_url_click({
|
||
let workspace = self.workspace.clone();
|
||
move |text, window, cx| {
|
||
Self::open_link(text, &workspace, window, cx);
|
||
}
|
||
})
|
||
}),
|
||
))
|
||
.children(content.map(|content| self.render_tool_call_content(content, window, cx)))
|
||
.child(
|
||
button_container
|
||
.child(
|
||
Button::new(
|
||
("always_allow", tool_call_id.0),
|
||
format!("Always Allow {root_command}"),
|
||
)
|
||
.icon(IconName::CheckDouble)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::AlwaysAllow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("allow", tool_call_id.0), "Allow")
|
||
.icon(IconName::Check)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Allow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("reject", tool_call_id.0), "Reject")
|
||
.icon(IconName::X)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Error)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Reject,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
),
|
||
)
|
||
.into_any(),
|
||
ToolCallConfirmation::Mcp {
|
||
server_name,
|
||
tool_name: _,
|
||
tool_display_name,
|
||
description,
|
||
} => confirmation_container
|
||
.child(
|
||
v_flex()
|
||
.px_2()
|
||
.pb_1p5()
|
||
.child(format!("{server_name} - {tool_display_name}"))
|
||
.children(description.clone().map(|description| {
|
||
self.render_markdown(
|
||
description,
|
||
default_markdown_style(false, window, cx),
|
||
)
|
||
})),
|
||
)
|
||
.children(content.map(|content| self.render_tool_call_content(content, window, cx)))
|
||
.child(
|
||
button_container
|
||
.child(
|
||
Button::new(
|
||
("always_allow_server", tool_call_id.0),
|
||
format!("Always Allow {server_name}"),
|
||
)
|
||
.icon(IconName::CheckDouble)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::AlwaysAllowMcpServer,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(
|
||
("always_allow_tool", tool_call_id.0),
|
||
format!("Always Allow {tool_display_name}"),
|
||
)
|
||
.icon(IconName::CheckDouble)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::AlwaysAllowTool,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("allow", tool_call_id.0), "Allow")
|
||
.icon(IconName::Check)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Allow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("reject", tool_call_id.0), "Reject")
|
||
.icon(IconName::X)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Error)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Reject,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
),
|
||
)
|
||
.into_any(),
|
||
ToolCallConfirmation::Fetch { description, urls } => confirmation_container
|
||
.child(
|
||
v_flex()
|
||
.px_2()
|
||
.pb_1p5()
|
||
.gap_1()
|
||
.children(urls.iter().map(|url| {
|
||
h_flex().child(
|
||
Button::new(url.clone(), url)
|
||
.icon(IconName::ArrowUpRight)
|
||
.icon_color(Color::Muted)
|
||
.icon_size(IconSize::XSmall)
|
||
.on_click({
|
||
let url = url.clone();
|
||
move |_, _, cx| cx.open_url(&url)
|
||
}),
|
||
)
|
||
}))
|
||
.children(description.clone().map(|description| {
|
||
self.render_markdown(
|
||
description,
|
||
default_markdown_style(false, window, cx),
|
||
)
|
||
})),
|
||
)
|
||
.children(content.map(|content| self.render_tool_call_content(content, window, cx)))
|
||
.child(
|
||
button_container
|
||
.child(
|
||
Button::new(("always_allow", tool_call_id.0), "Always Allow")
|
||
.icon(IconName::CheckDouble)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::AlwaysAllow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("allow", tool_call_id.0), "Allow")
|
||
.icon(IconName::Check)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Allow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("reject", tool_call_id.0), "Reject")
|
||
.icon(IconName::X)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Error)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Reject,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
),
|
||
)
|
||
.into_any(),
|
||
ToolCallConfirmation::Other { description } => confirmation_container
|
||
.child(v_flex().px_2().pb_1p5().child(self.render_markdown(
|
||
description.clone(),
|
||
default_markdown_style(false, window, cx),
|
||
)))
|
||
.children(content.map(|content| self.render_tool_call_content(content, window, cx)))
|
||
.child(
|
||
button_container
|
||
.child(
|
||
Button::new(("always_allow", tool_call_id.0), "Always Allow")
|
||
.icon(IconName::CheckDouble)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::AlwaysAllow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("allow", tool_call_id.0), "Allow")
|
||
.icon(IconName::Check)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Success)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Allow,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
)
|
||
.child(
|
||
Button::new(("reject", tool_call_id.0), "Reject")
|
||
.icon(IconName::X)
|
||
.icon_position(IconPosition::Start)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Error)
|
||
.label_size(LabelSize::Small)
|
||
.on_click(cx.listener({
|
||
let id = tool_call_id;
|
||
move |this, _, _, cx| {
|
||
this.authorize_tool_call(
|
||
id,
|
||
acp::ToolCallConfirmationOutcome::Reject,
|
||
cx,
|
||
);
|
||
}
|
||
})),
|
||
),
|
||
)
|
||
.into_any(),
|
||
}
|
||
}
|
||
|
||
fn render_diff_editor(&self, multibuffer: &Entity<MultiBuffer>, path: &Path) -> AnyElement {
|
||
v_flex()
|
||
.h_full()
|
||
.child(path.to_string_lossy().to_string())
|
||
.child(
|
||
if let Some(editor) = self.diff_editors.get(&multibuffer.entity_id()) {
|
||
editor.clone().into_any_element()
|
||
} else {
|
||
Empty.into_any()
|
||
},
|
||
)
|
||
.into_any()
|
||
}
|
||
|
||
fn render_gemini_logo(&self) -> AnyElement {
|
||
Icon::new(IconName::AiGemini)
|
||
.color(Color::Muted)
|
||
.size(IconSize::XLarge)
|
||
.into_any_element()
|
||
}
|
||
|
||
fn render_error_gemini_logo(&self) -> AnyElement {
|
||
let logo = Icon::new(IconName::AiGemini)
|
||
.color(Color::Muted)
|
||
.size(IconSize::XLarge)
|
||
.into_any_element();
|
||
|
||
h_flex()
|
||
.relative()
|
||
.justify_center()
|
||
.child(div().opacity(0.3).child(logo))
|
||
.child(
|
||
h_flex().absolute().right_1().bottom_0().child(
|
||
Icon::new(IconName::XCircle)
|
||
.color(Color::Error)
|
||
.size(IconSize::Small),
|
||
),
|
||
)
|
||
.into_any_element()
|
||
}
|
||
|
||
fn render_empty_state(&self, loading: bool, cx: &App) -> AnyElement {
|
||
v_flex()
|
||
.size_full()
|
||
.items_center()
|
||
.justify_center()
|
||
.child(
|
||
if loading {
|
||
h_flex()
|
||
.justify_center()
|
||
.child(self.render_gemini_logo())
|
||
.with_animation(
|
||
"pulsating_icon",
|
||
Animation::new(Duration::from_secs(2))
|
||
.repeat()
|
||
.with_easing(pulsating_between(0.4, 1.0)),
|
||
|icon, delta| icon.opacity(delta),
|
||
).into_any()
|
||
} else {
|
||
self.render_gemini_logo().into_any_element()
|
||
}
|
||
)
|
||
.child(
|
||
h_flex()
|
||
.mt_4()
|
||
.mb_1()
|
||
.justify_center()
|
||
.child(Headline::new(if loading {
|
||
"Connecting to Gemini…"
|
||
} else {
|
||
"Welcome to Gemini"
|
||
}).size(HeadlineSize::Medium)),
|
||
)
|
||
.child(
|
||
div()
|
||
.max_w_1_2()
|
||
.text_sm()
|
||
.text_center()
|
||
.map(|this| if loading {
|
||
this.invisible()
|
||
} else {
|
||
this.text_color(cx.theme().colors().text_muted)
|
||
})
|
||
.child("Ask questions, edit files, run commands.\nBe specific for the best results.")
|
||
)
|
||
.into_any()
|
||
}
|
||
|
||
fn render_pending_auth_state(&self) -> AnyElement {
|
||
v_flex()
|
||
.items_center()
|
||
.justify_center()
|
||
.child(self.render_error_gemini_logo())
|
||
.child(
|
||
h_flex()
|
||
.mt_4()
|
||
.mb_1()
|
||
.justify_center()
|
||
.child(Headline::new("Not Authenticated").size(HeadlineSize::Medium)),
|
||
)
|
||
.into_any()
|
||
}
|
||
|
||
fn render_error_state(&self, e: &LoadError, cx: &Context<Self>) -> AnyElement {
|
||
let mut container = v_flex()
|
||
.items_center()
|
||
.justify_center()
|
||
.child(self.render_error_gemini_logo())
|
||
.child(
|
||
v_flex()
|
||
.mt_4()
|
||
.mb_2()
|
||
.gap_0p5()
|
||
.text_center()
|
||
.items_center()
|
||
.child(Headline::new("Failed to launch").size(HeadlineSize::Medium))
|
||
.child(
|
||
Label::new(e.to_string())
|
||
.size(LabelSize::Small)
|
||
.color(Color::Muted),
|
||
),
|
||
);
|
||
|
||
if matches!(e, LoadError::Unsupported { .. }) {
|
||
container =
|
||
container.child(Button::new("upgrade", "Upgrade Gemini to Latest").on_click(
|
||
cx.listener(|this, _, window, cx| {
|
||
this.workspace
|
||
.update(cx, |workspace, cx| {
|
||
let project = workspace.project().read(cx);
|
||
let cwd = project.first_project_directory(cx);
|
||
let shell = project.terminal_settings(&cwd, cx).shell.clone();
|
||
let command =
|
||
"npm install -g @google/gemini-cli@latest".to_string();
|
||
let spawn_in_terminal = task::SpawnInTerminal {
|
||
id: task::TaskId("install".to_string()),
|
||
full_label: command.clone(),
|
||
label: command.clone(),
|
||
command: Some(command.clone()),
|
||
args: Vec::new(),
|
||
command_label: command.clone(),
|
||
cwd,
|
||
env: Default::default(),
|
||
use_new_terminal: true,
|
||
allow_concurrent_runs: true,
|
||
reveal: Default::default(),
|
||
reveal_target: Default::default(),
|
||
hide: Default::default(),
|
||
shell,
|
||
show_summary: true,
|
||
show_command: true,
|
||
show_rerun: false,
|
||
};
|
||
workspace
|
||
.spawn_in_terminal(spawn_in_terminal, window, cx)
|
||
.detach();
|
||
})
|
||
.ok();
|
||
}),
|
||
));
|
||
}
|
||
|
||
container.into_any()
|
||
}
|
||
|
||
fn render_message_editor(&mut self, cx: &mut Context<Self>) -> AnyElement {
|
||
let settings = ThemeSettings::get_global(cx);
|
||
let font_size = TextSize::Small
|
||
.rems(cx)
|
||
.to_pixels(settings.agent_font_size(cx));
|
||
let line_height = settings.buffer_line_height.value() * font_size;
|
||
|
||
let text_style = TextStyle {
|
||
color: cx.theme().colors().text,
|
||
font_family: settings.buffer_font.family.clone(),
|
||
font_fallbacks: settings.buffer_font.fallbacks.clone(),
|
||
font_features: settings.buffer_font.features.clone(),
|
||
font_size: font_size.into(),
|
||
line_height: line_height.into(),
|
||
..Default::default()
|
||
};
|
||
|
||
EditorElement::new(
|
||
&self.message_editor,
|
||
EditorStyle {
|
||
background: cx.theme().colors().editor_background,
|
||
local_player: cx.theme().players().local(),
|
||
text: text_style,
|
||
syntax: cx.theme().syntax().clone(),
|
||
..Default::default()
|
||
},
|
||
)
|
||
.into_any()
|
||
}
|
||
|
||
fn render_markdown(&self, markdown: Entity<Markdown>, style: MarkdownStyle) -> MarkdownElement {
|
||
let workspace = self.workspace.clone();
|
||
MarkdownElement::new(markdown, style).on_url_click(move |text, window, cx| {
|
||
Self::open_link(text, &workspace, window, cx);
|
||
})
|
||
}
|
||
|
||
fn open_link(
|
||
url: SharedString,
|
||
workspace: &WeakEntity<Workspace>,
|
||
window: &mut Window,
|
||
cx: &mut App,
|
||
) {
|
||
let Some(workspace) = workspace.upgrade() else {
|
||
cx.open_url(&url);
|
||
return;
|
||
};
|
||
|
||
if let Some(mention_path) = MentionPath::try_parse(&url) {
|
||
workspace.update(cx, |workspace, cx| {
|
||
let project = workspace.project();
|
||
let Some((path, entry)) = project.update(cx, |project, cx| {
|
||
let path = project.find_project_path(mention_path.path(), cx)?;
|
||
let entry = project.entry_for_path(&path, cx)?;
|
||
Some((path, entry))
|
||
}) else {
|
||
return;
|
||
};
|
||
|
||
if entry.is_dir() {
|
||
project.update(cx, |_, cx| {
|
||
cx.emit(project::Event::RevealInProjectPanel(entry.id));
|
||
});
|
||
} else {
|
||
workspace
|
||
.open_path(path, None, true, window, cx)
|
||
.detach_and_log_err(cx);
|
||
}
|
||
})
|
||
} else {
|
||
cx.open_url(&url);
|
||
}
|
||
}
|
||
|
||
pub fn open_thread_as_markdown(
|
||
&self,
|
||
workspace: Entity<Workspace>,
|
||
window: &mut Window,
|
||
cx: &mut App,
|
||
) -> Task<anyhow::Result<()>> {
|
||
let markdown_language_task = workspace
|
||
.read(cx)
|
||
.app_state()
|
||
.languages
|
||
.language_for_name("Markdown");
|
||
|
||
let (thread_summary, markdown) = match &self.thread_state {
|
||
ThreadState::Ready { thread, .. } | ThreadState::Unauthenticated { thread } => {
|
||
let thread = thread.read(cx);
|
||
(thread.title().to_string(), thread.to_markdown(cx))
|
||
}
|
||
ThreadState::Loading { .. } | ThreadState::LoadError(..) => return Task::ready(Ok(())),
|
||
};
|
||
|
||
window.spawn(cx, async move |cx| {
|
||
let markdown_language = markdown_language_task.await?;
|
||
|
||
workspace.update_in(cx, |workspace, window, cx| {
|
||
let project = workspace.project().clone();
|
||
|
||
if !project.read(cx).is_local() {
|
||
anyhow::bail!("failed to open active thread as markdown in remote project");
|
||
}
|
||
|
||
let buffer = project.update(cx, |project, cx| {
|
||
project.create_local_buffer(&markdown, Some(markdown_language), cx)
|
||
});
|
||
let buffer = cx.new(|cx| {
|
||
MultiBuffer::singleton(buffer, cx).with_title(thread_summary.clone())
|
||
});
|
||
|
||
workspace.add_item_to_active_pane(
|
||
Box::new(cx.new(|cx| {
|
||
let mut editor =
|
||
Editor::for_multibuffer(buffer, Some(project.clone()), window, cx);
|
||
editor.set_breadcrumb_header(thread_summary);
|
||
editor
|
||
})),
|
||
None,
|
||
true,
|
||
window,
|
||
cx,
|
||
);
|
||
|
||
anyhow::Ok(())
|
||
})??;
|
||
anyhow::Ok(())
|
||
})
|
||
}
|
||
|
||
fn scroll_to_top(&mut self, cx: &mut Context<Self>) {
|
||
self.list_state.scroll_to(ListOffset::default());
|
||
cx.notify();
|
||
}
|
||
}
|
||
|
||
impl Focusable for AcpThreadView {
|
||
fn focus_handle(&self, cx: &App) -> FocusHandle {
|
||
self.message_editor.focus_handle(cx)
|
||
}
|
||
}
|
||
|
||
impl Render for AcpThreadView {
|
||
fn render(&mut self, window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
|
||
let text = self.message_editor.read(cx).text(cx);
|
||
let is_editor_empty = text.is_empty();
|
||
let focus_handle = self.message_editor.focus_handle(cx);
|
||
|
||
let open_as_markdown = IconButton::new("open-as-markdown", IconName::DocumentText)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Ignored)
|
||
.tooltip(Tooltip::text("Open Thread as Markdown"))
|
||
.on_click(cx.listener(move |this, _, window, cx| {
|
||
if let Some(workspace) = this.workspace.upgrade() {
|
||
this.open_thread_as_markdown(workspace, window, cx)
|
||
.detach_and_log_err(cx);
|
||
}
|
||
}));
|
||
|
||
let scroll_to_top = IconButton::new("scroll_to_top", IconName::ArrowUpAlt)
|
||
.icon_size(IconSize::XSmall)
|
||
.icon_color(Color::Ignored)
|
||
.tooltip(Tooltip::text("Scroll To Top"))
|
||
.on_click(cx.listener(move |this, _, _, cx| {
|
||
this.scroll_to_top(cx);
|
||
}));
|
||
|
||
let feedback_container = h_flex()
|
||
.group("feedback_container")
|
||
.mt_1()
|
||
.py_2()
|
||
.px(RESPONSE_PADDING_X)
|
||
.mr_1()
|
||
.opacity(0.4)
|
||
.hover(|style| style.opacity(1.))
|
||
.gap_1p5()
|
||
.flex_wrap()
|
||
.justify_end()
|
||
.child(h_flex().child(open_as_markdown))
|
||
.child(scroll_to_top)
|
||
.into_any_element();
|
||
|
||
let show_controls = matches!(&self.thread_state, ThreadState::Ready { thread, .. } if thread.read(cx).status() == ThreadStatus::Idle);
|
||
|
||
v_flex()
|
||
.size_full()
|
||
.key_context("AcpThread")
|
||
.on_action(cx.listener(Self::chat))
|
||
.on_action(cx.listener(Self::previous_history_message))
|
||
.on_action(cx.listener(Self::next_history_message))
|
||
.child(match &self.thread_state {
|
||
ThreadState::Unauthenticated { .. } => v_flex()
|
||
.p_2()
|
||
.flex_1()
|
||
.items_center()
|
||
.justify_center()
|
||
.child(self.render_pending_auth_state())
|
||
.child(h_flex().mt_1p5().justify_center().child(
|
||
Button::new("sign-in", "Sign in to Gemini").on_click(
|
||
cx.listener(|this, _, window, cx| this.authenticate(window, cx)),
|
||
),
|
||
)),
|
||
ThreadState::Loading { .. } => {
|
||
v_flex().flex_1().child(self.render_empty_state(true, cx))
|
||
}
|
||
ThreadState::LoadError(e) => v_flex()
|
||
.p_2()
|
||
.flex_1()
|
||
.items_center()
|
||
.justify_center()
|
||
.child(self.render_error_state(e, cx)),
|
||
ThreadState::Ready { thread, .. } => v_flex().flex_1().map(|this| {
|
||
if self.list_state.item_count() > 0 {
|
||
this.child(
|
||
list(self.list_state.clone())
|
||
.with_sizing_behavior(gpui::ListSizingBehavior::Auto)
|
||
.flex_grow()
|
||
.into_any(),
|
||
)
|
||
.children(match thread.read(cx).status() {
|
||
ThreadStatus::Idle | ThreadStatus::WaitingForToolConfirmation => None,
|
||
ThreadStatus::Generating => div()
|
||
.px_5()
|
||
.py_2()
|
||
.child(LoadingLabel::new("").size(LabelSize::Small))
|
||
.into(),
|
||
})
|
||
} else {
|
||
this.child(self.render_empty_state(false, cx))
|
||
}
|
||
}),
|
||
})
|
||
.when(show_controls, |el| el.child(feedback_container))
|
||
.when_some(self.last_error.clone(), |el, error| {
|
||
el.child(
|
||
div()
|
||
.p_2()
|
||
.text_xs()
|
||
.border_t_1()
|
||
.border_color(cx.theme().colors().border)
|
||
.bg(cx.theme().status().error_background)
|
||
.child(
|
||
self.render_markdown(error, default_markdown_style(false, window, cx)),
|
||
),
|
||
)
|
||
})
|
||
.child(
|
||
v_flex()
|
||
.p_2()
|
||
.pt_3()
|
||
.gap_1()
|
||
.bg(cx.theme().colors().editor_background)
|
||
.border_t_1()
|
||
.border_color(cx.theme().colors().border)
|
||
.child(self.render_message_editor(cx))
|
||
.child({
|
||
let thread = self.thread();
|
||
|
||
h_flex().justify_end().child(
|
||
if thread.map_or(true, |thread| {
|
||
thread.read(cx).status() == ThreadStatus::Idle
|
||
}) {
|
||
IconButton::new("send-message", IconName::Send)
|
||
.icon_color(Color::Accent)
|
||
.style(ButtonStyle::Filled)
|
||
.disabled(thread.is_none() || is_editor_empty)
|
||
.on_click({
|
||
let focus_handle = focus_handle.clone();
|
||
move |_event, window, cx| {
|
||
focus_handle.dispatch_action(&Chat, window, cx);
|
||
}
|
||
})
|
||
.when(!is_editor_empty, |button| {
|
||
button.tooltip(move |window, cx| {
|
||
Tooltip::for_action("Send", &Chat, window, cx)
|
||
})
|
||
})
|
||
.when(is_editor_empty, |button| {
|
||
button.tooltip(Tooltip::text("Type a message to submit"))
|
||
})
|
||
} else {
|
||
IconButton::new("stop-generation", IconName::StopFilled)
|
||
.icon_color(Color::Error)
|
||
.style(ButtonStyle::Tinted(ui::TintColor::Error))
|
||
.tooltip(move |window, cx| {
|
||
Tooltip::for_action(
|
||
"Stop Generation",
|
||
&editor::actions::Cancel,
|
||
window,
|
||
cx,
|
||
)
|
||
})
|
||
.on_click(cx.listener(|this, _event, _, cx| this.cancel(cx)))
|
||
},
|
||
)
|
||
}),
|
||
)
|
||
}
|
||
}
|
||
|
||
fn user_message_markdown_style(window: &Window, cx: &App) -> MarkdownStyle {
|
||
let mut style = default_markdown_style(false, window, cx);
|
||
let mut text_style = window.text_style();
|
||
let theme_settings = ThemeSettings::get_global(cx);
|
||
|
||
let buffer_font = theme_settings.buffer_font.family.clone();
|
||
let buffer_font_size = TextSize::Small.rems(cx);
|
||
|
||
text_style.refine(&TextStyleRefinement {
|
||
font_family: Some(buffer_font),
|
||
font_size: Some(buffer_font_size.into()),
|
||
..Default::default()
|
||
});
|
||
|
||
style.base_text_style = text_style;
|
||
style.link_callback = Some(Rc::new(move |url, cx| {
|
||
if MentionPath::try_parse(url).is_some() {
|
||
let colors = cx.theme().colors();
|
||
Some(TextStyleRefinement {
|
||
background_color: Some(colors.element_background),
|
||
..Default::default()
|
||
})
|
||
} else {
|
||
None
|
||
}
|
||
}));
|
||
style
|
||
}
|
||
|
||
fn default_markdown_style(buffer_font: bool, window: &Window, cx: &App) -> MarkdownStyle {
|
||
let theme_settings = ThemeSettings::get_global(cx);
|
||
let colors = cx.theme().colors();
|
||
|
||
let buffer_font_size = TextSize::Small.rems(cx);
|
||
|
||
let mut text_style = window.text_style();
|
||
let line_height = buffer_font_size * 1.75;
|
||
|
||
let font_family = if buffer_font {
|
||
theme_settings.buffer_font.family.clone()
|
||
} else {
|
||
theme_settings.ui_font.family.clone()
|
||
};
|
||
|
||
let font_size = if buffer_font {
|
||
TextSize::Small.rems(cx)
|
||
} else {
|
||
TextSize::Default.rems(cx)
|
||
};
|
||
|
||
text_style.refine(&TextStyleRefinement {
|
||
font_family: Some(font_family),
|
||
font_fallbacks: theme_settings.ui_font.fallbacks.clone(),
|
||
font_features: Some(theme_settings.ui_font.features.clone()),
|
||
font_size: Some(font_size.into()),
|
||
line_height: Some(line_height.into()),
|
||
color: Some(cx.theme().colors().text),
|
||
..Default::default()
|
||
});
|
||
|
||
MarkdownStyle {
|
||
base_text_style: text_style.clone(),
|
||
syntax: cx.theme().syntax().clone(),
|
||
selection_background_color: cx.theme().colors().element_selection_background,
|
||
code_block_overflow_x_scroll: true,
|
||
table_overflow_x_scroll: true,
|
||
heading_level_styles: Some(HeadingLevelStyles {
|
||
h1: Some(TextStyleRefinement {
|
||
font_size: Some(rems(1.15).into()),
|
||
..Default::default()
|
||
}),
|
||
h2: Some(TextStyleRefinement {
|
||
font_size: Some(rems(1.1).into()),
|
||
..Default::default()
|
||
}),
|
||
h3: Some(TextStyleRefinement {
|
||
font_size: Some(rems(1.05).into()),
|
||
..Default::default()
|
||
}),
|
||
h4: Some(TextStyleRefinement {
|
||
font_size: Some(rems(1.).into()),
|
||
..Default::default()
|
||
}),
|
||
h5: Some(TextStyleRefinement {
|
||
font_size: Some(rems(0.95).into()),
|
||
..Default::default()
|
||
}),
|
||
h6: Some(TextStyleRefinement {
|
||
font_size: Some(rems(0.875).into()),
|
||
..Default::default()
|
||
}),
|
||
}),
|
||
code_block: StyleRefinement {
|
||
padding: EdgesRefinement {
|
||
top: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))),
|
||
left: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))),
|
||
right: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))),
|
||
bottom: Some(DefiniteLength::Absolute(AbsoluteLength::Pixels(Pixels(8.)))),
|
||
},
|
||
margin: EdgesRefinement {
|
||
top: Some(Length::Definite(Pixels(8.).into())),
|
||
left: Some(Length::Definite(Pixels(0.).into())),
|
||
right: Some(Length::Definite(Pixels(0.).into())),
|
||
bottom: Some(Length::Definite(Pixels(12.).into())),
|
||
},
|
||
border_style: Some(BorderStyle::Solid),
|
||
border_widths: EdgesRefinement {
|
||
top: Some(AbsoluteLength::Pixels(Pixels(1.))),
|
||
left: Some(AbsoluteLength::Pixels(Pixels(1.))),
|
||
right: Some(AbsoluteLength::Pixels(Pixels(1.))),
|
||
bottom: Some(AbsoluteLength::Pixels(Pixels(1.))),
|
||
},
|
||
border_color: Some(colors.border_variant),
|
||
background: Some(colors.editor_background.into()),
|
||
text: Some(TextStyleRefinement {
|
||
font_family: Some(theme_settings.buffer_font.family.clone()),
|
||
font_fallbacks: theme_settings.buffer_font.fallbacks.clone(),
|
||
font_features: Some(theme_settings.buffer_font.features.clone()),
|
||
font_size: Some(buffer_font_size.into()),
|
||
..Default::default()
|
||
}),
|
||
..Default::default()
|
||
},
|
||
inline_code: TextStyleRefinement {
|
||
font_family: Some(theme_settings.buffer_font.family.clone()),
|
||
font_fallbacks: theme_settings.buffer_font.fallbacks.clone(),
|
||
font_features: Some(theme_settings.buffer_font.features.clone()),
|
||
font_size: Some(buffer_font_size.into()),
|
||
background_color: Some(colors.editor_foreground.opacity(0.08)),
|
||
..Default::default()
|
||
},
|
||
link: TextStyleRefinement {
|
||
background_color: Some(colors.editor_foreground.opacity(0.025)),
|
||
underline: Some(UnderlineStyle {
|
||
color: Some(colors.text_accent.opacity(0.5)),
|
||
thickness: px(1.),
|
||
..Default::default()
|
||
}),
|
||
..Default::default()
|
||
},
|
||
..Default::default()
|
||
}
|
||
}
|