Add invite tokens models

4 files changed, 42 insertions, 0 deletions

diff --git a/entity/src/invite_tokens.rs b/entity/src/invite_tokens.rs
new file mode 100644
index 0000000..f239567
--- /dev/null
+++ b/entity/src/invite_tokens.rs
@@ -0,0 +1,32 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.14
+
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
+#[sea_orm(table_name = "invite_tokens")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub id: i64,
+    pub token: Uuid,
+    pub queue_id: i64,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::queues::Entity",
+        from = "Column::QueueId",
+        to = "super::queues::Column::Id",
+        on_update = "Cascade",
+        on_delete = "Cascade"
+    )]
+    Queues,
+}
+
+impl Related<super::queues::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Queues.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}
diff --git a/entity/src/lib.rs b/entity/src/lib.rs
index 1ca05d6..389846a 100644
--- a/entity/src/lib.rs
+++ b/entity/src/lib.rs
@@ -3,6 +3,7 @@
 pub mod prelude;
 
 pub mod access_to_queue;
+pub mod invite_tokens;
 pub mod queue_elements;
 pub mod queues;
 pub mod sea_orm_active_enums;
diff --git a/entity/src/prelude.rs b/entity/src/prelude.rs
index 60f72f9..c4f6a2c 100644
--- a/entity/src/prelude.rs
+++ b/entity/src/prelude.rs
@@ -1,6 +1,7 @@
 //! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.14
 
 pub use super::access_to_queue::Entity as AccessToQueue;
+pub use super::invite_tokens::Entity as InviteTokens;
 pub use super::queue_elements::Entity as QueueElements;
 pub use super::queues::Entity as Queues;
 pub use super::users::Entity as Users;
diff --git a/entity/src/queues.rs b/entity/src/queues.rs
index 34b358a..c6116df 100644
--- a/entity/src/queues.rs
+++ b/entity/src/queues.rs
@@ -15,6 +15,8 @@ pub struct Model {
 pub enum Relation {
     #[sea_orm(has_many = "super::access_to_queue::Entity")]
     AccessToQueue,
+    #[sea_orm(has_many = "super::invite_tokens::Entity")]
+    InviteTokens,
     #[sea_orm(has_many = "super::queue_elements::Entity")]
     QueueElements,
     #[sea_orm(
@@ -33,6 +35,12 @@ impl Related<super::access_to_queue::Entity> for Entity {
     }
 }
 
+impl Related<super::invite_tokens::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::InviteTokens.def()
+    }
+}
+
 impl Related<super::queue_elements::Entity> for Entity {
     fn to() -> RelationDef {
         Relation::QueueElements.def()