treewalk 0.1.4

Common utilities for exploring a file tree
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
54

use std::io::{self, ErrorKind};
use std::path::Path;
use std::path::PathBuf;

pub fn children(path: &Path) -> Result<Vec<PathBuf>, io::Error> {
    let mut children: Vec<PathBuf> = Vec::new();
    let path = path.read_dir();

    if let Ok(path) = path {
        for child in path.flatten() {
            let child_path = child.path();
            children.push(child_path);
        }
    } else if let Err(path) = path {
        //panic!("Attempt to read contents of directory has failed!")
        eprintln!("Attempt to read contents of {path} has failed!")
    }

    Ok(children)
}

/// returns a [Vec] of the children of a [Path] (recursively)
pub fn get_all_children(path: &Path) -> Result<Vec<PathBuf>, io::Error> {
    let mut children: Vec<PathBuf> = Vec::new();
    if path.try_exists()? {
        get_all_driver(path, &mut children);
        Ok(children)
    } else {
        Err(ErrorKind::NotFound.into())
    }
}

/// is a driver since passing in a mutable [Vec] and returning
/// nothing is pretty ugly
fn get_all_driver(path: &Path, children: &mut Vec<PathBuf>) {
    if path.is_dir() {
        let path = path.read_dir();
        if let Ok(path) = path {
            for child in path.flatten() {
                let child_path = child.path();
                children.push(child_path.to_path_buf());
                get_all_driver(&child_path, children);
            }
        } else if let Err(error) = path {
            // report and leave out the directories that cannot be read
            eprintln!(
                "{}: Attempt to read contents of {:?} has failed!",
                error.kind(),
                children.last().unwrap()
            );
            children.pop();
        }
    }
}