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

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

import javax.transaction.Transactional;
import java.util.ArrayList;
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 role repository. */
    @Autowired
    private RoleRepository roleRepository;

    /** Autowired configuration manager. */
    @Autowired
    private ConfigurationManager configurationManager;

    /** Autowired parameter manager. */
    @Autowired
    private ParameterManager parameterManager;

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

    /**
     * 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 all Assemblies from database ordered.
     * @return List of assemblies.
     */
    @Override
    public List<Assembly> getAssembliesOrdered() {
        return this.assemblyRepository.getByOrderByAssemblyOrderAsc();
    }

    /**
     * Get all Assemblies from database with role, ordered.
     * @return List of assemblies.
     */
    @Override
    public List<Assembly> getAssembliesWithRoleOrdered(Role role) {
        return this.assemblyRepository.getByRolesContainingAndIsPublicOrderByAssemblyOrderAsc(role, 1);
    }

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

    /**
     * Update assembly order.
     * @param data - sorted IDs.
     */
    @Override
    public void updateAssemblyOrder(String data) {
        String[] idOrder = data.split(",");
        for (int i = 0; i < idOrder.length; i++) {
            Assembly assembly = getAssemblyById(Long.valueOf(idOrder[i]));
            assembly.setAssemblyOrder(i + 1);
            this.assemblyRepository.save(assembly);
        }
    }

    /**
     * Update assembly.
     * @param updatedAssemblyValues - Assembly values.
     * @return ID of assembly.
     */
    @Override
    public Long updateAssembly(Assembly updatedAssemblyValues) {
        Assembly assembly = getAssemblyById(updatedAssemblyValues.getId());
        assembly.setName(updatedAssemblyValues.getName());
        assembly.setSQLQuery(updatedAssemblyValues.getSQLQuery());
        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) {
        Assembly assembly = getAssemblyById(id);
        List<Role> currentRoles = assembly.getRoles();

        if (updatedRoles != null) {
            // 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) {
        // Create new assembly and assign values.
        Assembly assembly = new Assembly();
        assembly.setName(assemblyValues.getName());
        assembly.setSQLQuery(assemblyValues.getSQLQuery());

        Assembly lastAssembly = this.assemblyRepository.findTopByOrderByAssemblyOrderDesc();
        int order = 1;

        if (lastAssembly != null) {
            order = lastAssembly.getAssemblyOrder() + 1;
        }

        assembly.setAssemblyOrder(order);
        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();
    }

    /**
     * Delete assembly with ID.
     * @param id - ID of assembly.
     * @return True if delete was successful, otherwise false.
     */
    @Override
    public boolean deleteAssembly(Long id) {
        // TODO: Do it as transactional.
        try {
            // Delete configurations, which using current assembly.
            this.configurationManager.deleteConfigurations(id);

            // Delete parameters in current assembly.
            this.parameterManager.deleteParameters(id);

            Assembly assembly = this.assemblyRepository.getById(id);

            // Remove roles from 'Sestava_ma_Role' table.
            List<Role> roleList = new ArrayList<>(assembly.getRoles());
            for (Role role : roleList) {
                assembly.removeRole(role);
            }

            this.assemblyRepository.save(assembly);

            // Delete assembly.
            this.assemblyRepository.delete(assembly);
            // TODO: Send message for all users using this assembly.
            return true;
        } catch (Exception e) {
            return false;
        }
    }
}