use capycrypt::{
    ecc::{encryptable::KeyEncryptable, keypair::KeyPair, signable::Signable},
    sha3::{aux_functions::byte_utils::get_random_bytes, encryptable::SpongeEncryptable},
    Message, SecParam,
    SecParam::D512,
};
use criterion::{criterion_group, criterion_main, Criterion};

const BIT_SECURITY: SecParam = D512;

/// Symmetric encrypt and decrypt roundtrip
fn sym_enc(pw: &[u8], mut msg: Message) {
    msg.sha3_encrypt(pw, BIT_SECURITY);
    let _ = msg.sha3_decrypt(pw);
}

/// Asymmetric encrypt and decrypt roundtrip + keygen
fn key_gen_enc_dec(pw: &[u8], mut msg: Message) {
    let key_pair = KeyPair::new(pw, "test key".to_string(), BIT_SECURITY);
    msg.key_encrypt(&key_pair.pub_key, BIT_SECURITY);
    let _ = msg.key_decrypt(&key_pair.priv_key);
}

/// Signature generation + verification roundtrip
pub fn sign_verify(key_pair: KeyPair, mut msg: Message) {
    msg.sign(&key_pair, BIT_SECURITY);
    let _ = msg.verify(&key_pair.pub_key);
}

fn bench_sign_verify(c: &mut Criterion) {
    c.bench_function("e448 + SHA3-512 Sign + Verify Roundtrip 5mb", |b| {
        b.iter(|| {
            sign_verify(
                KeyPair::new(&get_random_bytes(16), "test key".to_string(), BIT_SECURITY),
                Message::new(get_random_bytes(5242880)),
            )
        });
    });
}

fn bench_sym_enc(c: &mut Criterion) {
    c.bench_function("SHA3-512 Symmetric enc + dec 5mb", |b| {
        b.iter(|| {
            sym_enc(
                &get_random_bytes(64),
                Message::new(get_random_bytes(5242880)),
            )
        });
    });
}

fn bench_key_gen_enc_dec(c: &mut Criterion) {
    c.bench_function("e448 + SHA3-512 Asymmetric enc + dec 5mb", |b| {
        b.iter(|| {
            key_gen_enc_dec(
                &KeyPair::new(&get_random_bytes(32), "test key".to_string(), BIT_SECURITY).priv_key,
                Message::new(get_random_bytes(5242880)),
            )
        });
    });
}

criterion_group!(
    benches,
    bench_sym_enc,
    bench_key_gen_enc_dec,
    bench_sign_verify
);
criterion_main!(benches);