use pq_msg::{
exchange::pair::KEMPair, messaging::MessageSession, messaging::create_nonce,
messaging::gen_session_id, signatures::keypair::SignerPair,
};
fn main() {
let alice_kem = KEMPair::create();
let alice_signer = SignerPair::create();
let bob_kem = KEMPair::create();
let bob_signer = SignerPair::create();
let base_nonce = create_nonce(&gen_session_id(), 0);
let (mut alice_session, ciphertext) = MessageSession::new_initiator(
alice_kem,
alice_signer.clone(),
base_nonce,
&bob_kem.to_bytes().unwrap().0, &bob_signer.to_bytes().unwrap().0, )
.unwrap();
let mut bob_session = MessageSession::new_responder(
bob_kem,
bob_signer.clone(),
base_nonce,
&ciphertext,
&alice_signer.to_bytes().unwrap().0, )
.unwrap();
let message = b"Hello, Bob! This is a secret message.";
let encrypted_message = alice_session.craft_message(message).unwrap();
let raw_message = bob_session.validate_message(&encrypted_message).unwrap();
let message_str = String::from_utf8_lossy(message);
let raw_message_str = String::from_utf8_lossy(&raw_message);
println!("[1] Alice's message: {}", message_str);
println!("[2] Bob's decrypted message: {}", raw_message_str);
let reply = b"Hello, Alice! I received your message safely.";
let encrypted_reply = bob_session.craft_message(reply).unwrap();
let raw_reply = alice_session.validate_message(&encrypted_reply).unwrap();
let reply_str = String::from_utf8_lossy(reply);
let raw_reply_str = String::from_utf8_lossy(&raw_reply);
println!("[3] Bob's reply: {}", reply_str);
println!("[4] Alice's decrypted reply: {}", raw_reply_str);
}