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

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.BindingResult;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import vldc.aswi.domain.Assembly;
import vldc.aswi.domain.Configuration;
import vldc.aswi.domain.Location;
import vldc.aswi.domain.Operator;
import vldc.aswi.domain.parameter.Parameter;
import vldc.aswi.domain.parameter.ParameterInConfiguration;
import vldc.aswi.service.AssemblyManager;
import vldc.aswi.service.ConfigurationManager;
import vldc.aswi.service.SqlQueryManager;

import javax.validation.Valid;
import java.util.ArrayList;

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

    /**
     * Autowired SQL query manager
     */
    @Autowired
    private SqlQueryManager sqlQueryManager;

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

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

    /**
     * Opens an configuration page after assembly name was clicked
     * @param id ID of assembly
     * @return modelAndView of page to be shown
     */
    @GetMapping("/assembly")
    public ModelAndView assemblyIndex(@Valid @ModelAttribute("assemblyID") String id) {
        ModelAndView modelAndView = new ModelAndView("assembly");

        ModelMap modelMap = modelAndView.getModelMap();

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

        Configuration configuration = new Configuration();

        configuration.setAssembly(assembly);
        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());
            configuration.getParametersInConfiguration().add(parameterInConfiguration);
        }

        modelMap.addAttribute("configuration", configuration);
        modelMap.addAttribute("isAssembly", false);

        return modelAndView;
    }

    /**
     * Opens an configuration page after configuration name was clicked
     * @param id ID of configuration
     * @return modelAndView of page to be shown
     */
    @GetMapping("/configuration")
    public ModelAndView configurationIndex(@Valid @ModelAttribute("configurationID") String id) {
        ModelAndView modelAndView = new ModelAndView("assembly");

        ModelMap modelMap = modelAndView.getModelMap();

        Configuration configuration = configurationManager.getConfigurationById(Long.parseLong(id));

        initializeFields(configuration);

        modelMap.addAttribute("configuration", configuration);
        modelMap.addAttribute("isAssembly", "0");

        return modelAndView;
    }

    /**
     * Saves or edits configuration and redirects to the page with configuration
     * @param newConfiguration contained with configuration data
     * @param id ID of configuration, if empty new configuration is created
     * @param bindingResult binding result
     * @return modelAndView with redirection
     */
    @PostMapping("/configuration")
    public ModelAndView saveConfiguration(@Valid Configuration newConfiguration, BindingResult bindingResult,
                                          @RequestParam("configurationID") String id) {
        ModelAndView modelAndView = new ModelAndView();

        if (bindingResult.hasErrors()) {
            modelAndView.setViewName("redirect:/");

            return modelAndView;
        }

        Configuration configuration = configurationManager.saveConfiguration(newConfiguration, id);

        initializeFields(configuration);

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

        return modelAndView;
    }

    @PostMapping("/assembly")
    public ModelAndView indexPost(@Valid Assembly assembly, BindingResult bindingResult, RedirectAttributes atts) {
        ModelAndView modelAndView = new ModelAndView();

        ModelMap modelMap = modelAndView.getModelMap();

        long assemblyID = assembly.getId();
        System.out.println(assemblyID);
        Assembly assembly2 = assemblyManager.getAssemblyById(assemblyID);


        modelMap.addAttribute("assemblies", assemblyManager.getAssemblies());
        modelMap.addAttribute("assembly", assembly2);
        modelMap.addAttribute("contingencyTableRows", sqlQueryManager.getContingencyTableRow(assembly2.getSQLQuery()));

        return modelAndView;
    }

    @GetMapping("/assembly_edit")
    public ModelAndView assemblyEditIndex(@Valid @ModelAttribute("assemblyID") String id) {
        ModelAndView modelAndView = new ModelAndView("assembly_edit");

        ModelMap modelMap = modelAndView.getModelMap();

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

        modelMap.addAttribute("assemblies", assemblyManager.getAssemblies());
        modelMap.addAttribute("assembly", assembly);

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