aswi2020vldc-gitlab/src/main/java/vldc/aswi/service/UserManagerImpl.java @ 621e3fca

package vldc.aswi.service;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.context.event.EventListener;
import org.springframework.core.annotation.Order;
import org.springframework.dao.InvalidDataAccessResourceUsageException;
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.password.PasswordEncoder;
import org.springframework.stereotype.Service;
import vldc.aswi.dao.RoleRepository;
import vldc.aswi.dao.UserRepository;
import vldc.aswi.domain.Role;
import vldc.aswi.domain.User;
import vldc.aswi.model.login.WebCredentials;

import javax.transaction.Transactional;
import java.util.LinkedList;
import java.util.List;

/**
 * Manager for User.
 */
@Service
@Slf4j
public class UserManagerImpl implements UserManager, UserDetailsService {

    @Autowired
    private PasswordEncoder encoder;

    /** Autowired user repository. */
    @Autowired
    private UserRepository userRepository;

    /** Autowired role repository. */
    @Autowired
    private RoleRepository roleRepository;

    /**
     * Initialization setup for user manager.
     * Check if table "Uzivatel" exists.
     */
    @EventListener(classes = {
            ContextRefreshedEvent.class,
    })
    @Order(2)
    @Transactional
    public void setup() {
        try {
            if (this.userRepository.count() == 0) {
                // Just checking if table exists.
            }
        }
        catch (InvalidDataAccessResourceUsageException e) {
            log.error("Table \"Uzivatel\" did not exists in database!");
            System.exit(1);
        }
    }

    /**
     * Converts role name to Spring role name
     * @param role role which name should be converted
     * @return string with Spring role
     */
    private String toSpringRole(Role role) {
        return role.getName();
    }

    /**
     * Load user details by his username.
     * @param username Username of user.
     * @return UserDetails of user.
     * @throws UsernameNotFoundException Exception.
     */
    @Override
    @Transactional
    public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
        User user = this.userRepository.findByUsername(username);
        if (user == null) {
            throw new UsernameNotFoundException("Invalid username!");
        }

        WebCredentials credentials = new WebCredentials(user.getUsername(), user.getPassword());

        credentials.addRole(toSpringRole(user.getRole()));

        return credentials;
    }

    /**
     * Get all users from database.
     * @return List of users.
     */
    @Override
    public List<User> getUsers() {
        List<User> retVal = new LinkedList<>();
        this.userRepository.findAll().forEach(retVal::add);
        return retVal;
    }

    /**
     * Add new user into database.
     * @param username - login username for user.
     * @param password - password for user.
     * @return ID of newly created user.
     */
    @Override
    public Long addUser(String username, String password, Long roleId) {
        String hashed = this.encoder.encode(password);
        User user = new User(username, hashed);
        user.setRole(this.roleRepository.getById(roleId));
        this.userRepository.save(user);
        return user.getId();
    }
}