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

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.WebDataBinder;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.InitBinder;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import vldc.aswi.domain.Assembly;
import vldc.aswi.service.*;
import vldc.aswi.service.parameter.ParameterManager;
import vldc.aswi.service.parameter.ParameterTypeManager;
import vldc.aswi.validators.AssemblyValidator;

import javax.validation.Valid;

/**
 * Controller for assembly.
 */
@Controller
public class AssemblyController extends BasicController {

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

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

    /** 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 parameter manager. */
    @Autowired
    private ParameterManager parameterManager;

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

    /** 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";

    /** Name of thymeleaf parameter, which will contain error text from validator. */
    private String assemblyErrorName = "errorText";

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

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

        ModelMap modelMap = modelAndView.getModelMap();

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

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

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

        Assembly assembly2 = this.assemblyManager.getAssemblyById(assemblyID);

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

        return modelAndView;
    }

    /**
     * Get method for assembly edit.
     * @param id - Id of assembly.
     * @return ModelAndView for assembly edit.
     */
    @GetMapping("/assembly_edit")
    public ModelAndView assemblyEditIndex(@Valid @ModelAttribute("assemblyID") String id) {
        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());

        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 assemblyEditPostIndex(@Valid @ModelAttribute("assembly") Assembly assembly, BindingResult result,
                                              @Valid @ModelAttribute("checkboxPublic") String assemblyIsPublic) {

        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(assembly.getId());

            modelMap.addAttribute("assembly", updatedAssembly);
            modelMap.addAttribute(assemblyTitleName, assemblyEditTitleText + " " + updatedAssembly.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());

        return modelAndView;
    }

    /**
     * Get method for new assembly form.
     * @return ModelAndView for new assembly.
     */
    @GetMapping("/assembly_new")
    public ModelAndView assemblyNewIndex() {
        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 assemblyNewIndexPost(@Valid @ModelAttribute("assembly") Assembly assembly, BindingResult result) {
        ModelAndView modelAndView = new ModelAndView();

        if (result.hasErrors()) {
            ModelMap modelMap = modelAndView.getModelMap();
            modelAndView.setViewName("assembly_manage");

            modelMap.addAttribute("assembly", assembly);

            modelMap.addAttribute(assemblyTitleName, assemblyNewTitleText);
            modelMap.addAttribute(assemblyErrorName, getFullErrorMessage(result));

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

        ModelMap modelMap = modelAndView.getModelMap();

        // 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);

        return modelAndView;
    }
}