Shield SeaORM: Add support for entity model

packages/storage/shield-seaorm/development.md

@@ -3,5 +3,5 @@
 ```
 sea-orm-cli migrate fresh -u "sqlite:///tmp/shield-seaorm.sqlite?mode=rwc" -d ./examples/seaorm
 
-sea-orm-cli generate entity -u "sqlite:///tmp/shield-seaorm.sqlite?mode=rwc" -o packages/storage/shield-seaorm/src/entities --with-serde both
+sea-orm-cli generate entity -u "sqlite:///tmp/shield-seaorm.sqlite?mode=rwc" -o packages/storage/shield-seaorm/src/entities_template --with-serde both
 ```

packages/storage/shield-seaorm/src/entities.rs

@@ -0,0 +1,6 @@
+pub mod prelude;
+
+pub mod email_address;
+#[cfg(feature = "entity")]
+pub mod entity;
+pub mod user;

packages/storage/shield-seaorm/src/entities/email_address.rs

@@ -17,11 +17,24 @@ pub struct Model {
     pub verification_token: Option<String>,
     pub verification_token_expires_at: Option<DateTimeUtc>,
     pub verified_at: Option<DateTimeUtc>,
+    #[cfg(feature = "entity")]
+    pub entity_id: Uuid,
+    #[cfg(not(feature = "entity"))]
     pub user_id: Uuid,
 }
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
 pub enum Relation {
+    #[cfg(feature = "entity")]
+    #[sea_orm(
+        belongs_to = "super::entity::Entity",
+        from = "Column::EntityId",
+        to = "super::entity::Column::Id",
+        on_update = "Cascade",
+        on_delete = "Cascade"
+    )]
+    Entity,
+    #[cfg(not(feature = "entity"))]
     #[sea_orm(
         belongs_to = "super::user::Entity",
         from = "Column::UserId",
@@ -32,6 +45,14 @@ pub enum Relation {
     User,
 }
 
+#[cfg(feature = "entity")]
+impl Related<super::entity::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Entity.def()
+    }
+}
+
+#[cfg(not(feature = "entity"))]
 impl Related<super::user::Entity> for Entity {
     fn to() -> RelationDef {
         Relation::User.def()

packages/storage/shield-seaorm/src/entities/entity.rs

@@ -0,0 +1,37 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.2
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "entity")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub id: Uuid,
+    pub created_at: DateTimeUtc,
+    pub updated_at: DateTimeUtc,
+    #[sea_orm(column_type = "Text")]
+    pub name: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(has_many = "super::email_address::Entity")]
+    EmailAddress,
+    #[sea_orm(has_many = "super::user::Entity")]
+    User,
+}
+
+impl Related<super::email_address::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::EmailAddress.def()
+    }
+}
+
+impl Related<super::user::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::User.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

packages/storage/shield-seaorm/src/entities/prelude.rs

@@ -1,4 +1,6 @@
 //! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.2
 
 pub use super::email_address::Entity as EmailAddress;
+#[cfg(feature = "entity")]
+pub use super::entity::Entity;
 pub use super::user::Entity as User;

packages/storage/shield-seaorm/src/entities/user.rs

@@ -10,16 +10,37 @@ pub struct Model {
     pub id: Uuid,
     pub created_at: DateTimeUtc,
     pub updated_at: DateTimeUtc,
+    #[cfg(not(feature = "entity"))]
     #[sea_orm(column_type = "Text")]
     pub name: String,
+    #[cfg(feature = "entity")]
+    pub entity_id: Uuid,
 }
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
 pub enum Relation {
+    #[cfg(feature = "entity")]
+    #[sea_orm(
+        belongs_to = "super::entity::Entity",
+        from = "Column::EntityId",
+        to = "super::entity::Column::Id",
+        on_update = "Cascade",
+        on_delete = "Cascade"
+    )]
+    Entity,
+    #[cfg(not(feature = "entity"))]
     #[sea_orm(has_many = "super::email_address::Entity")]
     EmailAddress,
 }
 
+#[cfg(feature = "entity")]
+impl Related<super::entity::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Entity.def()
+    }
+}
+
+#[cfg(not(feature = "entity"))]
 impl Related<super::email_address::Entity> for Entity {
     fn to() -> RelationDef {
         Relation::EmailAddress.def()

packages/storage/shield-seaorm/src/migrations/m20241210_203135_create_user.rs

@@ -13,7 +13,7 @@ impl MigrationTrait for Migration {
         {
             manager
                 .create_table(
-                    BaseTable::create(Entity::Table)
+                    BaseTable::create(Entity::Table, manager)
                         .col(ColumnDef::new(Entity::Name).text().not_null())
                         .to_owned(),
                 )