use criterion::{criterion_group, criterion_main, Criterion};
use rust_i18n::t;

rust_i18n::i18n!("./tests/locales", minify_key = true, minify_key_len = 12);

pub fn bench_t(c: &mut Criterion) {
    c.bench_function("t", |b| b.iter(|| t!("hello")));

    c.bench_function("t_lorem_ipsum", |b| b.iter(|| t!(
            r#"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Quisque sed nisi leo. Donec commodo in ex at aliquam. Nunc in aliquam arcu. Fusce mollis metus orci, ut sagittis erat lobortis sed. Morbi quis arcu ultrices turpis finibus tincidunt non in purus. Donec gravida condimentum sapien. Duis iaculis fermentum congue. Quisque blandit libero a lacus auctor vestibulum. Nunc efficitur sollicitudin nisi, sit amet tristique lectus mollis non. Praesent sit amet erat volutpat, pharetra orci eget, rutrum felis. Sed elit augue, imperdiet eu facilisis vel, finibus vel urna. Duis quis neque metus.

            Mauris suscipit bibendum mattis. Vestibulum eu augue diam. Morbi dapibus tempus viverra. Sed aliquam turpis eget justo ornare maximus vitae et tortor. Donec semper neque sit amet sapien congue scelerisque. Maecenas bibendum imperdiet dolor interdum facilisis. Integer non diam tempus, pharetra ex at, euismod diam. Ut enim turpis, sagittis in iaculis ut, finibus et sem. Suspendisse a felis euismod neque euismod placerat. Praesent ipsum libero, porta vel egestas quis, aliquet vitae lorem. Nullam vel pharetra erat, sit amet sodales leo."#
        )));

    c.bench_function("t_with_locale", |b| b.iter(|| t!("hello", locale = "en")));

    c.bench_function("t_with_threads", |b| {
        let exit_loop = std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false));
        let mut handles = Vec::new();
        for _ in 0..4 {
            let exit_loop = exit_loop.clone();
            handles.push(std::thread::spawn(move || {
                while !exit_loop.load(std::sync::atomic::Ordering::SeqCst) {
                    criterion::black_box(t!("hello"));
                }
            }));
        }
        b.iter(|| t!("hello"));
        exit_loop.store(true, std::sync::atomic::Ordering::SeqCst);
        for handle in handles {
            handle.join().unwrap();
        }
    });

    c.bench_function("t_with_args", |b| {
        b.iter(|| {
            t!(
                "Hello, %{name}. Your message is: %{msg}",
                name = "Jason",
                msg = "Bla bla"
            )
        })
    });

    c.bench_function("t_with_args (str)", |b| {
        b.iter(|| {
            t!(
                "Hello, %{name}. Your message is: %{msg}",
                "name" = "Jason",
                "msg" = "Bla bla"
            )
        })
    });

    c.bench_function("t_with_args (many)", |b| {
        b.iter(|| {
            t!(
                r#"Hello %{name} %{surname}, your account id is %{id}, email address is %{email}.
        You live in %{city} %{zip}.
        Your website is %{website}."#,
                id = 123,
                name = "Marion",
                surname = "Christiansen",
                email = "Marion_Christiansen83@hotmail.com",
                city = "Litteltown",
                zip = 8408,
                website = "https://snoopy-napkin.name"
            )
        })
    });

    c.bench_function("t_with_args (many-dynamic)", |b| {
        let msg = r#"Hello %{name} %{surname}, your account id is %{id}, email address is %{email}.
        You live in %{city} %{zip}.
        Your website is %{website}."#
            .to_string();
        b.iter(|| {
            t!(
                &msg,
                id = 123,
                name = "Marion",
                surname = "Christiansen",
                email = "Marion_Christiansen83@hotmail.com",
                city = "Litteltown",
                zip = 8408,
                website = "https://snoopy-napkin.name"
            )
        })
    });

    c.bench_function("format! (many)", |b| {
        b.iter(|| {
            format!(
                r#"Hello {name} %{surname}, your account id is {id}, email address is {email}.
        You live in {city} {zip}.
        Your website is {website}."#,
                id = 123,
                name = "Marion",
                surname = "Christiansen",
                email = "Marion_Christiansen83@hotmail.com",
                city = "Litteltown",
                zip = 8408,
                website = "https://snoopy-napkin.name"
            );
        })
    });
}

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