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
use crate::config::TwmGlobal;
use crate::workspace::path_meets_workspace_conditions;
use jwalk::{
WalkDir,
rayon::{
current_num_threads,
iter::{ParallelBridge, ParallelIterator},
},
};
use nucleo::Injector;
pub fn find_workspaces_in_dir(dir: &str, config: &TwmGlobal, injector: Injector<String>) {
WalkDir::new(dir)
.max_depth(config.max_search_depth)
.skip_hidden(false)
.follow_links(config.follow_links)
.parallelism(jwalk::Parallelism::RayonNewPool(std::cmp::max(
1,
current_num_threads() - 1,
)))
.into_iter()
.par_bridge()
.filter_map(std::result::Result::ok)
.filter(|e| {
e.file_type().is_dir()
// this can definitely be improved in the future
&& !e.path().components().any(|c| match c.as_os_str().to_str() {
Some(s) => config.exclude_path_components.iter().any(|e| s == e),
None => true,
})
})
.for_each(|entry| {
for workspace_definition in &config.workspace_definitions {
if path_meets_workspace_conditions(&entry.path(), &workspace_definition.conditions)
{
// just skip the path if it's not valid utf-8 since we can't use it
// skip here instead of checking earlier because i don't expect people having a bunch of non-utf8 paths to be common, so defer the check only if we have a match in the first place
if let Some(utf8_path) = entry.path().to_str() {
// previously we also stored which workspace type we matched on, but i decided to change it because we only ever need to know the workspace type for the workspace we're opening anyways
// having to re-lookup the workspace type on user selection is surely better than the hashmap we were using before, but better would probably be to just keep track of which WorkspaceDefinition matched here
// main reason I haven't yet is because I'm not entirely sure how to make that work nicely with the fuzzy finders
injector.push(utf8_path.to_string(), |_, dst| {
dst[0] = utf8_path.to_string().into()
});
}
break;
}
}
});
}