1use anyhow::{anyhow, Result};
 2use std::fs;
 3
 4pub fn get_dotenv_vars() -> Result<Vec<(String, String)>> {
 5    let mut vars = Vec::new();
 6    let env_content = fs::read_to_string("./crates/collab/.env.toml")
 7        .map_err(|_| anyhow!("no .env.toml file found"))?;
 8
 9    add_vars(env_content, &mut vars)?;
10
11    if let Ok(secret_content) = fs::read_to_string("./crates/collab/.env.secret.toml") {
12        add_vars(secret_content, &mut vars)?;
13    }
14
15    Ok(vars)
16}
17
18pub fn load_dotenv() -> Result<()> {
19    for (key, value) in get_dotenv_vars()? {
20        std::env::set_var(key, value);
21    }
22    Ok(())
23}
24
25fn add_vars(env_content: String, vars: &mut Vec<(String, String)>) -> Result<()> {
26    let env: toml::map::Map<String, toml::Value> = toml::de::from_str(&env_content)?;
27    for (key, value) in env {
28        let value = match value {
29            toml::Value::String(value) => value,
30            toml::Value::Integer(value) => value.to_string(),
31            toml::Value::Float(value) => value.to_string(),
32            _ => panic!("unsupported TOML value in .env.toml for key {}", key),
33        };
34        vars.push((key, value));
35    }
36    Ok(())
37}