src/search/entry.rs


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

use ignore::{Walk, WalkBuilder};
use tracing::error;

use crate::{
    config::Entry,
    project::{path::PathMatcher, ProjectParser, ProjectParserGroup},
    search::ProjectItem,
};

pub struct EntryIter {
    parsers: ProjectParserGroup,
    iter: Walk,
}

impl EntryIter {
    fn new(config: &Entry) -> Self {
        let iter = WalkBuilder::new(&config.path_buf)
            .standard_filters(true)
            .max_depth(config.max_depth)
            .hidden(!config.hidden)
            .build();

        let mut parsers = ProjectParserGroup::new();

        if let Some(s) = config.pattern.as_ref() {
            parsers.push(Box::new(PathMatcher(s.to_owned())));
        };

        #[cfg(feature = "git")]
        if config.git {
            parsers.push(Box::new(crate::project::git::GitMatcher));
        };

        Self { parsers, iter }
    }
}

impl From<Entry> for EntryIter {
    fn from(config: Entry) -> Self {
        Self::new(&config)
    }
}

impl Iterator for EntryIter {
    type Item = ProjectItem;

    fn next(&mut self) -> Option<Self::Item> {
        match self.iter.next()? {
            Ok(dir_entry) => self.parsers.parse(dir_entry.into_path()),
            Err(err) => {
                error!(%err, "Ignoring errored path");
                None
            }
        }
        .or_else(|| self.next())
    }
}