org.createnet.raptor.auth.service.services.UserService.java Source code

Introduction

Here is the source code for org.createnet.raptor.auth.service.services.UserService.java

Source

/*
 * Copyright 2017 FBK/CREATE-NET
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.createnet.raptor.auth.service.services;

import java.util.stream.Collectors;
import org.createnet.raptor.models.auth.Role;
import org.createnet.raptor.models.auth.User;
import org.createnet.raptor.auth.service.repository.RoleRepository;
import org.createnet.raptor.auth.service.repository.UserRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.stereotype.Service;

/**
 *
 * @author Luca Capra <lcapra@fbk.eu>
 */
@Service
public class UserService {

    @Autowired
    private PasswordEncoder passwordEncoder;

    @Autowired
    private UserRepository userRepository;

    @Autowired
    private RoleRepository roleRepository;

    public Iterable<User> list() {
        return userRepository.findAll();
    }

    public User save(User user) {
        saveRoles(user);
        return userRepository.save(user);
    }

    public User getByUuid(String uuid) {
        return userRepository.findByUuid(uuid);
    }

    /**
     * Save new roles or load roles based on name to ensure roles are all managed
     */
    protected void saveRoles(User user) {
        user.setRoles(user.getRoles().stream().map((r) -> {

            if (r.getId() != null) {
                return r;
            }

            Role r1 = roleRepository.findByName(r.getName());

            if (r1 != null) {
                return r1;
            }

            r = roleRepository.save(r);
            return r;
        }).collect(Collectors.toSet()));
    }

    public User update(String uuid, User rawUser) {

        User user = userRepository.findByUuid(uuid);

        if (rawUser.getFirstname() != null && !rawUser.getFirstname().isEmpty()) {
            user.setFirstname(rawUser.getFirstname());
        }

        if (rawUser.getLastname() != null && !rawUser.getLastname().isEmpty()) {
            user.setLastname(rawUser.getLastname());
        }

        if (rawUser.getEmail() != null && !rawUser.getEmail().isEmpty()) {
            user.setEmail(rawUser.getEmail());
        }

        if (rawUser.getEnabled() != null) {
            user.setEnabled(rawUser.getEnabled());
        }

        //      // TODO add Role repository
        if (!rawUser.getRoles().isEmpty()) {
            rawUser.getRoles().stream().forEach(r -> user.addRole(r));
            saveRoles(user);
        }

        encodePassword(rawUser);

        return userRepository.save(user);
    }

    public User create(User rawUser) {

        encodePassword(rawUser);

        return userRepository.save(rawUser);
    }

    public void delete(User user) {
        userRepository.delete(user.getId());
    }

    protected void encodePassword(User user) {
        // TODO missing password validation
        if (user.getPassword() != null && !user.getPassword().isEmpty()) {
            user.setPassword(passwordEncoder.encode(user.getPassword()));
        }
    }

    public boolean exists(User rawUser) {

        User user = getByUuid(rawUser.getUuid());
        if (user != null)
            return true;

        return userRepository.findByUsername(rawUser.getUsername()) != null;
    }

}