1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
mod lines;
use std::{collections::HashMap, fs::read_to_string, io::Result, path::PathBuf};

// Just re-exporting to use as a standalone parser
pub use lines::{Line, Lines};

#[derive(Debug)]
pub struct Zenv {
    path: PathBuf,
    expand: bool,
}

impl Zenv {
    pub fn new(path: PathBuf, expand: bool) -> Self {
        Self { path, expand }
    }

    pub fn parse(&self) -> Result<HashMap<String, String>> {
        let lines = {
            let r = read_to_string(&self.path)?;
            Lines::from(r)
        };

        let hash = match self.expand {
            true => lines.expand(),
            false => lines.into_hash_map(),
        };

        Ok(hash)
    }

    pub fn configure(&self) -> Result<()> {
        let vars = self.parse()?;

        for (key, val) in vars {
            std::env::set_var(key, val);
        }

        Ok(())
    }
}