just/src/load_dotenv.rs

48 lines
1.1 KiB
Rust
Raw Normal View History

use super::*;
2018-03-05 13:21:35 -08:00
const DEFAULT_DOTENV_FILENAME: &str = ".env";
pub(crate) fn load_dotenv(
config: &Config,
settings: &Settings,
working_directory: &Path,
) -> RunResult<'static, BTreeMap<String, String>> {
if !settings.dotenv_load.unwrap_or(false)
&& config.dotenv_filename.is_none()
&& config.dotenv_path.is_none()
{
return Ok(BTreeMap::new());
}
if let Some(path) = &config.dotenv_path {
return load_from_file(path);
}
let filename = config
.dotenv_filename
.as_deref()
.unwrap_or(DEFAULT_DOTENV_FILENAME)
.to_owned();
for directory in working_directory.ancestors() {
2022-12-15 16:53:21 -08:00
let path = directory.join(filename.as_str());
if path.is_file() {
return load_from_file(&path);
}
2018-03-05 13:21:35 -08:00
}
Ok(BTreeMap::new())
2018-03-05 13:21:35 -08:00
}
fn load_from_file(path: &Path) -> RunResult<'static, BTreeMap<String, String>> {
2022-12-14 20:32:27 -08:00
let iter = dotenvy::from_path_iter(path)?;
let mut dotenv = BTreeMap::new();
for result in iter {
let (key, value) = result?;
if env::var_os(&key).is_none() {
dotenv.insert(key, value);
}
}
Ok(dotenv)
}