optimize: security app token response when user not found or disable. (…

…#620)
ikaros-dev · Jul 11, 2024 · fd0841a · fd0841a
1 parent 5e3f545
commit fd0841a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/server/src/main/java/run/ikaros/server/security/SecurityEndpoint.java b/server/src/main/java/run/ikaros/server/security/SecurityEndpoint.java
@@ -16,6 +16,8 @@
 import org.springframework.web.reactive.function.server.ServerResponse;
 import reactor.core.publisher.Mono;
 import run.ikaros.api.constant.OpenApiConst;
+import run.ikaros.api.infra.exception.security.UserAuthenticationException;
+import run.ikaros.api.infra.exception.user.UserNotFoundException;
 import run.ikaros.server.endpoint.CoreEndpoint;
 import run.ikaros.server.security.authentication.jwt.JwtApplyParam;
 import run.ikaros.server.security.authentication.jwt.JwtAuthenticationProvider;
@@ -71,6 +73,8 @@ private Mono<ServerResponse> applyJwtToken(ServerRequest request) {
             .flatMap(userDetailsService::findByUsername)
             .map(UserDetails::getUsername)
             .map(jwtAuthenticationProvider::generateToken)
-            .flatMap(token -> ServerResponse.ok().bodyValue(token));
+            .flatMap(token -> ServerResponse.ok().bodyValue(token))
+            .onErrorResume(UserNotFoundException.class,
+                e -> Mono.error(new UserAuthenticationException(e.getLocalizedMessage(), e)));
     }
 }