Add invite tokens models

6 files changed, 70 insertions, 0 deletions

diff --git a/Cargo.toml b/Cargo.toml
index 4c8f11f..3a07517 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -27,3 +27,4 @@ tracing-subscriber = "0.3.19"
 utoipa = { version = "5.4.0", features = ["axum_extras", "chrono", "preserve_order", "preserve_path_order"] }
 utoipa-axum = "0.2.0"
 utoipa-scalar = { version = "0.3.0", features = ["axum"] }
+uuid = { version = "1.18.1", features = ["v4", "serde"] }
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()
diff --git a/migration/src/m0_init_tables.rs b/migration/src/m0_init_tables.rs
index 5bde7cc..ddbc0fb 100644
--- a/migration/src/m0_init_tables.rs
+++ b/migration/src/m0_init_tables.rs
@@ -24,6 +24,14 @@ enum Queues {
 }
 
 #[derive(DeriveIden)]
+enum InviteTokens {
+    Table,
+    Id,
+    Token,
+    QueueId,
+}
+
+#[derive(DeriveIden)]
 enum AccessToQueue {
     Table,
     UserId,
@@ -97,6 +105,25 @@ impl MigrationTrait for Migration {
         manager
             .create_table(
                 Table::create()
+                    .table(InviteTokens::Table)
+                    .if_not_exists()
+                    .col(pk_auto(InviteTokens::Id).big_integer())
+                    .col(uuid(InviteTokens::Token))
+                    .col(big_integer(InviteTokens::QueueId))
+                    .foreign_key(
+                        ForeignKey::create()
+                            .from(InviteTokens::Table, InviteTokens::QueueId)
+                            .to(Queues::Table, Queues::Id)
+                            .on_delete(ForeignKeyAction::Cascade)
+                            .on_update(ForeignKeyAction::Cascade),
+                    )
+                    .to_owned(),
+            )
+            .await?;
+
+        manager
+            .create_table(
+                Table::create()
                     .table(AccessToQueue::Table)
                     .if_not_exists()
                     .col(big_integer(AccessToQueue::UserId))