rarity 0.3.0

Library of useful functions.
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use camino::Utf8Path;
use std::{fs, io};

/// Find files function, takes in a dir, and a closure for what to include.
pub fn find_files_in_dir<F>(
	dir: &str, recursive: bool, filter: Option<F>,
) -> io::Result<Vec<String>>
where
	F: Fn(&str) -> bool + Clone,
{
	let mut files = vec![];
	let paths = Utf8Path::read_dir_utf8(dir.into())?;
	for path in paths {
		let path = path?.path().to_string();
		let utf8_path = Utf8Path::new(&path);
		if utf8_path.is_dir() && recursive {
			files.extend(find_files_in_dir(&path, recursive, filter.clone())?);
		} else if utf8_path.is_file() {
			if let Some(ref filter) = filter {
				if filter(&path) {
					files.push(path);
				}
			} else {
				files.push(path);
			}
		}
	}
	Ok(files)
}

/// Find dirs function, takes in a dir, and a closure for what to include.
pub fn find_dirs_in_dir<F>(dir: &str, recursive: bool, filter: Option<F>) -> io::Result<Vec<String>>
where
	F: Fn(&str) -> bool + Clone,
{
	let mut dirs = vec![];
	let paths = Utf8Path::read_dir_utf8(dir.into())?;
	for path in paths {
		let path = path?.path().to_string();
		let utf8_path = Utf8Path::new(&path);
		if utf8_path.is_dir() {
			if let Some(ref filter) = filter {
				if filter(&path) {
					dirs.push(path);
					if recursive {
						dirs.extend(find_dirs_in_dir(dir, recursive, Some(filter))?)
					}
				}
			} else {
				dirs.push(path);
			}
		}
	}
	Ok(dirs)
}

/// Copy files function, takes a source and destination folder.
/// recursive option allows it to go deeper than the root source directory.
/// closure lets you filter which files get copied.
pub fn copy_files_from_dir<T, F>(
	source_dir: &str, destination_dir: &str, recursive: bool, filter: Option<F>,
) -> io::Result<()>
where
	F: Fn(&str) -> bool + Clone,
{
	let source = Utf8Path::new(source_dir);
	let dest = Utf8Path::new(destination_dir);
	let files = find_files_in_dir(source_dir, recursive, filter)?;
	for file in files {
		let destination = dest.join(Utf8Path::new(&file).strip_prefix(source).unwrap());
		fs::copy(file, destination)?;
	}
	Ok(())
}

/// Move files function, takes a source and destination folder.
/// recursive option allows it to go deeper than the root source directory.
/// closure lets you filter which files get copied.
pub fn move_files_from_dir<T, F>(
	source_dir: &str, destination_dir: &str, recursive: bool, filter: Option<F>,
) -> io::Result<()>
where
	F: Fn(&str) -> bool + Clone,
{
	let source = Utf8Path::new(source_dir);
	let dest = Utf8Path::new(destination_dir);
	let files = find_files_in_dir(source_dir, recursive, filter)?;
	for file in files {
		let destination = dest.join(Utf8Path::new(&file).strip_prefix(source).unwrap());
		fs::rename(file, destination)?;
	}
	Ok(())
}