loc-rs 0.1.6

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

// Author: kelexine (https://github.com/kelexine)
// export/mod.rs — Export dispatcher

pub mod csv;
pub mod html;
pub mod json;

use crate::models::ScanResult;
use anyhow::Result;
use std::path::Path;

pub enum ExportFormat {
    Json,
    Jsonl,
    Csv,
    Html,
}

impl ExportFormat {
    pub fn from_path(path: &Path) -> Option<Self> {
        let ext = path.extension()?.to_str()?.to_lowercase();
        match ext.as_str() {
            "json" => Some(Self::Json),
            "jsonl" => Some(Self::Jsonl),
            "csv" => Some(Self::Csv),
            "html" | "htm" => Some(Self::Html),
            _ => None,
        }
    }
}

pub fn export(result: &ScanResult, output_path: &str, extract_functions: bool) -> Result<()> {
    let path = Path::new(output_path);

    match ExportFormat::from_path(path) {
        Some(ExportFormat::Json) => json::export_json(result, path, extract_functions),
        Some(ExportFormat::Jsonl) => json::export_jsonl(result, path),
        Some(ExportFormat::Csv) => csv::export_csv(result, path, extract_functions),
        Some(ExportFormat::Html) => html::export_html(result, path, extract_functions),
        None => anyhow::bail!(
            "Unsupported export format '{}'. Use .json, .jsonl, .csv, or .html",
            path.extension().and_then(|e| e.to_str()).unwrap_or("?")
        ),
    }
}