diff --git a/packages/storage/shield-sea-orm/src/user.rs b/packages/storage/shield-sea-orm/src/user.rs index 13eda7a..23a51ba 100644 --- a/packages/storage/shield-sea-orm/src/user.rs +++ b/packages/storage/shield-sea-orm/src/user.rs @@ -1,3 +1,5 @@ +use std::ops::Deref; + use async_trait::async_trait; use sea_orm::{prelude::Uuid, DatabaseConnection, ModelTrait}; use shield::{EmailAddress, StorageError}; @@ -27,6 +29,19 @@ impl User { entity, } } + + #[cfg(feature = "entity")] + pub fn entity(&self) -> &entity::Model { + &self.entity + } +} + +impl Deref for User { + type Target = user::Model; + + fn deref(&self) -> &Self::Target { + &self.user + } } #[async_trait]