toolbelt 0.0.6

a collection of simple rust utilities
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
55
56
57
58
59
60

//! Helper functions for iterating over files and folders

use std::path::Path;
use walkdir::{WalkDir, DirEntry};

pub fn entries_in_path(path: &str) -> Result<Vec<DirEntry>, String> {
    let path = Path::new(path);
    if !path.exists() {
        Err(format!("Path does not exist: {}", path.to_str().unwrap()))
    }
    else {
        let entry_names = WalkDir::new(path).max_depth(1)
            .into_iter()
            .filter_map(Result::ok)
            .skip(1)
            .collect();
        Ok(entry_names)
    }
}

pub fn files_in_path(path: &str) -> Result<Vec<DirEntry>, String> {
    match entries_in_path(path) {
        Ok(entries) => Ok(entries.iter().cloned().filter(|e| !e.file_type().is_dir()).collect()),
        Err(e) => Err(e)
    }
}

pub fn directories_in_path(path: &str) -> Result<Vec<DirEntry>, String> {
    match entries_in_path(path) {
        Ok(entries) => Ok(entries.iter().cloned().filter(|e| e.file_type().is_dir()).collect()),
        Err(e) => Err(e)
    }
}

pub fn entry_names_in_path(path: &str) -> Result<Vec<String>, String> {
    match entries_in_path(path) {
        Ok(entries) => Ok(entries.iter().cloned().map(|e| e.path().file_name().unwrap().to_str().unwrap().to_string()).collect()),
        Err(e) => Err(e)
    }
}

pub fn file_names_in_path(path: &str) -> Result<Vec<String>, String> {
    match entries_in_path(path) {
        Ok(entries) => Ok(entries.iter().cloned()
                                 .filter(|e| !e.file_type().is_dir())
                                 .map(|e| e.path().file_name().unwrap().to_str().unwrap().to_string())
                                 .collect()),
        Err(e) => Err(e)
    }
}

pub fn directory_names_in_path(path: &str) -> Result<Vec<String>, String> {
    match entries_in_path(path) {
        Ok(entries) => Ok(entries.iter().cloned()
                                        .filter(|e| e.file_type().is_dir())
                                        .map(|e| e.path().file_name().unwrap().to_str().unwrap().to_string())
                                        .collect()),
        Err(e) => Err(e)
    }
}