#[macro_use]
extern crate criterion;

use criterion::Criterion;

use clap::{App, Arg};
use starship::context::Context;
use starship::modules;

fn char_segment(c: &mut Criterion) {
    let args = App::new("starship")
        .arg(Arg::with_name("status_code"))
        .get_matches_from(vec!["starship", "0"]);
    let context = Context::new_with_dir(args, "~");

    c.bench_function("char segment", move |b| {
        b.iter(|| modules::handle("char", &context))
    });
}

fn dir_segment(c: &mut Criterion) {
    let args = App::new("starship")
        .arg(Arg::with_name("status_code"))
        .get_matches_from(vec!["starship", "0"]);
    let context = Context::new_with_dir(args, "~");

    c.bench_function("dir segment", move |b| {
        b.iter(|| modules::handle("dir", &context))
    });
}

fn line_break_segment(c: &mut Criterion) {
    let args = App::new("starship")
        .arg(Arg::with_name("status_code"))
        .get_matches_from(vec!["starship", "0"]);
    let context = Context::new_with_dir(args, "~");

    c.bench_function("line break segment", move |b| {
        b.iter(|| modules::handle("line_break", &context))
    });
}

criterion_group!(benches, dir_segment, char_segment, line_break_segment);
criterion_main!(benches);