package com.rzyc.controller; import com.common.utils.model.Code; import com.common.utils.model.Message; import com.alibaba.fastjson.JSONArray; import com.common.utils.StringUtils; import com.common.utils.encryption.PasswdFactory; import com.common.utils.jwt.JwtUtil; import com.common.utils.model.SingleResult; import com.rzyc.bean.user.dto.LoginDto; import com.rzyc.model.ent.EntUser; import com.rzyc.service.PcBusinessService; import com.rzyc.service.UserLoginService; import com.rzyc.bean.user.dto.WeChartLoginDto; import com.rzyc.model.ent.SysEnterprise; import com.rzyc.model.user.SysUser; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import javax.validation.Valid; import java.util.List; import java.util.Objects; /** * 个人中心系统 * @author dong * @date 2022-09-16 14:19 * @Version V1.0 */ @Api(tags = "个人中心系统") @CrossOrigin("*") @RequestMapping("personal") @RestController @Validated public class PersonalController extends BaseController{ @Autowired UserLoginService userLoginService; @Autowired PcBusinessService pcBusinessService; /** * 用户登录 * @version v1.0 * @author dong * @date 2022/9/16 14:21 */ @ApiOperation(value = "用户登录", notes = "用户登录") @PostMapping(value = "/login") public SingleResult login(@Valid LoginDto loginDto)throws Exception{ SingleResult result = new SingleResult<>(); String generateCode = request.getSession().getAttribute(constantsConfigure.getGenerateCodeKey())+""; //验证码只能使用一次 request.getSession().removeAttribute(constantsConfigure.getGenerateCodeKey()); if(loginDto.getGenerateCode().equals(generateCode)) { String loginResult = userLoginService.login(loginDto.sysusername, loginDto.getSyspassword()); if (Objects.isNull(loginResult)) { result.setCode(Code.PASSWORD_OR_ACCOUNT_ERROR.getCode()); result.setMessage(Message.PASSWORD_OR_ACCOUNT_ERROR); } else { System.out.println("登陆成功"); result.setData(loginResult); } }else { result.setCode(Code.CODE_ERROT.getCode()); result.setMessage(Message.CODE_ERROT); } return result; } /** * 企业登录 * @param loginDto * @return * @throws Exception */ @ApiOperation(value = "企业登录", notes = "企业登录") @PostMapping(value = "/entlogin") @ResponseBody public SingleResult applogin(@Valid WeChartLoginDto loginDto)throws Exception{ SingleResult result = new SingleResult<>(); System.out.println("loginDto -> "+JSONArray.toJSONString(loginDto)); String sysusername = loginDto.getSysusername(); String syspassword = loginDto.getSyspassword(); System.out.println("sessionid -> "+request.getSession().getId()); SysUser sysUser = sysUserMapper.findBySysUserName(sysusername); if(null != sysUser && StringUtils.isNotBlank(sysUser.getUsertype())){ if("企业用户".equals(sysUser)){ //如果企业被禁用 直接返回登录失败 SysEnterprise sysEnterprise = sysEnterpriseMapper.selectByPrimaryKey(sysUser.getSysunitorentid()); if(null == sysEnterprise){ result.setCode(Code.TOKEN_EXPIRE.getCode()); result.setMessage(Message.TOKEN_EXPIRE); return result; } } String ps = PasswdFactory.encryptPasswd(sysUser.getSysuserid(), sysusername, syspassword); if(sysUser.getSyspassword().equals(ps) || "guest".equals(sysUser.getUsertype())){ sysUser.setSyspassword(""); sysUser.setSyspassword(""); //获取职务 sysUser = getUserDuty(sysUser); // if(StringUtils.isNotBlank()) //通过角色判断是否为安办 或者 部门管理员 if(StringUtils.isNotBlank(sysUser.getUserroles())){ Integer userRole = this.getUserRole(sysUser.getUserroles()); sysUser.setUserRole(userRole); } //用户权限 userAuth(sysUser); this.addLogAuth(sysUser.getSysuserid(),"登录","成功",""); String userToken = JwtUtil.createToken(sysUser.getSysuserid()); sysUser.setUserToken(userToken); result.setData(sysUser); }else{ this.addLogAuth(sysUser.getSysuserid(),"登录","失败",""); result.setCode(Code.PASSWORD_ERROR.getCode()); result.setMessage(Message.PASSWORD_ERROR); } }else{ result.setCode(Code.PASSWORD_ERROR.getCode()); result.setMessage(Message.PASSWORD_ERROR); } return result; } /** * 企业用户组织树 * @param enterpriseId 企业id * @param postId 企业用户id * @return 企业用户树 * @throws Exception */ @ApiOperation(value = "企业用户组织树", notes = "企业用户组织树") @ApiImplicitParams({ @ApiImplicitParam(name = "enterpriseId", value = "公司id", required = true, dataType = "string"), @ApiImplicitParam(name = "postId", value = "企业用户岗位id",required = false, dataType = "string"), }) @PostMapping(value = "/entUserTree") @PreAuthorize("hasRole('ADMIN')") @ResponseBody public SingleResult> entUserTree(String enterpriseId, String postId)throws Exception{ return pcBusinessService.entUserTree(enterpriseId,postId); } }