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

---
 entity/src/access_to_queue.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'entity/src/access_to_queue.rs')

diff --git a/entity/src/access_to_queue.rs b/entity/src/access_to_queue.rs
index 9de03d5..5244ed0 100644
--- a/entity/src/access_to_queue.rs
+++ b/entity/src/access_to_queue.rs
@@ -9,10 +9,19 @@ pub struct Model {
     pub user_id: i64,
     #[sea_orm(primary_key, auto_increment = false)]
     pub queue_id: i64,
+    pub invite_token_id: i64,
 }
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
 pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::invite_tokens::Entity",
+        from = "Column::InviteTokenId",
+        to = "super::invite_tokens::Column::Id",
+        on_update = "Cascade",
+        on_delete = "Cascade"
+    )]
+    InviteTokens,
     #[sea_orm(
         belongs_to = "super::queues::Entity",
         from = "Column::QueueId",
@@ -31,6 +40,12 @@ pub enum Relation {
     Users,
 }
 
+impl Related<super::invite_tokens::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::InviteTokens.def()
+    }
+}
+
 impl Related<super::queues::Entity> for Entity {
     fn to() -> RelationDef {
         Relation::Queues.def()
-- 
cgit v1.2.3