Load passdb \o/

examples/pass.toml

@@ -0,0 +1 @@
+Testuser = "secret"

src/db/pass.rs

@@ -1,4 +1,7 @@
 use std::sync::Arc;
+use std::path::Path;
+use std::fs;
+use std::collections::HashMap;
 
 use argon2;
 use lmdb::{Environment, Transaction, RwTransaction, Cursor};
@@ -53,4 +56,24 @@ impl PassDB {
         txn.put(self.db, &authcid.as_bytes(), &hash.as_bytes(), lmdb::WriteFlags::empty())
             .map_err(Into::into)
     }
+
+    pub fn insert_multiple(&self, vec: Vec<(String, String)>) -> Result<()> {
+        let mut txn = self.env.begin_rw_txn()?;
+        for (authcid, password) in vec.iter() {
+            self.store_with_txn(&mut txn, authcid.as_ref(), password.as_bytes())?;
+        }
+        txn.commit()?;
+
+        let v: Vec<&String> = vec.iter().map(|(a,_)| a).collect();
+        debug!(self.log, "Loaded passwords for: {:?}", v);
+
+        Ok(())
+    }
+
+    pub fn load_file<P: AsRef<Path>>(&self, path: P) -> Result<()> {
+        let f = fs::read(path)?;
+        let mut map: HashMap<String, String> = toml::from_slice(&f)?;
+
+        self.insert_multiple(map.drain().collect())
+    }
 }

src/main.rs

@@ -146,6 +146,10 @@ fn maybe(matches: clap::ArgMatches, log: Arc<Logger>) -> Result<(), Error> {
         db.access.internal.load_roles(&dir)?;
         dir.pop();
 
+        dir.push("pass.toml");
+        db.passdb.load_file(&dir);
+        dir.pop();
+
         Ok(())
     } else {
         let ex = Executor::new();