package com.xinquan.system.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.sun.org.apache.bcel.internal.generic.NEW; import com.xinquan.common.core.domain.R; import com.xinquan.common.core.utils.StringUtils; import com.xinquan.common.core.utils.page.PageDTO; import com.xinquan.common.core.utils.poi.ExcelUtil; import com.xinquan.common.core.web.controller.BaseController; import com.xinquan.common.core.web.domain.AjaxResult; import com.xinquan.common.core.web.page.PageInfo; import com.xinquan.common.log.annotation.Log; import com.xinquan.common.log.enums.BusinessType; import com.xinquan.common.security.annotation.InnerAuth; import com.xinquan.common.security.annotation.RequiresPermissions; import com.xinquan.common.security.service.TokenService; import com.xinquan.common.security.utils.SecurityUtils; import com.xinquan.system.api.domain.SysDept; import com.xinquan.system.api.domain.SysRole; import com.xinquan.system.api.domain.SysUser; import com.xinquan.system.api.domain.vo.SysUserVO; import com.xinquan.system.api.model.LoginUser; import com.xinquan.system.domain.Banner; import com.xinquan.system.domain.SysUserDept; import com.xinquan.system.domain.SysUserRole; import com.xinquan.system.domain.dto.SysUserQuery; import com.xinquan.system.domain.dto.SysUserUpdateStatusDTO; import com.xinquan.system.domain.dto.UpdatePwdDTO; import com.xinquan.system.service.*; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import java.io.IOException; import java.time.LocalDateTime; import java.util.*; import java.util.stream.Collectors; import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; import org.apache.commons.lang3.ArrayUtils; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; /** * 用户信息 * * @author ruoyi */ @Api(tags = "用户信息") @RestController @RequestMapping("/user") public class SysUserController extends BaseController { @Autowired private ISysUserService userService; @Autowired private ISysRoleService roleService; @Autowired private ISysDeptService deptService; @Autowired private TokenService tokenService; @Autowired private ISysUserRoleService sysUserRoleService; @Autowired private ISysPostService postService; @Autowired private ISysPermissionService permissionService; @Resource private ISysUserDeptService sysUserDeptService; @PostMapping("/removeByAppUserId/{id}") public AjaxResult removeByAppUserId(@PathVariable("id") Long id) { SysUser byId = userService.getById(id); userService.removeById(byId); return AjaxResult.success(); } /** * 获取用户列表 */ @ApiOperation(value = "获取用户列表") @PostMapping("/list") public AjaxResult> list(@RequestBody SysUserQuery query) { Page res = new Page<>(query.getPageCurr(), query.getPageSize()); PageInfo list = userService.pageList(query); for (SysUserVO record : list.getRecords()) { List collect1 = sysUserRoleService.lambdaQuery().eq(SysUserRole::getUserId, record.getUserId()) .list().stream().map(SysUserRole::getRoleId).collect(Collectors.toList()); List longs1 = new ArrayList<>(); for (Long l : collect1) { SysRole sysDept = roleService.selectRoleById(l); if (sysDept!=null){ longs1.add(sysDept.getRoleId()); } } record.setRole(longs1); List strings1 = new ArrayList<>(); if (record.getRole()!=null && (!record.getRole().isEmpty())){ for (Long l : collect1) { SysRole sysRole = roleService.selectRoleById(l); if (sysRole!=null){ strings1.add(sysRole.getRoleName()); } } } record.setRoleNames(strings1); List collect2 = sysUserDeptService.lambdaQuery().eq(SysUserDept::getUserId, record.getUserId()) .list().stream().map(SysUserDept::getDeptId).collect(Collectors.toList()); List longs2 = new ArrayList<>(); for (Long l : collect2) { SysDept sysDept = deptService.selectDeptById(l); if (sysDept!=null){ longs2.add(sysDept.getDeptId()); } } record.setDeptIds(longs2); List strings = new ArrayList<>(); if (record.getDeptIds()!=null && (!record.getDeptIds().isEmpty())){ for (Long deptId : record.getDeptIds()) { SysDept sysDept = deptService.selectDeptById(deptId); if (sysDept!=null){ strings.add(sysDept.getDeptName()); } } } record.setDeptNames(strings); } res.setRecords(list.getRecords()); PageDTO sysUserVOPageDTO = PageDTO.of(res, SysUserVO.class); return AjaxResult.success(sysUserVOPageDTO); } @ApiOperation(value = "新增用户管理") @PostMapping("/add") public AjaxResult add(@Validated @RequestBody SysUser user) { if (!userService.checkUserNameUnique(user)) { return error("新增用户'" + user.getUserName() + "'失败,登录账号已存在"); } else if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) { return error("新增用户'" + user.getUserName() + "'失败,手机号码已存在"); } user.setUserType("00"); user.setPassword("123456"); user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); userService.insertUser(user); return AjaxResult.success(); } @ApiOperation(value = "修改用户管理") @PostMapping("/edit") public AjaxResult edit(@Validated @RequestBody SysUser user) { // userService.checkUserAllowed(user); // userService.checkUserDataScope(user.getUserId()); if (!userService.checkUserNameUnique(user)) { return error("修改用户'" + user.getUserName() + "'失败,登录账号已存在"); } Long userid = tokenService.getLoginUser().getUserid(); SysUser sysUser = userService.selectUserById(userid); user.setUpdateBy(sysUser.getNickName()+"("+sysUser.getUserName()+")"); user.setHandleName(sysUser.getNickName()+"("+sysUser.getUserName()+")"); user.setUpdateTime(LocalDateTime.now()); if (user.getPassword()!=null){ user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); return AjaxResult.success(userService.resetPwd(user)); } return AjaxResult.success(userService.updateUser(user)); } @ApiOperation(value = "禁用启用") @PutMapping("/changeStatus") public AjaxResult changeStatus(@RequestBody SysUserUpdateStatusDTO dto) { SysUser user = new SysUser(); user.setUserId(dto.getUserId()); user.setStatus(String.valueOf(dto.getStatus())); user.setReason(dto.getReason()); user.setDisableTime(LocalDateTime.now()); Long userid = tokenService.getLoginUser().getUserid(); SysUser sysUser = userService.selectUserById(userid); user.setUpdateBy(sysUser.getNickName()+"("+sysUser.getUserName()+")"); user.setHandleName(sysUser.getNickName()+"("+sysUser.getUserName()+")"); return AjaxResult.success(userService.updateUserStatus(user)); } /** * 获取当前用户信息 */ @InnerAuth @GetMapping("/info/{username}") public R info(@PathVariable("username") String username) { SysUser sysUser = userService.selectUserByUserName(username); if (StringUtils.isNull(sysUser)) { return R.fail("用户名或密码错误"); } // 角色集合 Set roles = permissionService.getRolePermission(sysUser); // 权限集合 Set permissions = permissionService.getMenuPermission(sysUser); LoginUser sysUserVo = new LoginUser(); sysUserVo.setSysUser(sysUser); sysUserVo.setRoles(roles); sysUserVo.setPermissions(permissions); return R.ok(sysUserVo); } // @ApiOperation(value = "获取用户列表-不分页") // @PostMapping("/listNotPage") // public AjaxResult> listNotPage() // { // List list = userService.selectList(); // return AjaxResult.success(list); // } // // /** // * 获取用户黑名单列表 // */ //// @ApiOperation(value = "获取用户黑名单列表") //// @PostMapping("/blacklist") //// public AjaxResult blacklist(@RequestBody SysUserQuery query) //// { //// startPage(query.getPageNum(), query.getPageSize()); //// List list = userService.selectBlackPageList(query); //// return AjaxResult.success(getDataTable(list)); //// } // // /** // * 获取用户详情 // */ // @ApiOperation(value = "获取用户详情") // @GetMapping("/getDetail") // public AjaxResult getDetail(@RequestParam Long userId) // { // SysUser sysUser = userService.selectUserById(userId); // SysUserVO sysUserVO = new SysUserVO(); // BeanUtils.copyProperties(sysUser,sysUserVO); // // return AjaxResult.success(sysUser); // } // // // /** // * 获取用户数量统计 // */ // @ApiOperation(value = "获取用户数量统计") // @PostMapping("/getUserCount") // public AjaxResult> getUserCount() // { // Map map = new HashMap<>(); // // Integer userCountSum = userService.selectCount(null); // Integer normalCount = userService.selectCount(0);// 正常 // Integer stopCount = userService.selectCount(1);// 停用 // // map.put("all",userCountSum); // map.put("normal",normalCount); // map.put("stop",stopCount); // // return AjaxResult.success(map); // } // // /** // * 移除黑名单 // */ //// @GetMapping("/removeBlackList") //// public AjaxResult removeBlackList(@RequestParam String ids) //// { //// String[] split = ids.split(","); //// List id = new ArrayList<>(); //// for (String s : split) { //// id.add(Long.valueOf(s)); //// } //// userService.updateUserIfBlack(id); //// return AjaxResult.success(); //// } // // //// @Log(title = "用户管理", businessType = BusinessType.EXPORT) ////// // @PreAuthorize("@ss.hasPermi('system:user:export')") //// @PostMapping("/export") //// public void export(HttpServletResponse response, SysUser user) //// { //// List list = userService.selectUserList(user); //// ExcelUtil util = new ExcelUtil(SysUser.class); //// util.exportExcel(response, list, "用户数据"); //// } // //// @Log(title = "用户管理", businessType = BusinessType.IMPORT) ////// // @PreAuthorize("@ss.hasPermi('system:user:import')") //// @PostMapping("/importData") //// public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception //// { //// ExcelUtil util = new ExcelUtil(SysUser.class); //// List userList = util.importExcel(file.getInputStream()); //// String operName = getUsername(); //// String message = userService.importUser(userList, updateSupport, operName); //// return AjaxResult.success(message); //// } // //// @PostMapping("/importTemplate") //// public void importTemplate(HttpServletResponse response) //// { //// ExcelUtil util = new ExcelUtil(SysUser.class); //// util.importTemplateExcel(response, "用户数据"); //// } // // // /** // * 新增用户 // */ // // @PreAuthorize("@ss.hasPermi('system:user:add')") // @ApiOperation(value = "新增用户管理") // @Log(title = "用户信息-新增用户", businessType = BusinessType.INSERT) // @PostMapping("/add") // public AjaxResult add(@Validated @RequestBody SysUser user) // { // if (!userService.checkUserNameUnique(user)) // { // return error("新增用户'" + user.getUserName() + "'失败,登录账号已存在"); // } // else if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) // { // return error("新增用户'" + user.getUserName() + "'失败,手机号码已存在"); // } // user.setPassword("123456"); // user.setRePassword("123456"); // user.setCreateBy(getUsername()); // user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); // userService.insertUser(user); // return AjaxResult.success(); // } // // /** // * 修改用户 // */ // // @PreAuthorize("@ss.hasPermi('system:user:edit')") // @ApiOperation(value = "修改用户管理") // @Log(title = "用户信息-修改用户", businessType = BusinessType.UPDATE) // @PostMapping("/edit") // public AjaxResult edit(@Validated @RequestBody SysUser user) // { //// userService.checkUserAllowed(user); //// userService.checkUserDataScope(user.getUserId()); // if (!userService.checkUserNameUnique(user)) // { // return error("修改用户'" + user.getUserName() + "'失败,登录账号已存在"); // } // else if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) // { // return error("修改用户'" + user.getUserName() + "'失败,手机号码已存在"); // } // // user.setUpdateBy(getUsername()); //// user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); // return AjaxResult.success(userService.updateUser(user)); // } // // /** // * 删除用户 // */ // // @PreAuthorize("@ss.hasPermi('system:user:remove')") // @ApiOperation(value = "批量删除用户") // @Log(title = "用户信息-批量删除用户", businessType = BusinessType.DELETE) // @DeleteMapping("/deleteById/{ids}") // public AjaxResult remove(@PathVariable String ids) // { // String[] split = ids.split(","); // List userIds = new ArrayList<>(); // for (String s : split) { // userIds.add(Long.valueOf(s)); // } // if (userIds.contains(getUserId())) // { // return error("当前用户不能删除"); // } // return AjaxResult.success(userService.deleteUserByIds(userIds)); // } // // /** // * 重置密码 // */ // // @PreAuthorize("@ss.hasPermi('system:user:resetPwd')") // @ApiOperation(value = "重置密码") // @Log(title = "用户信息-重置密码", businessType = BusinessType.UPDATE) // @PostMapping("/resetPwd") // public AjaxResult resetPwd(@RequestBody SysUser user) // { // userService.checkUserAllowed(user); //// userService.checkUserDataScope(user.getUserId()); // user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); // user.setUpdateBy(getUsername()); // userService.resetPwd(user); // return AjaxResult.success(); // } // // /** // * 修改密码 // */ // // @PreAuthorize("@ss.hasPermi('system:user:resetPwd')") // @ApiOperation(value = "修改密码") // @Log(title = "用户信息-修改密码", businessType = BusinessType.UPDATE) // @PostMapping("/updatePwd") // public AjaxResult updatePwd(@RequestBody UpdatePwdDTO dto) // { // SysUser user = userService.selectUserByUserName(dto.getAccount()); // if(Objects.isNull(user)){ // return AjaxResult.error("未查询到该账号"); // } // userService.checkUserAllowed(user); // // 校验密码跟原密码是否匹配 // if (!SecurityUtils.matchesPassword(dto.getOldPassword(), user.getPassword())) { // throw new BadCredentialsException("输入原密码不正确"); // } // if (!dto.getPassword().equals(dto.getConfirmPassword())) { // throw new BadCredentialsException("两次输入密码不一致"); // } //// userService.checkUserDataScope(user.getUserId()); // user.setPassword(SecurityUtils.encryptPassword(dto.getPassword())); // user.setUpdateBy(getUsername()); // return AjaxResult.success(userService.resetPwd(user)); // } // // /** // * 状态修改 // */ // @ApiOperation(value = "状态修改") // @Log(title = "用户信息-状态修改", businessType = BusinessType.UPDATE) // @PutMapping("/changeStatus") // public AjaxResult changeStatus(@RequestBody SysUserUpdateStatusDTO dto) // { // SysUser user = new SysUser(); // user.setUserId(dto.getUserId()); // user.setStatus(String.valueOf(dto.getStatus())); // user.setDisableRemark(dto.getDisableRemark()); // user.setUpdateBy(getUsername()); // return AjaxResult.success(userService.updateUserStatus(user)); // } // // /** // * 根据用户编号获取授权角色 // */ // // @PreAuthorize("@ss.hasPermi('system:user:query')") // @GetMapping("/authRole/{userId}") // public AjaxResult authRole(@PathVariable("userId") Long userId) // { // AjaxResult ajax = AjaxResult.success(); // SysUser user = userService.selectUserById(userId); // List roles = roleService.selectRolesByUserId(userId); // ajax.put("user", user); // ajax.put("roles", SysUser.isAdmin(userId) ? roles : roles.stream().filter(r -> !r.isAdmin()).collect(Collectors.toList())); // return ajax; // } // // /** // * 用户授权角色 // */ // // @PreAuthorize("@ss.hasPermi('system:user:edit')") // @Log(title = "用户管理", businessType = BusinessType.GRANT) // @PutMapping("/authRole") // public AjaxResult insertAuthRole(Long userId, Long[] roleIds) // { // userService.checkUserDataScope(userId); // userService.insertUserAuth(userId, roleIds); // return AjaxResult.success(); // } // // /** // * 获取部门树列表 // */ // // @PreAuthorize("@ss.hasPermi('system:user:list')") // @GetMapping("/deptTree") // public AjaxResult deptTree(SysDept dept) // { // return AjaxResult.success(deptService.selectDeptTreeList(dept)); // } /** * 远程调用 根据系统用户id查询用户信息 * @return */ @PostMapping("/getSysUserById/{id}") public R getSysUserById(@PathVariable("id")String id) { SysUser byId = userService.getById(id); return R.ok(byId); } /** * 修改用户 */ // @Log(title = "用户管理", businessType = BusinessType.UPDATE) // @PutMapping("/update") // @ApiOperation("编辑管理员") // public AjaxResult edit(@Validated @RequestBody SysUser user) { // // user.setUserName(user.getPhonenumber()); // if(!org.springframework.util.StringUtils.hasLength(user.getNickName())){ // user.setNickName(user.getPhonenumber()); // } // userService.checkUserAllowed(user); // userService.checkUserDataScope(user.getUserId()); // SysUser sysUser = userService.getOne(Wrappers.lambdaQuery(SysUser.class) // .eq(SysUser::getUserName, user.getUserName()) // .eq(SysUser::getDelFlag,0) // .last("LIMIT 1")); // // if (StringUtils.isNotEmpty(user.getPhonenumber()) && (Objects.nonNull(sysUser) && !user.getUserId().equals(sysUser.getUserId()) )) { // return error("手机号已开通账号"); // } // user.setUpdateBy(SecurityUtils.getUsername()); // if (user.getPassword() != null && !"".equals(user.getPassword())) { // user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); // } // if (user.getPhonenumber() != null) { // user.setUserName(user.getPhonenumber()); // } // user.setPassword(null); // // return toAjax(userService.updateUser(user)); // } /** * 删除用户 */ @Log(title = "用户管理", businessType = BusinessType.DELETE) @DeleteMapping("/{userIds}") @ApiOperation("删除管理员") public AjaxResult remove(@PathVariable Long[] userIds) { if (ArrayUtils.contains(userIds, SecurityUtils.getUserId())) { return error("当前用户不能删除"); } return toAjax(userService.deleteUserByIds(userIds)); } @PostMapping("/getUserList") public R> getUserList(@RequestBody List userIds) { List list = userService.list(new LambdaQueryWrapper().in(SysUser::getUserId, userIds)); return R.ok(list); } @PostMapping("/getSysUser") public R getSysUser(@RequestBody Long userId) { try { SysUser sysUser = userService.selectUserById(userId); return R.ok(sysUser); } catch (Exception e) { e.printStackTrace(); return R.ok(); } } @PostMapping("/updateSysUser") public R updateSysUser(@RequestBody SysUser sysUser) { try { // sysUser.setUpdateBy(SecurityUtils.getUsername()); // sysUser.setUpdateTime(new Date()); userService.updateUser(sysUser); return R.ok(true); } catch (Exception e) { e.printStackTrace(); return R.ok(); } } @Log(title = "用户管理", businessType = BusinessType.EXPORT) @RequiresPermissions("system:user:export") @PostMapping("/export") public void export(HttpServletResponse response, SysUser user) { List list = userService.selectUserList(user); ExcelUtil util = new ExcelUtil(SysUser.class); util.exportExcel(response, list, "用户数据"); } @PostMapping("/importTemplate") public void importTemplate(HttpServletResponse response) throws IOException { ExcelUtil util = new ExcelUtil(SysUser.class); util.importTemplateExcel(response, "用户数据"); } @Log(title = "用户管理", businessType = BusinessType.IMPORT) @RequiresPermissions("system:user:import") @PostMapping("/importData") public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception { ExcelUtil util = new ExcelUtil(SysUser.class); List userList = util.importExcel(file.getInputStream()); String operName = SecurityUtils.getUsername(); String message = userService.importUser(userList, updateSupport, operName); return success(message); } /** * 获取当前用户信息 */ /** * 注册用户信息 */ @InnerAuth @PostMapping("/app-register") public R appRegister(@RequestBody SysUser sysUser) { String username = sysUser.getUserName(); //添加用户 userService.registerUser(sysUser); return R.ok(sysUser); } /** * 重置密码 */ @PostMapping("/resetPwd") public AjaxResult resetPwd(@RequestBody SysUser user) { userService.checkUserAllowed(user); user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); user.setUpdateBy(SecurityUtils.getUsername()); return toAjax(userService.resetPwd(user)); } /** * 根据用户编号获取授权角色 */ @RequiresPermissions("system:user:query") @GetMapping("/authRole/{userId}") public AjaxResult authRole(@PathVariable("userId") Long userId) { AjaxResult ajax = AjaxResult.success(); SysUser user = userService.selectUserById(userId); List roles = roleService.selectRolesByUserId(userId); ajax.put("user", user); ajax.put("roles", SysUser.isAdmin(userId) ? roles : roles.stream().filter(r -> !r.isAdmin()).collect(Collectors.toList())); return ajax; } /** * 用户授权角色 */ @RequiresPermissions("system:user:edit") @Log(title = "用户管理", businessType = BusinessType.GRANT) @PutMapping("/authRole") public AjaxResult insertAuthRole(Long userId, Long[] roleIds) { userService.checkUserDataScope(userId); userService.insertUserAuth(userId, roleIds); return success(); } /** * 获取部门树列表 */ @RequiresPermissions("system:user:list") @GetMapping("/deptTree") public AjaxResult deptTree(SysDept dept) { return success(deptService.selectDeptTreeList(dept)); } @PostMapping("/updateUser") public R updateUser(@RequestBody SysUser sysUser){ return R.ok(userService.updateById(sysUser)); } /** * 通过用户id查询角色对象 * @param userId * @return */ @GetMapping("/queryRoleByUserId/{userId}") public R queryRoleByUserId(@PathVariable("userId") Long userId){ SysUserRole one = sysUserRoleService.getOne(Wrappers.lambdaQuery(SysUserRole.class) .eq(SysUserRole::getUserId, userId) .last("LIMIT 1")); System.out.println("通过用户id查询角色对象"+one); if(Objects.nonNull(one)){ SysRole byId = roleService.getOne(Wrappers.lambdaQuery(SysRole.class) .eq(SysRole::getRoleId,one.getRoleId()) .last("LIMIT 1")); return R.ok(byId); } return R.ok(); } @GetMapping("/queryRoleByRoleId/{roleId}") public R queryRoleByRoleId(@PathVariable("roleId") Long roleId){ return R.ok( roleService.getOne(Wrappers.lambdaQuery(SysRole.class) .eq(SysRole::getRoleId, roleId) .last("LIMIT 1"))); } /** * 通过手机号集合查询用户 * @param phoneList * @return */ @PostMapping("/queryUserByPhoneList") public R> queryUserByPhoneList(@RequestBody List phoneList){ List list = userService.list(Wrappers.lambdaQuery(SysUser.class) .in(SysUser::getPhonenumber, phoneList) .ne(SysUser::getDelFlag,2)); return R.ok(list); } /** * 通过手机号查询用户 * @param phone * @return */ @PostMapping("/queryUserByPhone") public R queryUserByPhone(@RequestBody String phone){ SysUser user = userService.getOne(Wrappers.lambdaQuery(SysUser.class) .eq(SysUser::getPhonenumber, phone) .last("LIMIT 1")); return R.ok(user); } /** * 通过账号查询用户 * @param userName * @return */ @PostMapping("/queryUserByUserName") public R queryUserByUserName(@RequestBody String userName){ SysUser user = userService.getOne(Wrappers.lambdaQuery(SysUser.class) .eq(SysUser::getUserName, userName) .last("LIMIT 1")); return R.ok(user); } }