loc-rs 0.2.3

Advanced Lines of Code counter with function extraction, git integration, and parallel processing
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

#![allow(missing_docs)]

// loc — Advanced Lines of Code Counter
//
// Author : kelexine (https://github.com/kelexine)
// Version: See Cargo.toml
// License: MIT
//
// Features:
//   • True data-parallelism via Rayon (all CPU cores)
//   • Zero-copy byte-level line counting
//   • Pre-compiled regex patterns via once_cell::Lazy
//   • Richer function extraction (Rust structs/impls, Python decorators/docstrings)
//   • Filesystem mtime fallback when git is unavailable
//   • walkdir traversal (faster than os.walk)
//   • Typed errors via anyhow — no silent panics

mod cli;
mod config;
mod counter;
mod display;
mod export;
mod extractors;
mod language;
mod models;

use clap::Parser;
use colored::Colorize;
use std::process;

fn main() {
    let mut args = cli::Args::parse();

    // --func-analysis implicitly enables -f
    if args.func_analysis {
        args.functions = true;
    }

    let config = match counter::ScanConfig::from_args(&args) {
        Ok(c) => c,
        Err(e) => {
            eprintln!("{} {}", "[ERROR]".red().bold(), e);
            process::exit(1);
        }
    };

    let result = match counter::run_scan(&config) {
        Ok(r) => r,
        Err(e) => {
            eprintln!("{} {}", "[ERROR]".red().bold(), e);
            process::exit(1);
        }
    };

    // Display tree + summary
    display::display_results(
        &result,
        &config.target_dir,
        args.detailed,
        args.binary,
        args.tree,
        config.warn_size,
        config.extract_functions,
    );

    // Optional function analysis
    if args.func_analysis {
        display::display_function_analysis(&result, &config.target_dir);
    }

    // Optional export
    if let Some(ref output_file) = args.export
        && let Err(e) = export::export(&result, output_file, config.extract_functions)
    {
        eprintln!("{} {}", "[ERROR]".red().bold(), e);
        process::exit(1);
    }
}