From 413708f2baa3987b4b8c7c727eeee8cfd8d59f73 Mon Sep 17 00:00:00 2001
From: Tolmachev Igor <me@igorek.dev>
Date: Sat, 27 Sep 2025 13:48:50 +0300
Subject: Fix invite tokens models

---
 migration/src/m0_init_tables.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'migration/src/m0_init_tables.rs')

diff --git a/migration/src/m0_init_tables.rs b/migration/src/m0_init_tables.rs
index ddbc0fb..5e9a621 100644
--- a/migration/src/m0_init_tables.rs
+++ b/migration/src/m0_init_tables.rs
@@ -29,6 +29,8 @@ enum InviteTokens {
     Id,
     Token,
     QueueId,
+    Name,
+    IsRevoked,
 }
 
 #[derive(DeriveIden)]
@@ -36,6 +38,7 @@ enum AccessToQueue {
     Table,
     UserId,
     QueueId,
+    InviteTokenId,
 }
 
 #[derive(DeriveIden)]
@@ -117,6 +120,8 @@ impl MigrationTrait for Migration {
                             .on_delete(ForeignKeyAction::Cascade)
                             .on_update(ForeignKeyAction::Cascade),
                     )
+                    .col(string(InviteTokens::Name))
+                    .col(boolean(InviteTokens::IsRevoked).default(false))
                     .to_owned(),
             )
             .await?;
@@ -142,6 +147,14 @@ impl MigrationTrait for Migration {
                             .on_delete(ForeignKeyAction::Cascade)
                             .on_update(ForeignKeyAction::Cascade),
                     )
+                    .col(big_integer(AccessToQueue::InviteTokenId))
+                    .foreign_key(
+                        ForeignKey::create()
+                            .from(AccessToQueue::Table, AccessToQueue::InviteTokenId)
+                            .to(InviteTokens::Table, InviteTokens::Id)
+                            .on_delete(ForeignKeyAction::Cascade)
+                            .on_update(ForeignKeyAction::Cascade),
+                    )
                     .primary_key(
                         Index::create()
                             .col(AccessToQueue::UserId)
@@ -208,6 +221,10 @@ impl MigrationTrait for Migration {
             .drop_table(Table::drop().table(Queues::Table).to_owned())
             .await?;
 
+        manager
+            .drop_table(Table::drop().table(InviteTokens::Table).to_owned())
+            .await?;
+
         manager
             .drop_table(Table::drop().table(AccessToQueue::Table).to_owned())
             .await?;
-- 
cgit v1.2.3