use remote::Interactive; use std::{ any::Any, borrow::Borrow, path::{Path, PathBuf}, str::FromStr as _, sync::Arc, time::Duration, }; use anyhow::{Context as _, Result, bail}; use collections::HashMap; use fs::{Fs, RemoveOptions, RenameOptions}; use futures::StreamExt as _; use gpui::{ AppContext as _, AsyncApp, Context, Entity, EventEmitter, SharedString, Subscription, Task, }; use http_client::{HttpClient, github::AssetKind}; use node_runtime::NodeRuntime; use remote::RemoteClient; use rpc::{ AnyProtoClient, TypedEnvelope, proto::{self, ExternalExtensionAgent}, }; use schemars::JsonSchema; use semver::Version; use serde::{Deserialize, Serialize}; use settings::{RegisterSetting, SettingsStore}; use task::{Shell, SpawnInTerminal}; use util::{ResultExt as _, debug_panic}; use crate::ProjectEnvironment; use crate::agent_registry_store::{AgentRegistryStore, RegistryAgent, RegistryTargetConfig}; #[derive(Deserialize, Serialize, Clone, PartialEq, Eq, JsonSchema)] pub struct AgentServerCommand { #[serde(rename = "command")] pub path: PathBuf, #[serde(default)] pub args: Vec, pub env: Option>, } impl std::fmt::Debug for AgentServerCommand { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { let filtered_env = self.env.as_ref().map(|env| { env.iter() .map(|(k, v)| { ( k, if util::redact::should_redact(k) { "[REDACTED]" } else { v }, ) }) .collect::>() }); f.debug_struct("AgentServerCommand") .field("path", &self.path) .field("args", &self.args) .field("env", &filtered_env) .finish() } } #[derive(Clone, Debug, PartialEq, Eq, Hash)] pub struct ExternalAgentServerName(pub SharedString); impl std::fmt::Display for ExternalAgentServerName { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { write!(f, "{}", self.0) } } impl From<&'static str> for ExternalAgentServerName { fn from(value: &'static str) -> Self { ExternalAgentServerName(value.into()) } } impl From for SharedString { fn from(value: ExternalAgentServerName) -> Self { value.0 } } impl Borrow for ExternalAgentServerName { fn borrow(&self) -> &str { &self.0 } } #[derive(Clone, Copy, Debug, Default, PartialEq, Eq)] pub enum ExternalAgentSource { Builtin, #[default] Custom, Extension, Registry, } pub trait ExternalAgentServer { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, status_tx: Option>, new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>>; fn as_any_mut(&mut self) -> &mut dyn Any; } impl dyn ExternalAgentServer { fn downcast_mut(&mut self) -> Option<&mut T> { self.as_any_mut().downcast_mut() } } enum AgentServerStoreState { Local { node_runtime: NodeRuntime, fs: Arc, project_environment: Entity, downstream_client: Option<(u64, AnyProtoClient)>, settings: Option, http_client: Arc, extension_agents: Vec<( Arc, String, HashMap, HashMap, Option, Option, )>, _subscriptions: Vec, }, Remote { project_id: u64, upstream_client: Entity, }, Collab, } pub struct ExternalAgentEntry { server: Box, icon: Option, display_name: Option, pub source: ExternalAgentSource, } impl ExternalAgentEntry { pub fn new( server: Box, source: ExternalAgentSource, icon: Option, display_name: Option, ) -> Self { Self { server, icon, display_name, source, } } } pub struct AgentServerStore { state: AgentServerStoreState, pub external_agents: HashMap, } pub struct AgentServersUpdated; impl EventEmitter for AgentServerStore {} impl AgentServerStore { /// Synchronizes extension-provided agent servers with the store. pub fn sync_extension_agents<'a, I>( &mut self, manifests: I, extensions_dir: PathBuf, cx: &mut Context, ) where I: IntoIterator, { // Collect manifests first so we can iterate twice let manifests: Vec<_> = manifests.into_iter().collect(); // Remove all extension-provided agents // (They will be re-added below if they're in the currently installed extensions) self.external_agents .retain(|_, entry| entry.source != ExternalAgentSource::Extension); // Insert agent servers from extension manifests match &mut self.state { AgentServerStoreState::Local { extension_agents, .. } => { extension_agents.clear(); for (ext_id, manifest) in manifests { for (agent_name, agent_entry) in &manifest.agent_servers { let display_name = SharedString::from(agent_entry.name.clone()); let icon_path = agent_entry.icon.as_ref().and_then(|icon| { resolve_extension_icon_path(&extensions_dir, ext_id, icon) }); extension_agents.push(( agent_name.clone(), ext_id.to_owned(), agent_entry.targets.clone(), agent_entry.env.clone(), icon_path, Some(display_name), )); } } self.reregister_agents(cx); } AgentServerStoreState::Remote { project_id, upstream_client, } => { let mut agents = vec![]; for (ext_id, manifest) in manifests { for (agent_name, agent_entry) in &manifest.agent_servers { let display_name = SharedString::from(agent_entry.name.clone()); let icon_path = agent_entry.icon.as_ref().and_then(|icon| { resolve_extension_icon_path(&extensions_dir, ext_id, icon) }); let icon_shared = icon_path .as_ref() .map(|path| SharedString::from(path.clone())); let icon = icon_path; let agent_server_name = ExternalAgentServerName(agent_name.clone().into()); self.external_agents .entry(agent_server_name.clone()) .and_modify(|entry| { entry.icon = icon_shared.clone(); entry.display_name = Some(display_name.clone()); entry.source = ExternalAgentSource::Extension; }) .or_insert_with(|| { ExternalAgentEntry::new( Box::new(RemoteExternalAgentServer { project_id: *project_id, upstream_client: upstream_client.clone(), name: agent_server_name.clone(), status_tx: None, new_version_available_tx: None, }) as Box, ExternalAgentSource::Extension, icon_shared.clone(), Some(display_name.clone()), ) }); agents.push(ExternalExtensionAgent { name: agent_name.to_string(), icon_path: icon, extension_id: ext_id.to_string(), targets: agent_entry .targets .iter() .map(|(k, v)| (k.clone(), v.to_proto())) .collect(), env: agent_entry .env .iter() .map(|(k, v)| (k.clone(), v.clone())) .collect(), }); } } upstream_client .read(cx) .proto_client() .send(proto::ExternalExtensionAgentsUpdated { project_id: *project_id, agents, }) .log_err(); } AgentServerStoreState::Collab => { // Do nothing } } cx.emit(AgentServersUpdated); } pub fn agent_icon(&self, name: &ExternalAgentServerName) -> Option { self.external_agents .get(name) .and_then(|entry| entry.icon.clone()) } pub fn agent_source(&self, name: &ExternalAgentServerName) -> Option { self.external_agents.get(name).map(|entry| entry.source) } } /// Safely resolves an extension icon path, ensuring it stays within the extension directory. /// Returns `None` if the path would escape the extension directory (path traversal attack). pub fn resolve_extension_icon_path( extensions_dir: &Path, extension_id: &str, icon_relative_path: &str, ) -> Option { let extension_root = extensions_dir.join(extension_id); let icon_path = extension_root.join(icon_relative_path); // Canonicalize both paths to resolve symlinks and normalize the paths. // For the extension root, we need to handle the case where it might be a symlink // (common for dev extensions). let canonical_extension_root = extension_root.canonicalize().unwrap_or(extension_root); let canonical_icon_path = match icon_path.canonicalize() { Ok(path) => path, Err(err) => { log::warn!( "Failed to canonicalize icon path for extension '{}': {} (path: {})", extension_id, err, icon_relative_path ); return None; } }; // Verify the resolved icon path is within the extension directory if canonical_icon_path.starts_with(&canonical_extension_root) { Some(canonical_icon_path.to_string_lossy().to_string()) } else { log::warn!( "Icon path '{}' for extension '{}' escapes extension directory, ignoring for security", icon_relative_path, extension_id ); None } } impl AgentServerStore { pub fn agent_display_name(&self, name: &ExternalAgentServerName) -> Option { self.external_agents .get(name) .and_then(|entry| entry.display_name.clone()) } pub fn init_remote(session: &AnyProtoClient) { session.add_entity_message_handler(Self::handle_external_agents_updated); session.add_entity_message_handler(Self::handle_loading_status_updated); session.add_entity_message_handler(Self::handle_new_version_available); } pub fn init_headless(session: &AnyProtoClient) { session.add_entity_message_handler(Self::handle_external_extension_agents_updated); session.add_entity_request_handler(Self::handle_get_agent_server_command); } fn agent_servers_settings_changed(&mut self, cx: &mut Context) { let AgentServerStoreState::Local { settings: old_settings, .. } = &mut self.state else { debug_panic!( "should not be subscribed to agent server settings changes in non-local project" ); return; }; let new_settings = cx .global::() .get::(None) .clone(); if Some(&new_settings) == old_settings.as_ref() { return; } self.reregister_agents(cx); } fn reregister_agents(&mut self, cx: &mut Context) { let AgentServerStoreState::Local { node_runtime, fs, project_environment, downstream_client, settings: old_settings, http_client, extension_agents, .. } = &mut self.state else { debug_panic!("Non-local projects should never attempt to reregister. This is a bug!"); return; }; let new_settings = cx .global::() .get::(None) .clone(); // If we don't have agents from the registry loaded yet, trigger a // refresh, which will cause this function to be called again if new_settings.has_registry_agents() && let Some(registry) = AgentRegistryStore::try_global(cx) { registry.update(cx, |registry, cx| registry.refresh_if_stale(cx)); } self.external_agents.clear(); self.external_agents.insert( GEMINI_NAME.into(), ExternalAgentEntry::new( Box::new(LocalGemini { fs: fs.clone(), node_runtime: node_runtime.clone(), project_environment: project_environment.clone(), custom_command: new_settings .gemini .clone() .and_then(|settings| settings.custom_command()), settings_env: new_settings .gemini .as_ref() .and_then(|settings| settings.env.clone()), ignore_system_version: new_settings .gemini .as_ref() .and_then(|settings| settings.ignore_system_version) .unwrap_or(true), }), ExternalAgentSource::Builtin, None, None, ), ); self.external_agents.insert( CODEX_NAME.into(), ExternalAgentEntry::new( Box::new(LocalCodex { fs: fs.clone(), project_environment: project_environment.clone(), custom_command: new_settings .codex .clone() .and_then(|settings| settings.custom_command()), settings_env: new_settings .codex .as_ref() .and_then(|settings| settings.env.clone()), http_client: http_client.clone(), no_browser: downstream_client .as_ref() .is_some_and(|(_, client)| !client.has_wsl_interop()), }), ExternalAgentSource::Builtin, None, None, ), ); self.external_agents.insert( CLAUDE_CODE_NAME.into(), ExternalAgentEntry::new( Box::new(LocalClaudeCode { fs: fs.clone(), node_runtime: node_runtime.clone(), project_environment: project_environment.clone(), custom_command: new_settings .claude .clone() .and_then(|settings| settings.custom_command()), settings_env: new_settings .claude .as_ref() .and_then(|settings| settings.env.clone()), }), ExternalAgentSource::Builtin, None, None, ), ); let registry_store = AgentRegistryStore::try_global(cx); let registry_agents_by_id = registry_store .as_ref() .map(|store| { store .read(cx) .agents() .iter() .cloned() .map(|agent| (agent.id().to_string(), agent)) .collect::>() }) .unwrap_or_default(); // Insert extension agents before custom/registry so registry entries override extensions. for (agent_name, ext_id, targets, env, icon_path, display_name) in extension_agents.iter() { let name = ExternalAgentServerName(agent_name.clone().into()); let mut env = env.clone(); if let Some(settings_env) = new_settings .custom .get(agent_name.as_ref()) .and_then(|settings| match settings { CustomAgentServerSettings::Extension { env, .. } => Some(env.clone()), _ => None, }) { env.extend(settings_env); } let icon = icon_path .as_ref() .map(|path| SharedString::from(path.clone())); self.external_agents.insert( name.clone(), ExternalAgentEntry::new( Box::new(LocalExtensionArchiveAgent { fs: fs.clone(), http_client: http_client.clone(), node_runtime: node_runtime.clone(), project_environment: project_environment.clone(), extension_id: Arc::from(&**ext_id), targets: targets.clone(), env, agent_id: agent_name.clone(), }) as Box, ExternalAgentSource::Extension, icon, display_name.clone(), ), ); } for (name, settings) in &new_settings.custom { match settings { CustomAgentServerSettings::Custom { command, .. } => { let agent_name = ExternalAgentServerName(name.clone().into()); self.external_agents.insert( agent_name.clone(), ExternalAgentEntry::new( Box::new(LocalCustomAgent { command: command.clone(), project_environment: project_environment.clone(), }) as Box, ExternalAgentSource::Custom, None, None, ), ); } CustomAgentServerSettings::Registry { env, .. } => { let Some(agent) = registry_agents_by_id.get(name) else { if registry_store.is_some() { log::debug!("Registry agent '{}' not found in ACP registry", name); } continue; }; let agent_name = ExternalAgentServerName(name.clone().into()); match agent { RegistryAgent::Binary(agent) => { if !agent.supports_current_platform { log::warn!( "Registry agent '{}' has no compatible binary for this platform", name ); continue; } self.external_agents.insert( agent_name.clone(), ExternalAgentEntry::new( Box::new(LocalRegistryArchiveAgent { fs: fs.clone(), http_client: http_client.clone(), node_runtime: node_runtime.clone(), project_environment: project_environment.clone(), registry_id: Arc::from(name.as_str()), targets: agent.targets.clone(), env: env.clone(), }) as Box, ExternalAgentSource::Registry, agent.metadata.icon_path.clone(), Some(agent.metadata.name.clone()), ), ); } RegistryAgent::Npx(agent) => { self.external_agents.insert( agent_name.clone(), ExternalAgentEntry::new( Box::new(LocalRegistryNpxAgent { node_runtime: node_runtime.clone(), project_environment: project_environment.clone(), package: agent.package.clone(), args: agent.args.clone(), distribution_env: agent.env.clone(), settings_env: env.clone(), }) as Box, ExternalAgentSource::Registry, agent.metadata.icon_path.clone(), Some(agent.metadata.name.clone()), ), ); } } } CustomAgentServerSettings::Extension { .. } => {} } } *old_settings = Some(new_settings); if let Some((project_id, downstream_client)) = downstream_client { downstream_client .send(proto::ExternalAgentsUpdated { project_id: *project_id, names: self .external_agents .keys() .map(|name| name.to_string()) .collect(), }) .log_err(); } cx.emit(AgentServersUpdated); } pub fn node_runtime(&self) -> Option { match &self.state { AgentServerStoreState::Local { node_runtime, .. } => Some(node_runtime.clone()), _ => None, } } pub fn local( node_runtime: NodeRuntime, fs: Arc, project_environment: Entity, http_client: Arc, cx: &mut Context, ) -> Self { let mut subscriptions = vec![cx.observe_global::(|this, cx| { this.agent_servers_settings_changed(cx); })]; if let Some(registry_store) = AgentRegistryStore::try_global(cx) { subscriptions.push(cx.observe(®istry_store, |this, _, cx| { this.reregister_agents(cx); })); } let mut this = Self { state: AgentServerStoreState::Local { node_runtime, fs, project_environment, http_client, downstream_client: None, settings: None, extension_agents: vec![], _subscriptions: subscriptions, }, external_agents: Default::default(), }; if let Some(_events) = extension::ExtensionEvents::try_global(cx) {} this.agent_servers_settings_changed(cx); this } pub(crate) fn remote(project_id: u64, upstream_client: Entity) -> Self { // Set up the builtin agents here so they're immediately available in // remote projects--we know that the HeadlessProject on the other end // will have them. let external_agents: [(ExternalAgentServerName, ExternalAgentEntry); 3] = [ ( CLAUDE_CODE_NAME.into(), ExternalAgentEntry::new( Box::new(RemoteExternalAgentServer { project_id, upstream_client: upstream_client.clone(), name: CLAUDE_CODE_NAME.into(), status_tx: None, new_version_available_tx: None, }) as Box, ExternalAgentSource::Builtin, None, None, ), ), ( CODEX_NAME.into(), ExternalAgentEntry::new( Box::new(RemoteExternalAgentServer { project_id, upstream_client: upstream_client.clone(), name: CODEX_NAME.into(), status_tx: None, new_version_available_tx: None, }) as Box, ExternalAgentSource::Builtin, None, None, ), ), ( GEMINI_NAME.into(), ExternalAgentEntry::new( Box::new(RemoteExternalAgentServer { project_id, upstream_client: upstream_client.clone(), name: GEMINI_NAME.into(), status_tx: None, new_version_available_tx: None, }) as Box, ExternalAgentSource::Builtin, None, None, ), ), ]; Self { state: AgentServerStoreState::Remote { project_id, upstream_client, }, external_agents: external_agents.into_iter().collect(), } } pub fn collab() -> Self { Self { state: AgentServerStoreState::Collab, external_agents: Default::default(), } } pub fn shared(&mut self, project_id: u64, client: AnyProtoClient, cx: &mut Context) { match &mut self.state { AgentServerStoreState::Local { downstream_client, .. } => { *downstream_client = Some((project_id, client.clone())); // Send the current list of external agents downstream, but only after a delay, // to avoid having the message arrive before the downstream project's agent server store // sets up its handlers. cx.spawn(async move |this, cx| { cx.background_executor().timer(Duration::from_secs(1)).await; let names = this.update(cx, |this, _| { this.external_agents() .map(|name| name.to_string()) .collect() })?; client .send(proto::ExternalAgentsUpdated { project_id, names }) .log_err(); anyhow::Ok(()) }) .detach(); } AgentServerStoreState::Remote { .. } => { debug_panic!( "external agents over collab not implemented, remote project should not be shared" ); } AgentServerStoreState::Collab => { debug_panic!("external agents over collab not implemented, should not be shared"); } } } pub fn get_external_agent( &mut self, name: &ExternalAgentServerName, ) -> Option<&mut (dyn ExternalAgentServer + 'static)> { self.external_agents .get_mut(name) .map(|entry| entry.server.as_mut()) } pub fn external_agents(&self) -> impl Iterator { self.external_agents.keys() } async fn handle_get_agent_server_command( this: Entity, envelope: TypedEnvelope, mut cx: AsyncApp, ) -> Result { let (command, root_dir, login_command) = this .update(&mut cx, |this, cx| { let AgentServerStoreState::Local { downstream_client, .. } = &this.state else { debug_panic!("should not receive GetAgentServerCommand in a non-local project"); bail!("unexpected GetAgentServerCommand request in a non-local project"); }; let agent = this .external_agents .get_mut(&*envelope.payload.name) .map(|entry| entry.server.as_mut()) .with_context(|| format!("agent `{}` not found", envelope.payload.name))?; let (status_tx, new_version_available_tx) = downstream_client .clone() .map(|(project_id, downstream_client)| { let (status_tx, mut status_rx) = watch::channel(SharedString::from("")); let (new_version_available_tx, mut new_version_available_rx) = watch::channel(None); cx.spawn({ let downstream_client = downstream_client.clone(); let name = envelope.payload.name.clone(); async move |_, _| { while let Some(status) = status_rx.recv().await.ok() { downstream_client.send( proto::ExternalAgentLoadingStatusUpdated { project_id, name: name.clone(), status: status.to_string(), }, )?; } anyhow::Ok(()) } }) .detach_and_log_err(cx); cx.spawn({ let name = envelope.payload.name.clone(); async move |_, _| { if let Some(version) = new_version_available_rx.recv().await.ok().flatten() { downstream_client.send( proto::NewExternalAgentVersionAvailable { project_id, name: name.clone(), version, }, )?; } anyhow::Ok(()) } }) .detach_and_log_err(cx); (status_tx, new_version_available_tx) }) .unzip(); anyhow::Ok(agent.get_command( envelope.payload.root_dir.as_deref(), HashMap::default(), status_tx, new_version_available_tx, &mut cx.to_async(), )) })? .await?; Ok(proto::AgentServerCommand { path: command.path.to_string_lossy().into_owned(), args: command.args, env: command .env .map(|env| env.into_iter().collect()) .unwrap_or_default(), root_dir: root_dir, login: login_command.map(|cmd| cmd.to_proto()), }) } async fn handle_external_agents_updated( this: Entity, envelope: TypedEnvelope, mut cx: AsyncApp, ) -> Result<()> { this.update(&mut cx, |this, cx| { let AgentServerStoreState::Remote { project_id, upstream_client, } = &this.state else { debug_panic!( "handle_external_agents_updated should not be called for a non-remote project" ); bail!("unexpected ExternalAgentsUpdated message") }; let mut previous_entries = std::mem::take(&mut this.external_agents); let mut status_txs = HashMap::default(); let mut new_version_available_txs = HashMap::default(); let mut metadata = HashMap::default(); for (name, mut entry) in previous_entries.drain() { if let Some(agent) = entry.server.downcast_mut::() { status_txs.insert(name.clone(), agent.status_tx.take()); new_version_available_txs .insert(name.clone(), agent.new_version_available_tx.take()); } metadata.insert(name, (entry.icon, entry.display_name, entry.source)); } this.external_agents = envelope .payload .names .into_iter() .map(|name| { let agent_name = ExternalAgentServerName(name.clone().into()); let fallback_source = if name == GEMINI_NAME || name == CLAUDE_CODE_NAME || name == CODEX_NAME { ExternalAgentSource::Builtin } else { ExternalAgentSource::Custom }; let (icon, display_name, source) = metadata .remove(&agent_name) .or_else(|| { AgentRegistryStore::try_global(cx) .and_then(|store| store.read(cx).agent(&agent_name.0)) .map(|s| { ( s.icon_path().cloned(), Some(s.name().clone()), ExternalAgentSource::Registry, ) }) }) .unwrap_or((None, None, fallback_source)); let source = if fallback_source == ExternalAgentSource::Builtin { ExternalAgentSource::Builtin } else { source }; let agent = RemoteExternalAgentServer { project_id: *project_id, upstream_client: upstream_client.clone(), name: agent_name.clone(), status_tx: status_txs.remove(&agent_name).flatten(), new_version_available_tx: new_version_available_txs .remove(&agent_name) .flatten(), }; ( agent_name, ExternalAgentEntry::new( Box::new(agent) as Box, source, icon, display_name, ), ) }) .collect(); cx.emit(AgentServersUpdated); Ok(()) }) } async fn handle_external_extension_agents_updated( this: Entity, envelope: TypedEnvelope, mut cx: AsyncApp, ) -> Result<()> { this.update(&mut cx, |this, cx| { let AgentServerStoreState::Local { extension_agents, .. } = &mut this.state else { panic!( "handle_external_extension_agents_updated \ should not be called for a non-remote project" ); }; for ExternalExtensionAgent { name, icon_path, extension_id, targets, env, } in envelope.payload.agents { extension_agents.push(( Arc::from(&*name), extension_id, targets .into_iter() .map(|(k, v)| (k, extension::TargetConfig::from_proto(v))) .collect(), env.into_iter().collect(), icon_path, None, )); } this.reregister_agents(cx); cx.emit(AgentServersUpdated); Ok(()) }) } async fn handle_loading_status_updated( this: Entity, envelope: TypedEnvelope, mut cx: AsyncApp, ) -> Result<()> { this.update(&mut cx, |this, _| { if let Some(agent) = this.external_agents.get_mut(&*envelope.payload.name) && let Some(agent) = agent.server.downcast_mut::() && let Some(status_tx) = &mut agent.status_tx { status_tx.send(envelope.payload.status.into()).ok(); } }); Ok(()) } async fn handle_new_version_available( this: Entity, envelope: TypedEnvelope, mut cx: AsyncApp, ) -> Result<()> { this.update(&mut cx, |this, _| { if let Some(agent) = this.external_agents.get_mut(&*envelope.payload.name) && let Some(agent) = agent.server.downcast_mut::() && let Some(new_version_available_tx) = &mut agent.new_version_available_tx { new_version_available_tx .send(Some(envelope.payload.version)) .ok(); } }); Ok(()) } pub fn get_extension_id_for_agent( &mut self, name: &ExternalAgentServerName, ) -> Option> { self.external_agents.get_mut(name).and_then(|entry| { entry .server .as_any_mut() .downcast_ref::() .map(|ext_agent| ext_agent.extension_id.clone()) }) } } fn get_or_npm_install_builtin_agent( binary_name: SharedString, package_name: SharedString, entrypoint_path: PathBuf, minimum_version: Option, status_tx: Option>, new_version_available: Option>>, fs: Arc, node_runtime: NodeRuntime, cx: &mut AsyncApp, ) -> Task> { cx.spawn(async move |cx| { let node_path = node_runtime.binary_path().await?; let dir = paths::external_agents_dir().join(binary_name.as_str()); fs.create_dir(&dir).await?; let mut stream = fs.read_dir(&dir).await?; let mut versions = Vec::new(); let mut to_delete = Vec::new(); while let Some(entry) = stream.next().await { let Ok(entry) = entry else { continue }; let Some(file_name) = entry.file_name() else { continue; }; if let Some(name) = file_name.to_str() && let Some(version) = semver::Version::from_str(name).ok() && fs .is_file(&dir.join(file_name).join(&entrypoint_path)) .await { versions.push((version, file_name.to_owned())); } else { to_delete.push(file_name.to_owned()) } } versions.sort(); let newest_version = if let Some((version, _)) = versions.last().cloned() && minimum_version.is_none_or(|minimum_version| version >= minimum_version) { versions.pop() } else { None }; log::debug!("existing version of {package_name}: {newest_version:?}"); to_delete.extend(versions.into_iter().map(|(_, file_name)| file_name)); cx.background_spawn({ let fs = fs.clone(); let dir = dir.clone(); async move { for file_name in to_delete { fs.remove_dir( &dir.join(file_name), RemoveOptions { recursive: true, ignore_if_not_exists: false, }, ) .await .ok(); } } }) .detach(); let version = if let Some((version, file_name)) = newest_version { cx.background_spawn({ let dir = dir.clone(); let fs = fs.clone(); async move { let latest_version = node_runtime .npm_package_latest_version(&package_name) .await .ok(); if let Some(latest_version) = latest_version && latest_version != version { let download_result = download_latest_version( fs, dir.clone(), node_runtime, package_name.clone(), ) .await .log_err(); if let Some(mut new_version_available) = new_version_available && download_result.is_some() { new_version_available .send(Some(latest_version.to_string())) .ok(); } } } }) .detach(); file_name } else { if let Some(mut status_tx) = status_tx { status_tx.send("Installing…".into()).ok(); } let dir = dir.clone(); cx.background_spawn(download_latest_version( fs.clone(), dir.clone(), node_runtime, package_name.clone(), )) .await? .to_string() .into() }; let agent_server_path = dir.join(version).join(entrypoint_path); let agent_server_path_exists = fs.is_file(&agent_server_path).await; anyhow::ensure!( agent_server_path_exists, "Missing entrypoint path {} after installation", agent_server_path.to_string_lossy() ); anyhow::Ok(AgentServerCommand { path: node_path, args: vec![agent_server_path.to_string_lossy().into_owned()], env: None, }) }) } fn find_bin_in_path( bin_name: SharedString, root_dir: PathBuf, env: HashMap, cx: &mut AsyncApp, ) -> Task> { cx.background_executor().spawn(async move { let which_result = if cfg!(windows) { which::which(bin_name.as_str()) } else { let shell_path = env.get("PATH").cloned(); which::which_in(bin_name.as_str(), shell_path.as_ref(), &root_dir) }; if let Err(which::Error::CannotFindBinaryPath) = which_result { return None; } which_result.log_err() }) } async fn download_latest_version( fs: Arc, dir: PathBuf, node_runtime: NodeRuntime, package_name: SharedString, ) -> Result { log::debug!("downloading latest version of {package_name}"); let tmp_dir = tempfile::tempdir_in(&dir)?; node_runtime .npm_install_packages(tmp_dir.path(), &[(&package_name, "latest")]) .await?; let version = node_runtime .npm_package_installed_version(tmp_dir.path(), &package_name) .await? .context("expected package to be installed")?; fs.rename( &tmp_dir.keep(), &dir.join(version.to_string()), RenameOptions { ignore_if_exists: true, overwrite: true, create_parents: false, }, ) .await?; anyhow::Ok(version) } struct RemoteExternalAgentServer { project_id: u64, upstream_client: Entity, name: ExternalAgentServerName, status_tx: Option>, new_version_available_tx: Option>>, } impl ExternalAgentServer for RemoteExternalAgentServer { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, status_tx: Option>, new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let project_id = self.project_id; let name = self.name.to_string(); let upstream_client = self.upstream_client.downgrade(); let root_dir = root_dir.map(|root_dir| root_dir.to_owned()); self.status_tx = status_tx; self.new_version_available_tx = new_version_available_tx; cx.spawn(async move |cx| { let mut response = upstream_client .update(cx, |upstream_client, _| { upstream_client .proto_client() .request(proto::GetAgentServerCommand { project_id, name, root_dir: root_dir.clone(), }) })? .await?; let root_dir = response.root_dir; response.env.extend(extra_env); let command = upstream_client.update(cx, |client, _| { client.build_command_with_options( Some(response.path), &response.args, &response.env.into_iter().collect(), Some(root_dir.clone()), None, Interactive::No, ) })??; Ok(( AgentServerCommand { path: command.program.into(), args: command.args, env: Some(command.env), }, root_dir, response.login.map(SpawnInTerminal::from_proto), )) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } struct LocalGemini { fs: Arc, node_runtime: NodeRuntime, project_environment: Entity, custom_command: Option, settings_env: Option>, ignore_system_version: bool, } impl ExternalAgentServer for LocalGemini { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, status_tx: Option>, new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let fs = self.fs.clone(); let node_runtime = self.node_runtime.clone(); let project_environment = self.project_environment.downgrade(); let custom_command = self.custom_command.clone(); let settings_env = self.settings_env.clone(); let ignore_system_version = self.ignore_system_version; let root_dir: Arc = root_dir .map(|root_dir| Path::new(root_dir)) .unwrap_or(paths::home_dir()) .into(); cx.spawn(async move |cx| { let mut env = project_environment .update(cx, |project_environment, cx| { project_environment.local_directory_environment( &Shell::System, root_dir.clone(), cx, ) })? .await .unwrap_or_default(); env.extend(settings_env.unwrap_or_default()); let mut command = if let Some(mut custom_command) = custom_command { custom_command.env = Some(env); custom_command } else if !ignore_system_version && let Some(bin) = find_bin_in_path("gemini".into(), root_dir.to_path_buf(), env.clone(), cx).await { AgentServerCommand { path: bin, args: Vec::new(), env: Some(env), } } else { let mut command = get_or_npm_install_builtin_agent( GEMINI_NAME.into(), "@google/gemini-cli".into(), "node_modules/@google/gemini-cli/dist/index.js".into(), if cfg!(windows) { // v0.8.x on Windows has a bug that causes the initialize request to hang forever Some("0.9.0".parse().unwrap()) } else { Some("0.2.1".parse().unwrap()) }, status_tx, new_version_available_tx, fs, node_runtime, cx, ) .await?; command.env = Some(env); command }; // Gemini CLI doesn't seem to have a dedicated invocation for logging in--we just run it normally without any arguments. let login = task::SpawnInTerminal { command: Some(command.path.to_string_lossy().into_owned()), args: command.args.clone(), env: command.env.clone().unwrap_or_default(), label: "gemini /auth".into(), ..Default::default() }; command.env.get_or_insert_default().extend(extra_env); command.args.push("--experimental-acp".into()); Ok(( command, root_dir.to_string_lossy().into_owned(), Some(login), )) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } struct LocalClaudeCode { fs: Arc, node_runtime: NodeRuntime, project_environment: Entity, custom_command: Option, settings_env: Option>, } impl ExternalAgentServer for LocalClaudeCode { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, status_tx: Option>, new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let fs = self.fs.clone(); let node_runtime = self.node_runtime.clone(); let project_environment = self.project_environment.downgrade(); let custom_command = self.custom_command.clone(); let settings_env = self.settings_env.clone(); let root_dir: Arc = root_dir .map(|root_dir| Path::new(root_dir)) .unwrap_or(paths::home_dir()) .into(); cx.spawn(async move |cx| { let mut env = project_environment .update(cx, |project_environment, cx| { project_environment.local_directory_environment( &Shell::System, root_dir.clone(), cx, ) })? .await .unwrap_or_default(); env.insert("ANTHROPIC_API_KEY".into(), "".into()); env.extend(settings_env.unwrap_or_default()); let (mut command, login_command) = if let Some(mut custom_command) = custom_command { custom_command.env = Some(env); (custom_command, None) } else { let mut command = get_or_npm_install_builtin_agent( "claude-code-acp".into(), "@zed-industries/claude-code-acp".into(), "node_modules/@zed-industries/claude-code-acp/dist/index.js".into(), Some("0.5.2".parse().unwrap()), status_tx, new_version_available_tx, fs, node_runtime, cx, ) .await?; command.env = Some(env); let login = command .args .first() .and_then(|path| { path.strip_suffix("/@zed-industries/claude-code-acp/dist/index.js") }) .map(|path_prefix| task::SpawnInTerminal { command: Some(command.path.to_string_lossy().into_owned()), args: vec![ Path::new(path_prefix) .join("@anthropic-ai/claude-agent-sdk/cli.js") .to_string_lossy() .to_string(), "/login".into(), ], env: command.env.clone().unwrap_or_default(), label: "claude /login".into(), ..Default::default() }); (command, login) }; command.env.get_or_insert_default().extend(extra_env); Ok(( command, root_dir.to_string_lossy().into_owned(), login_command, )) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } struct LocalCodex { fs: Arc, project_environment: Entity, http_client: Arc, custom_command: Option, settings_env: Option>, no_browser: bool, } impl ExternalAgentServer for LocalCodex { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, mut status_tx: Option>, _new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let fs = self.fs.clone(); let project_environment = self.project_environment.downgrade(); let http = self.http_client.clone(); let custom_command = self.custom_command.clone(); let settings_env = self.settings_env.clone(); let root_dir: Arc = root_dir .map(|root_dir| Path::new(root_dir)) .unwrap_or(paths::home_dir()) .into(); let no_browser = self.no_browser; cx.spawn(async move |cx| { let mut env = project_environment .update(cx, |project_environment, cx| { project_environment.local_directory_environment( &Shell::System, root_dir.clone(), cx, ) })? .await .unwrap_or_default(); if no_browser { env.insert("NO_BROWSER".to_owned(), "1".to_owned()); } env.extend(settings_env.unwrap_or_default()); let mut command = if let Some(mut custom_command) = custom_command { custom_command.env = Some(env); custom_command } else { let dir = paths::external_agents_dir().join(CODEX_NAME); fs.create_dir(&dir).await?; let bin_name = if cfg!(windows) { "codex-acp.exe" } else { "codex-acp" }; let find_latest_local_version = async || -> Option { let mut local_versions: Vec<(semver::Version, String)> = Vec::new(); let mut stream = fs.read_dir(&dir).await.ok()?; while let Some(entry) = stream.next().await { let Ok(entry) = entry else { continue }; let Some(file_name) = entry.file_name() else { continue; }; let version_path = dir.join(&file_name); if fs.is_file(&version_path.join(bin_name)).await { let version_str = file_name.to_string_lossy(); if let Ok(version) = semver::Version::from_str(version_str.trim_start_matches('v')) { local_versions.push((version, version_str.into_owned())); } } } local_versions.sort_by(|(a, _), (b, _)| a.cmp(b)); local_versions.last().map(|(_, v)| dir.join(v)) }; let fallback_to_latest_local_version = async |err: anyhow::Error| -> Result { if let Some(local) = find_latest_local_version().await { log::info!( "Falling back to locally installed Codex version: {}", local.display() ); Ok(local) } else { Err(err) } }; let version_dir = match ::http_client::github::latest_github_release( CODEX_ACP_REPO, true, false, http.clone(), ) .await { Ok(release) => { let version_dir = dir.join(&release.tag_name); if !fs.is_dir(&version_dir).await { if let Some(ref mut status_tx) = status_tx { status_tx.send("Installing…".into()).ok(); } let tag = release.tag_name.clone(); let version_number = tag.trim_start_matches('v'); let asset_name = asset_name(version_number) .context("codex acp is not supported for this architecture")?; let asset = release .assets .into_iter() .find(|asset| asset.name == asset_name) .with_context(|| { format!("no asset found matching `{asset_name:?}`") })?; // Strip "sha256:" prefix from digest if present (GitHub API format) let digest = asset .digest .as_deref() .and_then(|d| d.strip_prefix("sha256:").or(Some(d))); match ::http_client::github_download::download_server_binary( &*http, &asset.browser_download_url, digest, &version_dir, if cfg!(target_os = "windows") && cfg!(target_arch = "x86_64") { AssetKind::Zip } else { AssetKind::TarGz }, ) .await { Ok(()) => { // remove older versions util::fs::remove_matching(&dir, |entry| entry != version_dir) .await; version_dir } Err(err) => { log::error!( "Failed to download Codex release {}: {err:#}", release.tag_name ); fallback_to_latest_local_version(err).await? } } } else { version_dir } } Err(err) => { log::error!("Failed to fetch Codex latest release: {err:#}"); fallback_to_latest_local_version(err).await? } }; let bin_path = version_dir.join(bin_name); anyhow::ensure!( fs.is_file(&bin_path).await, "Missing Codex binary at {} after installation", bin_path.to_string_lossy() ); let mut cmd = AgentServerCommand { path: bin_path, args: Vec::new(), env: None, }; cmd.env = Some(env); cmd }; command.env.get_or_insert_default().extend(extra_env); Ok((command, root_dir.to_string_lossy().into_owned(), None)) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } pub const CODEX_ACP_REPO: &str = "zed-industries/codex-acp"; fn get_platform_info() -> Option<(&'static str, &'static str, &'static str)> { let arch = if cfg!(target_arch = "x86_64") { "x86_64" } else if cfg!(target_arch = "aarch64") { "aarch64" } else { return None; }; let platform = if cfg!(target_os = "macos") { "apple-darwin" } else if cfg!(target_os = "windows") { "pc-windows-msvc" } else if cfg!(target_os = "linux") { "unknown-linux-gnu" } else { return None; }; // Windows uses .zip in release assets let ext = if cfg!(target_os = "windows") { "zip" } else { "tar.gz" }; Some((arch, platform, ext)) } fn asset_name(version: &str) -> Option { let (arch, platform, ext) = get_platform_info()?; Some(format!("codex-acp-{version}-{arch}-{platform}.{ext}")) } pub struct LocalExtensionArchiveAgent { pub fs: Arc, pub http_client: Arc, pub node_runtime: NodeRuntime, pub project_environment: Entity, pub extension_id: Arc, pub agent_id: Arc, pub targets: HashMap, pub env: HashMap, } impl ExternalAgentServer for LocalExtensionArchiveAgent { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, _status_tx: Option>, _new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let fs = self.fs.clone(); let http_client = self.http_client.clone(); let node_runtime = self.node_runtime.clone(); let project_environment = self.project_environment.downgrade(); let extension_id = self.extension_id.clone(); let agent_id = self.agent_id.clone(); let targets = self.targets.clone(); let base_env = self.env.clone(); let root_dir: Arc = root_dir .map(|root_dir| Path::new(root_dir)) .unwrap_or(paths::home_dir()) .into(); cx.spawn(async move |cx| { // Get project environment let mut env = project_environment .update(cx, |project_environment, cx| { project_environment.local_directory_environment( &Shell::System, root_dir.clone(), cx, ) })? .await .unwrap_or_default(); // Merge manifest env and extra env env.extend(base_env); env.extend(extra_env); let cache_key = format!("{}/{}", extension_id, agent_id); let dir = paths::external_agents_dir().join(&cache_key); fs.create_dir(&dir).await?; // Determine platform key let os = if cfg!(target_os = "macos") { "darwin" } else if cfg!(target_os = "linux") { "linux" } else if cfg!(target_os = "windows") { "windows" } else { anyhow::bail!("unsupported OS"); }; let arch = if cfg!(target_arch = "aarch64") { "aarch64" } else if cfg!(target_arch = "x86_64") { "x86_64" } else { anyhow::bail!("unsupported architecture"); }; let platform_key = format!("{}-{}", os, arch); let target_config = targets.get(&platform_key).with_context(|| { format!( "no target specified for platform '{}'. Available platforms: {}", platform_key, targets .keys() .map(|k| k.as_str()) .collect::>() .join(", ") ) })?; let archive_url = &target_config.archive; // Use URL as version identifier for caching // Hash the URL to get a stable directory name use std::collections::hash_map::DefaultHasher; use std::hash::{Hash, Hasher}; let mut hasher = DefaultHasher::new(); archive_url.hash(&mut hasher); let url_hash = hasher.finish(); let version_dir = dir.join(format!("v_{:x}", url_hash)); if !fs.is_dir(&version_dir).await { // Determine SHA256 for verification let sha256 = if let Some(provided_sha) = &target_config.sha256 { // Use provided SHA256 Some(provided_sha.clone()) } else if archive_url.starts_with("https://github.com/") { // Try to fetch SHA256 from GitHub API // Parse URL to extract repo and tag/file info // Format: https://github.com/owner/repo/releases/download/tag/file.zip if let Some(caps) = archive_url.strip_prefix("https://github.com/") { let parts: Vec<&str> = caps.split('/').collect(); if parts.len() >= 6 && parts[2] == "releases" && parts[3] == "download" { let repo = format!("{}/{}", parts[0], parts[1]); let tag = parts[4]; let filename = parts[5..].join("/"); // Try to get release info from GitHub if let Ok(release) = ::http_client::github::get_release_by_tag_name( &repo, tag, http_client.clone(), ) .await { // Find matching asset if let Some(asset) = release.assets.iter().find(|a| a.name == filename) { // Strip "sha256:" prefix if present asset.digest.as_ref().map(|d| { d.strip_prefix("sha256:") .map(|s| s.to_string()) .unwrap_or_else(|| d.clone()) }) } else { None } } else { None } } else { None } } else { None } } else { None }; // Determine archive type from URL let asset_kind = if archive_url.ends_with(".zip") { AssetKind::Zip } else if archive_url.ends_with(".tar.gz") || archive_url.ends_with(".tgz") { AssetKind::TarGz } else { anyhow::bail!("unsupported archive type in URL: {}", archive_url); }; // Download and extract ::http_client::github_download::download_server_binary( &*http_client, archive_url, sha256.as_deref(), &version_dir, asset_kind, ) .await?; } // Validate and resolve cmd path let cmd = &target_config.cmd; let cmd_path = if cmd == "node" { // Use Zed's managed Node.js runtime node_runtime.binary_path().await? } else { if cmd.contains("..") { anyhow::bail!("command path cannot contain '..': {}", cmd); } if cmd.starts_with("./") || cmd.starts_with(".\\") { // Relative to extraction directory let cmd_path = version_dir.join(&cmd[2..]); anyhow::ensure!( fs.is_file(&cmd_path).await, "Missing command {} after extraction", cmd_path.to_string_lossy() ); cmd_path } else { // On PATH anyhow::bail!("command must be relative (start with './'): {}", cmd); } }; let command = AgentServerCommand { path: cmd_path, args: target_config.args.clone(), env: Some(env), }; Ok((command, version_dir.to_string_lossy().into_owned(), None)) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } struct LocalRegistryArchiveAgent { fs: Arc, http_client: Arc, node_runtime: NodeRuntime, project_environment: Entity, registry_id: Arc, targets: HashMap, env: HashMap, } impl ExternalAgentServer for LocalRegistryArchiveAgent { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, _status_tx: Option>, _new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let fs = self.fs.clone(); let http_client = self.http_client.clone(); let node_runtime = self.node_runtime.clone(); let project_environment = self.project_environment.downgrade(); let registry_id = self.registry_id.clone(); let targets = self.targets.clone(); let settings_env = self.env.clone(); let root_dir: Arc = root_dir .map(|root_dir| Path::new(root_dir)) .unwrap_or(paths::home_dir()) .into(); cx.spawn(async move |cx| { let mut env = project_environment .update(cx, |project_environment, cx| { project_environment.local_directory_environment( &Shell::System, root_dir.clone(), cx, ) })? .await .unwrap_or_default(); let dir = paths::external_agents_dir() .join("registry") .join(registry_id.as_ref()); fs.create_dir(&dir).await?; let os = if cfg!(target_os = "macos") { "darwin" } else if cfg!(target_os = "linux") { "linux" } else if cfg!(target_os = "windows") { "windows" } else { anyhow::bail!("unsupported OS"); }; let arch = if cfg!(target_arch = "aarch64") { "aarch64" } else if cfg!(target_arch = "x86_64") { "x86_64" } else { anyhow::bail!("unsupported architecture"); }; let platform_key = format!("{}-{}", os, arch); let target_config = targets.get(&platform_key).with_context(|| { format!( "no target specified for platform '{}'. Available platforms: {}", platform_key, targets .keys() .map(|k| k.as_str()) .collect::>() .join(", ") ) })?; env.extend(target_config.env.clone()); env.extend(extra_env); env.extend(settings_env); let archive_url = &target_config.archive; use std::collections::hash_map::DefaultHasher; use std::hash::{Hash, Hasher}; let mut hasher = DefaultHasher::new(); archive_url.hash(&mut hasher); let url_hash = hasher.finish(); let version_dir = dir.join(format!("v_{:x}", url_hash)); if !fs.is_dir(&version_dir).await { let sha256 = if let Some(provided_sha) = &target_config.sha256 { Some(provided_sha.clone()) } else if archive_url.starts_with("https://github.com/") { if let Some(caps) = archive_url.strip_prefix("https://github.com/") { let parts: Vec<&str> = caps.split('/').collect(); if parts.len() >= 6 && parts[2] == "releases" && parts[3] == "download" { let repo = format!("{}/{}", parts[0], parts[1]); let tag = parts[4]; let filename = parts[5..].join("/"); if let Ok(release) = ::http_client::github::get_release_by_tag_name( &repo, tag, http_client.clone(), ) .await { if let Some(asset) = release.assets.iter().find(|a| a.name == filename) { asset.digest.as_ref().and_then(|d| { d.strip_prefix("sha256:") .map(|s| s.to_string()) .or_else(|| Some(d.clone())) }) } else { None } } else { None } } else { None } } else { None } } else { None }; let asset_kind = if archive_url.ends_with(".zip") { AssetKind::Zip } else if archive_url.ends_with(".tar.gz") || archive_url.ends_with(".tgz") { AssetKind::TarGz } else { anyhow::bail!("unsupported archive type in URL: {}", archive_url); }; ::http_client::github_download::download_server_binary( &*http_client, archive_url, sha256.as_deref(), &version_dir, asset_kind, ) .await?; } let cmd = &target_config.cmd; let cmd_path = if cmd == "node" { node_runtime.binary_path().await? } else { if cmd.contains("..") { anyhow::bail!("command path cannot contain '..': {}", cmd); } if cmd.starts_with("./") || cmd.starts_with(".\\") { let cmd_path = version_dir.join(&cmd[2..]); anyhow::ensure!( fs.is_file(&cmd_path).await, "Missing command {} after extraction", cmd_path.to_string_lossy() ); cmd_path } else { anyhow::bail!("command must be relative (start with './'): {}", cmd); } }; let command = AgentServerCommand { path: cmd_path, args: target_config.args.clone(), env: Some(env), }; Ok((command, version_dir.to_string_lossy().into_owned(), None)) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } struct LocalRegistryNpxAgent { node_runtime: NodeRuntime, project_environment: Entity, package: SharedString, args: Vec, distribution_env: HashMap, settings_env: HashMap, } impl ExternalAgentServer for LocalRegistryNpxAgent { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, _status_tx: Option>, _new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let node_runtime = self.node_runtime.clone(); let project_environment = self.project_environment.downgrade(); let package = self.package.clone(); let args = self.args.clone(); let distribution_env = self.distribution_env.clone(); let settings_env = self.settings_env.clone(); let env_root_dir: Arc = root_dir .map(|root_dir| Path::new(root_dir)) .unwrap_or(paths::home_dir()) .into(); cx.spawn(async move |cx| { let mut env = project_environment .update(cx, |project_environment, cx| { project_environment.local_directory_environment( &Shell::System, env_root_dir.clone(), cx, ) })? .await .unwrap_or_default(); let mut exec_args = Vec::new(); exec_args.push("--yes".to_string()); exec_args.push(package.to_string()); if !args.is_empty() { exec_args.push("--".to_string()); exec_args.extend(args); } let npm_command = node_runtime .npm_command( "exec", &exec_args.iter().map(|a| a.as_str()).collect::>(), ) .await?; env.extend(npm_command.env); env.extend(distribution_env); env.extend(extra_env); env.extend(settings_env); let command = AgentServerCommand { path: npm_command.path, args: npm_command.args, env: Some(env), }; Ok((command, env_root_dir.to_string_lossy().into_owned(), None)) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } struct LocalCustomAgent { project_environment: Entity, command: AgentServerCommand, } impl ExternalAgentServer for LocalCustomAgent { fn get_command( &mut self, root_dir: Option<&str>, extra_env: HashMap, _status_tx: Option>, _new_version_available_tx: Option>>, cx: &mut AsyncApp, ) -> Task)>> { let mut command = self.command.clone(); let root_dir: Arc = root_dir .map(|root_dir| Path::new(root_dir)) .unwrap_or(paths::home_dir()) .into(); let project_environment = self.project_environment.downgrade(); cx.spawn(async move |cx| { let mut env = project_environment .update(cx, |project_environment, cx| { project_environment.local_directory_environment( &Shell::System, root_dir.clone(), cx, ) })? .await .unwrap_or_default(); env.extend(command.env.unwrap_or_default()); env.extend(extra_env); command.env = Some(env); Ok((command, root_dir.to_string_lossy().into_owned(), None)) }) } fn as_any_mut(&mut self) -> &mut dyn Any { self } } pub const GEMINI_NAME: &'static str = "gemini"; pub const CLAUDE_CODE_NAME: &'static str = "claude"; pub const CODEX_NAME: &'static str = "codex"; #[derive(Default, Clone, JsonSchema, Debug, PartialEq, RegisterSetting)] pub struct AllAgentServersSettings { pub gemini: Option, pub claude: Option, pub codex: Option, pub custom: HashMap, } impl AllAgentServersSettings { pub fn has_registry_agents(&self) -> bool { self.custom .values() .any(|s| matches!(s, CustomAgentServerSettings::Registry { .. })) } } #[derive(Default, Clone, JsonSchema, Debug, PartialEq)] pub struct BuiltinAgentServerSettings { pub path: Option, pub args: Option>, pub env: Option>, pub ignore_system_version: Option, pub default_mode: Option, pub default_model: Option, pub favorite_models: Vec, pub default_config_options: HashMap, pub favorite_config_option_values: HashMap>, } impl BuiltinAgentServerSettings { fn custom_command(self) -> Option { self.path.map(|path| AgentServerCommand { path, args: self.args.unwrap_or_default(), // Settings env are always applied, so we don't need to supply them here as well env: None, }) } } impl From for BuiltinAgentServerSettings { fn from(value: settings::BuiltinAgentServerSettings) -> Self { BuiltinAgentServerSettings { path: value .path .map(|p| PathBuf::from(shellexpand::tilde(&p.to_string_lossy()).as_ref())), args: value.args, env: value.env, ignore_system_version: value.ignore_system_version, default_mode: value.default_mode, default_model: value.default_model, favorite_models: value.favorite_models, default_config_options: value.default_config_options, favorite_config_option_values: value.favorite_config_option_values, } } } impl From for BuiltinAgentServerSettings { fn from(value: AgentServerCommand) -> Self { BuiltinAgentServerSettings { path: Some(value.path), args: Some(value.args), env: value.env, ..Default::default() } } } #[derive(Clone, JsonSchema, Debug, PartialEq)] pub enum CustomAgentServerSettings { Custom { command: AgentServerCommand, /// The default mode to use for this agent. /// /// Note: Not only all agents support modes. /// /// Default: None default_mode: Option, /// The default model to use for this agent. /// /// This should be the model ID as reported by the agent. /// /// Default: None default_model: Option, /// The favorite models for this agent. /// /// Default: [] favorite_models: Vec, /// Default values for session config options. /// /// This is a map from config option ID to value ID. /// /// Default: {} default_config_options: HashMap, /// Favorited values for session config options. /// /// This is a map from config option ID to a list of favorited value IDs. /// /// Default: {} favorite_config_option_values: HashMap>, }, Extension { /// Additional environment variables to pass to the agent. /// /// Default: {} env: HashMap, /// The default mode to use for this agent. /// /// Note: Not only all agents support modes. /// /// Default: None default_mode: Option, /// The default model to use for this agent. /// /// This should be the model ID as reported by the agent. /// /// Default: None default_model: Option, /// The favorite models for this agent. /// /// Default: [] favorite_models: Vec, /// Default values for session config options. /// /// This is a map from config option ID to value ID. /// /// Default: {} default_config_options: HashMap, /// Favorited values for session config options. /// /// This is a map from config option ID to a list of favorited value IDs. /// /// Default: {} favorite_config_option_values: HashMap>, }, Registry { /// Additional environment variables to pass to the agent. /// /// Default: {} env: HashMap, /// The default mode to use for this agent. /// /// Note: Not only all agents support modes. /// /// Default: None default_mode: Option, /// The default model to use for this agent. /// /// This should be the model ID as reported by the agent. /// /// Default: None default_model: Option, /// The favorite models for this agent. /// /// Default: [] favorite_models: Vec, /// Default values for session config options. /// /// This is a map from config option ID to value ID. /// /// Default: {} default_config_options: HashMap, /// Favorited values for session config options. /// /// This is a map from config option ID to a list of favorited value IDs. /// /// Default: {} favorite_config_option_values: HashMap>, }, } impl CustomAgentServerSettings { pub fn command(&self) -> Option<&AgentServerCommand> { match self { CustomAgentServerSettings::Custom { command, .. } => Some(command), CustomAgentServerSettings::Extension { .. } | CustomAgentServerSettings::Registry { .. } => None, } } pub fn default_mode(&self) -> Option<&str> { match self { CustomAgentServerSettings::Custom { default_mode, .. } | CustomAgentServerSettings::Extension { default_mode, .. } | CustomAgentServerSettings::Registry { default_mode, .. } => default_mode.as_deref(), } } pub fn default_model(&self) -> Option<&str> { match self { CustomAgentServerSettings::Custom { default_model, .. } | CustomAgentServerSettings::Extension { default_model, .. } | CustomAgentServerSettings::Registry { default_model, .. } => default_model.as_deref(), } } pub fn favorite_models(&self) -> &[String] { match self { CustomAgentServerSettings::Custom { favorite_models, .. } | CustomAgentServerSettings::Extension { favorite_models, .. } | CustomAgentServerSettings::Registry { favorite_models, .. } => favorite_models, } } pub fn default_config_option(&self, config_id: &str) -> Option<&str> { match self { CustomAgentServerSettings::Custom { default_config_options, .. } | CustomAgentServerSettings::Extension { default_config_options, .. } | CustomAgentServerSettings::Registry { default_config_options, .. } => default_config_options.get(config_id).map(|s| s.as_str()), } } pub fn favorite_config_option_values(&self, config_id: &str) -> Option<&[String]> { match self { CustomAgentServerSettings::Custom { favorite_config_option_values, .. } | CustomAgentServerSettings::Extension { favorite_config_option_values, .. } | CustomAgentServerSettings::Registry { favorite_config_option_values, .. } => favorite_config_option_values .get(config_id) .map(|v| v.as_slice()), } } } impl From for CustomAgentServerSettings { fn from(value: settings::CustomAgentServerSettings) -> Self { match value { settings::CustomAgentServerSettings::Custom { path, args, env, default_mode, default_model, favorite_models, default_config_options, favorite_config_option_values, } => CustomAgentServerSettings::Custom { command: AgentServerCommand { path: PathBuf::from(shellexpand::tilde(&path.to_string_lossy()).as_ref()), args, env: Some(env), }, default_mode, default_model, favorite_models, default_config_options, favorite_config_option_values, }, settings::CustomAgentServerSettings::Extension { env, default_mode, default_model, default_config_options, favorite_models, favorite_config_option_values, } => CustomAgentServerSettings::Extension { env, default_mode, default_model, default_config_options, favorite_models, favorite_config_option_values, }, settings::CustomAgentServerSettings::Registry { env, default_mode, default_model, default_config_options, favorite_models, favorite_config_option_values, } => CustomAgentServerSettings::Registry { env, default_mode, default_model, default_config_options, favorite_models, favorite_config_option_values, }, } } } impl settings::Settings for AllAgentServersSettings { fn from_settings(content: &settings::SettingsContent) -> Self { let agent_settings = content.agent_servers.clone().unwrap(); Self { gemini: agent_settings.gemini.map(Into::into), claude: agent_settings.claude.map(Into::into), codex: agent_settings.codex.map(Into::into), custom: agent_settings .custom .into_iter() .map(|(k, v)| (k, v.into())) .collect(), } } }