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

package vldc.aswi.web.controller;

import oracle.jdbc.proxy.annotation.Post;
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.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import vldc.aswi.domain.Assembly;
import vldc.aswi.domain.Role;
import vldc.aswi.service.*;
import vldc.aswi.service.parameter.ParameterTypeManager;

import javax.validation.Valid;
import java.util.ArrayList;
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;

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

        modelMap.addAttribute("assemblies", 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 = 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_manage");

        ModelMap modelMap = modelAndView.getModelMap();

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

        modelMap.addAttribute("assembly", assembly);
        modelMap.addAttribute("roles", roleManager.getRoles());
        modelMap.addAttribute("rolesWithAccess", assembly.getRoles());
        modelMap.addAttribute("parameters", assembly.getParameters());
        modelMap.addAttribute("parameterTypes", parameterTypeManager.getParameterTypes());
        modelMap.addAttribute("functions", functionManager.getFunctions());
        modelMap.addAttribute("operators", operatorManager.getOperators());
        modelMap.addAttribute("locations", 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("roles", roleManager.getRoles());

        return modelAndView;
    }

    @PostMapping("/assembly_new")
    public ModelAndView assemblyNewIndexPost() {
        ModelAndView modelAndView = new ModelAndView("assembly_manage");

        ModelMap modelMap = modelAndView.getModelMap();

        modelMap.addAttribute("assembly", new Assembly());
        modelMap.addAttribute("roles", roleManager.getRoles());

        return modelAndView;
    }
}