fix(api): rebased

ScuffleCloud · Oct 16, 2023 · 4ee5a2e · 4ee5a2e
1 parent d5f2d04
commit 4ee5a2e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 16 deletions.
diff --git a/platform/api/src/api/v1/gql/guards/mod.rs b/platform/api/src/api/v1/gql/guards/mod.rs
@@ -18,7 +18,7 @@ pub async fn auth_guard<T>(
 ) -> Result<T> {
     let request_context = ctx.get_req_context();
 
-    let auth = request_context.auth().await;
+    let auth = request_context.auth().await?;
 
     if let Some(auth) = auth {
         if Ulid::from(auth.session.user_id) == user_id

diff --git a/platform/api/src/api/v1/gql/models/user.rs b/platform/api/src/api/v1/gql/models/user.rs
@@ -57,21 +57,7 @@ impl User {
     }
 
     async fn totp_enabled(&self, ctx: &Context<'_>) -> Result<bool> {
-        let request_context = ctx.get_req_context();
-        let auth = request_context.auth().await?;
-
-        if let Some(auth) = auth {
-            if Ulid::from(auth.session.user_id) == *self.id
-                || auth.user_permissions.has_permission(RolePermission::Admin)
-            {
-                return Ok(self.totp_enabled_);
-            }
-        }
-
-        Err(GqlError::Unauthorized {
-            field: "totpEnabled",
-        }
-        .into())
+        auth_guard(ctx, "totpEnabled", self.totp_enabled_, self.id.into()).await
     }
 }