1 files changed, 28 insertions, 1 deletions

diff --git a/src/models.rs b/src/models.rs
index 4821ec3..e815b73 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -1,6 +1,8 @@
-use entity::{queues, users};
+use chrono::NaiveDateTime;
+use entity::{invite_tokens, queues, users};
 use serde::Serialize;
 use utoipa::ToSchema;
+use uuid::Uuid;
 
 #[derive(Serialize, ToSchema)]
 #[schema(description = "Account information")]
@@ -42,3 +44,28 @@ impl From<queues::Model> for Queue {
         }
     }
 }
+
+#[derive(Serialize, ToSchema)]
+pub struct InviteToken {
+    #[schema(examples(1))]
+    pub id: i64,
+    pub token: Uuid,
+    #[schema(examples(1))]
+    pub queue_id: i64,
+    #[schema(examples("For classmates", "Для однокурсников"))]
+    pub name: String,
+    #[schema(examples(false))]
+    pub expiration_date: Option<NaiveDateTime>,
+}
+
+impl From<invite_tokens::Model> for InviteToken {
+    fn from(value: invite_tokens::Model) -> Self {
+        Self {
+            id: value.id,
+            token: value.token,
+            queue_id: value.queue_id,
+            name: value.name,
+            expiration_date: value.expiration_date,
+        }
+    }
+}