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

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.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import vldc.aswi.domain.Assembly;
import vldc.aswi.service.AssemblyManager;
import vldc.aswi.service.SqlQueryManager;

import javax.validation.Valid;

@Controller
public class AssemblyController {

    @Autowired
    private SqlQueryManager sqlQueryManager;

    @Autowired
    private AssemblyManager assemblyManager;

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


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

        ModelMap modelMap = modelAndView.getModelMap();

        Assembly assembly = new Assembly();

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

        return modelAndView;
    }
}