aswi2022code-of-duty-gitlab/backend/src/main/java/cz/zcu/kiv/backendapi/user/UserServiceImpl.java @ 87823897

package cz.zcu.kiv.backendapi.user;

import cz.zcu.kiv.backendapi.exception.ApiRequestException;
import cz.zcu.kiv.backendapi.password.PasswordDto;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UserDetailsService;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import cz.zcu.kiv.backendapi.permission.PermissionDto;

import java.util.List;
import java.util.stream.Collectors;

/**
 * User service implementation
 */
@Service
@Transactional
@RequiredArgsConstructor
@Slf4j
public class UserServiceImpl implements IUserService, UserDetailsService {
    /**
     * Message for exception when user is not found by username
     */
    private static final String USER_NOT_FOUND = "User with username %s not found";

    /**
     * User repository
     */
    private final UserRepository userRepository;

    /**
     * Password encoder
     */
    private final BCryptPasswordEncoder bCryptPasswordEncoder;


    @Override
    public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
        return userRepository.findByEmail(username)
                .orElseThrow(() -> {
                    log.error(String.format(USER_NOT_FOUND, username));
                    throw new UsernameNotFoundException(String.format(USER_NOT_FOUND, username));
                });
    }

    @Override
    public UserEntity getUserByName(String username) {
        return userRepository.findByEmail(username).orElseThrow(() -> {
            log.error(String.format(USER_NOT_FOUND, username));
            throw new UsernameNotFoundException(String.format(USER_NOT_FOUND, username));
        });
    }

    @Override
    public void registerNewUser(UserDto userDto) {
        if (userRepository.findByEmail(userDto.getEmail()).isPresent()) {
            log.error("Trying to register new user with username that is already taken: " + userDto.getEmail());
            throw new ApiRequestException(String.format("User with username %s already exists", userDto.getEmail()), HttpStatus.CONFLICT);
        }
        UserEntity userEntity = new UserEntity();
        convertDtoToEntity(userDto, userEntity);
        userRepository.save(userEntity);
    }

    @Override
    public void updatePermissions(String username, PermissionDto permissionDto) {
        UserEntity userEntity = userRepository.findByEmail(username).orElseThrow(() ->{
            log.error(String.format(USER_NOT_FOUND, username));
            throw new UsernameNotFoundException(String.format(USER_NOT_FOUND, username));
        });
        userEntity.setPermissions(getPermissionsFromDto(permissionDto));
        userEntity.setEmail(username);
        userRepository.save(userEntity);
    }

    @Override
    public void resetPassword(String username, String newPassword) {
        UserEntity userEntity = userRepository.findByEmail(username).orElseThrow(() ->{
            log.error(String.format(USER_NOT_FOUND, username));
            throw new UsernameNotFoundException(String.format(USER_NOT_FOUND, username));
        });
        userEntity.setPassword(bCryptPasswordEncoder.encode(newPassword));
        userRepository.save(userEntity);
    }

    //TODO maybe check if user is not deleting himself - or it might be ok
    @Override
    public void deleteUser(String username) {
        UserEntity userEntity = userRepository.findByEmail(username).orElseThrow(() ->{
            log.error(String.format(USER_NOT_FOUND, username));
            throw new UsernameNotFoundException(String.format(USER_NOT_FOUND, username));
        });
        userRepository.delete(userEntity);
    }

    @Override
    public List<UserDto> getAllUsers() {
        return userRepository.findAll().stream().map(this::convertEntityToDto).collect(Collectors.toList());
    }

    @Override
    public void changePassword(String oldPassword, PasswordDto passwordDto) {
        UserEntity loggedUser = getUserByName((String) SecurityContextHolder.getContext().getAuthentication().getPrincipal());
        if (!bCryptPasswordEncoder.matches(oldPassword, loggedUser.getPassword())) {
            throw new ApiRequestException("Old password does not match", HttpStatus.BAD_REQUEST);
        }
        loggedUser.setPassword(bCryptPasswordEncoder.encode(passwordDto.getPassword()));
        userRepository.save(loggedUser);
    }



    /**
     * Converts user DTO to user entity
     *
     * @param userDto    user DTO
     * @param userEntity user entity
     */
    private void convertDtoToEntity(UserDto userDto, UserEntity userEntity) {
        userEntity.setName(userDto.getName());
        userEntity.setEmail(userDto.getEmail());
        userEntity.setPermissions(getPermissionsFromDto(userDto.getPermissions()));
        userEntity.setPassword(bCryptPasswordEncoder.encode(userDto.getPasswords().getPassword()));
    }

    /**
     * Converts user entity to user DTO
     *
     * @param userEntity user entity
     * @return user DTO from user entity
     */
    private UserDto convertEntityToDto(UserEntity userEntity) {
        UserDto userDto = new UserDto();
        userDto.setName(userEntity.getName());
        userDto.setEmail(userEntity.getEmail());
        setPermissionsToDto(userDto, userEntity);
        return userDto;
    }

    /**
     * Sets permission to user DTO based on user entity
     *
     * @param userDto    user DTO
     * @param userEntity user entity
     */
    private void setPermissionsToDto(UserDto userDto, UserEntity userEntity) {
        byte userPermissions = userEntity.getPermissions();
        PermissionDto permissionDto = new PermissionDto();
        if ((userPermissions & Permission.READ.getBit()) == Permission.READ.getBit()) {
            permissionDto.setCanRead(true);
        }
        if ((userPermissions & Permission.WRITE.getBit()) == Permission.WRITE.getBit()) {
            permissionDto.setCanWrite(true);
        }
        if ((userPermissions & Permission.DELETE.getBit()) == Permission.DELETE.getBit()) {
            permissionDto.setCanDelete(true);
        }
        userDto.setPermissions(permissionDto);
    }

    /**
     * Returns permissions as byte from permission DTO
     *
     * @param permissionDto permission DTO
     * @return permissions as byte
     */
    private byte getPermissionsFromDto(PermissionDto permissionDto) {
        byte permissions = (byte) 0;
        if (permissionDto.isCanRead()) {
            permissions |= Permission.READ.getBit();
        }
        if (permissionDto.isCanWrite()) {
            permissions |= Permission.WRITE.getBit();
        }
        if (permissionDto.isCanDelete()) {
            permissions |= Permission.DELETE.getBit();
        }
        return permissions;
    }
}