cloc 0.6.2

Count, or compute differences of, lines of source code and comments.
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

use std::fs;
use std::path::{Path, PathBuf};
use std::sync::mpsc::{sync_channel, Receiver, SyncSender};

pub struct Explorer {
    ignore_list: Option<Vec<PathBuf>>,
    sender: SyncSender<PathBuf>,
}

impl Explorer {
    pub fn new(ignore_list: Option<Vec<PathBuf>>) -> (Self, Receiver<PathBuf>) {
        let (sender, receiver) = sync_channel::<PathBuf>(1024);
        let explorer = Self { ignore_list, sender };
        (explorer, receiver)
    }

    pub fn walk_directory<P: AsRef<Path>>(&mut self, entry: P) {
        self.walk_dir_impl(entry.as_ref());
    }

    fn walk_dir_impl(&mut self, path: &Path) {
        if path.is_file() && self.is_not_ignore_file(path) {
            let _ = self.sender.send(PathBuf::from(path));
        } else if path.is_dir() {
            if let Ok(entries) = fs::read_dir(path) {
                entries
                    .filter_map(|entry| entry.ok())
                    .for_each(|entry| self.walk_dir_impl(&entry.path()));
            }
        }
    }

    #[inline]
    fn is_not_ignore_file<P: AsRef<Path>>(&self, filename: P) -> bool {
        self.is_not_ignore_file_impl(filename.as_ref())
    }

    #[inline]
    fn is_not_ignore_file_impl(&self, filename: &Path) -> bool {
        if let Some(ignore_list) = &self.ignore_list {
            return ignore_list.iter().all(|path| !filename.starts_with(path));
        }

        true
    }
}