aswi2020vldc-gitlab/src/main/java/vldc/aswi/service/AssemblyManagerImpl.java @ ba8b306a

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.stereotype.Service;
import vldc.aswi.dao.AssemblyRepository;
import vldc.aswi.dao.RoleRepository;
import vldc.aswi.domain.Assembly;
import vldc.aswi.domain.Role;

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

/**
 * Manager for Assembly.
 */
@Service
@Slf4j
public class AssemblyManagerImpl implements AssemblyManager {

    /** Autowired assembly repository. */
    @Autowired
    private AssemblyRepository assemblyRepository;

    @Autowired
    private RoleRepository roleRepository;

    /**
     * Initialization setup for assembly manager. Check if assembly repository contains
     * records and if not, initialize default values.
     */
    @EventListener(classes = {
            ContextRefreshedEvent.class
    })
    @Order(1)
    @Transactional
    public void setup() {
        if (this.assemblyRepository.count() == 0) {
            log.info("No assembly present, creating items.");

        }
    }

    /**
     * Get all Assemblies from database.
     * @return List of assemblies.
     */
    @Override
    public List<Assembly> getAssemblies() {
        List<Assembly> retVal = new LinkedList<>();
        this.assemblyRepository.findAll().forEach(retVal::add);
        return retVal;
    }

    /**
     * Get assembly by id.
     * @param id - ID of assembly.
     * @return Assembly.
     */
    @Override
    public Assembly getAssemblyById(Long id) {
        return assemblyRepository.getById(id);
    }

    /**
     * Update assembly's order.
     * @param id - ID of assembly.
     * @param order - Assembly's order.
     */
    @Override
    public void updateAssemblyOrder(Long id, int order) {
        Assembly assembly = getAssemblyById(id);
        assembly.setAssemblyOrder(order);
        this.assemblyRepository.save(assembly);
    }

    /**
     * Update assembly.
     * @param updatedAssemblyValues - Assembly values.
     * @return ID of assembly.
     */
    @Override
    public Long updateAssembly(Assembly updatedAssemblyValues) {
        // TODO: test it.
        Assembly assembly = getAssemblyById(updatedAssemblyValues.getId());
        assembly.setName(updatedAssemblyValues.getName());
        assembly.setSQLQuery(updatedAssemblyValues.getSQLQuery());
        assembly.setAssemblyOrder(updatedAssemblyValues.getAssemblyOrder());
        assembly.setIsPublic(updatedAssemblyValues.getIsPublic());
        this.assemblyRepository.save(assembly);

        updateAssemblyRoles(updatedAssemblyValues.getId(), updatedAssemblyValues.getRoles());

        return assembly.getId();
    }

    /**
     * Update assembly's roles.
     * @param id - ID of assembly.
     * @param updatedRoles - List of roles.
     */
    private void updateAssemblyRoles(Long id, List<Role> updatedRoles) {
        // TODO: test it.
        Assembly assembly = getAssemblyById(id);
        List<Role> currentRoles = assembly.getRoles();

        // Find new roles.
        List<Role> currentRolesNew = new LinkedList<>(currentRoles);
        List<Role> updatedRolesNew = new LinkedList<>(updatedRoles);
        updatedRolesNew.removeAll(currentRolesNew);

        // Find deleted roles.
        List<Role> currentRolesDeleted = new LinkedList<>(currentRoles);
        List<Role> updatedRolesDeleted = new LinkedList<>(updatedRoles);
        currentRolesDeleted.removeAll(updatedRolesDeleted);

        // If no role has been deleted or added, then updating is unnecessary.
        if (updatedRolesNew.size() == 0 && currentRolesDeleted.size() == 0) return;

        // Add new roles and remove deleted roles.
        for (Role newRole : updatedRolesNew) {
            currentRoles.add(this.roleRepository.getById(newRole.getId()));
        }
        currentRoles.removeAll(currentRolesDeleted);

        // Save updates to database.
        this.assemblyRepository.save(assembly);
    }

    /**
     * Create new assembly.
     * @param assemblyValues - Assembly values.
     * @return ID of assembly.
     */
    @Override
    public Long createAssembly(Assembly assemblyValues) {
        // TODO: test it.
        // Create new assembly and assign values.
        Assembly assembly = new Assembly();
        assembly.setName(assemblyValues.getName());
        assembly.setSQLQuery(assemblyValues.getSQLQuery());
        assembly.setAssemblyOrder(this.assemblyRepository.findFirst1ByOrderByAssemblyOrder().getAssemblyOrder() + 1);
        assembly.setIsPublic(assemblyValues.getIsPublic());
        List<Role> roles = new LinkedList<>();

        // Add access for roles.
        if (assemblyValues.getRoles() != null && assemblyValues.getRoles().size() > 0) {
            for (Role newRole : assemblyValues.getRoles()) {
                roles.add(this.roleRepository.getById(newRole.getId()));
            }
        }

        assembly.setRoles(roles);

        // Create new record in database.
        this.assemblyRepository.save(assembly);

        return assembly.getId();
    }
}