1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use anyhow::anyhow;
use std::fs;

pub fn load_dotenv() -> anyhow::Result<()> {
    let env: toml::map::Map<String, toml::Value> = toml::de::from_str(
        &fs::read_to_string("./.env.toml").map_err(|_| anyhow!("no .env.toml file found"))?,
    )?;

    for (key, value) in env {
        let value = match value {
            toml::Value::String(value) => value,
            toml::Value::Integer(value) => value.to_string(),
            toml::Value::Float(value) => value.to_string(),
            _ => panic!("unsupported TOML value in .env.toml for key {}", key),
        };
        std::env::set_var(key, value);
    }

    Ok(())
}