Josh Piasecki
057c3c1206
I love keybindings. I spend way to much time thinking about them. I also REALLY like working in Zed. so far, however, I have found the key context system in Zed to be less flexible than in VSCode. the HUGE context that is available in VSCode helps you create keybindings for very specific targeted scenarios. the tree like structure of the Zed key context means you loose some information as focus moves throughout the application. For example, it is not currently possible to create a keybinding in the editor that will only work when one of the Docks is open, or if a specific dock is open. this would be useful in implementing solutions to ideas like #24222 we already have an action for moving focus to the dock, and we have an action for opening/closing the dock, but to my knowledge (very limited lol) we cannot determine if that dock *is open* unless we are focused on it. I think it is possible to create a more flexible key binding system by adding more context information to the higher up context ancestors. while: ``` Workspace right_dock=GitPanel Dock GitPanel Editor ``` may seem redundant, it actually communicates fundamentally different information than: ``` Workspace right_dock=GitPanel Pane Editor ``` the first says "the GitPanel is in the right hand dock AND IT IS FOCUSED", while the second means "Focus is on the Editor, and the GitPanel just happens to be open in the right hand dock" This change adds a new set of identifiers to the `Workspace` key_context that will indicate which docks are open and what is the specific panel that is currently visible in that dock. examples: - `left_dock=ProjectPanel` - `bottom_dock=TerminalPanel` - `right_dock=GitPanel` in my testing the following types of keybindings seem to be supported with this change: ```jsonc // match for any value of the identifier "context": "Workspace && bottom_dock" "context": "Workspace && !bottom_dock" // match only a specific value to an identifier "context": "Workspace && bottom_dock=TerminalPanel" // match only in a child context if the ancestor workspace has the correct identifier "context": "Workspace && !bottom_dock=DebugPanel > Editor" ``` some screen shots of the context matching in different circumstances: <img width="2032" height="1167" alt="Screenshot 2025-10-16 at 23 20 34" src="https://github.com/user-attachments/assets/116d0575-a1ae-4577-95b9-8415cda57e52" /> <img width="2032" height="1167" alt="Screenshot 2025-10-16 at 23 20 57" src="https://github.com/user-attachments/assets/000fdbb6-80bd-46e9-b668-f4b54ab708d2" /> <img width="2032" height="1167" alt="Screenshot 2025-10-16 at 23 21 37" src="https://github.com/user-attachments/assets/7b1c82da-b82f-4e14-a97c-3cd0e71bbca0" /> <img width="2032" height="1167" alt="Screenshot 2025-10-16 at 23 21 52" src="https://github.com/user-attachments/assets/1fd4b65a-09f7-47a9-a9b7-fdce4252aec3" /> <img width="2032" height="1167" alt="Screenshot 2025-10-16 at 23 22 38" src="https://github.com/user-attachments/assets/f4c2ac5c-e6f9-4e0e-b683-522b237e3328" /> the persistent_name values for `ProjectPanel` and `OutlinePanel` needed to be updated to not have a space in them in order to pass the `Identifier` check. all the other Panels already had names that did not include spaces, so it just makes these conform with the other ones. I think this is a great place to start with adding more context identifiers and i think this type of additional information will make it possible to create really dynamic keybindings! Release Notes: - Workspace key context now includes the state of the 3 docks
2629 lines
102 KiB
Rust
2629 lines
102 KiB
Rust
use std::ops::Range;
|
|
use std::path::Path;
|
|
use std::rc::Rc;
|
|
use std::sync::Arc;
|
|
|
|
use acp_thread::AcpThread;
|
|
use agent::{ContextServerRegistry, DbThreadMetadata, HistoryEntry, HistoryStore};
|
|
use db::kvp::{Dismissable, KEY_VALUE_STORE};
|
|
use project::agent_server_store::{
|
|
AgentServerCommand, AllAgentServersSettings, CLAUDE_CODE_NAME, CODEX_NAME, GEMINI_NAME,
|
|
};
|
|
use serde::{Deserialize, Serialize};
|
|
use settings::{
|
|
DefaultAgentView as DefaultView, LanguageModelProviderSetting, LanguageModelSelection,
|
|
};
|
|
use zed_actions::OpenBrowser;
|
|
use zed_actions::agent::{OpenClaudeCodeOnboardingModal, ReauthenticateAgent};
|
|
|
|
use crate::acp::{AcpThreadHistory, ThreadHistoryEvent};
|
|
use crate::context_store::ContextStore;
|
|
use crate::ui::{AcpOnboardingModal, ClaudeCodeOnboardingModal};
|
|
use crate::{
|
|
AddContextServer, AgentDiffPane, DeleteRecentlyOpenThread, Follow, InlineAssistant,
|
|
NewTextThread, NewThread, OpenActiveThreadAsMarkdown, OpenAgentDiff, OpenHistory,
|
|
ResetTrialEndUpsell, ResetTrialUpsell, ToggleNavigationMenu, ToggleNewThreadMenu,
|
|
ToggleOptionsMenu,
|
|
acp::AcpThreadView,
|
|
agent_configuration::{AgentConfiguration, AssistantConfigurationEvent},
|
|
slash_command::SlashCommandCompletionProvider,
|
|
text_thread_editor::{AgentPanelDelegate, TextThreadEditor, make_lsp_adapter_delegate},
|
|
ui::{AgentOnboardingModal, EndTrialUpsell},
|
|
};
|
|
use crate::{
|
|
ExternalAgent, NewExternalAgentThread, NewNativeAgentThreadFromSummary, placeholder_command,
|
|
};
|
|
use agent_settings::AgentSettings;
|
|
use ai_onboarding::AgentPanelOnboarding;
|
|
use anyhow::{Result, anyhow};
|
|
use assistant_context::{AssistantContext, ContextEvent, ContextSummary};
|
|
use assistant_slash_command::SlashCommandWorkingSet;
|
|
use client::{UserStore, zed_urls};
|
|
use cloud_llm_client::{Plan, PlanV1, PlanV2, UsageLimit};
|
|
use editor::{Anchor, AnchorRangeExt as _, Editor, EditorEvent, MultiBuffer};
|
|
use fs::Fs;
|
|
use gpui::{
|
|
Action, AnyElement, App, AsyncWindowContext, Corner, DismissEvent, Entity, EventEmitter,
|
|
ExternalPaths, FocusHandle, Focusable, KeyContext, Pixels, Subscription, Task, UpdateGlobal,
|
|
WeakEntity, prelude::*,
|
|
};
|
|
use language::LanguageRegistry;
|
|
use language_model::{ConfigurationError, LanguageModelRegistry};
|
|
use project::{Project, ProjectPath, Worktree};
|
|
use prompt_store::{PromptBuilder, PromptStore, UserPromptId};
|
|
use rules_library::{RulesLibrary, open_rules_library};
|
|
use search::{BufferSearchBar, buffer_search};
|
|
use settings::{Settings, SettingsStore, update_settings_file};
|
|
use theme::ThemeSettings;
|
|
use ui::utils::WithRemSize;
|
|
use ui::{
|
|
Callout, ContextMenu, ContextMenuEntry, KeyBinding, PopoverMenu, PopoverMenuHandle,
|
|
ProgressBar, Tab, Tooltip, prelude::*,
|
|
};
|
|
use util::ResultExt as _;
|
|
use workspace::{
|
|
CollaboratorId, DraggedSelection, DraggedTab, ToggleZoom, ToolbarItemView, Workspace,
|
|
dock::{DockPosition, Panel, PanelEvent},
|
|
};
|
|
use zed_actions::{
|
|
DecreaseBufferFontSize, IncreaseBufferFontSize, ResetBufferFontSize,
|
|
agent::{OpenAcpOnboardingModal, OpenOnboardingModal, OpenSettings, ResetOnboarding},
|
|
assistant::{OpenRulesLibrary, ToggleFocus},
|
|
};
|
|
|
|
const AGENT_PANEL_KEY: &str = "agent_panel";
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
struct SerializedAgentPanel {
|
|
width: Option<Pixels>,
|
|
selected_agent: Option<AgentType>,
|
|
}
|
|
|
|
pub fn init(cx: &mut App) {
|
|
cx.observe_new(
|
|
|workspace: &mut Workspace, _window, _cx: &mut Context<Workspace>| {
|
|
workspace
|
|
.register_action(|workspace, action: &NewThread, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
panel.update(cx, |panel, cx| panel.new_thread(action, window, cx));
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
}
|
|
})
|
|
.register_action(
|
|
|workspace, action: &NewNativeAgentThreadFromSummary, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_native_agent_thread_from_summary(action, window, cx)
|
|
});
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
}
|
|
},
|
|
)
|
|
.register_action(|workspace, _: &OpenHistory, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| panel.open_history(window, cx));
|
|
}
|
|
})
|
|
.register_action(|workspace, _: &OpenSettings, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| panel.open_configuration(window, cx));
|
|
}
|
|
})
|
|
.register_action(|workspace, _: &NewTextThread, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| panel.new_text_thread(window, cx));
|
|
}
|
|
})
|
|
.register_action(|workspace, action: &NewExternalAgentThread, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| {
|
|
panel.external_thread(action.agent.clone(), None, None, window, cx)
|
|
});
|
|
}
|
|
})
|
|
.register_action(|workspace, action: &OpenRulesLibrary, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| {
|
|
panel.deploy_rules_library(action, window, cx)
|
|
});
|
|
}
|
|
})
|
|
.register_action(|workspace, _: &Follow, window, cx| {
|
|
workspace.follow(CollaboratorId::Agent, window, cx);
|
|
})
|
|
.register_action(|workspace, _: &OpenAgentDiff, window, cx| {
|
|
let thread = workspace
|
|
.panel::<AgentPanel>(cx)
|
|
.and_then(|panel| panel.read(cx).active_thread_view().cloned())
|
|
.and_then(|thread_view| thread_view.read(cx).thread().cloned());
|
|
|
|
if let Some(thread) = thread {
|
|
AgentDiffPane::deploy_in_workspace(thread, workspace, window, cx);
|
|
}
|
|
})
|
|
.register_action(|workspace, _: &ToggleNavigationMenu, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| {
|
|
panel.toggle_navigation_menu(&ToggleNavigationMenu, window, cx);
|
|
});
|
|
}
|
|
})
|
|
.register_action(|workspace, _: &ToggleOptionsMenu, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| {
|
|
panel.toggle_options_menu(&ToggleOptionsMenu, window, cx);
|
|
});
|
|
}
|
|
})
|
|
.register_action(|workspace, _: &ToggleNewThreadMenu, window, cx| {
|
|
if let Some(panel) = workspace.panel::<AgentPanel>(cx) {
|
|
workspace.focus_panel::<AgentPanel>(window, cx);
|
|
panel.update(cx, |panel, cx| {
|
|
panel.toggle_new_thread_menu(&ToggleNewThreadMenu, window, cx);
|
|
});
|
|
}
|
|
})
|
|
.register_action(|workspace, _: &OpenOnboardingModal, window, cx| {
|
|
AgentOnboardingModal::toggle(workspace, window, cx)
|
|
})
|
|
.register_action(|workspace, _: &OpenAcpOnboardingModal, window, cx| {
|
|
AcpOnboardingModal::toggle(workspace, window, cx)
|
|
})
|
|
.register_action(|workspace, _: &OpenClaudeCodeOnboardingModal, window, cx| {
|
|
ClaudeCodeOnboardingModal::toggle(workspace, window, cx)
|
|
})
|
|
.register_action(|_workspace, _: &ResetOnboarding, window, cx| {
|
|
window.dispatch_action(workspace::RestoreBanner.boxed_clone(), cx);
|
|
window.refresh();
|
|
})
|
|
.register_action(|_workspace, _: &ResetTrialUpsell, _window, cx| {
|
|
OnboardingUpsell::set_dismissed(false, cx);
|
|
})
|
|
.register_action(|_workspace, _: &ResetTrialEndUpsell, _window, cx| {
|
|
TrialEndUpsell::set_dismissed(false, cx);
|
|
});
|
|
},
|
|
)
|
|
.detach();
|
|
}
|
|
|
|
enum ActiveView {
|
|
ExternalAgentThread {
|
|
thread_view: Entity<AcpThreadView>,
|
|
},
|
|
TextThread {
|
|
context_editor: Entity<TextThreadEditor>,
|
|
title_editor: Entity<Editor>,
|
|
buffer_search_bar: Entity<BufferSearchBar>,
|
|
_subscriptions: Vec<gpui::Subscription>,
|
|
},
|
|
History,
|
|
Configuration,
|
|
}
|
|
|
|
enum WhichFontSize {
|
|
AgentFont,
|
|
BufferFont,
|
|
None,
|
|
}
|
|
|
|
// TODO unify this with ExternalAgent
|
|
#[derive(Debug, Default, Clone, PartialEq, Serialize, Deserialize)]
|
|
pub enum AgentType {
|
|
#[default]
|
|
NativeAgent,
|
|
TextThread,
|
|
Gemini,
|
|
ClaudeCode,
|
|
Codex,
|
|
Custom {
|
|
name: SharedString,
|
|
command: AgentServerCommand,
|
|
},
|
|
}
|
|
|
|
impl AgentType {
|
|
fn label(&self) -> SharedString {
|
|
match self {
|
|
Self::NativeAgent | Self::TextThread => "Zed Agent".into(),
|
|
Self::Gemini => "Gemini CLI".into(),
|
|
Self::ClaudeCode => "Claude Code".into(),
|
|
Self::Codex => "Codex".into(),
|
|
Self::Custom { name, .. } => name.into(),
|
|
}
|
|
}
|
|
|
|
fn icon(&self) -> Option<IconName> {
|
|
match self {
|
|
Self::NativeAgent | Self::TextThread => None,
|
|
Self::Gemini => Some(IconName::AiGemini),
|
|
Self::ClaudeCode => Some(IconName::AiClaude),
|
|
Self::Codex => Some(IconName::AiOpenAi),
|
|
Self::Custom { .. } => Some(IconName::Terminal),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<ExternalAgent> for AgentType {
|
|
fn from(value: ExternalAgent) -> Self {
|
|
match value {
|
|
ExternalAgent::Gemini => Self::Gemini,
|
|
ExternalAgent::ClaudeCode => Self::ClaudeCode,
|
|
ExternalAgent::Codex => Self::Codex,
|
|
ExternalAgent::Custom { name, command } => Self::Custom { name, command },
|
|
ExternalAgent::NativeAgent => Self::NativeAgent,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ActiveView {
|
|
pub fn which_font_size_used(&self) -> WhichFontSize {
|
|
match self {
|
|
ActiveView::ExternalAgentThread { .. } | ActiveView::History => {
|
|
WhichFontSize::AgentFont
|
|
}
|
|
ActiveView::TextThread { .. } => WhichFontSize::BufferFont,
|
|
ActiveView::Configuration => WhichFontSize::None,
|
|
}
|
|
}
|
|
|
|
pub fn native_agent(
|
|
fs: Arc<dyn Fs>,
|
|
prompt_store: Option<Entity<PromptStore>>,
|
|
history_store: Entity<agent::HistoryStore>,
|
|
project: Entity<Project>,
|
|
workspace: WeakEntity<Workspace>,
|
|
window: &mut Window,
|
|
cx: &mut App,
|
|
) -> Self {
|
|
let thread_view = cx.new(|cx| {
|
|
crate::acp::AcpThreadView::new(
|
|
ExternalAgent::NativeAgent.server(fs, history_store.clone()),
|
|
None,
|
|
None,
|
|
workspace,
|
|
project,
|
|
history_store,
|
|
prompt_store,
|
|
window,
|
|
cx,
|
|
)
|
|
});
|
|
|
|
Self::ExternalAgentThread { thread_view }
|
|
}
|
|
|
|
pub fn text_thread(
|
|
context_editor: Entity<TextThreadEditor>,
|
|
acp_history_store: Entity<agent::HistoryStore>,
|
|
language_registry: Arc<LanguageRegistry>,
|
|
window: &mut Window,
|
|
cx: &mut App,
|
|
) -> Self {
|
|
let title = context_editor.read(cx).title(cx).to_string();
|
|
|
|
let editor = cx.new(|cx| {
|
|
let mut editor = Editor::single_line(window, cx);
|
|
editor.set_text(title, window, cx);
|
|
editor
|
|
});
|
|
|
|
// This is a workaround for `editor.set_text` emitting a `BufferEdited` event, which would
|
|
// cause a custom summary to be set. The presence of this custom summary would cause
|
|
// summarization to not happen.
|
|
let mut suppress_first_edit = true;
|
|
|
|
let subscriptions = vec![
|
|
window.subscribe(&editor, cx, {
|
|
{
|
|
let context_editor = context_editor.clone();
|
|
move |editor, event, window, cx| match event {
|
|
EditorEvent::BufferEdited => {
|
|
if suppress_first_edit {
|
|
suppress_first_edit = false;
|
|
return;
|
|
}
|
|
let new_summary = editor.read(cx).text(cx);
|
|
|
|
context_editor.update(cx, |context_editor, cx| {
|
|
context_editor
|
|
.context()
|
|
.update(cx, |assistant_context, cx| {
|
|
assistant_context.set_custom_summary(new_summary, cx);
|
|
})
|
|
})
|
|
}
|
|
EditorEvent::Blurred => {
|
|
if editor.read(cx).text(cx).is_empty() {
|
|
let summary = context_editor
|
|
.read(cx)
|
|
.context()
|
|
.read(cx)
|
|
.summary()
|
|
.or_default();
|
|
|
|
editor.update(cx, |editor, cx| {
|
|
editor.set_text(summary, window, cx);
|
|
});
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
}),
|
|
window.subscribe(&context_editor.read(cx).context().clone(), cx, {
|
|
let editor = editor.clone();
|
|
move |assistant_context, event, window, cx| match event {
|
|
ContextEvent::SummaryGenerated => {
|
|
let summary = assistant_context.read(cx).summary().or_default();
|
|
|
|
editor.update(cx, |editor, cx| {
|
|
editor.set_text(summary, window, cx);
|
|
})
|
|
}
|
|
ContextEvent::PathChanged { old_path, new_path } => {
|
|
acp_history_store.update(cx, |history_store, cx| {
|
|
if let Some(old_path) = old_path {
|
|
history_store
|
|
.replace_recently_opened_text_thread(old_path, new_path, cx);
|
|
} else {
|
|
history_store.push_recently_opened_entry(
|
|
agent::HistoryEntryId::TextThread(new_path.clone()),
|
|
cx,
|
|
);
|
|
}
|
|
});
|
|
}
|
|
_ => {}
|
|
}
|
|
}),
|
|
];
|
|
|
|
let buffer_search_bar =
|
|
cx.new(|cx| BufferSearchBar::new(Some(language_registry), window, cx));
|
|
buffer_search_bar.update(cx, |buffer_search_bar, cx| {
|
|
buffer_search_bar.set_active_pane_item(Some(&context_editor), window, cx)
|
|
});
|
|
|
|
Self::TextThread {
|
|
context_editor,
|
|
title_editor: editor,
|
|
buffer_search_bar,
|
|
_subscriptions: subscriptions,
|
|
}
|
|
}
|
|
}
|
|
|
|
pub struct AgentPanel {
|
|
workspace: WeakEntity<Workspace>,
|
|
loading: bool,
|
|
user_store: Entity<UserStore>,
|
|
project: Entity<Project>,
|
|
fs: Arc<dyn Fs>,
|
|
language_registry: Arc<LanguageRegistry>,
|
|
acp_history: Entity<AcpThreadHistory>,
|
|
history_store: Entity<agent::HistoryStore>,
|
|
text_thread_store: Entity<assistant_context::ContextStore>,
|
|
prompt_store: Option<Entity<PromptStore>>,
|
|
context_server_registry: Entity<ContextServerRegistry>,
|
|
inline_assist_context_store: Entity<ContextStore>,
|
|
configuration: Option<Entity<AgentConfiguration>>,
|
|
configuration_subscription: Option<Subscription>,
|
|
active_view: ActiveView,
|
|
previous_view: Option<ActiveView>,
|
|
new_thread_menu_handle: PopoverMenuHandle<ContextMenu>,
|
|
agent_panel_menu_handle: PopoverMenuHandle<ContextMenu>,
|
|
agent_navigation_menu_handle: PopoverMenuHandle<ContextMenu>,
|
|
agent_navigation_menu: Option<Entity<ContextMenu>>,
|
|
width: Option<Pixels>,
|
|
height: Option<Pixels>,
|
|
zoomed: bool,
|
|
pending_serialization: Option<Task<Result<()>>>,
|
|
onboarding: Entity<AgentPanelOnboarding>,
|
|
selected_agent: AgentType,
|
|
}
|
|
|
|
impl AgentPanel {
|
|
fn serialize(&mut self, cx: &mut Context<Self>) {
|
|
let width = self.width;
|
|
let selected_agent = self.selected_agent.clone();
|
|
self.pending_serialization = Some(cx.background_spawn(async move {
|
|
KEY_VALUE_STORE
|
|
.write_kvp(
|
|
AGENT_PANEL_KEY.into(),
|
|
serde_json::to_string(&SerializedAgentPanel {
|
|
width,
|
|
selected_agent: Some(selected_agent),
|
|
})?,
|
|
)
|
|
.await?;
|
|
anyhow::Ok(())
|
|
}));
|
|
}
|
|
|
|
pub fn load(
|
|
workspace: WeakEntity<Workspace>,
|
|
prompt_builder: Arc<PromptBuilder>,
|
|
mut cx: AsyncWindowContext,
|
|
) -> Task<Result<Entity<Self>>> {
|
|
let prompt_store = cx.update(|_window, cx| PromptStore::global(cx));
|
|
cx.spawn(async move |cx| {
|
|
let prompt_store = match prompt_store {
|
|
Ok(prompt_store) => prompt_store.await.ok(),
|
|
Err(_) => None,
|
|
};
|
|
let serialized_panel = if let Some(panel) = cx
|
|
.background_spawn(async move { KEY_VALUE_STORE.read_kvp(AGENT_PANEL_KEY) })
|
|
.await
|
|
.log_err()
|
|
.flatten()
|
|
{
|
|
serde_json::from_str::<SerializedAgentPanel>(&panel).log_err()
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let slash_commands = Arc::new(SlashCommandWorkingSet::default());
|
|
let text_thread_store = workspace
|
|
.update(cx, |workspace, cx| {
|
|
let project = workspace.project().clone();
|
|
assistant_context::ContextStore::new(
|
|
project,
|
|
prompt_builder,
|
|
slash_commands,
|
|
cx,
|
|
)
|
|
})?
|
|
.await?;
|
|
|
|
let panel = workspace.update_in(cx, |workspace, window, cx| {
|
|
let panel =
|
|
cx.new(|cx| Self::new(workspace, text_thread_store, prompt_store, window, cx));
|
|
|
|
panel.as_mut(cx).loading = true;
|
|
if let Some(serialized_panel) = serialized_panel {
|
|
panel.update(cx, |panel, cx| {
|
|
panel.width = serialized_panel.width.map(|w| w.round());
|
|
if let Some(selected_agent) = serialized_panel.selected_agent {
|
|
panel.selected_agent = selected_agent.clone();
|
|
panel.new_agent_thread(selected_agent, window, cx);
|
|
}
|
|
cx.notify();
|
|
});
|
|
} else {
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_agent_thread(AgentType::NativeAgent, window, cx);
|
|
});
|
|
}
|
|
panel.as_mut(cx).loading = false;
|
|
panel
|
|
})?;
|
|
|
|
Ok(panel)
|
|
})
|
|
}
|
|
|
|
fn new(
|
|
workspace: &Workspace,
|
|
text_thread_store: Entity<assistant_context::ContextStore>,
|
|
prompt_store: Option<Entity<PromptStore>>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) -> Self {
|
|
let fs = workspace.app_state().fs.clone();
|
|
let user_store = workspace.app_state().user_store.clone();
|
|
let project = workspace.project();
|
|
let language_registry = project.read(cx).languages().clone();
|
|
let client = workspace.client().clone();
|
|
let workspace = workspace.weak_handle();
|
|
|
|
let inline_assist_context_store = cx.new(|_cx| ContextStore::new(project.downgrade()));
|
|
let context_server_registry =
|
|
cx.new(|cx| ContextServerRegistry::new(project.read(cx).context_server_store(), cx));
|
|
|
|
let history_store = cx.new(|cx| agent::HistoryStore::new(text_thread_store.clone(), cx));
|
|
let acp_history = cx.new(|cx| AcpThreadHistory::new(history_store.clone(), window, cx));
|
|
cx.subscribe_in(
|
|
&acp_history,
|
|
window,
|
|
|this, _, event, window, cx| match event {
|
|
ThreadHistoryEvent::Open(HistoryEntry::AcpThread(thread)) => {
|
|
this.external_thread(
|
|
Some(crate::ExternalAgent::NativeAgent),
|
|
Some(thread.clone()),
|
|
None,
|
|
window,
|
|
cx,
|
|
);
|
|
}
|
|
ThreadHistoryEvent::Open(HistoryEntry::TextThread(thread)) => {
|
|
this.open_saved_text_thread(thread.path.clone(), window, cx)
|
|
.detach_and_log_err(cx);
|
|
}
|
|
},
|
|
)
|
|
.detach();
|
|
|
|
let panel_type = AgentSettings::get_global(cx).default_view;
|
|
let active_view = match panel_type {
|
|
DefaultView::Thread => ActiveView::native_agent(
|
|
fs.clone(),
|
|
prompt_store.clone(),
|
|
history_store.clone(),
|
|
project.clone(),
|
|
workspace.clone(),
|
|
window,
|
|
cx,
|
|
),
|
|
DefaultView::TextThread => {
|
|
let context = text_thread_store.update(cx, |store, cx| store.create(cx));
|
|
let lsp_adapter_delegate = make_lsp_adapter_delegate(&project.clone(), cx).unwrap();
|
|
let context_editor = cx.new(|cx| {
|
|
let mut editor = TextThreadEditor::for_context(
|
|
context,
|
|
fs.clone(),
|
|
workspace.clone(),
|
|
project.clone(),
|
|
lsp_adapter_delegate,
|
|
window,
|
|
cx,
|
|
);
|
|
editor.insert_default_prompt(window, cx);
|
|
editor
|
|
});
|
|
ActiveView::text_thread(
|
|
context_editor,
|
|
history_store.clone(),
|
|
language_registry.clone(),
|
|
window,
|
|
cx,
|
|
)
|
|
}
|
|
};
|
|
|
|
let weak_panel = cx.entity().downgrade();
|
|
|
|
window.defer(cx, move |window, cx| {
|
|
let panel = weak_panel.clone();
|
|
let agent_navigation_menu =
|
|
ContextMenu::build_persistent(window, cx, move |mut menu, _window, cx| {
|
|
if let Some(panel) = panel.upgrade() {
|
|
menu = Self::populate_recently_opened_menu_section(menu, panel, cx);
|
|
}
|
|
menu.action("View All", Box::new(OpenHistory))
|
|
.end_slot_action(DeleteRecentlyOpenThread.boxed_clone())
|
|
.fixed_width(px(320.).into())
|
|
.keep_open_on_confirm(false)
|
|
.key_context("NavigationMenu")
|
|
});
|
|
weak_panel
|
|
.update(cx, |panel, cx| {
|
|
cx.subscribe_in(
|
|
&agent_navigation_menu,
|
|
window,
|
|
|_, menu, _: &DismissEvent, window, cx| {
|
|
menu.update(cx, |menu, _| {
|
|
menu.clear_selected();
|
|
});
|
|
cx.focus_self(window);
|
|
},
|
|
)
|
|
.detach();
|
|
panel.agent_navigation_menu = Some(agent_navigation_menu);
|
|
})
|
|
.ok();
|
|
});
|
|
|
|
let onboarding = cx.new(|cx| {
|
|
AgentPanelOnboarding::new(
|
|
user_store.clone(),
|
|
client,
|
|
|_window, cx| {
|
|
OnboardingUpsell::set_dismissed(true, cx);
|
|
},
|
|
cx,
|
|
)
|
|
});
|
|
|
|
Self {
|
|
active_view,
|
|
workspace,
|
|
user_store,
|
|
project: project.clone(),
|
|
fs: fs.clone(),
|
|
language_registry,
|
|
text_thread_store,
|
|
prompt_store,
|
|
configuration: None,
|
|
configuration_subscription: None,
|
|
context_server_registry,
|
|
inline_assist_context_store,
|
|
previous_view: None,
|
|
new_thread_menu_handle: PopoverMenuHandle::default(),
|
|
agent_panel_menu_handle: PopoverMenuHandle::default(),
|
|
agent_navigation_menu_handle: PopoverMenuHandle::default(),
|
|
agent_navigation_menu: None,
|
|
width: None,
|
|
height: None,
|
|
zoomed: false,
|
|
pending_serialization: None,
|
|
onboarding,
|
|
acp_history,
|
|
history_store,
|
|
selected_agent: AgentType::default(),
|
|
loading: false,
|
|
}
|
|
}
|
|
|
|
pub fn toggle_focus(
|
|
workspace: &mut Workspace,
|
|
_: &ToggleFocus,
|
|
window: &mut Window,
|
|
cx: &mut Context<Workspace>,
|
|
) {
|
|
if workspace
|
|
.panel::<Self>(cx)
|
|
.is_some_and(|panel| panel.read(cx).enabled(cx))
|
|
{
|
|
workspace.toggle_panel_focus::<Self>(window, cx);
|
|
}
|
|
}
|
|
|
|
pub(crate) fn prompt_store(&self) -> &Option<Entity<PromptStore>> {
|
|
&self.prompt_store
|
|
}
|
|
|
|
pub(crate) fn inline_assist_context_store(&self) -> &Entity<ContextStore> {
|
|
&self.inline_assist_context_store
|
|
}
|
|
|
|
pub(crate) fn thread_store(&self) -> &Entity<HistoryStore> {
|
|
&self.history_store
|
|
}
|
|
|
|
pub(crate) fn context_server_registry(&self) -> &Entity<ContextServerRegistry> {
|
|
&self.context_server_registry
|
|
}
|
|
|
|
fn active_thread_view(&self) -> Option<&Entity<AcpThreadView>> {
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view, .. } => Some(thread_view),
|
|
ActiveView::TextThread { .. } | ActiveView::History | ActiveView::Configuration => None,
|
|
}
|
|
}
|
|
|
|
fn new_thread(&mut self, _action: &NewThread, window: &mut Window, cx: &mut Context<Self>) {
|
|
self.new_agent_thread(AgentType::NativeAgent, window, cx);
|
|
}
|
|
|
|
fn new_native_agent_thread_from_summary(
|
|
&mut self,
|
|
action: &NewNativeAgentThreadFromSummary,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
let Some(thread) = self
|
|
.history_store
|
|
.read(cx)
|
|
.thread_from_session_id(&action.from_session_id)
|
|
else {
|
|
return;
|
|
};
|
|
|
|
self.external_thread(
|
|
Some(ExternalAgent::NativeAgent),
|
|
None,
|
|
Some(thread.clone()),
|
|
window,
|
|
cx,
|
|
);
|
|
}
|
|
|
|
fn new_text_thread(&mut self, window: &mut Window, cx: &mut Context<Self>) {
|
|
telemetry::event!("Agent Thread Started", agent = "zed-text");
|
|
|
|
let context = self
|
|
.text_thread_store
|
|
.update(cx, |context_store, cx| context_store.create(cx));
|
|
let lsp_adapter_delegate = make_lsp_adapter_delegate(&self.project, cx)
|
|
.log_err()
|
|
.flatten();
|
|
|
|
let context_editor = cx.new(|cx| {
|
|
let mut editor = TextThreadEditor::for_context(
|
|
context,
|
|
self.fs.clone(),
|
|
self.workspace.clone(),
|
|
self.project.clone(),
|
|
lsp_adapter_delegate,
|
|
window,
|
|
cx,
|
|
);
|
|
editor.insert_default_prompt(window, cx);
|
|
editor
|
|
});
|
|
|
|
if self.selected_agent != AgentType::TextThread {
|
|
self.selected_agent = AgentType::TextThread;
|
|
self.serialize(cx);
|
|
}
|
|
|
|
self.set_active_view(
|
|
ActiveView::text_thread(
|
|
context_editor.clone(),
|
|
self.history_store.clone(),
|
|
self.language_registry.clone(),
|
|
window,
|
|
cx,
|
|
),
|
|
window,
|
|
cx,
|
|
);
|
|
context_editor.focus_handle(cx).focus(window);
|
|
}
|
|
|
|
fn external_thread(
|
|
&mut self,
|
|
agent_choice: Option<crate::ExternalAgent>,
|
|
resume_thread: Option<DbThreadMetadata>,
|
|
summarize_thread: Option<DbThreadMetadata>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
let workspace = self.workspace.clone();
|
|
let project = self.project.clone();
|
|
let fs = self.fs.clone();
|
|
let is_via_collab = self.project.read(cx).is_via_collab();
|
|
|
|
const LAST_USED_EXTERNAL_AGENT_KEY: &str = "agent_panel__last_used_external_agent";
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
struct LastUsedExternalAgent {
|
|
agent: crate::ExternalAgent,
|
|
}
|
|
|
|
let loading = self.loading;
|
|
let history = self.history_store.clone();
|
|
|
|
cx.spawn_in(window, async move |this, cx| {
|
|
let ext_agent = match agent_choice {
|
|
Some(agent) => {
|
|
cx.background_spawn({
|
|
let agent = agent.clone();
|
|
async move {
|
|
if let Some(serialized) =
|
|
serde_json::to_string(&LastUsedExternalAgent { agent }).log_err()
|
|
{
|
|
KEY_VALUE_STORE
|
|
.write_kvp(LAST_USED_EXTERNAL_AGENT_KEY.to_string(), serialized)
|
|
.await
|
|
.log_err();
|
|
}
|
|
}
|
|
})
|
|
.detach();
|
|
|
|
agent
|
|
}
|
|
None => {
|
|
if is_via_collab {
|
|
ExternalAgent::NativeAgent
|
|
} else {
|
|
cx.background_spawn(async move {
|
|
KEY_VALUE_STORE.read_kvp(LAST_USED_EXTERNAL_AGENT_KEY)
|
|
})
|
|
.await
|
|
.log_err()
|
|
.flatten()
|
|
.and_then(|value| {
|
|
serde_json::from_str::<LastUsedExternalAgent>(&value).log_err()
|
|
})
|
|
.map(|agent| agent.agent)
|
|
.unwrap_or(ExternalAgent::NativeAgent)
|
|
}
|
|
}
|
|
};
|
|
|
|
let server = ext_agent.server(fs, history);
|
|
|
|
if !loading {
|
|
telemetry::event!("Agent Thread Started", agent = server.telemetry_id());
|
|
}
|
|
|
|
this.update_in(cx, |this, window, cx| {
|
|
let selected_agent = ext_agent.into();
|
|
if this.selected_agent != selected_agent {
|
|
this.selected_agent = selected_agent;
|
|
this.serialize(cx);
|
|
}
|
|
|
|
let thread_view = cx.new(|cx| {
|
|
crate::acp::AcpThreadView::new(
|
|
server,
|
|
resume_thread,
|
|
summarize_thread,
|
|
workspace.clone(),
|
|
project,
|
|
this.history_store.clone(),
|
|
this.prompt_store.clone(),
|
|
window,
|
|
cx,
|
|
)
|
|
});
|
|
|
|
this.set_active_view(ActiveView::ExternalAgentThread { thread_view }, window, cx);
|
|
})
|
|
})
|
|
.detach_and_log_err(cx);
|
|
}
|
|
|
|
fn deploy_rules_library(
|
|
&mut self,
|
|
action: &OpenRulesLibrary,
|
|
_window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
open_rules_library(
|
|
self.language_registry.clone(),
|
|
Box::new(PromptLibraryInlineAssist::new(self.workspace.clone())),
|
|
Rc::new(|| {
|
|
Rc::new(SlashCommandCompletionProvider::new(
|
|
Arc::new(SlashCommandWorkingSet::default()),
|
|
None,
|
|
None,
|
|
))
|
|
}),
|
|
action
|
|
.prompt_to_select
|
|
.map(|uuid| UserPromptId(uuid).into()),
|
|
cx,
|
|
)
|
|
.detach_and_log_err(cx);
|
|
}
|
|
|
|
fn open_history(&mut self, window: &mut Window, cx: &mut Context<Self>) {
|
|
if matches!(self.active_view, ActiveView::History) {
|
|
if let Some(previous_view) = self.previous_view.take() {
|
|
self.set_active_view(previous_view, window, cx);
|
|
}
|
|
} else {
|
|
self.set_active_view(ActiveView::History, window, cx);
|
|
}
|
|
cx.notify();
|
|
}
|
|
|
|
pub(crate) fn open_saved_text_thread(
|
|
&mut self,
|
|
path: Arc<Path>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) -> Task<Result<()>> {
|
|
let context = self
|
|
.history_store
|
|
.update(cx, |store, cx| store.load_text_thread(path, cx));
|
|
cx.spawn_in(window, async move |this, cx| {
|
|
let context = context.await?;
|
|
this.update_in(cx, |this, window, cx| {
|
|
this.open_text_thread(context, window, cx);
|
|
})
|
|
})
|
|
}
|
|
|
|
pub(crate) fn open_text_thread(
|
|
&mut self,
|
|
context: Entity<AssistantContext>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
let lsp_adapter_delegate = make_lsp_adapter_delegate(&self.project.clone(), cx)
|
|
.log_err()
|
|
.flatten();
|
|
let editor = cx.new(|cx| {
|
|
TextThreadEditor::for_context(
|
|
context,
|
|
self.fs.clone(),
|
|
self.workspace.clone(),
|
|
self.project.clone(),
|
|
lsp_adapter_delegate,
|
|
window,
|
|
cx,
|
|
)
|
|
});
|
|
|
|
if self.selected_agent != AgentType::TextThread {
|
|
self.selected_agent = AgentType::TextThread;
|
|
self.serialize(cx);
|
|
}
|
|
|
|
self.set_active_view(
|
|
ActiveView::text_thread(
|
|
editor,
|
|
self.history_store.clone(),
|
|
self.language_registry.clone(),
|
|
window,
|
|
cx,
|
|
),
|
|
window,
|
|
cx,
|
|
);
|
|
}
|
|
|
|
pub fn go_back(&mut self, _: &workspace::GoBack, window: &mut Window, cx: &mut Context<Self>) {
|
|
match self.active_view {
|
|
ActiveView::Configuration | ActiveView::History => {
|
|
if let Some(previous_view) = self.previous_view.take() {
|
|
self.active_view = previous_view;
|
|
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view } => {
|
|
thread_view.focus_handle(cx).focus(window);
|
|
}
|
|
ActiveView::TextThread { context_editor, .. } => {
|
|
context_editor.focus_handle(cx).focus(window);
|
|
}
|
|
ActiveView::History | ActiveView::Configuration => {}
|
|
}
|
|
}
|
|
cx.notify();
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
pub fn toggle_navigation_menu(
|
|
&mut self,
|
|
_: &ToggleNavigationMenu,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
self.agent_navigation_menu_handle.toggle(window, cx);
|
|
}
|
|
|
|
pub fn toggle_options_menu(
|
|
&mut self,
|
|
_: &ToggleOptionsMenu,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
self.agent_panel_menu_handle.toggle(window, cx);
|
|
}
|
|
|
|
pub fn toggle_new_thread_menu(
|
|
&mut self,
|
|
_: &ToggleNewThreadMenu,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
self.new_thread_menu_handle.toggle(window, cx);
|
|
}
|
|
|
|
pub fn increase_font_size(
|
|
&mut self,
|
|
action: &IncreaseBufferFontSize,
|
|
_: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
self.handle_font_size_action(action.persist, px(1.0), cx);
|
|
}
|
|
|
|
pub fn decrease_font_size(
|
|
&mut self,
|
|
action: &DecreaseBufferFontSize,
|
|
_: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
self.handle_font_size_action(action.persist, px(-1.0), cx);
|
|
}
|
|
|
|
fn handle_font_size_action(&mut self, persist: bool, delta: Pixels, cx: &mut Context<Self>) {
|
|
match self.active_view.which_font_size_used() {
|
|
WhichFontSize::AgentFont => {
|
|
if persist {
|
|
update_settings_file(self.fs.clone(), cx, move |settings, cx| {
|
|
let agent_ui_font_size =
|
|
ThemeSettings::get_global(cx).agent_ui_font_size(cx) + delta;
|
|
let _ = settings
|
|
.theme
|
|
.agent_ui_font_size
|
|
.insert(theme::clamp_font_size(agent_ui_font_size).into());
|
|
});
|
|
} else {
|
|
theme::adjust_agent_ui_font_size(cx, |size| size + delta);
|
|
}
|
|
}
|
|
WhichFontSize::BufferFont => {
|
|
// Prompt editor uses the buffer font size, so allow the action to propagate to the
|
|
// default handler that changes that font size.
|
|
cx.propagate();
|
|
}
|
|
WhichFontSize::None => {}
|
|
}
|
|
}
|
|
|
|
pub fn reset_font_size(
|
|
&mut self,
|
|
action: &ResetBufferFontSize,
|
|
_: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
if action.persist {
|
|
update_settings_file(self.fs.clone(), cx, move |settings, _| {
|
|
settings.theme.agent_ui_font_size = None;
|
|
});
|
|
} else {
|
|
theme::reset_agent_ui_font_size(cx);
|
|
}
|
|
}
|
|
|
|
pub fn toggle_zoom(&mut self, _: &ToggleZoom, window: &mut Window, cx: &mut Context<Self>) {
|
|
if self.zoomed {
|
|
cx.emit(PanelEvent::ZoomOut);
|
|
} else {
|
|
if !self.focus_handle(cx).contains_focused(window, cx) {
|
|
cx.focus_self(window);
|
|
}
|
|
cx.emit(PanelEvent::ZoomIn);
|
|
}
|
|
}
|
|
|
|
pub(crate) fn open_configuration(&mut self, window: &mut Window, cx: &mut Context<Self>) {
|
|
let agent_server_store = self.project.read(cx).agent_server_store().clone();
|
|
let context_server_store = self.project.read(cx).context_server_store();
|
|
let fs = self.fs.clone();
|
|
|
|
self.set_active_view(ActiveView::Configuration, window, cx);
|
|
self.configuration = Some(cx.new(|cx| {
|
|
AgentConfiguration::new(
|
|
fs,
|
|
agent_server_store,
|
|
context_server_store,
|
|
self.context_server_registry.clone(),
|
|
self.language_registry.clone(),
|
|
self.workspace.clone(),
|
|
window,
|
|
cx,
|
|
)
|
|
}));
|
|
|
|
if let Some(configuration) = self.configuration.as_ref() {
|
|
self.configuration_subscription = Some(cx.subscribe_in(
|
|
configuration,
|
|
window,
|
|
Self::handle_agent_configuration_event,
|
|
));
|
|
|
|
configuration.focus_handle(cx).focus(window);
|
|
}
|
|
}
|
|
|
|
pub(crate) fn open_active_thread_as_markdown(
|
|
&mut self,
|
|
_: &OpenActiveThreadAsMarkdown,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
let Some(workspace) = self.workspace.upgrade() else {
|
|
return;
|
|
};
|
|
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view } => {
|
|
thread_view
|
|
.update(cx, |thread_view, cx| {
|
|
thread_view.open_thread_as_markdown(workspace, window, cx)
|
|
})
|
|
.detach_and_log_err(cx);
|
|
}
|
|
ActiveView::TextThread { .. } | ActiveView::History | ActiveView::Configuration => {}
|
|
}
|
|
}
|
|
|
|
fn handle_agent_configuration_event(
|
|
&mut self,
|
|
_entity: &Entity<AgentConfiguration>,
|
|
event: &AssistantConfigurationEvent,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
match event {
|
|
AssistantConfigurationEvent::NewThread(provider) => {
|
|
if LanguageModelRegistry::read_global(cx)
|
|
.default_model()
|
|
.is_none_or(|model| model.provider.id() != provider.id())
|
|
&& let Some(model) = provider.default_model(cx)
|
|
{
|
|
update_settings_file(self.fs.clone(), cx, move |settings, _| {
|
|
let provider = model.provider_id().0.to_string();
|
|
let model = model.id().0.to_string();
|
|
settings
|
|
.agent
|
|
.get_or_insert_default()
|
|
.set_model(LanguageModelSelection {
|
|
provider: LanguageModelProviderSetting(provider),
|
|
model,
|
|
})
|
|
});
|
|
}
|
|
|
|
self.new_thread(&NewThread, window, cx);
|
|
if let Some((thread, model)) = self
|
|
.active_native_agent_thread(cx)
|
|
.zip(provider.default_model(cx))
|
|
{
|
|
thread.update(cx, |thread, cx| {
|
|
thread.set_model(model, cx);
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub(crate) fn active_agent_thread(&self, cx: &App) -> Option<Entity<AcpThread>> {
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view, .. } => {
|
|
thread_view.read(cx).thread().cloned()
|
|
}
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub(crate) fn active_native_agent_thread(&self, cx: &App) -> Option<Entity<agent::Thread>> {
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view, .. } => {
|
|
thread_view.read(cx).as_native_thread(cx)
|
|
}
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub(crate) fn active_context_editor(&self) -> Option<Entity<TextThreadEditor>> {
|
|
match &self.active_view {
|
|
ActiveView::TextThread { context_editor, .. } => Some(context_editor.clone()),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
fn set_active_view(
|
|
&mut self,
|
|
new_view: ActiveView,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
let current_is_history = matches!(self.active_view, ActiveView::History);
|
|
let new_is_history = matches!(new_view, ActiveView::History);
|
|
|
|
let current_is_config = matches!(self.active_view, ActiveView::Configuration);
|
|
let new_is_config = matches!(new_view, ActiveView::Configuration);
|
|
|
|
let current_is_special = current_is_history || current_is_config;
|
|
let new_is_special = new_is_history || new_is_config;
|
|
|
|
match &new_view {
|
|
ActiveView::TextThread { context_editor, .. } => {
|
|
self.history_store.update(cx, |store, cx| {
|
|
if let Some(path) = context_editor.read(cx).context().read(cx).path() {
|
|
store.push_recently_opened_entry(
|
|
agent::HistoryEntryId::TextThread(path.clone()),
|
|
cx,
|
|
)
|
|
}
|
|
})
|
|
}
|
|
ActiveView::ExternalAgentThread { .. } => {}
|
|
ActiveView::History | ActiveView::Configuration => {}
|
|
}
|
|
|
|
if current_is_special && !new_is_special {
|
|
self.active_view = new_view;
|
|
} else if !current_is_special && new_is_special {
|
|
self.previous_view = Some(std::mem::replace(&mut self.active_view, new_view));
|
|
} else {
|
|
if !new_is_special {
|
|
self.previous_view = None;
|
|
}
|
|
self.active_view = new_view;
|
|
}
|
|
|
|
self.focus_handle(cx).focus(window);
|
|
}
|
|
|
|
fn populate_recently_opened_menu_section(
|
|
mut menu: ContextMenu,
|
|
panel: Entity<Self>,
|
|
cx: &mut Context<ContextMenu>,
|
|
) -> ContextMenu {
|
|
let entries = panel
|
|
.read(cx)
|
|
.history_store
|
|
.read(cx)
|
|
.recently_opened_entries(cx);
|
|
|
|
if entries.is_empty() {
|
|
return menu;
|
|
}
|
|
|
|
menu = menu.header("Recently Opened");
|
|
|
|
for entry in entries {
|
|
let title = entry.title().clone();
|
|
|
|
menu = menu.entry_with_end_slot_on_hover(
|
|
title,
|
|
None,
|
|
{
|
|
let panel = panel.downgrade();
|
|
let entry = entry.clone();
|
|
move |window, cx| {
|
|
let entry = entry.clone();
|
|
panel
|
|
.update(cx, move |this, cx| match &entry {
|
|
agent::HistoryEntry::AcpThread(entry) => this.external_thread(
|
|
Some(ExternalAgent::NativeAgent),
|
|
Some(entry.clone()),
|
|
None,
|
|
window,
|
|
cx,
|
|
),
|
|
agent::HistoryEntry::TextThread(entry) => this
|
|
.open_saved_text_thread(entry.path.clone(), window, cx)
|
|
.detach_and_log_err(cx),
|
|
})
|
|
.ok();
|
|
}
|
|
},
|
|
IconName::Close,
|
|
"Close Entry".into(),
|
|
{
|
|
let panel = panel.downgrade();
|
|
let id = entry.id();
|
|
move |_window, cx| {
|
|
panel
|
|
.update(cx, |this, cx| {
|
|
this.history_store.update(cx, |history_store, cx| {
|
|
history_store.remove_recently_opened_entry(&id, cx);
|
|
});
|
|
})
|
|
.ok();
|
|
}
|
|
},
|
|
);
|
|
}
|
|
|
|
menu = menu.separator();
|
|
|
|
menu
|
|
}
|
|
|
|
pub fn selected_agent(&self) -> AgentType {
|
|
self.selected_agent.clone()
|
|
}
|
|
|
|
pub fn new_agent_thread(
|
|
&mut self,
|
|
agent: AgentType,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
match agent {
|
|
AgentType::TextThread => {
|
|
window.dispatch_action(NewTextThread.boxed_clone(), cx);
|
|
}
|
|
AgentType::NativeAgent => self.external_thread(
|
|
Some(crate::ExternalAgent::NativeAgent),
|
|
None,
|
|
None,
|
|
window,
|
|
cx,
|
|
),
|
|
AgentType::Gemini => {
|
|
self.external_thread(Some(crate::ExternalAgent::Gemini), None, None, window, cx)
|
|
}
|
|
AgentType::ClaudeCode => {
|
|
self.selected_agent = AgentType::ClaudeCode;
|
|
self.serialize(cx);
|
|
self.external_thread(
|
|
Some(crate::ExternalAgent::ClaudeCode),
|
|
None,
|
|
None,
|
|
window,
|
|
cx,
|
|
)
|
|
}
|
|
AgentType::Codex => {
|
|
self.selected_agent = AgentType::Codex;
|
|
self.serialize(cx);
|
|
self.external_thread(Some(crate::ExternalAgent::Codex), None, None, window, cx)
|
|
}
|
|
AgentType::Custom { name, command } => self.external_thread(
|
|
Some(crate::ExternalAgent::Custom { name, command }),
|
|
None,
|
|
None,
|
|
window,
|
|
cx,
|
|
),
|
|
}
|
|
}
|
|
|
|
pub fn load_agent_thread(
|
|
&mut self,
|
|
thread: DbThreadMetadata,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
self.external_thread(
|
|
Some(ExternalAgent::NativeAgent),
|
|
Some(thread),
|
|
None,
|
|
window,
|
|
cx,
|
|
);
|
|
}
|
|
}
|
|
|
|
impl Focusable for AgentPanel {
|
|
fn focus_handle(&self, cx: &App) -> FocusHandle {
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view, .. } => thread_view.focus_handle(cx),
|
|
ActiveView::History => self.acp_history.focus_handle(cx),
|
|
ActiveView::TextThread { context_editor, .. } => context_editor.focus_handle(cx),
|
|
ActiveView::Configuration => {
|
|
if let Some(configuration) = self.configuration.as_ref() {
|
|
configuration.focus_handle(cx)
|
|
} else {
|
|
cx.focus_handle()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn agent_panel_dock_position(cx: &App) -> DockPosition {
|
|
AgentSettings::get_global(cx).dock.into()
|
|
}
|
|
|
|
impl EventEmitter<PanelEvent> for AgentPanel {}
|
|
|
|
impl Panel for AgentPanel {
|
|
fn persistent_name() -> &'static str {
|
|
"AgentPanel"
|
|
}
|
|
|
|
fn panel_key() -> &'static str {
|
|
AGENT_PANEL_KEY
|
|
}
|
|
|
|
fn position(&self, _window: &Window, cx: &App) -> DockPosition {
|
|
agent_panel_dock_position(cx)
|
|
}
|
|
|
|
fn position_is_valid(&self, position: DockPosition) -> bool {
|
|
position != DockPosition::Bottom
|
|
}
|
|
|
|
fn set_position(&mut self, position: DockPosition, _: &mut Window, cx: &mut Context<Self>) {
|
|
settings::update_settings_file(self.fs.clone(), cx, move |settings, _| {
|
|
settings
|
|
.agent
|
|
.get_or_insert_default()
|
|
.set_dock(position.into());
|
|
});
|
|
}
|
|
|
|
fn size(&self, window: &Window, cx: &App) -> Pixels {
|
|
let settings = AgentSettings::get_global(cx);
|
|
match self.position(window, cx) {
|
|
DockPosition::Left | DockPosition::Right => {
|
|
self.width.unwrap_or(settings.default_width)
|
|
}
|
|
DockPosition::Bottom => self.height.unwrap_or(settings.default_height),
|
|
}
|
|
}
|
|
|
|
fn set_size(&mut self, size: Option<Pixels>, window: &mut Window, cx: &mut Context<Self>) {
|
|
match self.position(window, cx) {
|
|
DockPosition::Left | DockPosition::Right => self.width = size,
|
|
DockPosition::Bottom => self.height = size,
|
|
}
|
|
self.serialize(cx);
|
|
cx.notify();
|
|
}
|
|
|
|
fn set_active(&mut self, _active: bool, _window: &mut Window, _cx: &mut Context<Self>) {}
|
|
|
|
fn remote_id() -> Option<proto::PanelId> {
|
|
Some(proto::PanelId::AssistantPanel)
|
|
}
|
|
|
|
fn icon(&self, _window: &Window, cx: &App) -> Option<IconName> {
|
|
(self.enabled(cx) && AgentSettings::get_global(cx).button).then_some(IconName::ZedAssistant)
|
|
}
|
|
|
|
fn icon_tooltip(&self, _window: &Window, _cx: &App) -> Option<&'static str> {
|
|
Some("Agent Panel")
|
|
}
|
|
|
|
fn toggle_action(&self) -> Box<dyn Action> {
|
|
Box::new(ToggleFocus)
|
|
}
|
|
|
|
fn activation_priority(&self) -> u32 {
|
|
3
|
|
}
|
|
|
|
fn enabled(&self, cx: &App) -> bool {
|
|
AgentSettings::get_global(cx).enabled(cx)
|
|
}
|
|
|
|
fn is_zoomed(&self, _window: &Window, _cx: &App) -> bool {
|
|
self.zoomed
|
|
}
|
|
|
|
fn set_zoomed(&mut self, zoomed: bool, _window: &mut Window, cx: &mut Context<Self>) {
|
|
self.zoomed = zoomed;
|
|
cx.notify();
|
|
}
|
|
}
|
|
|
|
impl AgentPanel {
|
|
fn render_title_view(&self, _window: &mut Window, cx: &Context<Self>) -> AnyElement {
|
|
const LOADING_SUMMARY_PLACEHOLDER: &str = "Loading Summary…";
|
|
|
|
let content = match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view } => {
|
|
if let Some(title_editor) = thread_view.read(cx).title_editor() {
|
|
div()
|
|
.w_full()
|
|
.on_action({
|
|
let thread_view = thread_view.downgrade();
|
|
move |_: &menu::Confirm, window, cx| {
|
|
if let Some(thread_view) = thread_view.upgrade() {
|
|
thread_view.focus_handle(cx).focus(window);
|
|
}
|
|
}
|
|
})
|
|
.on_action({
|
|
let thread_view = thread_view.downgrade();
|
|
move |_: &editor::actions::Cancel, window, cx| {
|
|
if let Some(thread_view) = thread_view.upgrade() {
|
|
thread_view.focus_handle(cx).focus(window);
|
|
}
|
|
}
|
|
})
|
|
.child(title_editor)
|
|
.into_any_element()
|
|
} else {
|
|
Label::new(thread_view.read(cx).title(cx))
|
|
.color(Color::Muted)
|
|
.truncate()
|
|
.into_any_element()
|
|
}
|
|
}
|
|
ActiveView::TextThread {
|
|
title_editor,
|
|
context_editor,
|
|
..
|
|
} => {
|
|
let summary = context_editor.read(cx).context().read(cx).summary();
|
|
|
|
match summary {
|
|
ContextSummary::Pending => Label::new(ContextSummary::DEFAULT)
|
|
.color(Color::Muted)
|
|
.truncate()
|
|
.into_any_element(),
|
|
ContextSummary::Content(summary) => {
|
|
if summary.done {
|
|
div()
|
|
.w_full()
|
|
.child(title_editor.clone())
|
|
.into_any_element()
|
|
} else {
|
|
Label::new(LOADING_SUMMARY_PLACEHOLDER)
|
|
.truncate()
|
|
.color(Color::Muted)
|
|
.into_any_element()
|
|
}
|
|
}
|
|
ContextSummary::Error => h_flex()
|
|
.w_full()
|
|
.child(title_editor.clone())
|
|
.child(
|
|
IconButton::new("retry-summary-generation", IconName::RotateCcw)
|
|
.icon_size(IconSize::Small)
|
|
.on_click({
|
|
let context_editor = context_editor.clone();
|
|
move |_, _window, cx| {
|
|
context_editor.update(cx, |context_editor, cx| {
|
|
context_editor.regenerate_summary(cx);
|
|
});
|
|
}
|
|
})
|
|
.tooltip(move |_window, cx| {
|
|
cx.new(|_| {
|
|
Tooltip::new("Failed to generate title")
|
|
.meta("Click to try again")
|
|
})
|
|
.into()
|
|
}),
|
|
)
|
|
.into_any_element(),
|
|
}
|
|
}
|
|
ActiveView::History => Label::new("History").truncate().into_any_element(),
|
|
ActiveView::Configuration => Label::new("Settings").truncate().into_any_element(),
|
|
};
|
|
|
|
h_flex()
|
|
.key_context("TitleEditor")
|
|
.id("TitleEditor")
|
|
.flex_grow()
|
|
.w_full()
|
|
.max_w_full()
|
|
.overflow_x_scroll()
|
|
.child(content)
|
|
.into_any()
|
|
}
|
|
|
|
fn render_panel_options_menu(
|
|
&self,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) -> impl IntoElement {
|
|
let user_store = self.user_store.read(cx);
|
|
let usage = user_store.model_request_usage();
|
|
let account_url = zed_urls::account_url(cx);
|
|
|
|
let focus_handle = self.focus_handle(cx);
|
|
|
|
let full_screen_label = if self.is_zoomed(window, cx) {
|
|
"Disable Full Screen"
|
|
} else {
|
|
"Enable Full Screen"
|
|
};
|
|
|
|
let selected_agent = self.selected_agent.clone();
|
|
|
|
PopoverMenu::new("agent-options-menu")
|
|
.trigger_with_tooltip(
|
|
IconButton::new("agent-options-menu", IconName::Ellipsis)
|
|
.icon_size(IconSize::Small),
|
|
{
|
|
let focus_handle = focus_handle.clone();
|
|
move |window, cx| {
|
|
Tooltip::for_action_in(
|
|
"Toggle Agent Menu",
|
|
&ToggleOptionsMenu,
|
|
&focus_handle,
|
|
window,
|
|
cx,
|
|
)
|
|
}
|
|
},
|
|
)
|
|
.anchor(Corner::TopRight)
|
|
.with_handle(self.agent_panel_menu_handle.clone())
|
|
.menu({
|
|
move |window, cx| {
|
|
Some(ContextMenu::build(window, cx, |mut menu, _window, _| {
|
|
menu = menu.context(focus_handle.clone());
|
|
if let Some(usage) = usage {
|
|
menu = menu
|
|
.header_with_link("Prompt Usage", "Manage", account_url.clone())
|
|
.custom_entry(
|
|
move |_window, cx| {
|
|
let used_percentage = match usage.limit {
|
|
UsageLimit::Limited(limit) => {
|
|
Some((usage.amount as f32 / limit as f32) * 100.)
|
|
}
|
|
UsageLimit::Unlimited => None,
|
|
};
|
|
|
|
h_flex()
|
|
.flex_1()
|
|
.gap_1p5()
|
|
.children(used_percentage.map(|percent| {
|
|
ProgressBar::new("usage", percent, 100., cx)
|
|
}))
|
|
.child(
|
|
Label::new(match usage.limit {
|
|
UsageLimit::Limited(limit) => {
|
|
format!("{} / {limit}", usage.amount)
|
|
}
|
|
UsageLimit::Unlimited => {
|
|
format!("{} / ∞", usage.amount)
|
|
}
|
|
})
|
|
.size(LabelSize::Small)
|
|
.color(Color::Muted),
|
|
)
|
|
.into_any_element()
|
|
},
|
|
move |_, cx| cx.open_url(&zed_urls::account_url(cx)),
|
|
)
|
|
.separator()
|
|
}
|
|
|
|
menu = menu
|
|
.header("MCP Servers")
|
|
.action(
|
|
"View Server Extensions",
|
|
Box::new(zed_actions::Extensions {
|
|
category_filter: Some(
|
|
zed_actions::ExtensionCategoryFilter::ContextServers,
|
|
),
|
|
id: None,
|
|
}),
|
|
)
|
|
.action("Add Custom Server…", Box::new(AddContextServer))
|
|
.separator();
|
|
|
|
menu = menu
|
|
.action("Rules…", Box::new(OpenRulesLibrary::default()))
|
|
.action("Settings", Box::new(OpenSettings))
|
|
.separator()
|
|
.action(full_screen_label, Box::new(ToggleZoom));
|
|
|
|
if selected_agent == AgentType::Gemini {
|
|
menu = menu.action("Reauthenticate", Box::new(ReauthenticateAgent))
|
|
}
|
|
|
|
menu
|
|
}))
|
|
}
|
|
})
|
|
}
|
|
|
|
fn render_recent_entries_menu(
|
|
&self,
|
|
icon: IconName,
|
|
corner: Corner,
|
|
cx: &mut Context<Self>,
|
|
) -> impl IntoElement {
|
|
let focus_handle = self.focus_handle(cx);
|
|
|
|
PopoverMenu::new("agent-nav-menu")
|
|
.trigger_with_tooltip(
|
|
IconButton::new("agent-nav-menu", icon).icon_size(IconSize::Small),
|
|
{
|
|
move |window, cx| {
|
|
Tooltip::for_action_in(
|
|
"Toggle Recent Threads",
|
|
&ToggleNavigationMenu,
|
|
&focus_handle,
|
|
window,
|
|
cx,
|
|
)
|
|
}
|
|
},
|
|
)
|
|
.anchor(corner)
|
|
.with_handle(self.agent_navigation_menu_handle.clone())
|
|
.menu({
|
|
let menu = self.agent_navigation_menu.clone();
|
|
move |window, cx| {
|
|
telemetry::event!("View Thread History Clicked");
|
|
|
|
if let Some(menu) = menu.as_ref() {
|
|
menu.update(cx, |_, cx| {
|
|
cx.defer_in(window, |menu, window, cx| {
|
|
menu.rebuild(window, cx);
|
|
});
|
|
})
|
|
}
|
|
menu.clone()
|
|
}
|
|
})
|
|
}
|
|
|
|
fn render_toolbar_back_button(&self, cx: &mut Context<Self>) -> impl IntoElement {
|
|
let focus_handle = self.focus_handle(cx);
|
|
|
|
IconButton::new("go-back", IconName::ArrowLeft)
|
|
.icon_size(IconSize::Small)
|
|
.on_click(cx.listener(|this, _, window, cx| {
|
|
this.go_back(&workspace::GoBack, window, cx);
|
|
}))
|
|
.tooltip({
|
|
move |window, cx| {
|
|
Tooltip::for_action_in("Go Back", &workspace::GoBack, &focus_handle, window, cx)
|
|
}
|
|
})
|
|
}
|
|
|
|
fn render_toolbar(&self, window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
|
|
let agent_server_store = self.project.read(cx).agent_server_store().clone();
|
|
let focus_handle = self.focus_handle(cx);
|
|
|
|
let active_thread = match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view } => {
|
|
thread_view.read(cx).as_native_thread(cx)
|
|
}
|
|
ActiveView::TextThread { .. } | ActiveView::History | ActiveView::Configuration => None,
|
|
};
|
|
|
|
let new_thread_menu = PopoverMenu::new("new_thread_menu")
|
|
.trigger_with_tooltip(
|
|
IconButton::new("new_thread_menu_btn", IconName::Plus).icon_size(IconSize::Small),
|
|
{
|
|
let focus_handle = focus_handle.clone();
|
|
move |window, cx| {
|
|
Tooltip::for_action_in(
|
|
"New…",
|
|
&ToggleNewThreadMenu,
|
|
&focus_handle,
|
|
window,
|
|
cx,
|
|
)
|
|
}
|
|
},
|
|
)
|
|
.anchor(Corner::TopRight)
|
|
.with_handle(self.new_thread_menu_handle.clone())
|
|
.menu({
|
|
let workspace = self.workspace.clone();
|
|
let is_via_collab = workspace
|
|
.update(cx, |workspace, cx| {
|
|
workspace.project().read(cx).is_via_collab()
|
|
})
|
|
.unwrap_or_default();
|
|
|
|
move |window, cx| {
|
|
telemetry::event!("New Thread Clicked");
|
|
|
|
let active_thread = active_thread.clone();
|
|
Some(ContextMenu::build(window, cx, |menu, _window, cx| {
|
|
menu
|
|
.context(focus_handle.clone())
|
|
.header("Zed Agent")
|
|
.when_some(active_thread, |this, active_thread| {
|
|
let thread = active_thread.read(cx);
|
|
|
|
if !thread.is_empty() {
|
|
let session_id = thread.id().clone();
|
|
this.item(
|
|
ContextMenuEntry::new("New From Summary")
|
|
.icon(IconName::ThreadFromSummary)
|
|
.icon_color(Color::Muted)
|
|
.handler(move |window, cx| {
|
|
window.dispatch_action(
|
|
Box::new(NewNativeAgentThreadFromSummary {
|
|
from_session_id: session_id.clone(),
|
|
}),
|
|
cx,
|
|
);
|
|
}),
|
|
)
|
|
} else {
|
|
this
|
|
}
|
|
})
|
|
.item(
|
|
ContextMenuEntry::new("New Thread")
|
|
.action(NewThread.boxed_clone())
|
|
.icon(IconName::Thread)
|
|
.icon_color(Color::Muted)
|
|
.handler({
|
|
let workspace = workspace.clone();
|
|
move |window, cx| {
|
|
if let Some(workspace) = workspace.upgrade() {
|
|
workspace.update(cx, |workspace, cx| {
|
|
if let Some(panel) =
|
|
workspace.panel::<AgentPanel>(cx)
|
|
{
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_agent_thread(
|
|
AgentType::NativeAgent,
|
|
window,
|
|
cx,
|
|
);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}),
|
|
)
|
|
.item(
|
|
ContextMenuEntry::new("New Text Thread")
|
|
.icon(IconName::TextThread)
|
|
.icon_color(Color::Muted)
|
|
.action(NewTextThread.boxed_clone())
|
|
.handler({
|
|
let workspace = workspace.clone();
|
|
move |window, cx| {
|
|
if let Some(workspace) = workspace.upgrade() {
|
|
workspace.update(cx, |workspace, cx| {
|
|
if let Some(panel) =
|
|
workspace.panel::<AgentPanel>(cx)
|
|
{
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_agent_thread(
|
|
AgentType::TextThread,
|
|
window,
|
|
cx,
|
|
);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}),
|
|
)
|
|
.separator()
|
|
.header("External Agents")
|
|
.item(
|
|
ContextMenuEntry::new("New Claude Code Thread")
|
|
.icon(IconName::AiClaude)
|
|
.disabled(is_via_collab)
|
|
.icon_color(Color::Muted)
|
|
.handler({
|
|
let workspace = workspace.clone();
|
|
move |window, cx| {
|
|
if let Some(workspace) = workspace.upgrade() {
|
|
workspace.update(cx, |workspace, cx| {
|
|
if let Some(panel) =
|
|
workspace.panel::<AgentPanel>(cx)
|
|
{
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_agent_thread(
|
|
AgentType::ClaudeCode,
|
|
window,
|
|
cx,
|
|
);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}),
|
|
)
|
|
.item(
|
|
ContextMenuEntry::new("New Codex Thread")
|
|
.icon(IconName::AiOpenAi)
|
|
.disabled(is_via_collab)
|
|
.icon_color(Color::Muted)
|
|
.handler({
|
|
let workspace = workspace.clone();
|
|
move |window, cx| {
|
|
if let Some(workspace) = workspace.upgrade() {
|
|
workspace.update(cx, |workspace, cx| {
|
|
if let Some(panel) =
|
|
workspace.panel::<AgentPanel>(cx)
|
|
{
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_agent_thread(
|
|
AgentType::Codex,
|
|
window,
|
|
cx,
|
|
);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}),
|
|
)
|
|
.item(
|
|
ContextMenuEntry::new("New Gemini CLI Thread")
|
|
.icon(IconName::AiGemini)
|
|
.icon_color(Color::Muted)
|
|
.disabled(is_via_collab)
|
|
.handler({
|
|
let workspace = workspace.clone();
|
|
move |window, cx| {
|
|
if let Some(workspace) = workspace.upgrade() {
|
|
workspace.update(cx, |workspace, cx| {
|
|
if let Some(panel) =
|
|
workspace.panel::<AgentPanel>(cx)
|
|
{
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_agent_thread(
|
|
AgentType::Gemini,
|
|
window,
|
|
cx,
|
|
);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}),
|
|
)
|
|
.map(|mut menu| {
|
|
let agent_names = agent_server_store
|
|
.read(cx)
|
|
.external_agents()
|
|
.filter(|name| {
|
|
name.0 != GEMINI_NAME && name.0 != CLAUDE_CODE_NAME && name.0 != CODEX_NAME
|
|
})
|
|
.cloned()
|
|
.collect::<Vec<_>>();
|
|
let custom_settings = cx.global::<SettingsStore>().get::<AllAgentServersSettings>(None).custom.clone();
|
|
for agent_name in agent_names {
|
|
menu = menu.item(
|
|
ContextMenuEntry::new(format!("New {} Thread", agent_name))
|
|
.icon(IconName::Terminal)
|
|
.icon_color(Color::Muted)
|
|
.disabled(is_via_collab)
|
|
.handler({
|
|
let workspace = workspace.clone();
|
|
let agent_name = agent_name.clone();
|
|
let custom_settings = custom_settings.clone();
|
|
move |window, cx| {
|
|
if let Some(workspace) = workspace.upgrade() {
|
|
workspace.update(cx, |workspace, cx| {
|
|
if let Some(panel) =
|
|
workspace.panel::<AgentPanel>(cx)
|
|
{
|
|
panel.update(cx, |panel, cx| {
|
|
panel.new_agent_thread(
|
|
AgentType::Custom {
|
|
name: agent_name.clone().into(),
|
|
command: custom_settings
|
|
.get(&agent_name.0)
|
|
.map(|settings| {
|
|
settings.command.clone()
|
|
})
|
|
.unwrap_or(placeholder_command()),
|
|
},
|
|
window,
|
|
cx,
|
|
);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}),
|
|
);
|
|
}
|
|
|
|
menu
|
|
})
|
|
.separator().link(
|
|
"Add Other Agents",
|
|
OpenBrowser {
|
|
url: zed_urls::external_agents_docs(cx),
|
|
}
|
|
.boxed_clone(),
|
|
)
|
|
}))
|
|
}
|
|
});
|
|
|
|
let selected_agent_label = self.selected_agent.label();
|
|
let selected_agent = div()
|
|
.id("selected_agent_icon")
|
|
.when_some(self.selected_agent.icon(), |this, icon| {
|
|
this.px(DynamicSpacing::Base02.rems(cx))
|
|
.child(Icon::new(icon).color(Color::Muted))
|
|
.tooltip(move |window, cx| {
|
|
Tooltip::with_meta(
|
|
selected_agent_label.clone(),
|
|
None,
|
|
"Selected Agent",
|
|
window,
|
|
cx,
|
|
)
|
|
})
|
|
})
|
|
.into_any_element();
|
|
|
|
h_flex()
|
|
.id("agent-panel-toolbar")
|
|
.h(Tab::container_height(cx))
|
|
.max_w_full()
|
|
.flex_none()
|
|
.justify_between()
|
|
.gap_2()
|
|
.bg(cx.theme().colors().tab_bar_background)
|
|
.border_b_1()
|
|
.border_color(cx.theme().colors().border)
|
|
.child(
|
|
h_flex()
|
|
.size_full()
|
|
.gap(DynamicSpacing::Base04.rems(cx))
|
|
.pl(DynamicSpacing::Base04.rems(cx))
|
|
.child(match &self.active_view {
|
|
ActiveView::History | ActiveView::Configuration => {
|
|
self.render_toolbar_back_button(cx).into_any_element()
|
|
}
|
|
_ => selected_agent.into_any_element(),
|
|
})
|
|
.child(self.render_title_view(window, cx)),
|
|
)
|
|
.child(
|
|
h_flex()
|
|
.flex_none()
|
|
.gap(DynamicSpacing::Base02.rems(cx))
|
|
.pl(DynamicSpacing::Base04.rems(cx))
|
|
.pr(DynamicSpacing::Base06.rems(cx))
|
|
.child(new_thread_menu)
|
|
.child(self.render_recent_entries_menu(
|
|
IconName::MenuAltTemp,
|
|
Corner::TopRight,
|
|
cx,
|
|
))
|
|
.child(self.render_panel_options_menu(window, cx)),
|
|
)
|
|
}
|
|
|
|
fn should_render_trial_end_upsell(&self, cx: &mut Context<Self>) -> bool {
|
|
if TrialEndUpsell::dismissed() {
|
|
return false;
|
|
}
|
|
|
|
match &self.active_view {
|
|
ActiveView::TextThread { .. } => {
|
|
if LanguageModelRegistry::global(cx)
|
|
.read(cx)
|
|
.default_model()
|
|
.is_some_and(|model| {
|
|
model.provider.id() != language_model::ZED_CLOUD_PROVIDER_ID
|
|
})
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
ActiveView::ExternalAgentThread { .. }
|
|
| ActiveView::History
|
|
| ActiveView::Configuration => return false,
|
|
}
|
|
|
|
let plan = self.user_store.read(cx).plan();
|
|
let has_previous_trial = self.user_store.read(cx).trial_started_at().is_some();
|
|
|
|
matches!(
|
|
plan,
|
|
Some(Plan::V1(PlanV1::ZedFree) | Plan::V2(PlanV2::ZedFree))
|
|
) && has_previous_trial
|
|
}
|
|
|
|
fn should_render_onboarding(&self, cx: &mut Context<Self>) -> bool {
|
|
if OnboardingUpsell::dismissed() {
|
|
return false;
|
|
}
|
|
|
|
let user_store = self.user_store.read(cx);
|
|
|
|
if user_store
|
|
.plan()
|
|
.is_some_and(|plan| matches!(plan, Plan::V1(PlanV1::ZedPro) | Plan::V2(PlanV2::ZedPro)))
|
|
&& user_store
|
|
.subscription_period()
|
|
.and_then(|period| period.0.checked_add_days(chrono::Days::new(1)))
|
|
.is_some_and(|date| date < chrono::Utc::now())
|
|
{
|
|
OnboardingUpsell::set_dismissed(true, cx);
|
|
return false;
|
|
}
|
|
|
|
match &self.active_view {
|
|
ActiveView::History | ActiveView::Configuration => false,
|
|
ActiveView::ExternalAgentThread { thread_view, .. }
|
|
if thread_view.read(cx).as_native_thread(cx).is_none() =>
|
|
{
|
|
false
|
|
}
|
|
_ => {
|
|
let history_is_empty = self.history_store.read(cx).is_empty(cx);
|
|
|
|
let has_configured_non_zed_providers = LanguageModelRegistry::read_global(cx)
|
|
.providers()
|
|
.iter()
|
|
.any(|provider| {
|
|
provider.is_authenticated(cx)
|
|
&& provider.id() != language_model::ZED_CLOUD_PROVIDER_ID
|
|
});
|
|
|
|
history_is_empty || !has_configured_non_zed_providers
|
|
}
|
|
}
|
|
}
|
|
|
|
fn render_onboarding(
|
|
&self,
|
|
_window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) -> Option<impl IntoElement> {
|
|
if !self.should_render_onboarding(cx) {
|
|
return None;
|
|
}
|
|
|
|
let text_thread_view = matches!(&self.active_view, ActiveView::TextThread { .. });
|
|
|
|
Some(
|
|
div()
|
|
.when(text_thread_view, |this| {
|
|
this.bg(cx.theme().colors().editor_background)
|
|
})
|
|
.child(self.onboarding.clone()),
|
|
)
|
|
}
|
|
|
|
fn render_trial_end_upsell(
|
|
&self,
|
|
_window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) -> Option<impl IntoElement> {
|
|
if !self.should_render_trial_end_upsell(cx) {
|
|
return None;
|
|
}
|
|
|
|
let plan = self.user_store.read(cx).plan()?;
|
|
|
|
Some(
|
|
v_flex()
|
|
.absolute()
|
|
.inset_0()
|
|
.size_full()
|
|
.bg(cx.theme().colors().panel_background)
|
|
.opacity(0.85)
|
|
.block_mouse_except_scroll()
|
|
.child(EndTrialUpsell::new(
|
|
plan,
|
|
Arc::new({
|
|
let this = cx.entity();
|
|
move |_, cx| {
|
|
this.update(cx, |_this, cx| {
|
|
TrialEndUpsell::set_dismissed(true, cx);
|
|
cx.notify();
|
|
});
|
|
}
|
|
}),
|
|
)),
|
|
)
|
|
}
|
|
|
|
fn render_configuration_error(
|
|
&self,
|
|
border_bottom: bool,
|
|
configuration_error: &ConfigurationError,
|
|
focus_handle: &FocusHandle,
|
|
window: &mut Window,
|
|
cx: &mut App,
|
|
) -> impl IntoElement {
|
|
let zed_provider_configured = AgentSettings::get_global(cx)
|
|
.default_model
|
|
.as_ref()
|
|
.is_some_and(|selection| selection.provider.0.as_str() == "zed.dev");
|
|
|
|
let callout = if zed_provider_configured {
|
|
Callout::new()
|
|
.icon(IconName::Warning)
|
|
.severity(Severity::Warning)
|
|
.when(border_bottom, |this| {
|
|
this.border_position(ui::BorderPosition::Bottom)
|
|
})
|
|
.title("Sign in to continue using Zed as your LLM provider.")
|
|
.actions_slot(
|
|
Button::new("sign_in", "Sign In")
|
|
.style(ButtonStyle::Tinted(ui::TintColor::Warning))
|
|
.label_size(LabelSize::Small)
|
|
.on_click({
|
|
let workspace = self.workspace.clone();
|
|
move |_, _, cx| {
|
|
let Ok(client) =
|
|
workspace.update(cx, |workspace, _| workspace.client().clone())
|
|
else {
|
|
return;
|
|
};
|
|
|
|
cx.spawn(async move |cx| {
|
|
client.sign_in_with_optional_connect(true, cx).await
|
|
})
|
|
.detach_and_log_err(cx);
|
|
}
|
|
}),
|
|
)
|
|
} else {
|
|
Callout::new()
|
|
.icon(IconName::Warning)
|
|
.severity(Severity::Warning)
|
|
.when(border_bottom, |this| {
|
|
this.border_position(ui::BorderPosition::Bottom)
|
|
})
|
|
.title(configuration_error.to_string())
|
|
.actions_slot(
|
|
Button::new("settings", "Configure")
|
|
.style(ButtonStyle::Tinted(ui::TintColor::Warning))
|
|
.label_size(LabelSize::Small)
|
|
.key_binding(
|
|
KeyBinding::for_action_in(&OpenSettings, focus_handle, window, cx)
|
|
.map(|kb| kb.size(rems_from_px(12.))),
|
|
)
|
|
.on_click(|_event, window, cx| {
|
|
window.dispatch_action(OpenSettings.boxed_clone(), cx)
|
|
}),
|
|
)
|
|
};
|
|
|
|
match configuration_error {
|
|
ConfigurationError::ModelNotFound
|
|
| ConfigurationError::ProviderNotAuthenticated(_)
|
|
| ConfigurationError::NoProvider => callout.into_any_element(),
|
|
}
|
|
}
|
|
|
|
fn render_text_thread(
|
|
&self,
|
|
context_editor: &Entity<TextThreadEditor>,
|
|
buffer_search_bar: &Entity<BufferSearchBar>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) -> Div {
|
|
let mut registrar = buffer_search::DivRegistrar::new(
|
|
|this, _, _cx| match &this.active_view {
|
|
ActiveView::TextThread {
|
|
buffer_search_bar, ..
|
|
} => Some(buffer_search_bar.clone()),
|
|
_ => None,
|
|
},
|
|
cx,
|
|
);
|
|
BufferSearchBar::register(&mut registrar);
|
|
registrar
|
|
.into_div()
|
|
.size_full()
|
|
.relative()
|
|
.map(|parent| {
|
|
buffer_search_bar.update(cx, |buffer_search_bar, cx| {
|
|
if buffer_search_bar.is_dismissed() {
|
|
return parent;
|
|
}
|
|
parent.child(
|
|
div()
|
|
.p(DynamicSpacing::Base08.rems(cx))
|
|
.border_b_1()
|
|
.border_color(cx.theme().colors().border_variant)
|
|
.bg(cx.theme().colors().editor_background)
|
|
.child(buffer_search_bar.render(window, cx)),
|
|
)
|
|
})
|
|
})
|
|
.child(context_editor.clone())
|
|
.child(self.render_drag_target(cx))
|
|
}
|
|
|
|
fn render_drag_target(&self, cx: &Context<Self>) -> Div {
|
|
let is_local = self.project.read(cx).is_local();
|
|
div()
|
|
.invisible()
|
|
.absolute()
|
|
.top_0()
|
|
.right_0()
|
|
.bottom_0()
|
|
.left_0()
|
|
.bg(cx.theme().colors().drop_target_background)
|
|
.drag_over::<DraggedTab>(|this, _, _, _| this.visible())
|
|
.drag_over::<DraggedSelection>(|this, _, _, _| this.visible())
|
|
.when(is_local, |this| {
|
|
this.drag_over::<ExternalPaths>(|this, _, _, _| this.visible())
|
|
})
|
|
.on_drop(cx.listener(move |this, tab: &DraggedTab, window, cx| {
|
|
let item = tab.pane.read(cx).item_for_index(tab.ix);
|
|
let project_paths = item
|
|
.and_then(|item| item.project_path(cx))
|
|
.into_iter()
|
|
.collect::<Vec<_>>();
|
|
this.handle_drop(project_paths, vec![], window, cx);
|
|
}))
|
|
.on_drop(
|
|
cx.listener(move |this, selection: &DraggedSelection, window, cx| {
|
|
let project_paths = selection
|
|
.items()
|
|
.filter_map(|item| this.project.read(cx).path_for_entry(item.entry_id, cx))
|
|
.collect::<Vec<_>>();
|
|
this.handle_drop(project_paths, vec![], window, cx);
|
|
}),
|
|
)
|
|
.on_drop(cx.listener(move |this, paths: &ExternalPaths, window, cx| {
|
|
let tasks = paths
|
|
.paths()
|
|
.iter()
|
|
.map(|path| {
|
|
Workspace::project_path_for_path(this.project.clone(), path, false, cx)
|
|
})
|
|
.collect::<Vec<_>>();
|
|
cx.spawn_in(window, async move |this, cx| {
|
|
let mut paths = vec![];
|
|
let mut added_worktrees = vec![];
|
|
let opened_paths = futures::future::join_all(tasks).await;
|
|
for entry in opened_paths {
|
|
if let Some((worktree, project_path)) = entry.log_err() {
|
|
added_worktrees.push(worktree);
|
|
paths.push(project_path);
|
|
}
|
|
}
|
|
this.update_in(cx, |this, window, cx| {
|
|
this.handle_drop(paths, added_worktrees, window, cx);
|
|
})
|
|
.ok();
|
|
})
|
|
.detach();
|
|
}))
|
|
}
|
|
|
|
fn handle_drop(
|
|
&mut self,
|
|
paths: Vec<ProjectPath>,
|
|
added_worktrees: Vec<Entity<Worktree>>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Self>,
|
|
) {
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view } => {
|
|
thread_view.update(cx, |thread_view, cx| {
|
|
thread_view.insert_dragged_files(paths, added_worktrees, window, cx);
|
|
});
|
|
}
|
|
ActiveView::TextThread { context_editor, .. } => {
|
|
context_editor.update(cx, |context_editor, cx| {
|
|
TextThreadEditor::insert_dragged_files(
|
|
context_editor,
|
|
paths,
|
|
added_worktrees,
|
|
window,
|
|
cx,
|
|
);
|
|
});
|
|
}
|
|
ActiveView::History | ActiveView::Configuration => {}
|
|
}
|
|
}
|
|
|
|
fn key_context(&self) -> KeyContext {
|
|
let mut key_context = KeyContext::new_with_defaults();
|
|
key_context.add("AgentPanel");
|
|
match &self.active_view {
|
|
ActiveView::ExternalAgentThread { .. } => key_context.add("acp_thread"),
|
|
ActiveView::TextThread { .. } => key_context.add("text_thread"),
|
|
ActiveView::History | ActiveView::Configuration => {}
|
|
}
|
|
key_context
|
|
}
|
|
}
|
|
|
|
impl Render for AgentPanel {
|
|
fn render(&mut self, window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
|
|
// WARNING: Changes to this element hierarchy can have
|
|
// non-obvious implications to the layout of children.
|
|
//
|
|
// If you need to change it, please confirm:
|
|
// - The message editor expands (cmd-option-esc) correctly
|
|
// - When expanded, the buttons at the bottom of the panel are displayed correctly
|
|
// - Font size works as expected and can be changed with cmd-+/cmd-
|
|
// - Scrolling in all views works as expected
|
|
// - Files can be dropped into the panel
|
|
let content = v_flex()
|
|
.relative()
|
|
.size_full()
|
|
.justify_between()
|
|
.key_context(self.key_context())
|
|
.on_action(cx.listener(|this, action: &NewThread, window, cx| {
|
|
this.new_thread(action, window, cx);
|
|
}))
|
|
.on_action(cx.listener(|this, _: &OpenHistory, window, cx| {
|
|
this.open_history(window, cx);
|
|
}))
|
|
.on_action(cx.listener(|this, _: &OpenSettings, window, cx| {
|
|
this.open_configuration(window, cx);
|
|
}))
|
|
.on_action(cx.listener(Self::open_active_thread_as_markdown))
|
|
.on_action(cx.listener(Self::deploy_rules_library))
|
|
.on_action(cx.listener(Self::go_back))
|
|
.on_action(cx.listener(Self::toggle_navigation_menu))
|
|
.on_action(cx.listener(Self::toggle_options_menu))
|
|
.on_action(cx.listener(Self::increase_font_size))
|
|
.on_action(cx.listener(Self::decrease_font_size))
|
|
.on_action(cx.listener(Self::reset_font_size))
|
|
.on_action(cx.listener(Self::toggle_zoom))
|
|
.on_action(cx.listener(|this, _: &ReauthenticateAgent, window, cx| {
|
|
if let Some(thread_view) = this.active_thread_view() {
|
|
thread_view.update(cx, |thread_view, cx| thread_view.reauthenticate(window, cx))
|
|
}
|
|
}))
|
|
.child(self.render_toolbar(window, cx))
|
|
.children(self.render_onboarding(window, cx))
|
|
.map(|parent| match &self.active_view {
|
|
ActiveView::ExternalAgentThread { thread_view, .. } => parent
|
|
.child(thread_view.clone())
|
|
.child(self.render_drag_target(cx)),
|
|
ActiveView::History => parent.child(self.acp_history.clone()),
|
|
ActiveView::TextThread {
|
|
context_editor,
|
|
buffer_search_bar,
|
|
..
|
|
} => {
|
|
let model_registry = LanguageModelRegistry::read_global(cx);
|
|
let configuration_error =
|
|
model_registry.configuration_error(model_registry.default_model(), cx);
|
|
parent
|
|
.map(|this| {
|
|
if !self.should_render_onboarding(cx)
|
|
&& let Some(err) = configuration_error.as_ref()
|
|
{
|
|
this.child(self.render_configuration_error(
|
|
true,
|
|
err,
|
|
&self.focus_handle(cx),
|
|
window,
|
|
cx,
|
|
))
|
|
} else {
|
|
this
|
|
}
|
|
})
|
|
.child(self.render_text_thread(
|
|
context_editor,
|
|
buffer_search_bar,
|
|
window,
|
|
cx,
|
|
))
|
|
}
|
|
ActiveView::Configuration => parent.children(self.configuration.clone()),
|
|
})
|
|
.children(self.render_trial_end_upsell(window, cx));
|
|
|
|
match self.active_view.which_font_size_used() {
|
|
WhichFontSize::AgentFont => {
|
|
WithRemSize::new(ThemeSettings::get_global(cx).agent_ui_font_size(cx))
|
|
.size_full()
|
|
.child(content)
|
|
.into_any()
|
|
}
|
|
_ => content.into_any(),
|
|
}
|
|
}
|
|
}
|
|
|
|
struct PromptLibraryInlineAssist {
|
|
workspace: WeakEntity<Workspace>,
|
|
}
|
|
|
|
impl PromptLibraryInlineAssist {
|
|
pub fn new(workspace: WeakEntity<Workspace>) -> Self {
|
|
Self { workspace }
|
|
}
|
|
}
|
|
|
|
impl rules_library::InlineAssistDelegate for PromptLibraryInlineAssist {
|
|
fn assist(
|
|
&self,
|
|
prompt_editor: &Entity<Editor>,
|
|
initial_prompt: Option<String>,
|
|
window: &mut Window,
|
|
cx: &mut Context<RulesLibrary>,
|
|
) {
|
|
InlineAssistant::update_global(cx, |assistant, cx| {
|
|
let Some(project) = self
|
|
.workspace
|
|
.upgrade()
|
|
.map(|workspace| workspace.read(cx).project().downgrade())
|
|
else {
|
|
return;
|
|
};
|
|
let prompt_store = None;
|
|
let thread_store = None;
|
|
let context_store = cx.new(|_| ContextStore::new(project.clone()));
|
|
assistant.assist(
|
|
prompt_editor,
|
|
self.workspace.clone(),
|
|
context_store,
|
|
project,
|
|
prompt_store,
|
|
thread_store,
|
|
initial_prompt,
|
|
window,
|
|
cx,
|
|
)
|
|
})
|
|
}
|
|
|
|
fn focus_agent_panel(
|
|
&self,
|
|
workspace: &mut Workspace,
|
|
window: &mut Window,
|
|
cx: &mut Context<Workspace>,
|
|
) -> bool {
|
|
workspace.focus_panel::<AgentPanel>(window, cx).is_some()
|
|
}
|
|
}
|
|
|
|
pub struct ConcreteAssistantPanelDelegate;
|
|
|
|
impl AgentPanelDelegate for ConcreteAssistantPanelDelegate {
|
|
fn active_context_editor(
|
|
&self,
|
|
workspace: &mut Workspace,
|
|
_window: &mut Window,
|
|
cx: &mut Context<Workspace>,
|
|
) -> Option<Entity<TextThreadEditor>> {
|
|
let panel = workspace.panel::<AgentPanel>(cx)?;
|
|
panel.read(cx).active_context_editor()
|
|
}
|
|
|
|
fn open_saved_context(
|
|
&self,
|
|
workspace: &mut Workspace,
|
|
path: Arc<Path>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Workspace>,
|
|
) -> Task<Result<()>> {
|
|
let Some(panel) = workspace.panel::<AgentPanel>(cx) else {
|
|
return Task::ready(Err(anyhow!("Agent panel not found")));
|
|
};
|
|
|
|
panel.update(cx, |panel, cx| {
|
|
panel.open_saved_text_thread(path, window, cx)
|
|
})
|
|
}
|
|
|
|
fn open_remote_context(
|
|
&self,
|
|
_workspace: &mut Workspace,
|
|
_context_id: assistant_context::ContextId,
|
|
_window: &mut Window,
|
|
_cx: &mut Context<Workspace>,
|
|
) -> Task<Result<Entity<TextThreadEditor>>> {
|
|
Task::ready(Err(anyhow!("opening remote context not implemented")))
|
|
}
|
|
|
|
fn quote_selection(
|
|
&self,
|
|
workspace: &mut Workspace,
|
|
selection_ranges: Vec<Range<Anchor>>,
|
|
buffer: Entity<MultiBuffer>,
|
|
window: &mut Window,
|
|
cx: &mut Context<Workspace>,
|
|
) {
|
|
let Some(panel) = workspace.panel::<AgentPanel>(cx) else {
|
|
return;
|
|
};
|
|
|
|
if !panel.focus_handle(cx).contains_focused(window, cx) {
|
|
workspace.toggle_panel_focus::<AgentPanel>(window, cx);
|
|
}
|
|
|
|
panel.update(cx, |_, cx| {
|
|
// Wait to create a new context until the workspace is no longer
|
|
// being updated.
|
|
cx.defer_in(window, move |panel, window, cx| {
|
|
if let Some(thread_view) = panel.active_thread_view() {
|
|
thread_view.update(cx, |thread_view, cx| {
|
|
thread_view.insert_selections(window, cx);
|
|
});
|
|
} else if let Some(context_editor) = panel.active_context_editor() {
|
|
let snapshot = buffer.read(cx).snapshot(cx);
|
|
let selection_ranges = selection_ranges
|
|
.into_iter()
|
|
.map(|range| range.to_point(&snapshot))
|
|
.collect::<Vec<_>>();
|
|
|
|
context_editor.update(cx, |context_editor, cx| {
|
|
context_editor.quote_ranges(selection_ranges, snapshot, window, cx)
|
|
});
|
|
}
|
|
});
|
|
});
|
|
}
|
|
}
|
|
|
|
struct OnboardingUpsell;
|
|
|
|
impl Dismissable for OnboardingUpsell {
|
|
const KEY: &'static str = "dismissed-trial-upsell";
|
|
}
|
|
|
|
struct TrialEndUpsell;
|
|
|
|
impl Dismissable for TrialEndUpsell {
|
|
const KEY: &'static str = "dismissed-trial-end-upsell";
|
|
}
|