refactor: update user creation response type and user mapper function

Author: UmaxCode

src/main/java/org/umaxcode/controller/UserAuthController.java

@@ -23,9 +23,10 @@ public class UserAuthController {
     @ResponseStatus(HttpStatus.CREATED)
     public SuccessResponse signup(@RequestBody UserCreationDto request) {
 
-        String message = userAuthService.register(request);
+        UserDto registeredUser = userAuthService.register(request);
         return SuccessResponse.builder()
-                .message(message)
+                .message("User created successfully")
+                .data(registeredUser)
                 .build();
     }
 

src/main/java/org/umaxcode/domain/dto/response/UserDto.java

@@ -5,6 +5,8 @@
 @Builder
 public record UserDto(
         String userId,
-        String email
+        String username,
+        String email,
+        String role
 ) {
 }

src/main/java/org/umaxcode/mapper/UserMapper.java

@@ -19,8 +19,15 @@ public static List<UserDto> toUserDto(List<UserType> users) {
                         .email(u.attributes().stream()
                                 .filter(a -> "email".equalsIgnoreCase(a.name()))
                                 .map(AttributeType::value)
-                                .findFirst().orElse(null)
-                        )
+                                .findFirst().orElse(null))
+                        .username(u.attributes().stream()
+                                .filter(a -> "name".equalsIgnoreCase(a.name()))
+                                .map(AttributeType::value)
+                                .findFirst().orElse(null))
+                        .role(u.attributes().stream()
+                                .filter(a -> "custom:role".equalsIgnoreCase(a.name()))
+                                .map(AttributeType::value)
+                                .findFirst().orElse(null))
                         .build()
         ).toList();
     }

src/main/java/org/umaxcode/service/UserAuthService.java

@@ -7,7 +7,7 @@
 
 public interface UserAuthService {
 
-    String register(UserCreationDto request);
+    UserDto register(UserCreationDto request);
 
     List<UserDto> fetchAllUsers();
 }

src/main/java/org/umaxcode/service/impl/UserAuthServiceImpl.java

@@ -32,7 +32,7 @@ public class UserAuthServiceImpl implements UserAuthService {
     private String userPoolId;
 
     @Override
-    public String register(UserCreationDto request) {
+    public UserDto register(UserCreationDto request) {
 
         try {
             AdminCreateUserRequest adminRequest = AdminCreateUserRequest.builder()
@@ -51,7 +51,14 @@ public String register(UserCreationDto request) {
             AdminCreateUserResponse response = cognitoClient.adminCreateUser(adminRequest);
             startStateMachineForSNSSub(request.email());
             System.out.println("User" + response.user());
-            return "User created: " + response.user().username();
+
+            return UserDto.builder()
+                    .userId(request.username())
+                    .email(request.email())
+                    .username(request.username())
+                    .role(Role.USER.toString())
+                    .build();
+
         } catch (CognitoIdentityProviderException e) {
             throw new UserAuthException("Failed to create user: " + e.getMessage());
         }