mescloud/src/main/java/com/jeesite/modules/mes/web/EmpUserCustomController.java

package com.jeesite.modules.mes.web;

import com.jeesite.common.config.Global;
import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.web.BaseController;
import com.jeesite.modules.sys.entity.EmpUser;
import com.jeesite.modules.sys.entity.EmployeeOffice;
import com.jeesite.modules.sys.entity.Role;
import com.jeesite.modules.sys.service.EmpUserService;
import com.jeesite.modules.sys.web.user.EmpUserController;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.List;

@Controller
@RequestMapping(value = "${adminPath}/mes/empUserCustom")
public class EmpUserCustomController extends BaseController {

    @Autowired
    private EmpUserController empUserController;

    @Autowired
    private EmpUserService empUserService;

    @ModelAttribute
    public EmpUser get(String userCode, boolean isNewRecord) {
        return empUserService.get(userCode, isNewRecord);
    }

    @RequestMapping(value = "form")
    public String form(EmpUser empUser, String op, Model model) {
        if ("copy".equals(op)) {
            String view = empUserController.form(empUser, "edit", model);
            
            EmpUser fullUser = (EmpUser) model.asMap().get("empUser");
            if (fullUser != null) {
                fullUser.setIsNewRecord(true);
                fullUser.setUserCode(null);
                
                if (fullUser.getEmployee() != null) {
                    fullUser.getEmployee().setIsNewRecord(true);
                    fullUser.getEmployee().setEmpCode(null);
                    
                    List<EmployeeOffice> officeList = fullUser.getEmployee().getEmployeeOfficeList();
                    if (officeList != null && !officeList.isEmpty()) {
                        for (EmployeeOffice eo : officeList) {
                            eo.setIsNewRecord(true);
                            eo.setId(null);
                        }
                    }
                }
                
                List<Role> roleList = fullUser.getRoleList();
                if (roleList != null && !roleList.isEmpty()) {
                    List<String> roleCodes = new ArrayList<>();
                    for (Role role : roleList) {
                        roleCodes.add(role.getRoleCode());
                    }
                    fullUser.setUserRoleString(StringUtils.join(roleCodes, ","));
                }
                
                model.addAttribute("empUser", fullUser);
                model.addAttribute("roleList", roleList);
                model.addAttribute("op", "add");
            }
            
            return "modules/sys/user/empUserForm";
        }
        
        if ("add".equals(op) || "edit".equals(op)) {
            empUserController.form(empUser, op, model);
            return "modules/sys/user/empUserForm";
        }
        
        return empUserController.form(empUser, op, model);
    }

    @RequestMapping(value = "save")
    @ResponseBody
    public String save(@Validated EmpUser empUser, String oldLoginCode, HttpServletRequest request) {
        EmpUser existUser = new EmpUser();
        existUser.setLoginCode(empUser.getLoginCode());
        List<EmpUser> uList = empUserService.findList(existUser);
        if (uList != null && !uList.isEmpty()) {
            return renderResult(Global.FALSE, "保存用户失败，登录账号已存在（不能与原账号相同），请修改登录账号！");
        }
        
        if (StringUtils.isNotBlank(empUser.getUserName())) {
            EmpUser where = new EmpUser();
            where.setUserName(empUser.getUserName());
            List<EmpUser> list = empUserService.findList(where);
            if (list != null && !list.isEmpty()) {
                return renderResult(Global.FALSE, "保存用户失败，用户昵称已存在（不能与原昵称相同），请修改用户昵称！");
            }
        }
        
        return empUserController.save(empUser, oldLoginCode, request);
    }
}