aswi2020vldc-gitlab/src/main/java/vldc/aswi/web/controller/IndexController.java @ 509b3231

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.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;
import vldc.aswi.domain.Assembly;
import vldc.aswi.domain.Role;
import vldc.aswi.domain.parameter.Parameter;
import vldc.aswi.service.AssemblyManager;
import vldc.aswi.service.ConfigurationManager;
import vldc.aswi.service.RoleManager;
import vldc.aswi.service.SqlQueryManager;
import vldc.aswi.utils.AuthControl;

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

/**
 * Controller for index page
 */
@Controller
public class IndexController extends BasicController {

	/**
	 * Length when attribute string generation stops after its exceeded
	 */
	private static final int ATTRIBUTES_LENGTH = 50;

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

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

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

	/**
	 * Autowired role manager
	 */
	@Autowired
	private RoleManager roleManager;

	/**
	 * Shows index page
	 * @return modelAndView with index page
	 */
	@GetMapping("/")
	public ModelAndView index(@ModelAttribute("successText") String successMsg, @ModelAttribute("errorText") String errorMsg) {
		ModelAndView modelAndView = new ModelAndView("index");

		ModelMap modelMap = modelAndView.getModelMap();

		String roleName = AuthControl.getRoleName();

		if (roleName == null) {
			// TODO: 04.05.2020 error message, user not authenticated
		}

		Role role = roleManager.getRole(roleName);

		List<Assembly> assemblies;
		if (role.getName().equals("Administrátor")) {
			assemblies = assemblyManager.getAssembliesOrdered();
		}
		else {
			assemblies = assemblyManager.getAssembliesWithRoleOrdered(role);
		}

		modelMap.addAttribute("assemblies", assemblies);
		modelMap.addAttribute("attributes", createAttributesString(assemblies));
		modelMap.addAttribute("configurations", configurationManager.getConfigurations());

		modelMap.addAttribute(assemblySuccessName, successMsg);
		modelMap.addAttribute(assemblyErrorName, errorMsg);


		return modelAndView;
	}

	@PostMapping("/")
	public ModelAndView indexPost(@Valid @ModelAttribute("assemblyID") String id) {
		ModelAndView modelAndView = new ModelAndView();
		modelAndView.setViewName("redirect:/assembly");

		return modelAndView;
	}

	/**
	 * Generates list of strings with attributes for each assembly
	 * @param assemblies list of assemblies
	 * @return list of strings with attributes
	 */
	private List<String> createAttributesString(List<Assembly> assemblies) {
		List<String> attributes = new ArrayList<>();
		for(Assembly assembly : assemblies) {
			int i = 0;
			StringBuilder stringBuilder = new StringBuilder();
			for(Parameter parameter : assembly.getParameters()) {
				stringBuilder.append(parameter.getName());
				i++;
				if (i == assembly.getParameters().size()) {
					break;
				}
				else if (stringBuilder.toString().length() > ATTRIBUTES_LENGTH) {
					stringBuilder.append(",...");
					break;
				}
				else {
					stringBuilder.append(", ");
				}
			}
			attributes.add(stringBuilder.toString());
		}

		return attributes;
	}

	@ResponseBody
	@RequestMapping("/saveOrder")
	public String saveOrder(@RequestBody String data) {
		assemblyManager.updateAssemblyOrder(data);

		return "Pořadí sestav bylo uloženo";
	}

}