1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

//! Trait for a KeyProvider

use aes_gcm::aes::cipher::consts::U16;
use aes_gcm::Key;
use async_trait::async_trait;
use zeroize::Zeroize;

use crate::errors::{KeyDecryptionError, KeyGenerationError};

#[derive(Debug, Clone, Zeroize)]
pub struct DataKey {
    pub key: Key<U16>,
    // TODO: Maybe make a type for EncryptedKey
    pub encrypted_key: Vec<u8>,
    pub key_id: String,
}

#[async_trait(?Send)]
pub trait KeyProvider {
    async fn generate_data_key(&self) -> Result<DataKey, KeyGenerationError>;
    async fn decrypt_data_key(
        &self,
        encrypted_key: &Vec<u8>,
    ) -> Result<Key<U16>, KeyDecryptionError>;
}

#[async_trait(?Send)]
impl KeyProvider for Box<dyn KeyProvider> {
    async fn generate_data_key(&self) -> Result<DataKey, KeyGenerationError> {
        (**self).generate_data_key().await
    }

    async fn decrypt_data_key(
        &self,
        encrypted_key: &Vec<u8>,
    ) -> Result<Key<U16>, KeyDecryptionError> {
        (**self).decrypt_data_key(encrypted_key).await
    }
}