aswi2020vldc-gitlab/src/main/java/vldc/aswi/web/controller/AssemblyController.java @ d90fc9de

package vldc.aswi.web.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.validation.BeanPropertyBindingResult;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.WebDataBinder;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import vldc.aswi.domain.*;
import vldc.aswi.domain.parameter.Parameter;
import vldc.aswi.domain.parameter.ParameterInConfiguration;
import vldc.aswi.model.ExportType;
import vldc.aswi.model.table.contingencyTable.ContingencyTableRow;
import vldc.aswi.service.*;
import vldc.aswi.service.parameter.ParameterManager;
import vldc.aswi.service.parameter.ParameterTypeManager;
import vldc.aswi.utils.AuthControl;
import vldc.aswi.validators.AssemblyValidator;
import vldc.aswi.utils.Utils;
import vldc.aswi.validators.ConfigurationValidator;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;

/**
 * Controller for assemblies and configurations
 */
@Controller
public class AssemblyController extends BasicController {

    /** Autowired sql query manager. */
    @Autowired
    private SqlQueryManager sqlQueryManager;

    /** Autowired assembly manager. */
    @Autowired
    private AssemblyManager assemblyManager;

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

    /** Autowired export manager. */
    @Autowired
    private ExportManager exportManager;

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

    /** Autowired role manager. */
    @Autowired
    private RoleManager roleManager;

    /** Autowired parameterType manager. */
    @Autowired
    private ParameterTypeManager parameterTypeManager;

    /** Autowired operator manager. */
    @Autowired
    private OperatorManager operatorManager;

    /** Autowired function manager. */
    @Autowired
    private FunctionManager functionManager;

    /** Autowired location manager. */
    @Autowired
    private LocationManager locationManager;

    /** Autowired assembly validator */
    @Autowired
    private AssemblyValidator assemblyValidator;

    /** Autowired configuration validator */
    @Autowired
    private ConfigurationValidator configurationValidator;

    /** Name of thymeleaf parameter, which will contain text in assembly form. */
    private String assemblyTitleName = "title";

    /** Title text for assembly edit form.  */
    private String assemblyEditTitleText = "Editace - ";

    /** Title text for new assembly form. */
    private String assemblyNewTitleText = "Vytvoření nové sestavy";

    /** Title text for table generation form.  */
    private String generateTableTitleText = "Generování tabulky - ";

    /**
     * Bind assembly validator.
     * @param binder Binder.
     */
    @InitBinder("assembly")
    protected void userBinder(WebDataBinder binder) {
        binder.addValidators(this.assemblyValidator);
    }

    /**
     * Get method for form, where configuration is created from assembly.
     * @param id - ID of assembly.
     * @return ModelAndView for assembly.
     */
    @GetMapping("/assembly")
    public ModelAndView assemblyGet(@RequestParam("assemblyID") String id) {
        ModelAndView modelAndView = new ModelAndView("assembly");

        ModelMap modelMap = modelAndView.getModelMap();

        Assembly assembly = this.assemblyManager.getAssemblyById(Long.parseLong(id));

        // TODO: 04.05.2020 error page when id doesn't exist

        String roleName = AuthControl.getRoleName();

        if (roleName == null) {
            // TODO: 04.05.2020 error message, user not authenticated
        }

        Role role = roleManager.getRole(roleName);

        if (!assembly.getRoles().contains(role) && !role.getName().equals("Administrátor")) {
            // TODO: 04.05.2020 Error page, wrong role
            return new ModelAndView("redirect:/");
        }

        Configuration configuration = new Configuration();

        configuration.setAssembly(assembly);
        configuration.setTableName(assembly.getName());
        configuration.setParametersInConfiguration(new ArrayList<>());
        for(Parameter parameter : assembly.getParameters()) {
            ParameterInConfiguration parameterInConfiguration = new ParameterInConfiguration();
            parameterInConfiguration.setParameter(parameter);
            parameterInConfiguration.setConfiguration(configuration);
            parameterInConfiguration.setOperator(new Operator());
            parameterInConfiguration.setLocation(new Location());
            parameterInConfiguration.setParameterOrder(parameter.getParameterOrder());
            configuration.getParametersInConfiguration().add(parameterInConfiguration);
        }

        Comparator<ParameterInConfiguration> comparator = Comparator.comparingInt(o -> o.getParameter().getParameterOrder());

        boolean isNotRemoveEmpty = false;
        modelMap.addAttribute("isNotRemoveEmpty", isNotRemoveEmpty);
        modelMap.addAttribute("configuration", configuration);
        modelMap.addAttribute("comparator", comparator);
        modelMap.addAttribute("formAction", "/assembly?assemblyID=" + assembly.getId());

        return modelAndView;
    }

    /**
     * Post method for form, where configuration is created from assembly.
     * @param newConfiguration - Configuration values.
     * @param bindingResult - Error results from assembly validators.
     * @return ModelAndView for assembly.
     */
    @PostMapping("/assembly")
    public ModelAndView assemblyPost(Configuration newConfiguration,
                                     BindingResult bindingResult,
                                     RedirectAttributes atts,
                                     @RequestParam("isNotRemoveEmpty") String isNotRemoveEmptyString,
                                     @RequestParam("assemblyID") String id,
                                     @RequestParam(required=false, value="generateTable") String generateTable,
                                     @RequestParam(required=false, value="exportXlsx") String exportXlsx,
                                     @RequestParam(required=false, value="exportPdf") String exportPdf,
                                     @RequestParam(required=false, value="saveConfiguration") String saveConfiguration,
                                     RedirectAttributes redirectAttributes,
                                     HttpServletResponse response
                                    )
    {

        ModelAndView modelAndView = new ModelAndView();
        ModelMap modelMap = modelAndView.getModelMap();
        boolean isNotRemoveEmpty = isNotRemoveEmptyString == null || !isNotRemoveEmptyString.equals("");

        if (bindingResult.hasErrors()) {
            // TODO: 04.05.2020 Error message
            modelAndView.setViewName("redirect:/");

            return modelAndView;
        }

        if (generateTable != null)
        {
            Assembly assembly = this.assemblyManager.getAssemblyById(Long.parseLong(id));
            newConfiguration.setAssembly(assembly);
            newConfiguration.setTableName(assembly.getName());

            // If configuration name is empty, then replace it by assembly name.
            if (newConfiguration.getName().equals("")) {
                newConfiguration.setName(assembly.getName());
            }

            prepareForTable(newConfiguration);
            BindingResult errors = new BeanPropertyBindingResult(newConfiguration, newConfiguration.getClass().getName());
            configurationValidator.validate(newConfiguration, errors);
            if (errors.hasErrors()) {
                addConfigurationDataIntoModelAndView(newConfiguration, modelAndView, modelMap);

                modelMap.addAttribute(assemblyErrorName, getFullErrorMessage(errors));
                modelMap.addAttribute(assemblyTitleName, generateTableTitleText + " " + newConfiguration.getName());
            }
            else {
                List<ContingencyTableRow> rows = this.sqlQueryManager.getContingencyTableRow(newConfiguration.getAssembly(),
                        newConfiguration.getParametersInConfiguration(), isNotRemoveEmpty);

                modelMap.addAttribute("configurationID", id);
                modelMap.addAttribute("contingencyTableRows", rows);
                modelMap.addAttribute("isNotRemoveEmpty", isNotRemoveEmpty);
                addConfigurationDataIntoModelAndView(newConfiguration, modelAndView, modelMap);
            }
        }
        else if (exportXlsx != null)
        {
            prepareForTable(newConfiguration);
            this.exportManager.exportContingencyTable(newConfiguration, isNotRemoveEmpty, response, ExportType.XLSX);

            // Return null, because we want to continue on our page with selected parameters / attributes.
            return null;
        }
        else if (exportPdf != null)
        {
            prepareForTable(newConfiguration);
            this.exportManager.exportContingencyTable(newConfiguration, isNotRemoveEmpty, response, ExportType.PDF);

            // Return null, because we want to continue on our page with selected parameters / attributes.
            return null;

        }
        else if (saveConfiguration != null)
        {
            Configuration configuration = configurationManager.saveConfiguration(newConfiguration, "");

            initializeFields(configuration);

            redirectAttributes.addFlashAttribute(assemblySuccessName, "Šablona byla úspěšně uložena");

            modelAndView.setViewName("redirect:/configuration?configurationID=" + configuration.getId());
        }

        return modelAndView;
    }

    /**
     * Adds configuration and other data to modelAndView.
     * @param newConfiguration Configuration to be added to the modelAndView.
     * @param modelAndView ModelAndView to which to which modelMap belongs.
     * @param modelMap ModelMap to which data is added.
     */
    private void addConfigurationDataIntoModelAndView(Configuration newConfiguration, ModelAndView modelAndView, ModelMap modelMap) {
        Comparator<ParameterInConfiguration> comparator = Comparator.comparingInt(o -> o.getParameter().getParameterOrder());
        initializeFields(newConfiguration);
        modelMap.addAttribute("configuration", newConfiguration);
        modelMap.addAttribute("comparator", comparator);
        modelMap.addAttribute("formAction", "/assembly?assemblyID=" + newConfiguration.getAssembly().getId());
        modelAndView.setViewName("assembly");
    }

    /**
     * Get method for assembly delete.
     * @param id - ID of assembly.
     * @return ModelAndView for index.
     */
    @GetMapping("/assembly_delete")
    public ModelAndView assemblyDeleteGet(@RequestParam("assemblyID") String id, RedirectAttributes redirectAttributes) {
        ModelAndView modelAndView = new ModelAndView("redirect:/");

        Long assemblyId = Utils.tryParseLong(id);

        if (assemblyId == null) {
            redirectAttributes.addFlashAttribute(assemblyErrorName, "Sestavu se nepodařilo odstranit.");
        }
        boolean success = this.assemblyManager.deleteAssembly(assemblyId);

        if (success)
        {
            redirectAttributes.addFlashAttribute(assemblySuccessName, "Sestava byla úspěšně odstraněna");
        }
        else
        {
            redirectAttributes.addFlashAttribute(assemblyErrorName, "Sestavu se nepodařilo odstranit");
        }

        return modelAndView;
    }

    /**
     * Get method for assembly edit.
     * @param id - Id of assembly.
     * @return ModelAndView for assembly edit.
     */
    @GetMapping("/assembly_edit")
    public ModelAndView assemblyEditGet(@RequestParam("assemblyID") String id) {
        long startTime = System.currentTimeMillis();
        ModelAndView modelAndView = new ModelAndView("assembly_manage");

        ModelMap modelMap = modelAndView.getModelMap();

        Assembly assembly = this.assemblyManager.getAssemblyById(Long.parseLong(id));

        modelMap.addAttribute("assembly", assembly);

        modelMap.addAttribute(assemblyTitleName, assemblyEditTitleText + " " + assembly.getName());

        modelMap.addAttribute("allRoles", this.roleManager.getRoles());
        modelMap.addAttribute("allParameterTypes", this.parameterTypeManager.getParameterTypes());
        modelMap.addAttribute("allFunctions", this.functionManager.getFunctions());
        modelMap.addAttribute("allOperators", this.operatorManager.getOperators());
        modelMap.addAttribute("allLocations", this.locationManager.getLocations());
        long estimatedTime = System.currentTimeMillis() - startTime;
        System.out.println("EDIT GET estimated - " + ((double)estimatedTime / 1000) + " sekund ");

        return modelAndView;
    }

    /**
     * Post method for assembly edit.
     * @param assembly - Edited assembly values.
     * @param result - Validator results.
     * @param assemblyIsPublic - Attribute indicating if assembly is public or not.
     * @return ModelAndView for assembly edit.
     */
    @PostMapping("/assembly_edit")
    public ModelAndView assemblyEditPost(@Valid @ModelAttribute("assembly") Assembly assembly, BindingResult result,
                                              @Valid @ModelAttribute("checkboxPublic") String assemblyIsPublic) {
        long startTime = System.currentTimeMillis();
        ModelAndView modelAndView = new ModelAndView("assembly_manage");
        ModelMap modelMap = modelAndView.getModelMap();

        if (result.hasErrors()) {
            modelMap.addAttribute("assembly", assembly);

            modelMap.addAttribute(assemblyErrorName, getFullErrorMessage(result));
            modelMap.addAttribute(assemblyTitleName, assemblyEditTitleText + " " + assembly.getName());
        }
        else {
            if (assemblyIsPublic.equals("on")) assembly.setIsPublic(1);
            else assembly.setIsPublic(0);

            // TODO: chybí defaultValue.
            Long assemblyID = this.assemblyManager.updateAssembly(assembly);

            this.parameterManager.updateParameters(assemblyID, assembly.getParameters());
            Assembly updatedAssembly = this.assemblyManager.getAssemblyById(assemblyID);

            modelMap.addAttribute("assembly", updatedAssembly);
            modelMap.addAttribute(assemblyTitleName, assemblyEditTitleText + " " + updatedAssembly.getName());
            modelMap.addAttribute(assemblySuccessName, "Úspěšně uloženo");
        }

        modelMap.addAttribute("allRoles", this.roleManager.getRoles());
        modelMap.addAttribute("allParameterTypes", this.parameterTypeManager.getParameterTypes());
        modelMap.addAttribute("allFunctions", this.functionManager.getFunctions());
        modelMap.addAttribute("allOperators", this.operatorManager.getOperators());
        modelMap.addAttribute("allLocations", this.locationManager.getLocations());
        long estimatedTime = System.currentTimeMillis() - startTime;
        System.out.println("EDIT POST estimated - " + ((double)estimatedTime / 1000) + " sekund ");

        return modelAndView;
    }

    /**
     * Get method for new assembly form.
     * @return ModelAndView for new assembly.
     */
    @GetMapping("/assembly_new")
    public ModelAndView assemblyNewGet() {
        ModelAndView modelAndView = new ModelAndView("assembly_manage");

        ModelMap modelMap = modelAndView.getModelMap();

        modelMap.addAttribute("assembly", new Assembly());

        modelMap.addAttribute(assemblyTitleName, assemblyNewTitleText);

        modelMap.addAttribute("allRoles", this.roleManager.getRoles());
        modelMap.addAttribute("allParameterTypes", this.parameterTypeManager.getParameterTypes());
        modelMap.addAttribute("allFunctions", this.functionManager.getFunctions());
        modelMap.addAttribute("allOperators", this.operatorManager.getOperators());
        modelMap.addAttribute("allLocations", this.locationManager.getLocations());

        return modelAndView;
    }

    /**
     * Post method for new assembly form.
     * @param assembly - Assembly values.
     * @param result - Validator results.
     * @return
     */
    @PostMapping("/assembly_new")
    public ModelAndView assemblyNewPost(@Valid @ModelAttribute("assembly") Assembly assembly, BindingResult result) {
        ModelAndView modelAndView = new ModelAndView();
        ModelMap modelMap = modelAndView.getModelMap();

        if (result.hasErrors()) {
            modelAndView.setViewName("assembly_manage");
            modelMap.addAttribute("assembly", assembly);

            modelMap.addAttribute(assemblyErrorName, getFullErrorMessage(result));
            modelMap.addAttribute(assemblyTitleName, assemblyNewTitleText);
        }
        else {
            // TODO: chybí získání default value.
            Long assemblyId = this.assemblyManager.createAssembly(assembly);
            this.parameterManager.addParameters(assemblyId, assembly.getParameters());

            modelAndView.setViewName("redirect:/assembly_edit?assemblyID=" + assemblyId);
            modelMap.addAttribute(assemblySuccessName, "Sestava byla úspěšně vytvořena");
        }

        modelMap.addAttribute("allRoles", this.roleManager.getRoles());
        modelMap.addAttribute("allParameterTypes", this.parameterTypeManager.getParameterTypes());
        modelMap.addAttribute("allFunctions", this.functionManager.getFunctions());
        modelMap.addAttribute("allOperators", this.operatorManager.getOperators());
        modelMap.addAttribute("allLocations", this.locationManager.getLocations());

        return modelAndView;
    }

    /**
     * Initializes fields of a given configuration
     * @param configuration configuration which fields should be initialized
     */
    private void initializeFields(Configuration configuration) {
        for(ParameterInConfiguration parameterInConfiguration : configuration.getParametersInConfiguration()) {
            if(parameterInConfiguration.getLocation() == null) {
                parameterInConfiguration.setLocation(new Location());
            }
            if(parameterInConfiguration.getOperator() == null) {
                parameterInConfiguration.setOperator(new Operator());
            }
            if(parameterInConfiguration.getFunctions() == null) {
                parameterInConfiguration.setFunctions(new ArrayList<>());
            }
        }
    }

    /**
     * Prepares the configuration for contingency table generation. Initializes assembly and parameters of parameters in
     * configuration.
     * @param configuration configuration to be prepared.
     */
    private void prepareForTable(Configuration configuration) {
        Assembly assembly = assemblyManager.getAssemblyById(configuration.getAssembly().getId());
        configuration.setAssembly(assembly);

        for(int i = 0; i < configuration.getParametersInConfiguration().size(); i++) {
            ParameterInConfiguration parameterInConfiguration = configuration.getParametersInConfiguration().get(i);
            parameterInConfiguration.setParameter(parameterManager.getParameterById(parameterInConfiguration.getParameter().getId()));
            if (parameterInConfiguration.getLocation() != null) {
                parameterInConfiguration.setLocation(locationManager.getLocationById(parameterInConfiguration.getLocation().getId()));
            } else {
                parameterInConfiguration.setLocation(new Location());
            }
            parameterInConfiguration.setConfiguration(configuration);
        }

        Comparator<ParameterInConfiguration> comparator = Comparator.comparingInt(o -> o.getParameter().getParameterOrder());
        configuration.getParametersInConfiguration().sort(comparator);
    }
}