Return user_id on JWT creation

2 files changed, 5 insertions, 2 deletions

diff --git a/src/graphql/mutation.rs b/src/graphql/mutation.rs
index 140d98a..682d9b6 100644
--- a/src/graphql/mutation.rs
+++ b/src/graphql/mutation.rs
@@ -39,7 +39,7 @@ impl Mutation {
             // Create a new claim using the found ID
             let claims = jwt::Claims::new(id[0]);
             let token = claims.get_token().unwrap();
-            Ok(jwt::AuthBody::new(token))
+            Ok(jwt::AuthBody::new(token, id[0]))
         } else {
             Err(Error::new("Invalid email or password"))
         }
diff --git a/src/graphql/types/jwt.rs b/src/graphql/types/jwt.rs
index 475b1bd..df815aa 100644
--- a/src/graphql/types/jwt.rs
+++ b/src/graphql/types/jwt.rs
@@ -77,13 +77,16 @@ pub struct AuthBody {
     access_token: String,
     /// "Bearer" string
     token_type: String,
+    /// User id
+    user_id: i32,
 }
 
 impl AuthBody {
-    pub fn new(access_token: String) -> Self {
+    pub fn new(access_token: String, user_id: i32) -> Self {
         Self {
             access_token,
             token_type: "Bearer".to_string(),
+            user_id,
         }
     }
 }