use bumpalo::Bump;
use criterion::{black_box, criterion_group, criterion_main, Criterion};
use roc_parse::{
    ast::Defs,
    header::{self, parse_module_defs},
    state::State,
};
use std::path::PathBuf;

pub fn parse_benchmark(c: &mut Criterion) {
    c.bench_function("parse false-interpreter", |b| {
        let mut path = PathBuf::from(std::env!("ROC_WORKSPACE_DIR"));
        path.push("examples");
        path.push("cli");
        path.push("false-interpreter");
        path.push("main.roc");
        let src = std::fs::read_to_string(&path).unwrap();

        b.iter(|| {
            let arena = Bump::new();

            let (_actual, state) =
                header::parse_header(&arena, State::new(src.as_bytes())).unwrap();

            let res = parse_module_defs(&arena, state, Defs::default()).unwrap();

            black_box(res.len());
        })
    });

    c.bench_function("parse Num builtin", |b| {
        let mut path = PathBuf::from(std::env!("ROC_WORKSPACE_DIR"));
        path.push("crates");
        path.push("compiler");
        path.push("builtins");
        path.push("roc");
        path.push("Num.roc");
        let src = std::fs::read_to_string(&path).unwrap();

        b.iter(|| {
            let arena = Bump::new();

            let (_actual, state) =
                header::parse_header(&arena, State::new(src.as_bytes())).unwrap();

            let res = parse_module_defs(&arena, state, Defs::default()).unwrap();

            black_box(res.len());
        })
    });
}

criterion_group!(benches, parse_benchmark);
criterion_main!(benches);