cargo-warloc 0.1.1

Smart LOC counter for Rust projects
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

mod cli;
mod output;
mod visitor;
mod warlocs;

use std::{
    collections::BTreeMap,
    path::{Path, PathBuf},
};

use clap::Parser;
use cli::{CargoCli, Cli};
use ignore::Walk;
use visitor::Visitor;
use warlocs::Warlocs;

use crate::output::{output_multiple_file_stats, output_total_stats};

fn main() {
    let CargoCli::Command(args) = CargoCli::parse();

    let root_dir = PathBuf::from(".");

    let files_stats: BTreeMap<PathBuf, Warlocs> = enumerate_rust_files(&root_dir)
        .map(|p| (p.clone(), calculate_file_stats(&p, &args)))
        .collect();

    if args.by_file {
        output_multiple_file_stats(&args.output_format, files_stats);
    } else {
        let file_count = files_stats.len();
        let total_stats = files_stats.into_values().sum();
        output_total_stats(file_count as u64, &total_stats, &args.output_format);
    }
}

fn calculate_file_stats(file_path: impl AsRef<Path>, args: &Cli) -> Warlocs {
    Visitor::new(&file_path, args.debug).visit_file()
}

fn enumerate_rust_files(root: impl AsRef<Path>) -> impl Iterator<Item = PathBuf> {
    Walk::new(root)
        .filter_map(|entry| entry.ok())
        .filter(|entry| {
            entry.file_type().is_some_and(|e| e.is_file())
                && entry
                    .file_name()
                    .to_str()
                    .map(|name| name.ends_with(".rs"))
                    .unwrap_or(false)
        })
        .map(|entry| entry.into_path())
}