use acp_thread::{ AcpThread, AcpThreadEvent, AgentThreadEntry, AssistantMessage, AssistantMessageChunk, LoadError, MentionUri, ThreadStatus, ToolCall, ToolCallContent, ToolCallStatus, UserMessageId, }; use acp_thread::{AgentConnection, Plan}; use action_log::ActionLog; use agent::{TextThreadStore, ThreadStore}; use agent_client_protocol::{self as acp}; use agent_servers::AgentServer; use agent_settings::{AgentSettings, NotifyWhenAgentWaiting}; use anyhow::bail; use audio::{Audio, Sound}; use buffer_diff::BufferDiff; use collections::{HashMap, HashSet}; use editor::scroll::Autoscroll; use editor::{Editor, EditorMode, MultiBuffer, PathKey, SelectionEffects}; use file_icons::FileIcons; use gpui::{ Action, Animation, AnimationExt, App, BorderStyle, ClickEvent, EdgesRefinement, Empty, Entity, FocusHandle, Focusable, Hsla, Length, ListOffset, ListState, MouseButton, PlatformDisplay, SharedString, Stateful, StyleRefinement, Subscription, Task, TextStyle, TextStyleRefinement, Transformation, UnderlineStyle, WeakEntity, Window, WindowHandle, div, linear_color_stop, linear_gradient, list, percentage, point, prelude::*, pulsating_between, }; use language::Buffer; use markdown::{HeadingLevelStyles, Markdown, MarkdownElement, MarkdownStyle}; use project::Project; use prompt_store::PromptId; use rope::Point; use settings::{Settings as _, SettingsStore}; use std::{collections::BTreeMap, process::ExitStatus, rc::Rc, time::Duration}; use text::Anchor; use theme::ThemeSettings; use ui::{ Disclosure, Divider, DividerColor, KeyBinding, PopoverMenuHandle, Scrollbar, ScrollbarState, Tooltip, prelude::*, }; use util::{ResultExt, size::format_file_size, time::duration_alt_display}; use workspace::{CollaboratorId, Workspace}; use zed_actions::agent::{Chat, ToggleModelSelector}; use zed_actions::assistant::OpenRulesLibrary; use super::entry_view_state::EntryViewState; use crate::acp::AcpModelSelectorPopover; use crate::acp::message_editor::{MessageEditor, MessageEditorEvent}; use crate::agent_diff::AgentDiff; use crate::ui::{AgentNotification, AgentNotificationEvent}; use crate::{ AgentDiffPane, AgentPanel, ExpandMessageEditor, Follow, KeepAll, OpenAgentDiff, RejectAll, }; const RESPONSE_PADDING_X: Pixels = px(19.); pub const MIN_EDITOR_LINES: usize = 4; pub const MAX_EDITOR_LINES: usize = 8; pub struct AcpThreadView { agent: Rc, workspace: WeakEntity, project: Entity, thread_store: Entity, text_thread_store: Entity, thread_state: ThreadState, entry_view_state: EntryViewState, message_editor: Entity, model_selector: Option>, notifications: Vec>, notification_subscriptions: HashMap, Vec>, last_error: Option>, list_state: ListState, scrollbar_state: ScrollbarState, auth_task: Option>, expanded_tool_calls: HashSet, expanded_thinking_blocks: HashSet<(usize, usize)>, edits_expanded: bool, plan_expanded: bool, editor_expanded: bool, terminal_expanded: bool, editing_message: Option, _cancel_task: Option>, _subscriptions: [Subscription; 2], } struct EditingMessage { index: usize, message_id: UserMessageId, editor: Entity, _subscription: Subscription, } enum ThreadState { Loading { _task: Task<()>, }, Ready { thread: Entity, _subscription: [Subscription; 2], }, LoadError(LoadError), Unauthenticated { connection: Rc, }, ServerExited { status: ExitStatus, }, } impl AcpThreadView { pub fn new( agent: Rc, workspace: WeakEntity, project: Entity, thread_store: Entity, text_thread_store: Entity, window: &mut Window, cx: &mut Context, ) -> Self { let message_editor = cx.new(|cx| { MessageEditor::new( workspace.clone(), project.clone(), thread_store.clone(), text_thread_store.clone(), editor::EditorMode::AutoHeight { min_lines: MIN_EDITOR_LINES, max_lines: Some(MAX_EDITOR_LINES), }, window, cx, ) }); let list_state = ListState::new(0, gpui::ListAlignment::Bottom, px(2048.0)); let subscriptions = [ cx.observe_global_in::(window, Self::settings_changed), cx.subscribe_in(&message_editor, window, Self::on_message_editor_event), ]; Self { agent: agent.clone(), workspace: workspace.clone(), project: project.clone(), thread_store, text_thread_store, thread_state: Self::initial_state(agent, workspace, project, window, cx), message_editor, model_selector: None, notifications: Vec::new(), notification_subscriptions: HashMap::default(), entry_view_state: EntryViewState::default(), list_state: list_state.clone(), scrollbar_state: ScrollbarState::new(list_state).parent_entity(&cx.entity()), last_error: None, auth_task: None, expanded_tool_calls: HashSet::default(), expanded_thinking_blocks: HashSet::default(), editing_message: None, edits_expanded: false, plan_expanded: false, editor_expanded: false, terminal_expanded: true, _subscriptions: subscriptions, _cancel_task: None, } } fn initial_state( agent: Rc, workspace: WeakEntity, project: Entity, window: &mut Window, cx: &mut Context, ) -> 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 connect_task = agent.connect(&root_dir, &project, cx); let load_task = cx.spawn_in(window, async move |this, cx| { let connection = match connect_task.await { Ok(connection) => connection, Err(err) => { this.update(cx, |this, cx| { this.handle_load_error(err, cx); cx.notify(); }) .log_err(); return; } }; // this.update_in(cx, |_this, _window, cx| { // let status = connection.exit_status(cx); // cx.spawn(async move |this, cx| { // let status = status.await.ok(); // this.update(cx, |this, cx| { // this.thread_state = ThreadState::ServerExited { status }; // cx.notify(); // }) // .ok(); // }) // .detach(); // }) // .ok(); let Some(result) = cx .update(|_, cx| { connection .clone() .new_thread(project.clone(), &root_dir, cx) }) .log_err() else { return; }; let result = match result.await { Err(e) => { let mut cx = cx.clone(); if e.is::() { this.update(&mut cx, |this, cx| { this.thread_state = ThreadState::Unauthenticated { connection }; cx.notify(); }) .ok(); return; } else { Err(e) } } Ok(thread) => Ok(thread), }; this.update_in(cx, |this, window, cx| { match result { Ok(thread) => { let thread_subscription = cx.subscribe_in(&thread, window, Self::handle_thread_event); let action_log = thread.read(cx).action_log().clone(); let action_log_subscription = cx.observe(&action_log, |_, _, cx| cx.notify()); this.list_state .splice(0..0, thread.read(cx).entries().len()); AgentDiff::set_active_thread(&workspace, thread.clone(), window, cx); this.model_selector = thread .read(cx) .connection() .model_selector() .map(|selector| { cx.new(|cx| { AcpModelSelectorPopover::new( thread.read(cx).session_id().clone(), selector, PopoverMenuHandle::default(), this.focus_handle(cx), window, cx, ) }) }); this.thread_state = ThreadState::Ready { thread, _subscription: [thread_subscription, action_log_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) { if let Some(load_err) = err.downcast_ref::() { self.thread_state = ThreadState::LoadError(load_err.clone()); } else { self.thread_state = ThreadState::LoadError(LoadError::Other(err.to_string().into())) } cx.notify(); } pub fn thread(&self) -> Option<&Entity> { match &self.thread_state { ThreadState::Ready { thread, .. } => Some(thread), ThreadState::Unauthenticated { .. } | ThreadState::Loading { .. } | ThreadState::LoadError(..) | ThreadState::ServerExited { .. } => 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(), ThreadState::ServerExited { .. } => "Server exited unexpectedly".into(), } } pub fn cancel_generation(&mut self, cx: &mut Context) { self.last_error.take(); if let Some(thread) = self.thread() { self._cancel_task = Some(thread.update(cx, |thread, cx| thread.cancel(cx))); } } pub fn expand_message_editor( &mut self, _: &ExpandMessageEditor, _window: &mut Window, cx: &mut Context, ) { self.set_editor_is_expanded(!self.editor_expanded, cx); cx.notify(); } fn set_editor_is_expanded(&mut self, is_expanded: bool, cx: &mut Context) { self.editor_expanded = is_expanded; self.message_editor.update(cx, |editor, cx| { if is_expanded { editor.set_mode( EditorMode::Full { scale_ui_elements_with_buffer_font_size: false, show_active_line_background: false, sized_by_content: false, }, cx, ) } else { editor.set_mode( EditorMode::AutoHeight { min_lines: MIN_EDITOR_LINES, max_lines: Some(MAX_EDITOR_LINES), }, cx, ) } }); cx.notify(); } pub fn on_message_editor_event( &mut self, _: &Entity, event: &MessageEditorEvent, window: &mut Window, cx: &mut Context, ) { match event { MessageEditorEvent::Send => self.send(window, cx), MessageEditorEvent::Cancel => self.cancel_generation(cx), } } fn send(&mut self, window: &mut Window, cx: &mut Context) { let contents = self .message_editor .update(cx, |message_editor, cx| message_editor.contents(window, cx)); self.send_impl(contents, window, cx) } fn send_impl( &mut self, contents: Task>>, window: &mut Window, cx: &mut Context, ) { self.last_error.take(); self.editing_message.take(); let Some(thread) = self.thread().cloned() else { return; }; let task = cx.spawn_in(window, async move |this, cx| { let contents = contents.await?; if contents.is_empty() { return Ok(()); } this.update_in(cx, |this, window, cx| { this.set_editor_is_expanded(false, cx); this.scroll_to_bottom(cx); this.message_editor.update(cx, |message_editor, cx| { message_editor.clear(window, cx); }); })?; let send = thread.update(cx, |thread, cx| thread.send(contents, cx))?; send.await }); cx.spawn(async move |this, cx| { if let Err(e) = task.await { this.update(cx, |this, cx| { this.last_error = Some(cx.new(|cx| Markdown::new(e.to_string().into(), None, None, cx))); cx.notify() }) .ok(); } }) .detach(); } fn cancel_editing(&mut self, _: &ClickEvent, _window: &mut Window, cx: &mut Context) { self.editing_message.take(); cx.notify(); } fn regenerate(&mut self, _: &ClickEvent, window: &mut Window, cx: &mut Context) { let Some(editing_message) = self.editing_message.take() else { return; }; let Some(thread) = self.thread().cloned() else { return; }; let rewind = thread.update(cx, |thread, cx| { thread.rewind(editing_message.message_id, cx) }); let contents = editing_message .editor .update(cx, |message_editor, cx| message_editor.contents(window, cx)); let task = cx.foreground_executor().spawn(async move { rewind.await?; contents.await }); self.send_impl(task, window, cx); } fn open_agent_diff(&mut self, _: &OpenAgentDiff, window: &mut Window, cx: &mut Context) { if let Some(thread) = self.thread() { AgentDiffPane::deploy(thread.clone(), self.workspace.clone(), window, cx).log_err(); } } fn open_edited_buffer( &mut self, buffer: &Entity, window: &mut Window, cx: &mut Context, ) { let Some(thread) = self.thread() else { return; }; let Some(diff) = AgentDiffPane::deploy(thread.clone(), self.workspace.clone(), window, cx).log_err() else { return; }; diff.update(cx, |diff, cx| { diff.move_to_path(PathKey::for_buffer(&buffer, cx), window, cx) }) } fn handle_thread_event( &mut self, thread: &Entity, event: &AcpThreadEvent, window: &mut Window, cx: &mut Context, ) { match event { AcpThreadEvent::NewEntry => { let len = thread.read(cx).entries().len(); let index = len - 1; self.entry_view_state.sync_entry( self.workspace.clone(), thread.clone(), index, window, cx, ); self.list_state.splice(index..index, 1); } AcpThreadEvent::EntryUpdated(index) => { self.entry_view_state.sync_entry( self.workspace.clone(), thread.clone(), *index, window, cx, ); self.list_state.splice(*index..index + 1, 1); } AcpThreadEvent::EntriesRemoved(range) => { self.entry_view_state.remove(range.clone()); self.list_state.splice(range.clone(), 0); } AcpThreadEvent::ToolAuthorizationRequired => { self.notify_with_sound("Waiting for tool confirmation", IconName::Info, window, cx); } AcpThreadEvent::Stopped => { let used_tools = thread.read(cx).used_tools_since_last_user_message(); self.notify_with_sound( if used_tools { "Finished running tools" } else { "New message" }, IconName::ZedAssistant, window, cx, ); } AcpThreadEvent::Error => { self.notify_with_sound( "Agent stopped due to an error", IconName::Warning, window, cx, ); } AcpThreadEvent::ServerExited(status) => { self.thread_state = ThreadState::ServerExited { status: *status }; } } cx.notify(); } fn authenticate( &mut self, method: acp::AuthMethodId, window: &mut Window, cx: &mut Context, ) { let ThreadState::Unauthenticated { ref connection } = self.thread_state else { return; }; self.last_error.take(); let authenticate = connection.authenticate(method, cx); self.auth_task = Some(cx.spawn_in(window, { let project = self.project.clone(); let agent = self.agent.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( agent, this.workspace.clone(), project.clone(), window, cx, ) } this.auth_task.take() }) .ok(); } })); } fn authorize_tool_call( &mut self, tool_call_id: acp::ToolCallId, option_id: acp::PermissionOptionId, option_kind: acp::PermissionOptionKind, cx: &mut Context, ) { let Some(thread) = self.thread() else { return; }; thread.update(cx, |thread, cx| { thread.authorize_tool_call(tool_call_id, option_id, option_kind, cx); }); cx.notify(); } fn rewind(&mut self, message_id: &UserMessageId, cx: &mut Context) { let Some(thread) = self.thread() else { return; }; thread .update(cx, |thread, cx| thread.rewind(message_id.clone(), cx)) .detach_and_log_err(cx); cx.notify(); } fn render_entry( &self, entry_ix: usize, total_entries: usize, entry: &AgentThreadEntry, window: &mut Window, cx: &Context, ) -> AnyElement { let primary = match &entry { AgentThreadEntry::UserMessage(message) => div() .id(("user_message", entry_ix)) .py_4() .px_2() .children(message.id.clone().and_then(|message_id| { message.checkpoint.as_ref()?; Some( Button::new("restore-checkpoint", "Restore Checkpoint") .icon(IconName::Undo) .icon_size(IconSize::XSmall) .icon_position(IconPosition::Start) .label_size(LabelSize::XSmall) .on_click(cx.listener(move |this, _, _window, cx| { this.rewind(&message_id, cx); })), ) })) .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() .id("message") .on_click(cx.listener({ move |this, _, window, cx| { this.start_editing_message(entry_ix, window, cx) } })) .children( if let Some(editing) = self.editing_message.as_ref() && Some(&editing.message_id) == message.id.as_ref() { Some( self.render_edit_message_editor(editing, cx) .into_any_element(), ) } else { message.content.markdown().map(|md| { self.render_markdown( md.clone(), user_message_markdown_style(window, cx), ) .into_any_element() }) }, ), ) .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().filter_map( |(chunk_ix, chunk)| match chunk { AssistantMessageChunk::Message { block } => { block.markdown().map(|md| { self.render_markdown(md.clone(), style.clone()) .into_any_element() }) } AssistantMessageChunk::Thought { block } => { block.markdown().map(|md| { self.render_thinking_block( entry_ix, chunk_ix, md.clone(), window, cx, ) .into_any_element() }) } }, )) .into_any(); v_flex() .px_5() .py_1() .when(entry_ix + 1 == total_entries, |this| this.pb_4()) .w_full() .text_ui(cx) .child(message_body) .into_any() } AgentThreadEntry::ToolCall(tool_call) => { let has_terminals = tool_call.terminals().next().is_some(); div().w_full().py_1p5().px_5().map(|this| { if has_terminals { this.children(tool_call.terminals().map(|terminal| { self.render_terminal_tool_call( entry_ix, terminal, tool_call, window, cx, ) })) } else { this.child(self.render_tool_call(entry_ix, tool_call, window, cx)) } }) } .into_any(), }; let Some(thread) = self.thread() else { return primary; }; let is_generating = matches!(thread.read(cx).status(), ThreadStatus::Generating); let primary = if entry_ix == total_entries - 1 && !is_generating { v_flex() .w_full() .child(primary) .child(self.render_thread_controls(cx)) .into_any_element() } else { primary }; if let Some(editing) = self.editing_message.as_ref() && editing.index < entry_ix { let backdrop = div() .id(("backdrop", entry_ix)) .size_full() .absolute() .inset_0() .bg(cx.theme().colors().panel_background) .opacity(0.8) .block_mouse_except_scroll() .on_click(cx.listener(Self::cancel_editing)); div() .relative() .child(backdrop) .child(primary) .into_any_element() } else { primary } } fn tool_card_header_bg(&self, cx: &Context) -> Hsla { cx.theme() .colors() .element_background .blend(cx.theme().colors().editor_foreground.opacity(0.025)) } fn tool_card_border_color(&self, cx: &Context) -> 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, window: &Window, cx: &Context, ) -> AnyElement { let header_id = SharedString::from(format!("thinking-block-header-{}", entry_ix)); let card_header_id = SharedString::from("inner-card-header"); let key = (entry_ix, chunk_ix); let is_open = self.expanded_thinking_blocks.contains(&key); v_flex() .child( h_flex() .id(header_id) .group(&card_header_id) .relative() .w_full() .gap_1p5() .opacity(0.8) .hover(|style| style.opacity(1.)) .child( h_flex() .size_4() .justify_center() .child( div() .group_hover(&card_header_id, |s| s.invisible().w_0()) .child( Icon::new(IconName::ToolThink) .size(IconSize::Small) .color(Color::Muted), ), ) .child( h_flex() .absolute() .inset_0() .invisible() .justify_center() .group_hover(&card_header_id, |s| s.visible()) .child( Disclosure::new(("expand", entry_ix), is_open) .opened_icon(IconName::ChevronUp) .closed_icon(IconName::ChevronRight) .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(); } })), ), ), ) .child( div() .text_size(self.tool_name_font_size()) .child("Thinking"), ) .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_icon( &self, group_name: SharedString, entry_ix: usize, is_collapsible: bool, is_open: bool, tool_call: &ToolCall, cx: &Context, ) -> Div { let tool_icon = Icon::new(match tool_call.kind { acp::ToolKind::Read => IconName::ToolRead, acp::ToolKind::Edit => IconName::ToolPencil, acp::ToolKind::Delete => IconName::ToolDeleteFile, acp::ToolKind::Move => IconName::ArrowRightLeft, acp::ToolKind::Search => IconName::ToolSearch, acp::ToolKind::Execute => IconName::ToolTerminal, acp::ToolKind::Think => IconName::ToolThink, acp::ToolKind::Fetch => IconName::ToolWeb, acp::ToolKind::Other => IconName::ToolHammer, }) .size(IconSize::Small) .color(Color::Muted); let base_container = h_flex().size_4().justify_center(); if is_collapsible { base_container .child( div() .group_hover(&group_name, |s| s.invisible().w_0()) .child(tool_icon), ) .child( h_flex() .absolute() .inset_0() .invisible() .justify_center() .group_hover(&group_name, |s| s.visible()) .child( Disclosure::new(("expand", entry_ix), is_open) .opened_icon(IconName::ChevronUp) .closed_icon(IconName::ChevronRight) .on_click(cx.listener({ let id = tool_call.id.clone(); move |this: &mut Self, _, _, cx: &mut Context| { if is_open { this.expanded_tool_calls.remove(&id); } else { this.expanded_tool_calls.insert(id.clone()); } cx.notify(); } })), ), ) } else { base_container.child(tool_icon) } } fn render_tool_call( &self, entry_ix: usize, tool_call: &ToolCall, window: &Window, cx: &Context, ) -> Div { let header_id = SharedString::from(format!("outer-tool-call-header-{}", entry_ix)); let card_header_id = SharedString::from("inner-tool-call-header"); let status_icon = match &tool_call.status { ToolCallStatus::Allowed { status: acp::ToolCallStatus::Pending, } | ToolCallStatus::WaitingForConfirmation { .. } => None, ToolCallStatus::Allowed { status: acp::ToolCallStatus::InProgress, .. } => 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::Completed, .. } => None, ToolCallStatus::Rejected | ToolCallStatus::Canceled | ToolCallStatus::Allowed { status: acp::ToolCallStatus::Failed, .. } => Some( Icon::new(IconName::Close) .color(Color::Error) .size(IconSize::Small) .into_any_element(), ), }; let needs_confirmation = matches!( tool_call.status, ToolCallStatus::WaitingForConfirmation { .. } ); let is_edit = matches!(tool_call.kind, acp::ToolKind::Edit); let has_diff = tool_call .content .iter() .any(|content| matches!(content, ToolCallContent::Diff { .. })); let has_nonempty_diff = tool_call.content.iter().any(|content| match content { ToolCallContent::Diff(diff) => diff.read(cx).has_revealed_range(cx), _ => false, }); let use_card_layout = needs_confirmation || is_edit || has_diff; let is_collapsible = !tool_call.content.is_empty() && !use_card_layout; let is_open = tool_call.content.is_empty() || needs_confirmation || has_nonempty_diff || self.expanded_tool_calls.contains(&tool_call.id); let gradient_overlay = |color: Hsla| { div() .absolute() .top_0() .right_0() .w_12() .h_full() .bg(linear_gradient( 90., linear_color_stop(color, 1.), linear_color_stop(color.opacity(0.2), 0.), )) }; let gradient_color = if use_card_layout { self.tool_card_header_bg(cx) } else { cx.theme().colors().panel_background }; let tool_output_display = match &tool_call.status { ToolCallStatus::WaitingForConfirmation { options, .. } => v_flex() .w_full() .children(tool_call.content.iter().map(|content| { div() .child( self.render_tool_call_content(entry_ix, content, tool_call, window, cx), ) .into_any_element() })) .child(self.render_permission_buttons( options, entry_ix, tool_call.id.clone(), tool_call.content.is_empty(), cx, )), ToolCallStatus::Allowed { .. } | ToolCallStatus::Canceled => v_flex() .w_full() .children(tool_call.content.iter().map(|content| { div() .child( self.render_tool_call_content(entry_ix, content, tool_call, window, cx), ) .into_any_element() })), ToolCallStatus::Rejected => v_flex().size_0(), }; v_flex() .when(use_card_layout, |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 use_card_layout { this.pl_2() .pr_1() .py_1() .rounded_t_md() .bg(self.tool_card_header_bg(cx)) } else { this.opacity(0.8).hover(|style| style.opacity(1.)) } }) .child( h_flex() .group(&card_header_id) .relative() .w_full() .text_size(self.tool_name_font_size()) .child(self.render_tool_call_icon( card_header_id, entry_ix, is_collapsible, is_open, tool_call, cx, )) .child(if tool_call.locations.len() == 1 { let name = tool_call.locations[0] .path .file_name() .unwrap_or_default() .display() .to_string(); h_flex() .id(("open-tool-call-location", entry_ix)) .w_full() .max_w_full() .px_1p5() .rounded_sm() .overflow_x_scroll() .opacity(0.8) .hover(|label| { label.opacity(1.).bg(cx .theme() .colors() .element_hover .opacity(0.5)) }) .child(name) .tooltip(Tooltip::text("Jump to File")) .on_click(cx.listener(move |this, _, window, cx| { this.open_tool_call_location(entry_ix, 0, window, cx); })) .into_any_element() } else { h_flex() .id("non-card-label-container") .w_full() .relative() .ml_1p5() .overflow_hidden() .child( h_flex() .id("non-card-label") .pr_8() .w_full() .overflow_x_scroll() .child(self.render_markdown( tool_call.label.clone(), default_markdown_style( needs_confirmation || is_edit || has_diff, window, cx, ), )), ) .child(gradient_overlay(gradient_color)) .on_click(cx.listener({ let id = tool_call.id.clone(); move |this: &mut Self, _, _, cx: &mut Context| { if is_open { this.expanded_tool_calls.remove(&id); } else { this.expanded_tool_calls.insert(id.clone()); } cx.notify(); } })) .into_any() }), ) .children(status_icon), ) .when(is_open, |this| this.child(tool_output_display)) } fn render_tool_call_content( &self, entry_ix: usize, content: &ToolCallContent, tool_call: &ToolCall, window: &Window, cx: &Context, ) -> AnyElement { match content { ToolCallContent::ContentBlock(content) => { if let Some(resource_link) = content.resource_link() { self.render_resource_link(resource_link, cx) } else if let Some(markdown) = content.markdown() { self.render_markdown_output(markdown.clone(), tool_call.id.clone(), window, cx) } else { Empty.into_any_element() } } ToolCallContent::Diff(diff) => { self.render_diff_editor(entry_ix, &diff.read(cx).multibuffer(), cx) } ToolCallContent::Terminal(terminal) => { self.render_terminal_tool_call(entry_ix, terminal, tool_call, window, cx) } } } fn render_markdown_output( &self, markdown: Entity, tool_call_id: acp::ToolCallId, window: &Window, cx: &Context, ) -> AnyElement { let button_id = SharedString::from(format!("tool_output-{:?}", tool_call_id.clone())); v_flex() .mt_1p5() .ml(px(7.)) .px_3p5() .gap_2() .border_l_1() .border_color(self.tool_card_border_color(cx)) .text_sm() .text_color(cx.theme().colors().text_muted) .child(self.render_markdown(markdown, default_markdown_style(false, window, cx))) .child( Button::new(button_id, "Collapse Output") .full_width() .style(ButtonStyle::Outlined) .label_size(LabelSize::Small) .icon(IconName::ChevronUp) .icon_color(Color::Muted) .icon_position(IconPosition::Start) .on_click(cx.listener({ let id = tool_call_id.clone(); move |this: &mut Self, _, _, cx: &mut Context| { this.expanded_tool_calls.remove(&id); cx.notify(); } })), ) .into_any_element() } fn render_resource_link( &self, resource_link: &acp::ResourceLink, cx: &Context, ) -> AnyElement { let uri: SharedString = resource_link.uri.clone().into(); let label: SharedString = if let Some(path) = resource_link.uri.strip_prefix("file://") { path.to_string().into() } else { uri.clone() }; let button_id = SharedString::from(format!("item-{}", uri.clone())); div() .ml(px(7.)) .pl_2p5() .border_l_1() .border_color(self.tool_card_border_color(cx)) .overflow_hidden() .child( Button::new(button_id, label) .label_size(LabelSize::Small) .color(Color::Muted) .icon(IconName::ArrowUpRight) .icon_size(IconSize::XSmall) .icon_color(Color::Muted) .truncate(true) .on_click(cx.listener({ let workspace = self.workspace.clone(); move |_, _, window, cx: &mut Context| { Self::open_link(uri.clone(), &workspace, window, cx); } })), ) .into_any_element() } fn render_permission_buttons( &self, options: &[acp::PermissionOption], entry_ix: usize, tool_call_id: acp::ToolCallId, empty_content: bool, cx: &Context, ) -> Div { h_flex() .py_1() .pl_2() .pr_1() .gap_1() .justify_between() .flex_wrap() .when(!empty_content, |this| { this.border_t_1() .border_color(self.tool_card_border_color(cx)) }) .child( div() .min_w(rems_from_px(145.)) .child(LoadingLabel::new("Waiting for Confirmation").size(LabelSize::Small)), ) .child(h_flex().gap_0p5().children(options.iter().map(|option| { let option_id = SharedString::from(option.id.0.clone()); Button::new((option_id, entry_ix), option.name.clone()) .map(|this| match option.kind { acp::PermissionOptionKind::AllowOnce => { this.icon(IconName::Check).icon_color(Color::Success) } acp::PermissionOptionKind::AllowAlways => { this.icon(IconName::CheckDouble).icon_color(Color::Success) } acp::PermissionOptionKind::RejectOnce => { this.icon(IconName::Close).icon_color(Color::Error) } acp::PermissionOptionKind::RejectAlways => { this.icon(IconName::Close).icon_color(Color::Error) } }) .icon_position(IconPosition::Start) .icon_size(IconSize::XSmall) .label_size(LabelSize::Small) .on_click(cx.listener({ let tool_call_id = tool_call_id.clone(); let option_id = option.id.clone(); let option_kind = option.kind; move |this, _, _, cx| { this.authorize_tool_call( tool_call_id.clone(), option_id.clone(), option_kind, cx, ); } })) }))) } fn render_diff_editor( &self, entry_ix: usize, multibuffer: &Entity, cx: &Context, ) -> AnyElement { v_flex() .h_full() .border_t_1() .border_color(self.tool_card_border_color(cx)) .child( if let Some(entry) = self.entry_view_state.entry(entry_ix) && let Some(editor) = entry.editor_for_diff(&multibuffer) { editor.clone().into_any_element() } else { Empty.into_any() }, ) .into_any() } fn render_terminal_tool_call( &self, entry_ix: usize, terminal: &Entity, tool_call: &ToolCall, window: &Window, cx: &Context, ) -> AnyElement { let terminal_data = terminal.read(cx); let working_dir = terminal_data.working_dir(); let command = terminal_data.command(); let started_at = terminal_data.started_at(); let tool_failed = matches!( &tool_call.status, ToolCallStatus::Rejected | ToolCallStatus::Canceled | ToolCallStatus::Allowed { status: acp::ToolCallStatus::Failed, .. } ); let output = terminal_data.output(); let command_finished = output.is_some(); let truncated_output = output.is_some_and(|output| output.was_content_truncated); let output_line_count = output.map(|output| output.content_line_count).unwrap_or(0); let command_failed = command_finished && output.is_some_and(|o| o.exit_status.is_none_or(|status| !status.success())); let time_elapsed = if let Some(output) = output { output.ended_at.duration_since(started_at) } else { started_at.elapsed() }; let header_bg = cx .theme() .colors() .element_background .blend(cx.theme().colors().editor_foreground.opacity(0.025)); let border_color = cx.theme().colors().border.opacity(0.6); let working_dir = working_dir .as_ref() .map(|path| format!("{}", path.display())) .unwrap_or_else(|| "current directory".to_string()); let header = h_flex() .id(SharedString::from(format!( "terminal-tool-header-{}", terminal.entity_id() ))) .flex_none() .gap_1() .justify_between() .rounded_t_md() .child( div() .id(("command-target-path", terminal.entity_id())) .w_full() .max_w_full() .overflow_x_scroll() .child( Label::new(working_dir) .buffer_font(cx) .size(LabelSize::XSmall) .color(Color::Muted), ), ) .when(!command_finished, |header| { header .gap_1p5() .child( Button::new( SharedString::from(format!("stop-terminal-{}", terminal.entity_id())), "Stop", ) .icon(IconName::Stop) .icon_position(IconPosition::Start) .icon_size(IconSize::Small) .icon_color(Color::Error) .label_size(LabelSize::Small) .tooltip(move |window, cx| { Tooltip::with_meta( "Stop This Command", None, "Also possible by placing your cursor inside the terminal and using regular terminal bindings.", window, cx, ) }) .on_click({ let terminal = terminal.clone(); cx.listener(move |_this, _event, _window, cx| { let inner_terminal = terminal.read(cx).inner().clone(); inner_terminal.update(cx, |inner_terminal, _cx| { inner_terminal.kill_active_task(); }); }) }), ) .child(Divider::vertical()) .child( Icon::new(IconName::ArrowCircle) .size(IconSize::XSmall) .color(Color::Info) .with_animation( "arrow-circle", Animation::new(Duration::from_secs(2)).repeat(), |icon, delta| { icon.transform(Transformation::rotate(percentage(delta))) }, ), ) }) .when(tool_failed || command_failed, |header| { header.child( div() .id(("terminal-tool-error-code-indicator", terminal.entity_id())) .child( Icon::new(IconName::Close) .size(IconSize::Small) .color(Color::Error), ) .when_some(output.and_then(|o| o.exit_status), |this, status| { this.tooltip(Tooltip::text(format!( "Exited with code {}", status.code().unwrap_or(-1), ))) }), ) }) .when(truncated_output, |header| { let tooltip = if let Some(output) = output { if output_line_count + 10 > terminal::MAX_SCROLL_HISTORY_LINES { "Output exceeded terminal max lines and was \ truncated, the model received the first 16 KB." .to_string() } else { format!( "Output is {} long—to avoid unexpected token usage, \ only 16 KB was sent back to the model.", format_file_size(output.original_content_len as u64, true), ) } } else { "Output was truncated".to_string() }; header.child( h_flex() .id(("terminal-tool-truncated-label", terminal.entity_id())) .gap_1() .child( Icon::new(IconName::Info) .size(IconSize::XSmall) .color(Color::Ignored), ) .child( Label::new("Truncated") .color(Color::Muted) .size(LabelSize::XSmall), ) .tooltip(Tooltip::text(tooltip)), ) }) .when(time_elapsed > Duration::from_secs(10), |header| { header.child( Label::new(format!("({})", duration_alt_display(time_elapsed))) .buffer_font(cx) .color(Color::Muted) .size(LabelSize::XSmall), ) }) .child( Disclosure::new( SharedString::from(format!( "terminal-tool-disclosure-{}", terminal.entity_id() )), self.terminal_expanded, ) .opened_icon(IconName::ChevronUp) .closed_icon(IconName::ChevronDown) .on_click(cx.listener(move |this, _event, _window, _cx| { this.terminal_expanded = !this.terminal_expanded; })), ); let terminal_view = self .entry_view_state .entry(entry_ix) .and_then(|entry| entry.terminal(&terminal)); let show_output = self.terminal_expanded && terminal_view.is_some(); v_flex() .mb_2() .border_1() .when(tool_failed || command_failed, |card| card.border_dashed()) .border_color(border_color) .rounded_lg() .overflow_hidden() .child( v_flex() .py_1p5() .pl_2() .pr_1p5() .gap_0p5() .bg(header_bg) .text_xs() .child(header) .child( MarkdownElement::new( command.clone(), terminal_command_markdown_style(window, cx), ) .code_block_renderer( markdown::CodeBlockRenderer::Default { copy_button: false, copy_button_on_hover: true, border: false, }, ), ), ) .when(show_output, |this| { this.child( div() .pt_2() .border_t_1() .when(tool_failed || command_failed, |card| card.border_dashed()) .border_color(border_color) .bg(cx.theme().colors().editor_background) .rounded_b_md() .text_ui_sm(cx) .children(terminal_view.clone()), ) }) .into_any() } fn render_agent_logo(&self) -> AnyElement { Icon::new(self.agent.logo()) .color(Color::Muted) .size(IconSize::XLarge) .into_any_element() } fn render_error_agent_logo(&self) -> AnyElement { let logo = Icon::new(self.agent.logo()) .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, cx: &App) -> AnyElement { let loading = matches!(&self.thread_state, ThreadState::Loading { .. }); v_flex() .size_full() .items_center() .justify_center() .child(if loading { h_flex() .justify_center() .child(self.render_agent_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_agent_logo().into_any_element() }) .child(h_flex().mt_4().mb_1().justify_center().child(if loading { div() .child(LoadingLabel::new("").size(LabelSize::Large)) .into_any_element() } else { Headline::new(self.agent.empty_state_headline()) .size(HeadlineSize::Medium) .into_any_element() })) .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(self.agent.empty_state_message()), ) .into_any() } fn render_pending_auth_state(&self) -> AnyElement { v_flex() .items_center() .justify_center() .child(self.render_error_agent_logo()) .child( h_flex() .mt_4() .mb_1() .justify_center() .child(Headline::new("Not Authenticated").size(HeadlineSize::Medium)), ) .into_any() } fn render_server_exited(&self, status: ExitStatus, _cx: &Context) -> AnyElement { v_flex() .items_center() .justify_center() .child(self.render_error_agent_logo()) .child( v_flex() .mt_4() .mb_2() .gap_0p5() .text_center() .items_center() .child(Headline::new("Server exited unexpectedly").size(HeadlineSize::Medium)) .child( Label::new(format!("Exit status: {}", status.code().unwrap_or(-127))) .size(LabelSize::Small) .color(Color::Muted), ), ) .into_any_element() } fn render_load_error(&self, e: &LoadError, cx: &Context) -> AnyElement { let mut container = v_flex() .items_center() .justify_center() .child(self.render_error_agent_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 let LoadError::Unsupported { upgrade_message, upgrade_command, .. } = &e { let upgrade_message = upgrade_message.clone(); let upgrade_command = upgrade_command.clone(); container = container.child(Button::new("upgrade", upgrade_message).on_click( cx.listener(move |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 spawn_in_terminal = task::SpawnInTerminal { id: task::TaskId("install".to_string()), full_label: upgrade_command.clone(), label: upgrade_command.clone(), command: Some(upgrade_command.clone()), args: Vec::new(), command_label: upgrade_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_activity_bar( &self, thread_entity: &Entity, window: &mut Window, cx: &Context, ) -> Option { let thread = thread_entity.read(cx); let action_log = thread.action_log(); let changed_buffers = action_log.read(cx).changed_buffers(cx); let plan = thread.plan(); if changed_buffers.is_empty() && plan.is_empty() { return None; } let editor_bg_color = cx.theme().colors().editor_background; let active_color = cx.theme().colors().element_selected; let bg_edit_files_disclosure = editor_bg_color.blend(active_color.opacity(0.3)); let pending_edits = thread.has_pending_edit_tool_calls(); v_flex() .mt_1() .mx_2() .bg(bg_edit_files_disclosure) .border_1() .border_b_0() .border_color(cx.theme().colors().border) .rounded_t_md() .shadow(vec![gpui::BoxShadow { color: gpui::black().opacity(0.15), offset: point(px(1.), px(-1.)), blur_radius: px(3.), spread_radius: px(0.), }]) .when(!plan.is_empty(), |this| { this.child(self.render_plan_summary(plan, window, cx)) .when(self.plan_expanded, |parent| { parent.child(self.render_plan_entries(plan, window, cx)) }) }) .when(!plan.is_empty() && !changed_buffers.is_empty(), |this| { this.child(Divider::horizontal().color(DividerColor::Border)) }) .when(!changed_buffers.is_empty(), |this| { this.child(self.render_edits_summary( action_log, &changed_buffers, self.edits_expanded, pending_edits, window, cx, )) .when(self.edits_expanded, |parent| { parent.child(self.render_edited_files( action_log, &changed_buffers, pending_edits, cx, )) }) }) .into_any() .into() } fn render_plan_summary(&self, plan: &Plan, window: &mut Window, cx: &Context) -> Div { let stats = plan.stats(); let title = if let Some(entry) = stats.in_progress_entry && !self.plan_expanded { h_flex() .w_full() .cursor_default() .gap_1() .text_xs() .text_color(cx.theme().colors().text_muted) .justify_between() .child( h_flex() .gap_1() .child( Label::new("Current:") .size(LabelSize::Small) .color(Color::Muted), ) .child(MarkdownElement::new( entry.content.clone(), plan_label_markdown_style(&entry.status, window, cx), )), ) .when(stats.pending > 0, |this| { this.child( Label::new(format!("{} left", stats.pending)) .size(LabelSize::Small) .color(Color::Muted) .mr_1(), ) }) } else { let status_label = if stats.pending == 0 { "All Done".to_string() } else if stats.completed == 0 { format!("{} Tasks", plan.entries.len()) } else { format!("{}/{}", stats.completed, plan.entries.len()) }; h_flex() .w_full() .gap_1() .justify_between() .child( Label::new("Plan") .size(LabelSize::Small) .color(Color::Muted), ) .child( Label::new(status_label) .size(LabelSize::Small) .color(Color::Muted) .mr_1(), ) }; h_flex() .p_1() .justify_between() .when(self.plan_expanded, |this| { this.border_b_1().border_color(cx.theme().colors().border) }) .child( h_flex() .id("plan_summary") .w_full() .gap_1() .child(Disclosure::new("plan_disclosure", self.plan_expanded)) .child(title) .on_click(cx.listener(|this, _, _, cx| { this.plan_expanded = !this.plan_expanded; cx.notify(); })), ) } fn render_plan_entries(&self, plan: &Plan, window: &mut Window, cx: &Context) -> Div { v_flex().children(plan.entries.iter().enumerate().flat_map(|(index, entry)| { let element = h_flex() .py_1() .px_2() .gap_2() .justify_between() .bg(cx.theme().colors().editor_background) .when(index < plan.entries.len() - 1, |parent| { parent.border_color(cx.theme().colors().border).border_b_1() }) .child( h_flex() .id(("plan_entry", index)) .gap_1p5() .max_w_full() .overflow_x_scroll() .text_xs() .text_color(cx.theme().colors().text_muted) .child(match entry.status { acp::PlanEntryStatus::Pending => Icon::new(IconName::TodoPending) .size(IconSize::Small) .color(Color::Muted) .into_any_element(), acp::PlanEntryStatus::InProgress => Icon::new(IconName::TodoProgress) .size(IconSize::Small) .color(Color::Accent) .with_animation( "running", Animation::new(Duration::from_secs(2)).repeat(), |icon, delta| { icon.transform(Transformation::rotate(percentage(delta))) }, ) .into_any_element(), acp::PlanEntryStatus::Completed => Icon::new(IconName::TodoComplete) .size(IconSize::Small) .color(Color::Success) .into_any_element(), }) .child(MarkdownElement::new( entry.content.clone(), plan_label_markdown_style(&entry.status, window, cx), )), ); Some(element) })) } fn render_edits_summary( &self, action_log: &Entity, changed_buffers: &BTreeMap, Entity>, expanded: bool, pending_edits: bool, window: &mut Window, cx: &Context, ) -> Div { const EDIT_NOT_READY_TOOLTIP_LABEL: &str = "Wait until file edits are complete."; let focus_handle = self.focus_handle(cx); h_flex() .p_1() .justify_between() .when(expanded, |this| { this.border_b_1().border_color(cx.theme().colors().border) }) .child( h_flex() .id("edits-container") .w_full() .gap_1() .child(Disclosure::new("edits-disclosure", expanded)) .map(|this| { if pending_edits { this.child( Label::new(format!( "Editing {} {}…", changed_buffers.len(), if changed_buffers.len() == 1 { "file" } else { "files" } )) .color(Color::Muted) .size(LabelSize::Small) .with_animation( "edit-label", Animation::new(Duration::from_secs(2)) .repeat() .with_easing(pulsating_between(0.3, 0.7)), |label, delta| label.alpha(delta), ), ) } else { this.child( Label::new("Edits") .size(LabelSize::Small) .color(Color::Muted), ) .child(Label::new("•").size(LabelSize::XSmall).color(Color::Muted)) .child( Label::new(format!( "{} {}", changed_buffers.len(), if changed_buffers.len() == 1 { "file" } else { "files" } )) .size(LabelSize::Small) .color(Color::Muted), ) } }) .on_click(cx.listener(|this, _, _, cx| { this.edits_expanded = !this.edits_expanded; cx.notify(); })), ) .child( h_flex() .gap_1() .child( IconButton::new("review-changes", IconName::ListTodo) .icon_size(IconSize::Small) .tooltip({ let focus_handle = focus_handle.clone(); move |window, cx| { Tooltip::for_action_in( "Review Changes", &OpenAgentDiff, &focus_handle, window, cx, ) } }) .on_click(cx.listener(|_, _, window, cx| { window.dispatch_action(OpenAgentDiff.boxed_clone(), cx); })), ) .child(Divider::vertical().color(DividerColor::Border)) .child( Button::new("reject-all-changes", "Reject All") .label_size(LabelSize::Small) .disabled(pending_edits) .when(pending_edits, |this| { this.tooltip(Tooltip::text(EDIT_NOT_READY_TOOLTIP_LABEL)) }) .key_binding( KeyBinding::for_action_in( &RejectAll, &focus_handle.clone(), window, cx, ) .map(|kb| kb.size(rems_from_px(10.))), ) .on_click({ let action_log = action_log.clone(); cx.listener(move |_, _, _, cx| { action_log.update(cx, |action_log, cx| { action_log.reject_all_edits(cx).detach(); }) }) }), ) .child( Button::new("keep-all-changes", "Keep All") .label_size(LabelSize::Small) .disabled(pending_edits) .when(pending_edits, |this| { this.tooltip(Tooltip::text(EDIT_NOT_READY_TOOLTIP_LABEL)) }) .key_binding( KeyBinding::for_action_in(&KeepAll, &focus_handle, window, cx) .map(|kb| kb.size(rems_from_px(10.))), ) .on_click({ let action_log = action_log.clone(); cx.listener(move |_, _, _, cx| { action_log.update(cx, |action_log, cx| { action_log.keep_all_edits(cx); }) }) }), ), ) } fn render_edited_files( &self, action_log: &Entity, changed_buffers: &BTreeMap, Entity>, pending_edits: bool, cx: &Context, ) -> Div { let editor_bg_color = cx.theme().colors().editor_background; v_flex().children(changed_buffers.into_iter().enumerate().flat_map( |(index, (buffer, _diff))| { let file = buffer.read(cx).file()?; let path = file.path(); let file_path = path.parent().and_then(|parent| { let parent_str = parent.to_string_lossy(); if parent_str.is_empty() { None } else { Some( Label::new(format!("/{}{}", parent_str, std::path::MAIN_SEPARATOR_STR)) .color(Color::Muted) .size(LabelSize::XSmall) .buffer_font(cx), ) } }); let file_name = path.file_name().map(|name| { Label::new(name.to_string_lossy().to_string()) .size(LabelSize::XSmall) .buffer_font(cx) }); let file_icon = FileIcons::get_icon(&path, cx) .map(Icon::from_path) .map(|icon| icon.color(Color::Muted).size(IconSize::Small)) .unwrap_or_else(|| { Icon::new(IconName::File) .color(Color::Muted) .size(IconSize::Small) }); let overlay_gradient = linear_gradient( 90., linear_color_stop(editor_bg_color, 1.), linear_color_stop(editor_bg_color.opacity(0.2), 0.), ); let element = h_flex() .group("edited-code") .id(("file-container", index)) .relative() .py_1() .pl_2() .pr_1() .gap_2() .justify_between() .bg(editor_bg_color) .when(index < changed_buffers.len() - 1, |parent| { parent.border_color(cx.theme().colors().border).border_b_1() }) .child( h_flex() .id(("file-name", index)) .pr_8() .gap_1p5() .max_w_full() .overflow_x_scroll() .child(file_icon) .child(h_flex().gap_0p5().children(file_name).children(file_path)) .on_click({ let buffer = buffer.clone(); cx.listener(move |this, _, window, cx| { this.open_edited_buffer(&buffer, window, cx); }) }), ) .child( h_flex() .gap_1() .visible_on_hover("edited-code") .child( Button::new("review", "Review") .label_size(LabelSize::Small) .on_click({ let buffer = buffer.clone(); cx.listener(move |this, _, window, cx| { this.open_edited_buffer(&buffer, window, cx); }) }), ) .child(Divider::vertical().color(DividerColor::BorderVariant)) .child( Button::new("reject-file", "Reject") .label_size(LabelSize::Small) .disabled(pending_edits) .on_click({ let buffer = buffer.clone(); let action_log = action_log.clone(); move |_, _, cx| { action_log.update(cx, |action_log, cx| { action_log .reject_edits_in_ranges( buffer.clone(), vec![Anchor::MIN..Anchor::MAX], cx, ) .detach_and_log_err(cx); }) } }), ) .child( Button::new("keep-file", "Keep") .label_size(LabelSize::Small) .disabled(pending_edits) .on_click({ let buffer = buffer.clone(); let action_log = action_log.clone(); move |_, _, cx| { action_log.update(cx, |action_log, cx| { action_log.keep_edits_in_range( buffer.clone(), Anchor::MIN..Anchor::MAX, cx, ); }) } }), ), ) .child( div() .id("gradient-overlay") .absolute() .h_full() .w_12() .top_0() .bottom_0() .right(px(152.)) .bg(overlay_gradient), ); Some(element) }, )) } fn render_message_editor(&mut self, window: &mut Window, cx: &mut Context) -> AnyElement { let focus_handle = self.message_editor.focus_handle(cx); let editor_bg_color = cx.theme().colors().editor_background; let (expand_icon, expand_tooltip) = if self.editor_expanded { (IconName::Minimize, "Minimize Message Editor") } else { (IconName::Maximize, "Expand Message Editor") }; v_flex() .on_action(cx.listener(Self::expand_message_editor)) .on_action(cx.listener(|this, _: &ToggleModelSelector, window, cx| { if let Some(model_selector) = this.model_selector.as_ref() { model_selector .update(cx, |model_selector, cx| model_selector.toggle(window, cx)); } })) .p_2() .gap_2() .border_t_1() .border_color(cx.theme().colors().border) .bg(editor_bg_color) .when(self.editor_expanded, |this| { this.h(vh(0.8, window)).size_full().justify_between() }) .child( v_flex() .relative() .size_full() .pt_1() .pr_2p5() .child(self.message_editor.clone()) .child( h_flex() .absolute() .top_0() .right_0() .opacity(0.5) .hover(|this| this.opacity(1.0)) .child( IconButton::new("toggle-height", expand_icon) .icon_size(IconSize::Small) .icon_color(Color::Muted) .tooltip({ let focus_handle = focus_handle.clone(); move |window, cx| { Tooltip::for_action_in( expand_tooltip, &ExpandMessageEditor, &focus_handle, window, cx, ) } }) .on_click(cx.listener(|_, _, window, cx| { window.dispatch_action(Box::new(ExpandMessageEditor), cx); })), ), ), ) .child( h_flex() .flex_none() .justify_between() .child(self.render_follow_toggle(cx)) .child( h_flex() .gap_1() .children(self.model_selector.clone()) .child(self.render_send_button(cx)), ), ) .into_any() } fn start_editing_message(&mut self, index: usize, window: &mut Window, cx: &mut Context) { let Some(thread) = self.thread() else { return; }; let Some(AgentThreadEntry::UserMessage(message)) = thread.read(cx).entries().get(index) else { return; }; let Some(message_id) = message.id.clone() else { return; }; self.list_state.scroll_to_reveal_item(index); let chunks = message.chunks.clone(); let editor = cx.new(|cx| { let mut editor = MessageEditor::new( self.workspace.clone(), self.project.clone(), self.thread_store.clone(), self.text_thread_store.clone(), editor::EditorMode::AutoHeight { min_lines: 1, max_lines: None, }, window, cx, ); editor.set_message(chunks, window, cx); editor }); let subscription = cx.subscribe_in(&editor, window, |this, _, event, window, cx| match event { MessageEditorEvent::Send => { this.regenerate(&Default::default(), window, cx); } MessageEditorEvent::Cancel => { this.cancel_editing(&Default::default(), window, cx); } }); editor.focus_handle(cx).focus(window); self.editing_message.replace(EditingMessage { index: index, message_id: message_id.clone(), editor, _subscription: subscription, }); cx.notify(); } fn render_edit_message_editor(&self, editing: &EditingMessage, cx: &Context) -> Div { v_flex() .w_full() .gap_2() .child(editing.editor.clone()) .child( h_flex() .gap_1() .child( Icon::new(IconName::Warning) .color(Color::Warning) .size(IconSize::XSmall), ) .child( Label::new("Editing will restart the thread from this point.") .color(Color::Muted) .size(LabelSize::XSmall), ) .child(self.render_editing_message_editor_buttons(editing, cx)), ) } fn render_editing_message_editor_buttons( &self, editing: &EditingMessage, cx: &Context, ) -> Div { h_flex() .gap_0p5() .flex_1() .justify_end() .child( IconButton::new("cancel-edit-message", IconName::Close) .shape(ui::IconButtonShape::Square) .icon_color(Color::Error) .icon_size(IconSize::Small) .tooltip({ let focus_handle = editing.editor.focus_handle(cx); move |window, cx| { Tooltip::for_action_in( "Cancel Edit", &menu::Cancel, &focus_handle, window, cx, ) } }) .on_click(cx.listener(Self::cancel_editing)), ) .child( IconButton::new("confirm-edit-message", IconName::Return) .disabled(editing.editor.read(cx).is_empty(cx)) .shape(ui::IconButtonShape::Square) .icon_color(Color::Muted) .icon_size(IconSize::Small) .tooltip({ let focus_handle = editing.editor.focus_handle(cx); move |window, cx| { Tooltip::for_action_in( "Regenerate", &menu::Confirm, &focus_handle, window, cx, ) } }) .on_click(cx.listener(Self::regenerate)), ) } fn render_send_button(&self, cx: &mut Context) -> AnyElement { if self.thread().map_or(true, |thread| { thread.read(cx).status() == ThreadStatus::Idle }) { let is_editor_empty = self.message_editor.read(cx).is_empty(cx); IconButton::new("send-message", IconName::Send) .icon_color(Color::Accent) .style(ButtonStyle::Filled) .disabled(self.thread().is_none() || is_editor_empty) .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")) }) .on_click(cx.listener(|this, _, window, cx| { this.send(window, cx); })) .into_any_element() } else { IconButton::new("stop-generation", IconName::Stop) .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_generation(cx))) .into_any_element() } } fn render_follow_toggle(&self, cx: &mut Context) -> impl IntoElement { let following = self .workspace .read_with(cx, |workspace, _| { workspace.is_being_followed(CollaboratorId::Agent) }) .unwrap_or(false); IconButton::new("follow-agent", IconName::Crosshair) .icon_size(IconSize::Small) .icon_color(Color::Muted) .toggle_state(following) .selected_icon_color(Some(Color::Custom(cx.theme().players().agent().cursor))) .tooltip(move |window, cx| { if following { Tooltip::for_action("Stop Following Agent", &Follow, window, cx) } else { Tooltip::with_meta( "Follow Agent", Some(&Follow), "Track the agent's location as it reads and edits files.", window, cx, ) } }) .on_click(cx.listener(move |this, _, window, cx| { this.workspace .update(cx, |workspace, cx| { if following { workspace.unfollow(CollaboratorId::Agent, window, cx); } else { workspace.follow(CollaboratorId::Agent, window, cx); } }) .ok(); })) } fn render_markdown(&self, markdown: Entity, 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, window: &mut Window, cx: &mut App, ) { let Some(workspace) = workspace.upgrade() else { cx.open_url(&url); return; }; if let Some(mention) = MentionUri::parse(&url).log_err() { workspace.update(cx, |workspace, cx| match mention { MentionUri::File { abs_path, .. } => { let project = workspace.project(); let Some((path, entry)) = project.update(cx, |project, cx| { let path = project.find_project_path(abs_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); } } MentionUri::Symbol { path, line_range, .. } | MentionUri::Selection { path, line_range } => { let project = workspace.project(); let Some((path, _)) = project.update(cx, |project, cx| { let path = project.find_project_path(path, cx)?; let entry = project.entry_for_path(&path, cx)?; Some((path, entry)) }) else { return; }; let item = workspace.open_path(path, None, true, window, cx); window .spawn(cx, async move |cx| { let Some(editor) = item.await?.downcast::() else { return Ok(()); }; let range = Point::new(line_range.start, 0)..Point::new(line_range.start, 0); editor .update_in(cx, |editor, window, cx| { editor.change_selections( SelectionEffects::scroll(Autoscroll::center()), window, cx, |s| s.select_ranges(vec![range]), ); }) .ok(); anyhow::Ok(()) }) .detach_and_log_err(cx); } MentionUri::Thread { id, .. } => { if let Some(panel) = workspace.panel::(cx) { panel.update(cx, |panel, cx| { panel .open_thread_by_id(&id, window, cx) .detach_and_log_err(cx) }); } } MentionUri::TextThread { path, .. } => { if let Some(panel) = workspace.panel::(cx) { panel.update(cx, |panel, cx| { panel .open_saved_prompt_editor(path.as_path().into(), window, cx) .detach_and_log_err(cx); }); } } MentionUri::Rule { id, .. } => { let PromptId::User { uuid } = id else { return; }; window.dispatch_action( Box::new(OpenRulesLibrary { prompt_to_select: Some(uuid.0), }), cx, ) } MentionUri::Fetch { url } => { cx.open_url(url.as_str()); } }) } else { cx.open_url(&url); } } fn open_tool_call_location( &self, entry_ix: usize, location_ix: usize, window: &mut Window, cx: &mut Context, ) -> Option<()> { let (tool_call_location, agent_location) = self .thread()? .read(cx) .entries() .get(entry_ix)? .location(location_ix)?; let project_path = self .project .read(cx) .find_project_path(&tool_call_location.path, cx)?; let open_task = self .workspace .update(cx, |workspace, cx| { workspace.open_path(project_path, None, true, window, cx) }) .log_err()?; window .spawn(cx, async move |cx| { let item = open_task.await?; let Some(active_editor) = item.downcast::() else { return anyhow::Ok(()); }; active_editor.update_in(cx, |editor, window, cx| { let multibuffer = editor.buffer().read(cx); let buffer = multibuffer.as_singleton(); if agent_location.buffer.upgrade() == buffer { let excerpt_id = multibuffer.excerpt_ids().first().cloned(); let anchor = editor::Anchor::in_buffer( excerpt_id.unwrap(), buffer.unwrap().read(cx).remote_id(), agent_location.position, ); editor.change_selections(Default::default(), window, cx, |selections| { selections.select_anchor_ranges([anchor..anchor]); }) } else { let row = tool_call_location.line.unwrap_or_default(); editor.change_selections(Default::default(), window, cx, |selections| { selections.select_ranges([Point::new(row, 0)..Point::new(row, 0)]); }) } })?; anyhow::Ok(()) }) .detach_and_log_err(cx); None } pub fn open_thread_as_markdown( &self, workspace: Entity, window: &mut Window, cx: &mut App, ) -> Task> { let markdown_language_task = workspace .read(cx) .app_state() .languages .language_for_name("Markdown"); let (thread_summary, markdown) = if let Some(thread) = self.thread() { let thread = thread.read(cx); (thread.title().to_string(), thread.to_markdown(cx)) } else { 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() { 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.list_state.scroll_to(ListOffset::default()); cx.notify(); } pub fn scroll_to_bottom(&mut self, cx: &mut Context) { if let Some(thread) = self.thread() { let entry_count = thread.read(cx).entries().len(); self.list_state.reset(entry_count); cx.notify(); } } fn notify_with_sound( &mut self, caption: impl Into, icon: IconName, window: &mut Window, cx: &mut Context, ) { self.play_notification_sound(window, cx); self.show_notification(caption, icon, window, cx); } fn play_notification_sound(&self, window: &Window, cx: &mut App) { let settings = AgentSettings::get_global(cx); if settings.play_sound_when_agent_done && !window.is_window_active() { Audio::play_sound(Sound::AgentDone, cx); } } fn show_notification( &mut self, caption: impl Into, icon: IconName, window: &mut Window, cx: &mut Context, ) { if window.is_window_active() || !self.notifications.is_empty() { return; } let title = self.title(cx); match AgentSettings::get_global(cx).notify_when_agent_waiting { NotifyWhenAgentWaiting::PrimaryScreen => { if let Some(primary) = cx.primary_display() { self.pop_up(icon, caption.into(), title, window, primary, cx); } } NotifyWhenAgentWaiting::AllScreens => { let caption = caption.into(); for screen in cx.displays() { self.pop_up(icon, caption.clone(), title.clone(), window, screen, cx); } } NotifyWhenAgentWaiting::Never => { // Don't show anything } } } fn pop_up( &mut self, icon: IconName, caption: SharedString, title: SharedString, window: &mut Window, screen: Rc, cx: &mut Context, ) { let options = AgentNotification::window_options(screen, cx); let project_name = self.workspace.upgrade().and_then(|workspace| { workspace .read(cx) .project() .read(cx) .visible_worktrees(cx) .next() .map(|worktree| worktree.read(cx).root_name().to_string()) }); if let Some(screen_window) = cx .open_window(options, |_, cx| { cx.new(|_| { AgentNotification::new(title.clone(), caption.clone(), icon, project_name) }) }) .log_err() { if let Some(pop_up) = screen_window.entity(cx).log_err() { self.notification_subscriptions .entry(screen_window) .or_insert_with(Vec::new) .push(cx.subscribe_in(&pop_up, window, { |this, _, event, window, cx| match event { AgentNotificationEvent::Accepted => { let handle = window.window_handle(); cx.activate(true); let workspace_handle = this.workspace.clone(); // If there are multiple Zed windows, activate the correct one. cx.defer(move |cx| { handle .update(cx, |_view, window, _cx| { window.activate_window(); if let Some(workspace) = workspace_handle.upgrade() { workspace.update(_cx, |workspace, cx| { workspace.focus_panel::(window, cx); }); } }) .log_err(); }); this.dismiss_notifications(cx); } AgentNotificationEvent::Dismissed => { this.dismiss_notifications(cx); } } })); self.notifications.push(screen_window); // If the user manually refocuses the original window, dismiss the popup. self.notification_subscriptions .entry(screen_window) .or_insert_with(Vec::new) .push({ let pop_up_weak = pop_up.downgrade(); cx.observe_window_activation(window, move |_, window, cx| { if window.is_window_active() { if let Some(pop_up) = pop_up_weak.upgrade() { pop_up.update(cx, |_, cx| { cx.emit(AgentNotificationEvent::Dismissed); }); } } }) }); } } } fn dismiss_notifications(&mut self, cx: &mut Context) { for window in self.notifications.drain(..) { window .update(cx, |_, window, _| { window.remove_window(); }) .ok(); self.notification_subscriptions.remove(&window); } } fn render_thread_controls(&self, cx: &Context) -> impl IntoElement { let open_as_markdown = IconButton::new("open-as-markdown", IconName::FileMarkdown) .shape(ui::IconButtonShape::Square) .icon_size(IconSize::Small) .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::ArrowUp) .shape(ui::IconButtonShape::Square) .icon_size(IconSize::Small) .icon_color(Color::Ignored) .tooltip(Tooltip::text("Scroll To Top")) .on_click(cx.listener(move |this, _, _, cx| { this.scroll_to_top(cx); })); h_flex() .w_full() .mr_1() .pb_2() .px(RESPONSE_PADDING_X) .opacity(0.4) .hover(|style| style.opacity(1.)) .flex_wrap() .justify_end() .child(open_as_markdown) .child(scroll_to_top) } fn render_vertical_scrollbar(&self, cx: &mut Context) -> Stateful
{ div() .id("acp-thread-scrollbar") .occlude() .on_mouse_move(cx.listener(|_, _, _, cx| { cx.notify(); cx.stop_propagation() })) .on_hover(|_, _, cx| { cx.stop_propagation(); }) .on_any_mouse_down(|_, _, cx| { cx.stop_propagation(); }) .on_mouse_up( MouseButton::Left, cx.listener(|_, _, _, cx| { cx.stop_propagation(); }), ) .on_scroll_wheel(cx.listener(|_, _, _, cx| { cx.notify(); })) .h_full() .absolute() .right_1() .top_1() .bottom_0() .w(px(12.)) .cursor_default() .children(Scrollbar::vertical(self.scrollbar_state.clone()).map(|s| s.auto_hide(cx))) } fn settings_changed(&mut self, _window: &mut Window, cx: &mut Context) { self.entry_view_state.settings_changed(cx); } pub(crate) fn insert_dragged_files( &self, paths: Vec, added_worktrees: Vec>, window: &mut Window, cx: &mut Context, ) { self.message_editor.update(cx, |message_editor, cx| { message_editor.insert_dragged_files(paths, window, cx); drop(added_worktrees); }) } } 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) -> impl IntoElement { let has_messages = self.list_state.item_count() > 0; v_flex() .size_full() .key_context("AcpThread") .on_action(cx.listener(Self::open_agent_diff)) .bg(cx.theme().colors().panel_background) .child(match &self.thread_state { ThreadState::Unauthenticated { connection } => v_flex() .p_2() .flex_1() .items_center() .justify_center() .child(self.render_pending_auth_state()) .child(h_flex().mt_1p5().justify_center().children( connection.auth_methods().into_iter().map(|method| { Button::new( SharedString::from(method.id.0.clone()), method.name.clone(), ) .on_click({ let method_id = method.id.clone(); cx.listener(move |this, _, window, cx| { this.authenticate(method_id.clone(), window, cx) }) }) }), )), ThreadState::Loading { .. } => v_flex().flex_1().child(self.render_empty_state(cx)), ThreadState::LoadError(e) => v_flex() .p_2() .flex_1() .items_center() .justify_center() .child(self.render_load_error(e, cx)), ThreadState::ServerExited { status } => v_flex() .p_2() .flex_1() .items_center() .justify_center() .child(self.render_server_exited(*status, cx)), ThreadState::Ready { thread, .. } => { let thread_clone = thread.clone(); v_flex().flex_1().map(|this| { if has_messages { this.child( list( self.list_state.clone(), cx.processor(|this, 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) }), ) .with_sizing_behavior(gpui::ListSizingBehavior::Auto) .flex_grow() .into_any(), ) .child(self.render_vertical_scrollbar(cx)) .children( match thread_clone.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(cx)) } }) } }) // The activity bar is intentionally rendered outside of the ThreadState::Ready match // above so that the scrollbar doesn't render behind it. The current setup allows // the scrollbar to stop exactly at the activity bar start. .when(has_messages, |this| match &self.thread_state { ThreadState::Ready { thread, .. } => { this.children(self.render_activity_bar(thread, window, cx)) } _ => this, }) .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(self.render_message_editor(window, 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 MentionUri::parse(url).is_ok() { 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() } } fn plan_label_markdown_style( status: &acp::PlanEntryStatus, window: &Window, cx: &App, ) -> MarkdownStyle { let default_md_style = default_markdown_style(false, window, cx); MarkdownStyle { base_text_style: TextStyle { color: cx.theme().colors().text_muted, strikethrough: if matches!(status, acp::PlanEntryStatus::Completed) { Some(gpui::StrikethroughStyle { thickness: px(1.), color: Some(cx.theme().colors().text_muted.opacity(0.8)), }) } else { None }, ..default_md_style.base_text_style }, ..default_md_style } } fn terminal_command_markdown_style(window: &Window, cx: &App) -> MarkdownStyle { let default_md_style = default_markdown_style(true, window, cx); MarkdownStyle { base_text_style: TextStyle { ..default_md_style.base_text_style }, selection_background_color: cx.theme().colors().element_selection_background, ..Default::default() } } #[cfg(test)] pub(crate) mod tests { use std::path::Path; use acp_thread::StubAgentConnection; use agent::{TextThreadStore, ThreadStore}; use agent_client_protocol::SessionId; use editor::EditorSettings; use fs::FakeFs; use gpui::{SemanticVersion, TestAppContext, VisualTestContext}; use project::Project; use serde_json::json; use settings::SettingsStore; use super::*; #[gpui::test] async fn test_drop(cx: &mut TestAppContext) { init_test(cx); let (thread_view, _cx) = setup_thread_view(StubAgentServer::default(), cx).await; let weak_view = thread_view.downgrade(); drop(thread_view); assert!(!weak_view.is_upgradable()); } #[gpui::test] async fn test_notification_for_stop_event(cx: &mut TestAppContext) { init_test(cx); let (thread_view, cx) = setup_thread_view(StubAgentServer::default(), cx).await; let message_editor = cx.read(|cx| thread_view.read(cx).message_editor.clone()); message_editor.update_in(cx, |editor, window, cx| { editor.set_text("Hello", window, cx); }); cx.deactivate_window(); thread_view.update_in(cx, |thread_view, window, cx| { thread_view.send(window, cx); }); cx.run_until_parked(); assert!( cx.windows() .iter() .any(|window| window.downcast::().is_some()) ); } #[gpui::test] async fn test_notification_for_error(cx: &mut TestAppContext) { init_test(cx); let (thread_view, cx) = setup_thread_view(StubAgentServer::new(SaboteurAgentConnection), cx).await; let message_editor = cx.read(|cx| thread_view.read(cx).message_editor.clone()); message_editor.update_in(cx, |editor, window, cx| { editor.set_text("Hello", window, cx); }); cx.deactivate_window(); thread_view.update_in(cx, |thread_view, window, cx| { thread_view.send(window, cx); }); cx.run_until_parked(); assert!( cx.windows() .iter() .any(|window| window.downcast::().is_some()) ); } #[gpui::test] async fn test_notification_for_tool_authorization(cx: &mut TestAppContext) { init_test(cx); let tool_call_id = acp::ToolCallId("1".into()); let tool_call = acp::ToolCall { id: tool_call_id.clone(), title: "Label".into(), kind: acp::ToolKind::Edit, status: acp::ToolCallStatus::Pending, content: vec!["hi".into()], locations: vec![], raw_input: None, raw_output: None, }; let connection = StubAgentConnection::new().with_permission_requests(HashMap::from_iter([( tool_call_id, vec![acp::PermissionOption { id: acp::PermissionOptionId("1".into()), name: "Allow".into(), kind: acp::PermissionOptionKind::AllowOnce, }], )])); connection.set_next_prompt_updates(vec![acp::SessionUpdate::ToolCall(tool_call)]); let (thread_view, cx) = setup_thread_view(StubAgentServer::new(connection), cx).await; let message_editor = cx.read(|cx| thread_view.read(cx).message_editor.clone()); message_editor.update_in(cx, |editor, window, cx| { editor.set_text("Hello", window, cx); }); cx.deactivate_window(); thread_view.update_in(cx, |thread_view, window, cx| { thread_view.send(window, cx); }); cx.run_until_parked(); assert!( cx.windows() .iter() .any(|window| window.downcast::().is_some()) ); } async fn setup_thread_view( agent: impl AgentServer + 'static, cx: &mut TestAppContext, ) -> (Entity, &mut VisualTestContext) { let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let thread_store = cx.update(|_window, cx| cx.new(|cx| ThreadStore::fake(project.clone(), cx))); let text_thread_store = cx.update(|_window, cx| cx.new(|cx| TextThreadStore::fake(project.clone(), cx))); let thread_view = cx.update(|window, cx| { cx.new(|cx| { AcpThreadView::new( Rc::new(agent), workspace.downgrade(), project, thread_store.clone(), text_thread_store.clone(), window, cx, ) }) }); cx.run_until_parked(); (thread_view, cx) } struct StubAgentServer { connection: C, } impl StubAgentServer { fn new(connection: C) -> Self { Self { connection } } } impl StubAgentServer { fn default() -> Self { Self::new(StubAgentConnection::default()) } } impl AgentServer for StubAgentServer where C: 'static + AgentConnection + Send + Clone, { fn logo(&self) -> ui::IconName { unimplemented!() } fn name(&self) -> &'static str { unimplemented!() } fn empty_state_headline(&self) -> &'static str { unimplemented!() } fn empty_state_message(&self) -> &'static str { unimplemented!() } fn connect( &self, _root_dir: &Path, _project: &Entity, _cx: &mut App, ) -> Task>> { Task::ready(Ok(Rc::new(self.connection.clone()))) } } #[derive(Clone)] struct SaboteurAgentConnection; impl AgentConnection for SaboteurAgentConnection { fn new_thread( self: Rc, project: Entity, _cwd: &Path, cx: &mut gpui::App, ) -> Task>> { Task::ready(Ok(cx.new(|cx| { AcpThread::new( "SaboteurAgentConnection", self, project, SessionId("test".into()), cx, ) }))) } fn auth_methods(&self) -> &[acp::AuthMethod] { &[] } fn authenticate( &self, _method_id: acp::AuthMethodId, _cx: &mut App, ) -> Task> { unimplemented!() } fn prompt( &self, _id: Option, _params: acp::PromptRequest, _cx: &mut App, ) -> Task> { Task::ready(Err(anyhow::anyhow!("Error prompting"))) } fn cancel(&self, _session_id: &acp::SessionId, _cx: &mut App) { unimplemented!() } } pub(crate) fn init_test(cx: &mut TestAppContext) { cx.update(|cx| { let settings_store = SettingsStore::test(cx); cx.set_global(settings_store); language::init(cx); Project::init_settings(cx); AgentSettings::register(cx); workspace::init_settings(cx); ThemeSettings::register(cx); release_channel::init(SemanticVersion::default(), cx); EditorSettings::register(cx); }); } #[gpui::test] async fn test_rewind_views(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); fs.insert_tree( "/project", json!({ "test1.txt": "old content 1", "test2.txt": "old content 2" }), ) .await; let project = Project::test(fs, [Path::new("/project")], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let thread_store = cx.update(|_window, cx| cx.new(|cx| ThreadStore::fake(project.clone(), cx))); let text_thread_store = cx.update(|_window, cx| cx.new(|cx| TextThreadStore::fake(project.clone(), cx))); let connection = Rc::new(StubAgentConnection::new()); let thread_view = cx.update(|window, cx| { cx.new(|cx| { AcpThreadView::new( Rc::new(StubAgentServer::new(connection.as_ref().clone())), workspace.downgrade(), project.clone(), thread_store.clone(), text_thread_store.clone(), window, cx, ) }) }); cx.run_until_parked(); let thread = thread_view .read_with(cx, |view, _| view.thread().cloned()) .unwrap(); // First user message connection.set_next_prompt_updates(vec![acp::SessionUpdate::ToolCall(acp::ToolCall { id: acp::ToolCallId("tool1".into()), title: "Edit file 1".into(), kind: acp::ToolKind::Edit, status: acp::ToolCallStatus::Completed, content: vec![acp::ToolCallContent::Diff { diff: acp::Diff { path: "/project/test1.txt".into(), old_text: Some("old content 1".into()), new_text: "new content 1".into(), }, }], locations: vec![], raw_input: None, raw_output: None, })]); thread .update(cx, |thread, cx| thread.send_raw("Give me a diff", cx)) .await .unwrap(); cx.run_until_parked(); thread.read_with(cx, |thread, _| { assert_eq!(thread.entries().len(), 2); }); thread_view.read_with(cx, |view, _| { assert_eq!(view.entry_view_state.entry(0).unwrap().len(), 0); assert_eq!(view.entry_view_state.entry(1).unwrap().len(), 1); }); // Second user message connection.set_next_prompt_updates(vec![acp::SessionUpdate::ToolCall(acp::ToolCall { id: acp::ToolCallId("tool2".into()), title: "Edit file 2".into(), kind: acp::ToolKind::Edit, status: acp::ToolCallStatus::Completed, content: vec![acp::ToolCallContent::Diff { diff: acp::Diff { path: "/project/test2.txt".into(), old_text: Some("old content 2".into()), new_text: "new content 2".into(), }, }], locations: vec![], raw_input: None, raw_output: None, })]); thread .update(cx, |thread, cx| thread.send_raw("Another one", cx)) .await .unwrap(); cx.run_until_parked(); let second_user_message_id = thread.read_with(cx, |thread, _| { assert_eq!(thread.entries().len(), 4); let AgentThreadEntry::UserMessage(user_message) = thread.entries().get(2).unwrap() else { panic!(); }; user_message.id.clone().unwrap() }); thread_view.read_with(cx, |view, _| { assert_eq!(view.entry_view_state.entry(0).unwrap().len(), 0); assert_eq!(view.entry_view_state.entry(1).unwrap().len(), 1); assert_eq!(view.entry_view_state.entry(2).unwrap().len(), 0); assert_eq!(view.entry_view_state.entry(3).unwrap().len(), 1); }); // Rewind to first message thread .update(cx, |thread, cx| thread.rewind(second_user_message_id, cx)) .await .unwrap(); cx.run_until_parked(); thread.read_with(cx, |thread, _| { assert_eq!(thread.entries().len(), 2); }); thread_view.read_with(cx, |view, _| { assert_eq!(view.entry_view_state.entry(0).unwrap().len(), 0); assert_eq!(view.entry_view_state.entry(1).unwrap().len(), 1); // Old views should be dropped assert!(view.entry_view_state.entry(2).is_none()); assert!(view.entry_view_state.entry(3).is_none()); }); } }