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

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.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.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 user repository contains records and if not, initialize default values.
     */
    @EventListener(classes = {
            ContextRefreshedEvent.class
    })
    @Order(1)
    @Transactional
    public void setup() {
        if (this.userRepository.count() == 0) {
            log.info("No user present");
            /*
            this.addUser("administrator", "demo", (long) 1);
            this.addUser("fakulta", "demo", (long) 2);
            this.addUser("referentka", "demo", (long) 3);
            this.addUser("prorektor", "demo", (long) 4);
             */
        }
    }

    /**
     * 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(user.getRole().getName());

        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();
    }
}