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

package vldc.aswi.web.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
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.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import vldc.aswi.domain.Assembly;
import vldc.aswi.domain.Function;
import vldc.aswi.domain.Location;
import vldc.aswi.domain.parameter.Parameter;
import vldc.aswi.service.*;
import vldc.aswi.service.parameter.ParameterManager;
import vldc.aswi.service.parameter.ParameterTypeManager;
import vldc.aswi.service.parameter.ParameterValueManager;

import javax.validation.Valid;
import java.beans.PropertyEditorSupport;
import java.util.LinkedList;
import java.util.List;

@Controller
public class AssemblyController {

    @Autowired
    private SqlQueryManager sqlQueryManager;

    @Autowired
    private AssemblyManager assemblyManager;

    @Autowired
    private RoleManager roleManager;

    @Autowired
    private ParameterTypeManager parameterTypeManager;

    @Autowired
    private OperatorManager operatorManager;

    @Autowired
    private FunctionManager functionManager;

    @Autowired
    private LocationManager locationManager;

    @Autowired
    private ParameterManager parameterManager;

    @Autowired
    private ParameterValueManager parameterValueManager;

    @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;
    }

    @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("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;
    }

    @PostMapping("/assembly_edit")
    public ModelAndView assemblyEditPostIndex(@Valid @ModelAttribute("assembly") Assembly assembly, @Valid @ModelAttribute("checkboxPublic") String assemblyIsPublic) {

        ModelAndView modelAndView = new ModelAndView("assembly_manage");

        ModelMap modelMap = modelAndView.getModelMap();
        if (assemblyIsPublic.equals("on")) assembly.setIsPublic(1);
        else assembly.setIsPublic(0);

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

        // TODO: vyzkoušet jestli se dobře vrací ID parameterType a zda se vypisuje název v selectboxu namísto ID !!
        // TODO: chybí defaultValue.


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

    @GetMapping("/assembly_new")
    public ModelAndView assemblyNewIndex() {
        ModelAndView modelAndView = new ModelAndView("assembly_manage");

        ModelMap modelMap = modelAndView.getModelMap();

        modelMap.addAttribute("assembly", new Assembly());
        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;
    }

    @PostMapping("/assembly_new")
    public ModelAndView assemblyNewIndexPost(@Valid @ModelAttribute("assembly") Assembly assembly) {
        ModelAndView modelAndView = new ModelAndView();

        ModelMap modelMap = modelAndView.getModelMap();

        Long assemblyId = this.assemblyManager.createAssembly(assembly);
        this.parameterManager.addParameters(assemblyId, assembly.getParameters());

        modelAndView.setViewName("redirect:/assembly_edit?assemblyID=" + assembly.getId());
        return modelAndView;
    }
}