use std::{env, fs};
use std::io::stdout;
use std::time::Instant;

use minify_html::{Cfg, minify};

fn main() {
    let iterations = env::var("MHB_ITERATIONS").unwrap().parse::<usize>().unwrap();
    let input_dir = env::var("MHB_INPUT_DIR").unwrap();
    let html_only = env::var("MHB_HTML_ONLY").unwrap() == "1";

    let tests = fs::read_dir(input_dir).unwrap().map(|d| d.unwrap());

    let mut results: Vec<(String, f64)> = Vec::new();
    let mut cfg = Cfg::new();
    if !html_only {
        cfg.minify_css = true;
        cfg.minify_js = true;
    };

    for t in tests {
        let source = fs::read(t.path()).unwrap();
        let start = Instant::now();
        for _ in 0..iterations {
            let data = source.to_vec();
            minify(&source, &cfg);
        };
        let elapsed = start.elapsed().as_secs_f64();
        results.push((t.file_name().into_string().unwrap(), elapsed));
    };

    serde_json::to_writer(stdout(), &results).unwrap();
}