From 0b52ca8eefbe5fe48b3bcddbc0122f8c0f127740 Mon Sep 17 00:00:00 2001
From: Denys Konovalov <denys.konovalov@protonmail.com>
Date: Mon, 9 Aug 2021 22:09:20 +0200
Subject: [PATCH] fix userinfo endpoint

---
 src/main.rs | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 100e0ac..9b7bed2 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -22,7 +22,6 @@ use keycloak::KeycloakError;
 use rocket::fs::{relative, FileServer};
 use rocket::http::Status;
 use rocket::response::status;
-use rocket::http;
 use rocket::request::{FromRequest, Outcome, Request};
 use std::error::Error;
 use std::time::{Duration, SystemTime, UNIX_EPOCH};
@@ -245,7 +244,7 @@ async fn get_userinfo(credentials: Json<Credentials>) -> Result<Json<KeycloakUse
         credentials.otp.clone(),
     )
     .await;
-    let mut token = match keycloak_resp {
+    let token = match keycloak_resp {
         Ok(token) => Token {
             outcome: TokenOutcome {
                 status: TokenStatus::Success,
@@ -280,8 +279,12 @@ async fn get_userinfo(credentials: Json<Credentials>) -> Result<Json<KeycloakUse
             }
         }
     };
+    let outcome = match token.outcome.status {
+        TokenStatus::Success =>  Ok(Json(get_keycloak_userinfo(token.token.clone()).await.unwrap())),
+        _ => Err(status::Unauthorized::<()>(None))
+    };
 
-    Ok(Json(get_keycloak_userinfo(token.token.clone()).await.unwrap()))
+    outcome
 }