#![allow(non_snake_case)]
#![cfg(feature = "alloc")]

use serde::Deserialize;
use signature::Keypair;
use slh_dsa::*;

const KEYGEN_KAT_JSON: &str = include_str!("acvp/SLH-DSA-keyGen-FIPS205/internalProjection.json");

#[derive(Deserialize, Debug)]
struct TestCase {
    #[serde(with = "hex::serde")]
    skSeed: Vec<u8>,
    #[serde(with = "hex::serde")]
    skPrf: Vec<u8>,
    #[serde(with = "hex::serde")]
    pkSeed: Vec<u8>,
    #[serde(with = "hex::serde")]
    sk: Vec<u8>,
    #[serde(with = "hex::serde")]
    pk: Vec<u8>,
}

#[derive(Deserialize, Debug)]
struct TestGroup {
    parameterSet: String,
    tests: Vec<TestCase>,
}

#[derive(Deserialize, Debug)]
struct TestFile {
    testGroups: Vec<TestGroup>,
}

macro_rules! parameter_case {
    ($param:ident, $test_case:expr) => {{
        let sk = SigningKey::<$param>::slh_keygen_internal(
            &$test_case.skSeed,
            &$test_case.skPrf,
            &$test_case.pkSeed,
        );
        let vk = sk.verifying_key();
        assert_eq!(sk.to_vec(), $test_case.sk);
        assert_eq!(vk.to_vec(), $test_case.pk);
    }};
}

#[test]
fn test_keygen_cvp() {
    let mut i = 0;
    let test_file: TestFile = serde_json::from_str(KEYGEN_KAT_JSON).unwrap();
    for test_group in test_file.testGroups {
        let p = test_group.parameterSet;
        for test_case in test_group.tests {
            match p.as_str() {
                Shake128f::NAME => parameter_case!(Shake128f, test_case),
                Shake128s::NAME => parameter_case!(Shake128s, test_case),
                Shake192f::NAME => parameter_case!(Shake192f, test_case),
                Shake192s::NAME => parameter_case!(Shake192s, test_case),
                Shake256f::NAME => parameter_case!(Shake256f, test_case),
                Shake256s::NAME => parameter_case!(Shake256s, test_case),
                Sha2_128f::NAME => parameter_case!(Sha2_128f, test_case),
                Sha2_128s::NAME => parameter_case!(Sha2_128s, test_case),
                Sha2_192f::NAME => parameter_case!(Sha2_192f, test_case),
                Sha2_192s::NAME => parameter_case!(Sha2_192s, test_case),
                Sha2_256f::NAME => parameter_case!(Sha2_256f, test_case),
                Sha2_256s::NAME => parameter_case!(Sha2_256s, test_case),
                _ => panic!("Unknown parameter set: {}", p),
            }
            i += 1;
        }
    }
    print!("Number of test cases: {}", i);
}