aboutsummaryrefslogtreecommitdiff
path: root/bin/src/main.rs
blob: ae45bfe141ebf6f81622df557c2049dc9e1d9c65 (plain)
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
79
80
81
82
use std::{
    env, fs,
    path::{Path, PathBuf},
};

use anyhow::{Context, Result};
use ariadne::{Color, Fmt, Label, Report as CliReport, ReportKind as CliReportKind, Source};
use lib::{Report, LINTS};
use rnix::WalkEvent;

fn analyze(file: &str) -> Result<Vec<Report>> {
    let parsed = rnix::parse(file).as_result()?;

    Ok(parsed
        .node()
        .preorder_with_tokens()
        .filter_map(|event| match event {
            WalkEvent::Enter(child) => LINTS.get(&child.kind()).map(|rules| {
                rules
                    .iter()
                    .filter_map(|rule| rule.validate(&child))
                    .collect::<Vec<_>>()
            }),
            _ => None,
        })
        .flatten()
        .collect())
}

fn print_report(report: Report, file_src: &str, file_path: &Path) -> Result<()> {
    let src_id = file_path.to_str().unwrap_or("<unknown>");
    let offset = report
        .diagnostics
        .iter()
        .map(|d| d.at.start().into())
        .min()
        .unwrap_or(0usize);
    report
        .diagnostics
        .iter()
        .fold(
            CliReport::build(CliReportKind::Warning, src_id, offset),
            |cli_report, diagnostic| {
                let range = {
                    let at = diagnostic.at;
                    at.start().into()..at.end().into()
                };
                cli_report.with_label(
                    Label::new((src_id, range))
                        .with_message(diagnostic.message.as_str().fg(Color::Yellow)),
                )
            },
        )
        .finish()
        .print((src_id, Source::from(file_src)))
        .context("failed to print report to stdout")
}

fn _main() -> Result<()> {
    let args = env::args();
    for (file_src, file_path, reports) in args
        .map(|s| PathBuf::from(&s))
        .filter(|p| p.is_file())
        .filter_map(|path| {
            let s = fs::read_to_string(&path).ok()?;
            analyze(&s)
                .map(|analysis_result| (s, path, analysis_result))
                .ok()
        })
    {
        for r in reports {
            print_report(r, &file_src, &file_path)?
        }
    }
    Ok(())
}
fn main() {
    match _main() {
        Err(e) => eprintln!("{}", e),
        _ => {}
    }
}