hack to allow impersonator to impersonate target

1 files changed, 45 insertions, 8 deletions

diff --git a/crates/secd/src/client/store/mod.rs b/crates/secd/src/client/store/mod.rs
index 6c42dba..f08aa41 100644
--- a/crates/secd/src/client/store/mod.rs
+++ b/crates/secd/src/client/store/mod.rs
@@ -1,4 +1,4 @@
-pub(crate) mod sql_db;
+pub mod sql_db;
 
 use async_trait::async_trait;
 use sqlx::{Postgres, Sqlite};
@@ -7,7 +7,7 @@ use uuid::Uuid;
 
 use crate::{
     Address, AddressType, AddressValidation, Credential, CredentialId, CredentialType, Identity,
-    IdentityId,
+    IdentityId, Impersonator,
 };
 
 use self::sql_db::SqlClient;
@@ -19,6 +19,7 @@ pub enum StoreError {
     ParseError(#[from] strum::ParseError),
     StoreValueCannotBeParsedInvariant,
     IdempotentCheckAlreadyExists,
+    ExpectedEntity,
 }
 
 #[async_trait]
@@ -32,7 +33,7 @@ pub enum StoreType {
 }
 
 #[async_trait]
-pub(crate) trait Storable<'a> {
+pub trait Storable<'a> {
     type Item;
     type Lens;
 
@@ -43,33 +44,39 @@ pub(crate) trait Storable<'a> {
     ) -> Result<Vec<Self::Item>, StoreError>;
 }
 
-pub(crate) trait Lens {}
+pub trait Lens {}
 
-pub(crate) struct AddressLens<'a> {
+pub struct AddressLens<'a> {
     pub id: Option<&'a Uuid>,
     pub t: Option<&'a AddressType>,
 }
 impl<'a> Lens for AddressLens<'a> {}
 
-pub(crate) struct AddressValidationLens<'a> {
+pub struct AddressValidationLens<'a> {
     pub id: Option<&'a Uuid>,
 }
 impl<'a> Lens for AddressValidationLens<'a> {}
 
-pub(crate) struct IdentityLens<'a> {
+pub struct IdentityLens<'a> {
     pub id: Option<&'a Uuid>,
     pub address_type: Option<&'a AddressType>,
     pub validated_address: Option<bool>,
 }
 impl<'a> Lens for IdentityLens<'a> {}
 
-pub(crate) struct CredentialLens<'a> {
+pub struct CredentialLens<'a> {
     pub id: Option<CredentialId>,
     pub identity_id: Option<IdentityId>,
     pub t: Option<&'a CredentialType>,
 }
 impl<'a> Lens for CredentialLens<'a> {}
 
+pub struct ImpersonatorLens<'a> {
+    pub impersonator_id: Option<&'a IdentityId>,
+    pub target_id: Option<&'a IdentityId>,
+}
+impl<'a> Lens for ImpersonatorLens<'a> {}
+
 #[async_trait]
 impl<'a> Storable<'a> for Address {
     type Item = Address;
@@ -179,3 +186,33 @@ impl<'a> Storable<'a> for Credential {
         })
     }
 }
+
+#[async_trait]
+impl<'a> Storable<'a> for Impersonator {
+    type Item = Impersonator;
+    type Lens = ImpersonatorLens<'a>;
+
+    async fn write(&self, store: Arc<dyn Store>) -> Result<(), StoreError> {
+        match store.get_type() {
+            StoreType::Postgres { c } => c.write_impersonator(self).await?,
+            StoreType::Sqlite { c } => c.write_impersonator(self).await?,
+        }
+        Ok(())
+    }
+
+    async fn find(
+        store: Arc<dyn Store>,
+        lens: &'a Self::Lens,
+    ) -> Result<Vec<Self::Item>, StoreError> {
+        Ok(match store.get_type() {
+            StoreType::Postgres { c } => {
+                c.find_impersonator(lens.impersonator_id, lens.target_id)
+                    .await?
+            }
+            StoreType::Sqlite { c } => {
+                c.find_impersonator(lens.impersonator_id, lens.target_id)
+                    .await?
+            }
+        })
+    }
+}