1use crate::environment::EnvironmentErrorMessage;
 2use std::process::ExitStatus;
 3
 4#[cfg(not(any(target_os = "windows", test, feature = "test-support")))]
 5use {collections::HashMap, std::path::Path, util::ResultExt};
 6
 7#[derive(Clone)]
 8pub enum DirenvError {
 9    NotFound,
10    FailedRun,
11    NonZeroExit(ExitStatus, Vec<u8>),
12    EmptyOutput,
13    InvalidJson,
14}
15
16impl From<DirenvError> for Option<EnvironmentErrorMessage> {
17    fn from(value: DirenvError) -> Self {
18        match value {
19            DirenvError::NotFound => None,
20            DirenvError::FailedRun | DirenvError::NonZeroExit(_, _) => {
21                Some(EnvironmentErrorMessage(String::from(
22                    "Failed to run direnv. See logs for more info",
23                )))
24            }
25            DirenvError::EmptyOutput => None,
26            DirenvError::InvalidJson => Some(EnvironmentErrorMessage(String::from(
27                "Direnv returned invalid json. See logs for more info",
28            ))),
29        }
30    }
31}
32
33#[cfg(not(any(target_os = "windows", test, feature = "test-support")))]
34pub async fn load_direnv_environment(
35    env: &HashMap<String, String>,
36    dir: &Path,
37) -> Result<HashMap<String, String>, DirenvError> {
38    let Ok(direnv_path) = which::which("direnv") else {
39        return Err(DirenvError::NotFound);
40    };
41
42    let Some(direnv_output) = smol::process::Command::new(direnv_path)
43        .args(["export", "json"])
44        .envs(env)
45        .env("TERM", "dumb")
46        .current_dir(dir)
47        .output()
48        .await
49        .log_err()
50    else {
51        return Err(DirenvError::FailedRun);
52    };
53
54    if !direnv_output.status.success() {
55        log::error!(
56            "Loading direnv environment failed ({}), stderr: {}",
57            direnv_output.status,
58            String::from_utf8_lossy(&direnv_output.stderr)
59        );
60        return Err(DirenvError::NonZeroExit(
61            direnv_output.status,
62            direnv_output.stderr,
63        ));
64    }
65
66    let output = String::from_utf8_lossy(&direnv_output.stdout);
67    if output.is_empty() {
68        return Err(DirenvError::EmptyOutput);
69    }
70
71    let Some(env) = serde_json::from_str(&output).log_err() else {
72        return Err(DirenvError::InvalidJson);
73    };
74
75    Ok(env)
76}