Now with better file loading stuffs

src/db/access/internal.rs

@@ -113,39 +113,10 @@ impl Internal {
    }
 
    fn dump_roles<T: Transaction>(&mut self, txn: &T, mut path: PathBuf) -> Result<()> {
-       let mut role_cursor = txn.open_ro_cursor(self.roledb)?;
+       // TODO implement this for the new format
-       for buf in role_cursor.iter_start() {
+       unimplemented!()
-           let (kbuf, vbuf) = buf?;
-           let (kbytes, _rest) = kbuf.split_at(std::mem::size_of::<u64>());
-           let roleID = u64::from_ne_bytes(kbytes.try_into().unwrap());
-           let role: Role = flexbuffers::from_slice(vbuf)?;
-           let filename = format!("{:x}.toml", roleID);
-           path.set_file_name(filename);
-           let mut fp = std::fs::File::create(&path)?;
-           let out = toml::to_vec(&role)?;
-           fp.write_all(&out)?;
-       }
-
-       Ok(())
    }
 
-   //fn dump_perms<T: Transaction>(&mut self, txn: &T, mut path: PathBuf) -> Result<()> {
-   //    let mut perm_cursor = txn.open_ro_cursor(self.permdb)?;
-   //    for buf in perm_cursor.iter_start() {
-   //        let (kbuf, vbuf) = buf?;
-   //        let (kbytes, _rest) = kbuf.split_at(std::mem::size_of::<u64>());
-   //        let permID = u64::from_ne_bytes(kbytes.try_into().unwrap());
-   //        let perm: Perm = flexbuffers::from_slice(vbuf)?;
-   //        let filename = format!("{:x}.toml", permID);
-   //        path.set_file_name(filename);
-   //        let mut fp = std::fs::File::create(&path)?;
-   //        let out = toml::to_vec(&perm)?;
-   //        fp.write_all(&out)?;
-   //    }
-
-   //    Ok(())
-   //}
-
    pub fn load_db(&mut self, txn: &mut RwTransaction, mut path: PathBuf) -> Result<()> {
        path.push("roles");
        if !path.is_dir() {
@@ -158,43 +129,19 @@ impl Internal {
    }
 
    fn load_roles(&mut self, txn: &mut RwTransaction, path: &Path) -> Result<()> {
-       for entry in std::fs::read_dir(path)? {
+       if path.is_file() {
-           let entry = entry?;
+           let roles = Role::load_file(path)?;
-           let path = entry.path();
+
-           if path.is_file() {
+           for (k,v) in roles.iter() {
-               // will only ever be none if the path has no file name and then how is it a file?!
+               self.put_role(txn, k, v.clone())?;
-               let roleID_str = path
+           }
-                   .file_stem().expect("Found a file with no filename?")
+       } else {
-                   .to_str().expect("Found an OsStr that isn't valid Unicode. Fix your OS!");
+           for entry in std::fs::read_dir(path)? {
-               let roleID = match str::parse(roleID_str) {
+               let roles = Role::load_file(entry?.path())?;
-                   Ok(i) => i,
+
-                   Err(e) => {
+               for (k,v) in roles.iter() {
-                       warn!(self.log, "File {} had a invalid name.", path.display());
+                   self.put_role(txn, k, v.clone())?;
-                       continue;
+               }
-                   }
-               };
-               let s = match fs::read_to_string(path.as_path()) {
-                   Ok(s) => s,
-                   Err(e) => {
-                       warn!(self.log, "Failed to open file {}: {}, skipping!"
-                            , path.display()
-                            , e);
-                       continue;
-                   }
-               };
-               let role: Role = match toml::from_str(&s) {
-                   Ok(r) => r,
-                   Err(e) => {
-                       warn!(self.log, "Failed to parse role at path {}: {}, skipping!"
-                            , path.display()
-                            , e);
-                       continue;
-                   }
-               };
-               self.put_role(txn, &roleID, role)?;
-               debug!(self.log, "Loaded role {}", &roleID);
-           } else {
-               warn!(self.log, "Path {} is not a file, skipping!", path.display());
            }
        }
 

src/main.rs

@@ -149,16 +149,12 @@ fn main() -> Result<(), Error> {
     if matches.is_present("load") {
         if let Some(pathstr) = matches.value_of("load") {
             let path = std::path::Path::new(pathstr);
-            if !path.is_dir() {
-                error!(log, "The provided path is not a directory or does not exist");
-                return Ok(())
-            }
 
             let mut txn = env.begin_rw_txn()?;
             let path = path.to_path_buf();
             pdb?.load_db(&mut txn, path.clone())?;
             mdb?.load_db(&mut txn, path)?;
-            txn.commit();
+            txn.commit()?;
         } else {
             error!(log, "You must provide a directory path to load from");
         }