confignet 0.0.3

Classifier to detect CI/CD config files based on file name and MIME label.
Documentation 
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

use std::{env, path::Path, process};
use confignet::ConfigClassifier;
use anyhow::Result;
use serde_json::{json, Value};

fn main() -> Result<()> {
// 1. Get file_path and mime_type from args
let args: Vec<String> = env::args().collect();
if args.len() != 3 {
eprintln!("Usage: confignet <file_path> <mime_type>");
process::exit(1);
}

let file_path = args[1].clone();
let mime_type = args[2].clone();
let file_name = Path::new(&file_path)
    .file_name()
    .and_then(|n| n.to_str())
    .unwrap_or("")
    .to_string();

// 2. Load classifier
let classifier = ConfigClassifier::from_csv("data/labeled/ci_cd.csv")?;

// 3. Classify
let result: Value = match classifier.classify(&file_name, &mime_type) {
    Some(confignet::ClassifiedResult { file_name, is_ci_cd, .. }) => {
        let path = if Path::new(&file_path).parent().unwrap_or_else(|| Path::new("")) == Path::new("") {
            format!("./{}", file_name)
        } else {
            file_path.clone()
        };

        json!({
            "file_name": file_name,
            "file_path": path,
            "is_ci_cd": is_ci_cd
        })
    }
    None => {
        let path = if Path::new(&file_path).parent().unwrap_or_else(|| Path::new("")) == Path::new("") {
            format!("./{}", file_name)
        } else {
            file_path.clone()
        };

        json!({
            "file_name": file_name,
            "file_path": path,
            "is_ci_cd": false
        })
    }
};

// Instead of println! send this `result` to your parser
// For now, just printing nicely for testing
println!("{}", serde_json::to_string_pretty(&result)?);

Ok(())

}