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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
use crate::file_handler::{File, FileHandler};
use std::fs::read_dir;
use std::path::PathBuf;
pub fn find_files(directory_path: &PathBuf) -> Vec<File> {
find_files_recursively(directory_path, directory_path)
}
fn find_files_recursively(root_path: &PathBuf, directory_path: &PathBuf) -> Vec<File> {
let mut result: Vec<File> = vec![];
let file_handler = FileHandler::default();
let entries = read_dir(directory_path).unwrap();
for entry in entries {
let path = entry.unwrap().path();
let relative_path = path.strip_prefix(root_path).unwrap();
if relative_path.starts_with(".git") || relative_path.starts_with(".gitignore") {
continue;
}
if path.is_dir() {
result.append(&mut find_files_recursively(root_path, &path));
} else {
let file_type = file_handler.identify(&path);
result.push(File { path, file_type });
}
}
result
}
#[cfg(test)]
mod tests {
use std::collections::HashSet;
use std::fs::create_dir_all;
use std::path::PathBuf;
use super::*;
use crate::file_handler::File;
use crate::file_handler::FileType;
use test_utilities::*;
fn get_paths(root_directory: &PathBuf, files: &[File]) -> HashSet<String> {
files
.iter()
.map(|file| {
file.path
.strip_prefix(root_directory)
.unwrap()
.to_string_lossy()
.to_string()
})
.collect()
}
#[test]
fn finds_all_files() {
let test_dir = setup_test_dir();
create_dir_all(test_dir.join("nested")).expect("Could not create nested test directory");
create_dir_all(test_dir.join("assets")).expect("Could not create assets test directory");
create_test_file(&test_dir.join("test1.gmi"), "");
create_test_file(&test_dir.join("_layout.html"), "");
create_test_file(&test_dir.join("nested/nested.gmi"), "");
create_test_file(&test_dir.join("assets/style.css"), "");
create_test_file(&test_dir.join("image.png"), "");
let files = find_files(&test_dir);
let paths = get_paths(&test_dir, &files);
assert!(paths.contains("test1.gmi"));
assert!(paths.contains("_layout.html"));
assert!(paths.contains("nested/nested.gmi"));
assert!(paths.contains("assets/style.css"));
assert!(paths.contains("image.png"));
}
#[test]
fn identifies_correct_file_types() {
let test_dir = setup_test_dir();
create_dir_all(test_dir.join("nested")).expect("Could not create nested test directory");
create_dir_all(test_dir.join("assets")).expect("Could not create assets test directory");
create_test_file(&test_dir.join("_layout.html"), "");
create_test_file(&test_dir.join("nested/nested.gmi"), "");
create_test_file(&test_dir.join("assets/style.css"), "");
create_test_file(&test_dir.join("image.png"), "");
let files = find_files(&test_dir);
for file in files {
let extension = file.path.extension().and_then(|e| e.to_str());
match extension {
Some("gmi") => assert_eq!(file.file_type, FileType::Gemini),
Some("html") => {
if file.path.ends_with("_layout.html") {
assert_eq!(file.file_type, FileType::Layout)
} else {
assert_eq!(file.file_type, FileType::File)
}
}
_ => assert_eq!(file.file_type, FileType::File),
}
}
}
#[test]
fn ignores_git_directory() {
let test_dir = setup_test_dir();
create_dir_all(test_dir.join("nested")).expect("Could not create nested test directory");
create_dir_all(test_dir.join("assets")).expect("Could not create assets test directory");
create_dir_all(test_dir.join(".git")).expect("Could not create git test directory");
create_test_file(&test_dir.join("_layout.html"), "");
create_test_file(&test_dir.join("nested/nested.gmi"), "");
create_test_file(&test_dir.join("assets/style.css"), "");
create_test_file(&test_dir.join("image.png"), "");
create_test_file(&test_dir.join(".git/config"), "");
let files = find_files(&test_dir);
let paths = get_paths(&test_dir, &files);
assert!(!paths.iter().any(|p| p.starts_with(".git/")));
assert!(!paths.contains(".gitignore"));
}
#[test]
fn returns_empty_for_empty_directory() {
let test_dir = setup_test_dir();
let files = find_files(&test_dir);
assert!(files.is_empty());
}
}
|
