fact-rs 0.1.0

System facts, shipped to your door!
Documentation 
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
42
43
44
45
46
47
48
49
50
51
52
53

use std::path::{Path, PathBuf};

use anyhow::{Context, Result};

pub fn slurp(path: impl AsRef<Path>) -> Result<String> {
    std::fs::read_to_string(&path)
        .with_context(|| format!("reading {}", path.as_ref().display()))
        .map(|s| s.trim().to_string())
}

#[allow(dead_code)]
pub fn get_dirs_in_path(path: PathBuf) -> Result<Vec<String>> {
    let mut dirs = Vec::new();
    let entries = std::fs::read_dir(&path)
        .with_context(|| format!("listing contents of {}", path.display()))?;
    for entry in entries {
        if let Ok(entry) = entry {
            if let Ok(file_type) = entry.file_type() {
                if file_type.is_dir() {
                    if let Some(name) = entry.file_name().to_str() {
                        dirs.push(name.to_string());
                    }
                }
            }
        }
    }
    Ok(dirs)
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::{fs::File, io::Write};
    use tempfile::tempdir;

    #[test]
    fn test_slurp() {
        let dir = tempdir().unwrap();
        let root = dir.path();

        let file1path = root.join("test_slurp1");
        let mut file1 = File::create(file1path.clone()).unwrap();
        let _ = file1.write_all(b"testcontent1");
        let content = slurp(file1path.clone()).unwrap();
        assert_eq!(content, "testcontent1");

        let file2path = root.join("test_slurp2");
        let mut file2 = File::create(file2path.clone()).unwrap();
        let _ = file2.write_all(b"1\\^%(*!#_)FA");
        let content = slurp(file2path.clone()).unwrap();
        assert_eq!(content, "1\\^%(*!#_)FA");
    }
}