use safe_pqc_kyber::*;

fn main () -> Result<(), KyberError> {
  let mut rng = rand::thread_rng();

  // Alice generates a keypair
  let alice_keys = keypair(&mut rng);

  // Bob encapsulates a shared secret
  let (ciphertext, shared_secret_bob) = encapsulate(&alice_keys.public, &mut rng)?;
  
  // Alice decapsulates the shared secret
  let shared_secret_alice = decapsulate(&ciphertext, &alice_keys.secret)?;
  
  // Both can now communicate symetrically
  assert_eq!(shared_secret_alice, shared_secret_bob);
  Ok(())
}