NCBACK-162

2026-05-29 18:17:04 +07:00
parent bc26b39d86
commit 5e494ffc4b
60 changed files with 670 additions and 837 deletions
@@ -1,5 +1,6 @@
 package ru.soune.nocopy.controller;

+import jakarta.servlet.http.HttpServletRequest;
 import lombok.RequiredArgsConstructor;
 import org.springframework.http.ResponseEntity;
 import org.springframework.security.crypto.password.PasswordEncoder;
@@ -14,23 +15,19 @@ import ru.soune.nocopy.dto.user.UserDTO;
 import ru.soune.nocopy.dto.user.UserRequest;
 import ru.soune.nocopy.entity.tarif.Tariff;
 import ru.soune.nocopy.entity.tarif.TariffInfo;
-import ru.soune.nocopy.entity.user.AuthToken;
 import ru.soune.nocopy.entity.user.User;
 import ru.soune.nocopy.exception.InvalidUserEmail;
-import ru.soune.nocopy.exception.NotFoundAuthToken;
 import ru.soune.nocopy.exception.NotValidFieldException;
+import ru.soune.nocopy.exception.UserNotFoundException;
 import ru.soune.nocopy.mapper.UserMapper;
-import ru.soune.nocopy.repository.AuthTokenRepository;
 import ru.soune.nocopy.repository.UserRepository;
 import ru.soune.nocopy.service.file.FileStatsService;
-import ru.soune.nocopy.service.register.AuthService;
 import ru.soune.nocopy.service.tariff.TariffService;
 import ru.soune.nocopy.service.user.UserService;
+import ru.soune.nocopy.util.JwtUtil;

 import java.util.Arrays;
-import java.util.List;
 import java.util.Map;
-import java.util.Optional;

@RestController
@RequestMapping("v1/api/user")
@@ -41,88 +38,79 @@ public class UserController {

    private final UserService userService;

-    private final AuthTokenRepository authTokenRepository;
-
    private final UserMapper userMapper;

    private final PasswordEncoder passwordEncoder;

-    private final AuthService authService;
-
    private final TariffService tariffService;

    private final FileStatsService fileStatsService;

+    private final JwtUtil jwtUtil;
+
+    private final HttpServletRequest httpServletRequest;
+

    @GetMapping
-    public ResponseEntity<UserDTO> getUser(@RequestParam("email") String email,
-                                           @RequestHeader("Authorization") String tokenHeader) {
-        String token = tokenHeader.replace("Bearer ", "");
+    public ResponseEntity<UserDTO> getUser(@RequestParam("email") String email) {
+        User user = userRepository.findByEmail(email);

-        Optional<AuthToken> tokenOptional = authTokenRepository.findByToken(token);
+        if (user != null) {
+            UserDTO userDTO = userMapper.toDTO(user);
+            userDTO.setEmail(email);
+            userDTO.setFullName(user.getFullName());
+            userDTO.setVerifiedStatus(user.getVerificationStatus());

-        if (tokenOptional.isPresent()) {
-            User user = userRepository.findByEmail(email);
-
-            if (user != null) {
-                UserDTO userDTO = userMapper.toDTO(user);
-                userDTO.setEmail(email);
-                userDTO.setFullName(user.getFullName());
-                userDTO.setVerifiedStatus(user.getVerificationStatus());
-
-                if (user.getCompany() != null) {
-                    userDTO.setCompany(user.getCompany().getCompanyName());
-                }
-
-                if (user.canManageCompanySettings() || user.canLogin() && user.getCompany() == null) {
-                    userDTO.setTariffs(tariffService.getTariffByAccountType(user));
-
-                    Long fileOnDisk = user.canManageCompanySettings() ?
-                            fileStatsService.calculateCompanyFileSize(user.getId()):
-                            fileStatsService.calculateUserFileSize(user.getId());
-
-                    Integer filesCount =  user.canManageCompanySettings() ?
-                            fileStatsService.filesCompanyCount(user.getId()):
-                            fileStatsService.filesUserCount(user.getId());
-
-                    TariffInfo personalTariffInfo;
-
-                    if (user.getCompany() == null) {
-                        personalTariffInfo = user.getPersonalTariffInfo();
-                    } else {
-                        personalTariffInfo = user.getCompany().getTariffInfo();
-                    }
-
-                    Tariff tariff = personalTariffInfo.getTariff();
-
-                    TariffInfoDTO infoDTO = TariffInfoDTO.builder().tokens(personalTariffInfo.getTokens())
-                            .id(personalTariffInfo.getId())
-                            .tariffName(tariff.getName())
-                            .status(personalTariffInfo.getStatus().name())
-                            .tariffId(tariff.getId())
-                            .currentFileCounts(filesCount)
-                            .maxFileCounts(tariff.getMaxFilesCount())
-                            .currentFileOnDisk(fileOnDisk)
-                            .maxFileOnDisk(tariff.getDiskSize())
-                            .startTariff(personalTariffInfo.getStartTariff())
-                            .endTariff(personalTariffInfo.getEndTariff())
-                            .tokens(personalTariffInfo.getTokens() + personalTariffInfo.getBoughtTokens())
-                            .build();
-
-                    userDTO.setTariffInfo(infoDTO);
-                }
-
-                userDTO.setPhone(user.getPhone());
-                userDTO.setGenderType(user.getGenderType());
-                userDTO.setBirthday(user.getBirthday());
-                userDTO.setCreatedAt(user.getCreatedAt());
-                userDTO.setActive(user.isActive());
-                userDTO.setPermission(user.getUserPermissions());
-
-                return ResponseEntity.ok(userDTO);
+            if (user.getCompany() != null) {
+                userDTO.setCompany(user.getCompany().getCompanyName());
            }
-        } else {
-            return ResponseEntity.notFound().build();
+
+            if (user.canManageCompanySettings() || user.canLogin() && user.getCompany() == null) {
+                userDTO.setTariffs(tariffService.getTariffByAccountType(user));
+
+                Long fileOnDisk = user.canManageCompanySettings() ?
+                        fileStatsService.calculateCompanyFileSize(user.getId()):
+                        fileStatsService.calculateUserFileSize(user.getId());
+
+                Integer filesCount =  user.canManageCompanySettings() ?
+                        fileStatsService.filesCompanyCount(user.getId()):
+                        fileStatsService.filesUserCount(user.getId());
+
+                TariffInfo personalTariffInfo;
+
+                if (user.getCompany() == null) {
+                    personalTariffInfo = user.getPersonalTariffInfo();
+                } else {
+                    personalTariffInfo = user.getCompany().getTariffInfo();
+                }
+
+                Tariff tariff = personalTariffInfo.getTariff();
+
+                TariffInfoDTO infoDTO = TariffInfoDTO.builder().tokens(personalTariffInfo.getTokens())
+                        .id(personalTariffInfo.getId())
+                        .tariffName(tariff.getName())
+                        .status(personalTariffInfo.getStatus().name())
+                        .tariffId(tariff.getId())
+                        .currentFileCounts(filesCount)
+                        .maxFileCounts(tariff.getMaxFilesCount())
+                        .currentFileOnDisk(fileOnDisk)
+                        .maxFileOnDisk(tariff.getDiskSize())
+                        .startTariff(personalTariffInfo.getStartTariff())
+                        .endTariff(personalTariffInfo.getEndTariff())
+                        .tokens(personalTariffInfo.getTokens() + personalTariffInfo.getBoughtTokens())
+                        .build();
+
+                userDTO.setTariffInfo(infoDTO);
+            }
+
+            userDTO.setPhone(user.getPhone());
+            userDTO.setGenderType(user.getGenderType());
+            userDTO.setBirthday(user.getBirthday());
+            userDTO.setCreatedAt(user.getCreatedAt());
+            userDTO.setActive(user.isActive());
+            userDTO.setPermission(user.getUserPermissions());
+
+            return ResponseEntity.ok(userDTO);
        }

        return ResponseEntity.notFound().build();
@@ -130,15 +118,13 @@ public class UserController {

    //TODO fix mapper,doesnot exist all fields
    @PostMapping("/change-password")
-    public ResponseEntity<UserDTO> updateUser(@RequestBody ChangePasswordRequest changePasswordRequest,
-                                              @RequestHeader("Authorization") String tokenHeader) {
-        String token = tokenHeader.replace("Bearer ", "");
+    public ResponseEntity<UserDTO> updateUser(@RequestBody ChangePasswordRequest changePasswordRequest) {
+        Long userId = (Long) httpServletRequest.getAttribute("userId");
+        User user = userService.getUserById(userId).orElse(null);

-        Optional<AuthToken> tokenOptional = authTokenRepository.findByToken(token);
-
-        AuthToken authToken = tokenOptional.orElseThrow(() -> new NotFoundAuthToken("Token: " + token + "not found"));
-
-        User user = authToken.getUser();
+        if (user == null) {
+            throw  new UserNotFoundException("User not found by id: " + userId);
+        }

        if (!changePasswordRequest.getEmail().equals(user.getEmail())) {
            throw new InvalidUserEmail("Email is not valid: " + changePasswordRequest.getEmail() + "not found");
@@ -150,21 +136,19 @@ public class UserController {
    }

    @PatchMapping("/user-update")
-    public ResponseEntity<UserDTO> updateUser(@RequestBody UserRequest userRequest,
-                                              @RequestHeader("Authorization") String tokenHeader) {
-        String token = tokenHeader.replace("Bearer ", "");
+    public ResponseEntity<UserDTO> updateUser(@RequestBody UserRequest userRequest) {
+        Long userId = (Long) httpServletRequest.getAttribute("userId");
+        User user = userService.getUserById(userId).orElse(null);

-        Optional<AuthToken> tokenOptional = authTokenRepository.findByToken(token);
-
-        AuthToken authToken = tokenOptional.orElseThrow(() -> new NotFoundAuthToken("Token: " + token + "not found"));
-
-        User user = authToken.getUser();
+        if (user == null) {
+            throw  new UserNotFoundException("User not found by id: " + userId);
+        }

        return ResponseEntity.ok(userService.updateUser(userRequest, user));
    }

    @PostMapping("/create-user")
-    public ResponseEntity<User> addUser(@RequestBody RegRequest registerRequest) {
+    public ResponseEntity<?> addUser(@RequestBody RegRequest registerRequest) {

        if (userRepository.existsByEmail(registerRequest.getEmail()) ||
                userRepository.existsByPhone(registerRequest.getPhone())) {
@@ -191,11 +175,11 @@ public class UserController {

        User savedUser = userRepository.save(user);

-        AuthToken authToken = authService.generateAuthToken(savedUser);
+        String authToken = jwtUtil.generateToken(savedUser.getId(), savedUser.getEmail());

-        authTokenRepository.save(authToken);
-
-        return ResponseEntity.ok().body(savedUser);
+        return ResponseEntity.ok().body(Map.of(
+                "user", savedUser,
+                "token", authToken));
    }

    @DeleteMapping("/delete-user/{userId}")