Fix invite tokens models

1 files changed, 10 insertions, 0 deletions

diff --git a/entity/src/invite_tokens.rs b/entity/src/invite_tokens.rs
index f239567..c0b59f7 100644
--- a/entity/src/invite_tokens.rs
+++ b/entity/src/invite_tokens.rs
@@ -9,10 +9,14 @@ pub struct Model {
     pub id: i64,
     pub token: Uuid,
     pub queue_id: i64,
+    pub name: String,
+    pub is_revoked: bool,
 }
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
 pub enum Relation {
+    #[sea_orm(has_many = "super::access_to_queue::Entity")]
+    AccessToQueue,
     #[sea_orm(
         belongs_to = "super::queues::Entity",
         from = "Column::QueueId",
@@ -23,6 +27,12 @@ pub enum Relation {
     Queues,
 }
 
+impl Related<super::access_to_queue::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::AccessToQueue.def()
+    }
+}
+
 impl Related<super::queues::Entity> for Entity {
     fn to() -> RelationDef {
         Relation::Queues.def()