1use futures::{FutureExt, future::Shared};
2use language::Buffer;
3use std::{path::Path, sync::Arc};
4use util::ResultExt;
5use worktree::Worktree;
6
7use collections::HashMap;
8use gpui::{AppContext as _, Context, Entity, EventEmitter, Task};
9use settings::Settings as _;
10
11use crate::{
12 project_settings::{DirenvSettings, ProjectSettings},
13 worktree_store::WorktreeStore,
14};
15
16pub struct ProjectEnvironment {
17 cli_environment: Option<HashMap<String, String>>,
18 environments: HashMap<Arc<Path>, Shared<Task<Option<HashMap<String, String>>>>>,
19 environment_error_messages: HashMap<Arc<Path>, EnvironmentErrorMessage>,
20}
21
22pub enum ProjectEnvironmentEvent {
23 ErrorsUpdated,
24}
25
26impl EventEmitter<ProjectEnvironmentEvent> for ProjectEnvironment {}
27
28impl ProjectEnvironment {
29 pub fn new(cli_environment: Option<HashMap<String, String>>) -> Self {
30 Self {
31 cli_environment,
32 environments: Default::default(),
33 environment_error_messages: Default::default(),
34 }
35 }
36
37 /// Returns the inherited CLI environment, if this project was opened from the Zed CLI.
38 pub(crate) fn get_cli_environment(&self) -> Option<HashMap<String, String>> {
39 if let Some(mut env) = self.cli_environment.clone() {
40 set_origin_marker(&mut env, EnvironmentOrigin::Cli);
41 Some(env)
42 } else {
43 None
44 }
45 }
46
47 /// Returns an iterator over all pairs `(abs_path, error_message)` of
48 /// environment errors associated with this project environment.
49 pub(crate) fn environment_errors(
50 &self,
51 ) -> impl Iterator<Item = (&Arc<Path>, &EnvironmentErrorMessage)> {
52 self.environment_error_messages.iter()
53 }
54
55 pub(crate) fn remove_environment_error(&mut self, abs_path: &Path, cx: &mut Context<Self>) {
56 self.environment_error_messages.remove(abs_path);
57 cx.emit(ProjectEnvironmentEvent::ErrorsUpdated);
58 }
59
60 pub(crate) fn get_buffer_environment(
61 &mut self,
62 buffer: &Entity<Buffer>,
63 worktree_store: &Entity<WorktreeStore>,
64 cx: &mut Context<Self>,
65 ) -> Shared<Task<Option<HashMap<String, String>>>> {
66 if cfg!(any(test, feature = "test-support")) {
67 return Task::ready(Some(HashMap::default())).shared();
68 }
69
70 if let Some(cli_environment) = self.get_cli_environment() {
71 log::debug!("using project environment variables from CLI");
72 return Task::ready(Some(cli_environment)).shared();
73 }
74
75 let Some(worktree) = buffer
76 .read(cx)
77 .file()
78 .map(|f| f.worktree_id(cx))
79 .and_then(|worktree_id| worktree_store.read(cx).worktree_for_id(worktree_id, cx))
80 else {
81 return Task::ready(None).shared();
82 };
83
84 self.get_worktree_environment(worktree, cx)
85 }
86
87 pub fn get_worktree_environment(
88 &mut self,
89 worktree: Entity<Worktree>,
90 cx: &mut Context<Self>,
91 ) -> Shared<Task<Option<HashMap<String, String>>>> {
92 if cfg!(any(test, feature = "test-support")) {
93 return Task::ready(Some(HashMap::default())).shared();
94 }
95
96 if let Some(cli_environment) = self.get_cli_environment() {
97 log::debug!("using project environment variables from CLI");
98 return Task::ready(Some(cli_environment)).shared();
99 }
100
101 let mut abs_path = worktree.read(cx).abs_path();
102 if !worktree.read(cx).is_local() {
103 log::error!(
104 "attempted to get project environment for a non-local worktree at {abs_path:?}"
105 );
106 return Task::ready(None).shared();
107 } else if worktree.read(cx).is_single_file() {
108 let Some(parent) = abs_path.parent() else {
109 return Task::ready(None).shared();
110 };
111 abs_path = parent.into();
112 }
113
114 self.get_directory_environment(abs_path, cx)
115 }
116
117 /// Returns the project environment, if possible.
118 /// If the project was opened from the CLI, then the inherited CLI environment is returned.
119 /// If it wasn't opened from the CLI, and an absolute path is given, then a shell is spawned in
120 /// that directory, to get environment variables as if the user has `cd`'d there.
121 pub fn get_directory_environment(
122 &mut self,
123 abs_path: Arc<Path>,
124 cx: &mut Context<Self>,
125 ) -> Shared<Task<Option<HashMap<String, String>>>> {
126 if cfg!(any(test, feature = "test-support")) {
127 return Task::ready(Some(HashMap::default())).shared();
128 }
129
130 if let Some(cli_environment) = self.get_cli_environment() {
131 log::debug!("using project environment variables from CLI");
132 return Task::ready(Some(cli_environment)).shared();
133 }
134
135 self.environments
136 .entry(abs_path.clone())
137 .or_insert_with(|| get_directory_env_impl(abs_path.clone(), cx).shared())
138 .clone()
139 }
140}
141
142fn set_origin_marker(env: &mut HashMap<String, String>, origin: EnvironmentOrigin) {
143 env.insert(ZED_ENVIRONMENT_ORIGIN_MARKER.to_string(), origin.into());
144}
145
146const ZED_ENVIRONMENT_ORIGIN_MARKER: &str = "ZED_ENVIRONMENT";
147
148enum EnvironmentOrigin {
149 Cli,
150 WorktreeShell,
151}
152
153impl From<EnvironmentOrigin> for String {
154 fn from(val: EnvironmentOrigin) -> Self {
155 match val {
156 EnvironmentOrigin::Cli => "cli".into(),
157 EnvironmentOrigin::WorktreeShell => "worktree-shell".into(),
158 }
159 }
160}
161
162#[derive(Debug)]
163pub struct EnvironmentErrorMessage(pub String);
164
165impl std::fmt::Display for EnvironmentErrorMessage {
166 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
167 write!(f, "{}", self.0)
168 }
169}
170
171impl EnvironmentErrorMessage {
172 #[allow(dead_code)]
173 fn from_str(s: &str) -> Self {
174 Self(String::from(s))
175 }
176}
177
178async fn load_directory_shell_environment(
179 abs_path: &Path,
180 load_direnv: &DirenvSettings,
181) -> (
182 Option<HashMap<String, String>>,
183 Option<EnvironmentErrorMessage>,
184) {
185 match smol::fs::metadata(abs_path).await {
186 Ok(meta) => {
187 let dir = if meta.is_dir() {
188 abs_path
189 } else if let Some(parent) = abs_path.parent() {
190 parent
191 } else {
192 return (
193 None,
194 Some(EnvironmentErrorMessage(format!(
195 "Failed to load shell environment in {}: not a directory",
196 abs_path.display()
197 ))),
198 );
199 };
200
201 load_shell_environment(dir, load_direnv).await
202 }
203 Err(err) => (
204 None,
205 Some(EnvironmentErrorMessage(format!(
206 "Failed to load shell environment in {}: {}",
207 abs_path.display(),
208 err
209 ))),
210 ),
211 }
212}
213
214#[cfg(any(test, feature = "test-support"))]
215async fn load_shell_environment(
216 _dir: &Path,
217 _load_direnv: &DirenvSettings,
218) -> (
219 Option<HashMap<String, String>>,
220 Option<EnvironmentErrorMessage>,
221) {
222 let fake_env = [("ZED_FAKE_TEST_ENV".into(), "true".into())]
223 .into_iter()
224 .collect();
225 (Some(fake_env), None)
226}
227
228#[cfg(all(target_os = "windows", not(any(test, feature = "test-support"))))]
229async fn load_shell_environment(
230 dir: &Path,
231 _load_direnv: &DirenvSettings,
232) -> (
233 Option<HashMap<String, String>>,
234 Option<EnvironmentErrorMessage>,
235) {
236 use util::shell_env;
237
238 let envs = match shell_env::capture(dir).await {
239 Ok(envs) => envs,
240 Err(err) => {
241 util::log_err(&err);
242 return (
243 None,
244 Some(EnvironmentErrorMessage(format!(
245 "Failed to load environment variables: {}",
246 err
247 ))),
248 );
249 }
250 };
251
252 // Note: direnv is not available on Windows, so we skip direnv processing
253 // and just return the shell environment
254 (Some(envs), None)
255}
256
257#[cfg(not(any(target_os = "windows", test, feature = "test-support")))]
258async fn load_shell_environment(
259 dir: &Path,
260 load_direnv: &DirenvSettings,
261) -> (
262 Option<HashMap<String, String>>,
263 Option<EnvironmentErrorMessage>,
264) {
265 use crate::direnv::load_direnv_environment;
266 use util::shell_env;
267
268 let dir_ = dir.to_owned();
269 let mut envs = match shell_env::capture(&dir_).await {
270 Ok(envs) => envs,
271 Err(err) => {
272 util::log_err(&err);
273 return (
274 None,
275 Some(EnvironmentErrorMessage::from_str(
276 "Failed to load environment variables. See log for details",
277 )),
278 );
279 }
280 };
281
282 // If the user selects `Direct` for direnv, it would set an environment
283 // variable that later uses to know that it should not run the hook.
284 // We would include in `.envs` call so it is okay to run the hook
285 // even if direnv direct mode is enabled.
286 let (direnv_environment, direnv_error) = match load_direnv {
287 DirenvSettings::ShellHook => (None, None),
288 DirenvSettings::Direct => match load_direnv_environment(&envs, dir).await {
289 Ok(env) => (Some(env), None),
290 Err(err) => (None, err.into()),
291 },
292 };
293 if let Some(direnv_environment) = direnv_environment {
294 for (key, value) in direnv_environment {
295 if let Some(value) = value {
296 envs.insert(key, value);
297 } else {
298 envs.remove(&key);
299 }
300 }
301 }
302
303 (Some(envs), direnv_error)
304}
305
306fn get_directory_env_impl(
307 abs_path: Arc<Path>,
308 cx: &Context<ProjectEnvironment>,
309) -> Task<Option<HashMap<String, String>>> {
310 let load_direnv = ProjectSettings::get_global(cx).load_direnv.clone();
311
312 cx.spawn(async move |this, cx| {
313 let (mut shell_env, error_message) = cx
314 .background_spawn({
315 let abs_path = abs_path.clone();
316 async move { load_directory_shell_environment(&abs_path, &load_direnv).await }
317 })
318 .await;
319
320 if let Some(shell_env) = shell_env.as_mut() {
321 let path = shell_env
322 .get("PATH")
323 .map(|path| path.as_str())
324 .unwrap_or_default();
325 log::info!(
326 "using project environment variables shell launched in {:?}. PATH={:?}",
327 abs_path,
328 path
329 );
330
331 set_origin_marker(shell_env, EnvironmentOrigin::WorktreeShell);
332 }
333
334 if let Some(error) = error_message {
335 this.update(cx, |this, cx| {
336 log::error!("{error}",);
337 this.environment_error_messages.insert(abs_path, error);
338 cx.emit(ProjectEnvironmentEvent::ErrorsUpdated)
339 })
340 .log_err();
341 }
342
343 shell_env
344 })
345}