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
47
48
49
50
51
52
53
54
55
56
57
58

use std::fs;
use std::path::PathBuf;
use std::thread::spawn;

use crate::calculator::Calculator;
use crate::explorer::Explorer;
use crate::reporter::{Report, Reporter};
use crate::spinner::Spinner;

pub struct Engine {
    entry: PathBuf,

    explorer: Explorer,
    calculator: Calculator,
    reporter: Reporter,
}

impl Engine {
    pub fn new(entry: PathBuf, ignore_file: Option<PathBuf>) -> Self {
        let ignore_list = Self::read_ignore_list(ignore_file);

        let (explorer, filename_receiver) = Explorer::new(ignore_list);
        let (calculator, detail_receiver) = Calculator::new(filename_receiver);
        let reporter = Reporter::new(detail_receiver);

        Self {
            entry,
            explorer,
            calculator,
            reporter,
        }
    }

    pub fn serve(self) -> Report {
        let Self {
            entry,
            mut explorer,
            calculator,
            reporter,
        } = self;
        let spinner = Spinner::new();

        spinner.start();
        spawn(move || explorer.walk_directory(entry));
        spawn(|| calculator.calculate());
        let report = reporter.research();
        spinner.stop();

        report
    }

    #[rustfmt::skip]
    fn read_ignore_list(filename: Option<PathBuf>) -> Option<Vec<PathBuf>> {
        filename
            .and_then(|filename| fs::read_to_string(filename).ok())
            .map(|content| content.lines().filter_map(|path| fs::canonicalize(path).ok()).collect::<Vec<_>>())
    }
}