Add helpers for sealer records

Author: Bennett Hardwick

src/crypto/sealer.rs

@@ -213,7 +213,7 @@ impl Sealer {
 
             table_entries.push((
                 PrimaryKeyParts { pk, sk },
-                record.unsealed.unprotected(),
+                record.unsealed.unprotected().clone(),
                 terms,
             ));
         }

src/crypto/unsealed.rs

@@ -37,6 +37,14 @@ impl Unsealed {
         }
     }
 
+    pub fn protected(&self) -> &HashMap<String, (Plaintext, String)> {
+        &self.protected
+    }
+
+    pub fn unprotected(&self) -> &HashMap<String, TableAttribute> {
+        &self.unprotected
+    }
+
     pub fn get_protected(&self, name: &str) -> Result<&Plaintext, SealError> {
         let (plaintext, _) = self
             .protected
@@ -63,10 +71,6 @@ impl Unsealed {
         self.unprotected.insert(name.into(), attribute);
     }
 
-    pub(crate) fn unprotected(&self) -> HashMap<String, TableAttribute> {
-        self.unprotected.clone()
-    }
-
     /// Remove and return a protected value along with its descriptor.
     pub(crate) fn remove_protected_with_descriptor(
         &mut self,

src/encrypted_table/mod.rs

@@ -19,7 +19,7 @@ use cipherstash_client::{
         service_credentials::{ServiceCredentials, ServiceToken},
         Credentials,
     },
-    encryption::Encryption,
+    encryption::{Encryption, Plaintext},
     zero_kms::ZeroKMS,
 };
 use log::info;
@@ -137,6 +137,24 @@ impl PreparedRecord {
         }
     }
 
+    /// Get all the [`Plaintext`] protected attributes from the [`PreparedRecord`]
+    pub fn protected(&self) -> impl Iterator<Item = (&str, &Plaintext)> {
+        self.sealer
+            .unsealed
+            .protected()
+            .iter()
+            .map(|(key, (plaintext, _descriptor))| (key.as_str(), plaintext))
+    }
+
+    /// Get all the unprotected attributes from the [`PreparedRecord`]
+    pub fn unprotected(&self) -> impl Iterator<Item = (&str, &TableAttribute)> {
+        self.sealer
+            .unsealed
+            .unprotected()
+            .iter()
+            .map(|(key, attr)| (key.as_str(), attr))
+    }
+
     pub fn prepare_record<R>(record: R) -> Result<Self, SealError>
     where
         R: Searchable + Identifiable,