use crate::{errors::EncryptionError, traits::SignatureScheme};
use snarkvm_utilities::{
bytes::{FromBytes, ToBytes},
rand::UniformRand,
};
use rand::Rng;
use std::{fmt::Debug, hash::Hash};
pub trait EncryptionScheme: Sized + Clone + From<<Self as EncryptionScheme>::Parameters> + SignatureScheme {
type Parameters: Clone + Debug + Eq + ToBytes + FromBytes;
type PrivateKey: Clone + Debug + Default + Eq + Hash + ToBytes + FromBytes + UniformRand;
type PublicKey: Clone + Debug + Default + Eq + ToBytes + FromBytes;
type Text: Clone + Debug + Default + Eq + ToBytes + FromBytes;
type Randomness: Clone + Debug + Default + Eq + Hash + ToBytes + FromBytes + UniformRand;
type BlindingExponent: Clone + Debug + Default + Eq + Hash + ToBytes;
fn setup<R: Rng>(rng: &mut R) -> Self;
fn generate_private_key<R: Rng>(&self, rng: &mut R) -> <Self as EncryptionScheme>::PrivateKey;
fn generate_public_key(
&self,
private_key: &<Self as EncryptionScheme>::PrivateKey,
) -> Result<<Self as EncryptionScheme>::PublicKey, EncryptionError>;
fn generate_randomness<R: Rng>(
&self,
public_key: &<Self as EncryptionScheme>::PublicKey,
rng: &mut R,
) -> Result<Self::Randomness, EncryptionError>;
fn generate_blinding_exponents(
&self,
public_key: &<Self as EncryptionScheme>::PublicKey,
randomness: &Self::Randomness,
message_length: usize,
) -> Result<Vec<Self::BlindingExponent>, EncryptionError>;
fn encrypt(
&self,
public_key: &<Self as EncryptionScheme>::PublicKey,
randomness: &Self::Randomness,
message: &[Self::Text],
) -> Result<Vec<Self::Text>, EncryptionError>;
fn decrypt(
&self,
private_key: &<Self as EncryptionScheme>::PrivateKey,
ciphertext: &[Self::Text],
) -> Result<Vec<Self::Text>, EncryptionError>;
fn parameters(&self) -> &<Self as EncryptionScheme>::Parameters;
fn private_key_size_in_bits() -> usize;
}