package com.ruoyi.system.controller; import java.io.IOException; import java.util.*; import java.util.stream.Collectors; import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; import com.alibaba.nacos.shaded.org.checkerframework.checker.units.qual.A; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.ruoyi.common.core.utils.StringUtils; import com.ruoyi.common.core.web.page.PageInfo; import com.ruoyi.system.domain.dto.IndexDto; import com.ruoyi.system.domain.vo.UserRoleVO; import com.ruoyi.system.api.model.*; import com.ruoyi.system.domain.SysUserRole; import com.ruoyi.system.domain.dto.AddCompanyUserDto; import com.ruoyi.system.domain.vo.CompanyUserVo; import com.ruoyi.system.domain.vo.TCompanyToUserVo; import com.ruoyi.system.service.*; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.apache.commons.lang3.ArrayUtils; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.transaction.annotation.Propagation; import org.springframework.transaction.annotation.Transactional; import org.springframework.util.CollectionUtils; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import com.ruoyi.common.core.domain.R; import com.ruoyi.common.core.utils.poi.ExcelUtil; import com.ruoyi.common.core.web.controller.BaseController; import com.ruoyi.common.core.web.domain.AjaxResult; import com.ruoyi.common.log.annotation.Log; import com.ruoyi.common.log.enums.BusinessType; import com.ruoyi.common.security.annotation.InnerAuth; import com.ruoyi.common.security.annotation.RequiresPermissions; import com.ruoyi.common.security.utils.SecurityUtils; import com.ruoyi.system.api.domain.SysDept; import com.ruoyi.system.api.domain.SysRole; import com.ruoyi.system.api.domain.SysUser; /** * 用户信息 * * @author ruoyi */ @RestController @RequestMapping("/user") @Api(tags = "用户信息") public class SysUserController extends BaseController { @Autowired private ISysUserService userService; @Autowired private ISysRoleService roleService; @Autowired private ISysDeptService deptService; @Autowired private ISysPostService postService; @Autowired private ISysPermissionService permissionService; @Autowired private ISysConfigService configService; @Autowired private ISysUserRoleService userRoleService; @Autowired private ISysUserRoleService sysUserRoleService; /** * 获取用户列表 */ @GetMapping("/list") @ApiOperation("管理员列表") public AjaxResult list(String phonenumber, int pageNumber, int pageSize) { PageInfo pageInfo = new PageInfo<>(pageNumber, pageSize); PageInfo page = userService.getList(pageInfo, phonenumber); return AjaxResult.success(page); } /** * 新增用户 */ @Log(title = "用户管理", businessType = BusinessType.INSERT) @PostMapping("/add") @ApiOperation("添加管理员") public AjaxResult add(@Validated @RequestBody SysUser user) { user.setUserName(user.getPhonenumber()); if(!org.springframework.util.StringUtils.hasLength(user.getNickName())){ user.setNickName(user.getPhonenumber()); } if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) { return error("手机号已开通账号"); } user.setCreateBy(SecurityUtils.getUsername()); user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); user.setRoleType(1); int i = userService.insertUser(user); SysUserRole sysUserRole = new SysUserRole(); sysUserRole.setRoleId(1l); sysUserRole.setUserId(user.getUserId()); int i1 = userRoleService.insertSysUserRole(sysUserRole); return AjaxResult.success(i1); } /** * 根据用户编号获取详细信息 */ @ApiOperation("管理员详情") @GetMapping(value = {"/", "/{userId}"}) public AjaxResult getInfo(@PathVariable(value = "userId", required = false) Long userId) { userService.checkUserDataScope(userId); AjaxResult ajax = AjaxResult.success(); List roles = roleService.selectRoleAll(); ajax.put("roles", SysUser.isAdmin(userId) ? roles : roles.stream().filter(r -> !r.isAdmin()).collect(Collectors.toList())); ajax.put("posts", postService.selectPostAll()); if (StringUtils.isNotNull(userId)) { SysUser sysUser = userService.selectUserById(userId); ajax.put(AjaxResult.DATA_TAG, sysUser); ajax.put("postIds", postService.selectPostListByUserId(userId)); ajax.put("roleIds", sysUser.getRoles().stream().map(SysRole::getRoleId).collect(Collectors.toList())); } return AjaxResult.success(ajax); } /** * 修改用户 */ @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()); } R admin = this.isAdmin(user.getUserId()); Integer data = admin.getData(); if(data == null || data != 1){ SysUserRole one = sysUserRoleService.getOne(new LambdaQueryWrapper().eq(SysUserRole::getUserId, user.getUserId())); one.setRoleId(user.getRoleId()); sysUserRoleService.updateSysUserRole(one); } userService.checkUserAllowed(user); userService.checkUserDataScope(user.getUserId()); SysUser sysUser = userService.getOne(Wrappers.lambdaQuery(SysUser.class) .eq(SysUser::getPhonenumber, user.getPhonenumber()) .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())); user.setPassWordUpdate(new Date()); } if (user.getPhonenumber() != null) { user.setUserName(user.getPhonenumber()); } user.setPassword(null); user.setUpdateBy(SecurityUtils.getUsername()); user.setUpdateTime(new Date()); 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)); } @Log(title = "账号管理", businessType = BusinessType.INSERT) @PostMapping("/addCompanyUser") @ApiOperation("账号管理--添加账号") @Transactional(propagation = Propagation.REQUIRES_NEW) public AjaxResult addCompanyUser(@Validated @RequestBody AddCompanyUserDto dto) { SysUser user = dto.getSysUser(); if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) { return error("手机号已存在,请重新输入"); } user.setCreateBy(SecurityUtils.getUsername()); user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); user.setRoleType(3); user.setUserName(user.getPhonenumber()); if(!org.springframework.util.StringUtils.hasLength(user.getNickName())){ user.setNickName(user.getPhonenumber()); } int i = userService.insertUser(user); if (i > 0) { // 添加role SysUserRole sysUserRole = new SysUserRole(); sysUserRole.setRoleId(dto.getRoleId()); sysUserRole.setUserId(dto.getSysUser().getUserId()); userRoleService.insertSysUserRole(sysUserRole); } return AjaxResult.success(); } @Log(title = "账号管理", businessType = BusinessType.INSERT) @PostMapping("/companyUserInfo") @ApiOperation("账号管理--账号详情") @Transactional(propagation = Propagation.REQUIRES_NEW) public AjaxResult companyUserInfo(@RequestParam Long userId) { SysUser sysUser = userService.selectUserById(userId); SysUserRole sysUserRole = userRoleService.getOne(new LambdaQueryWrapper().eq(SysUserRole::getUserId, userId)); CompanyUserVo companyUserVo = new CompanyUserVo(); companyUserVo.setRoleId(sysUserRole.getRoleId()); companyUserVo.setSysUser(sysUser); return AjaxResult.success(companyUserVo); } @Log(title = "账号管理", businessType = BusinessType.INSERT) @PostMapping("/updateCompanyUser") @ApiOperation("账号管理--修改账号") @Transactional(propagation = Propagation.REQUIRES_NEW) public AjaxResult updateCompanyUser(@Validated @RequestBody AddCompanyUserDto dto) { SysUser user = dto.getSysUser(); if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) { return error("手机号已存在,请重新输入"); } user.setUserName(user.getPhonenumber()); if(!org.springframework.util.StringUtils.hasLength(user.getNickName())){ user.setNickName(user.getPhonenumber()); } if (user.getPassword() != null && !"".equals(user.getPassword())) { user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); } user.setRoleType(1); user.setUpdateBy(SecurityUtils.getUsername()); user.setUpdateTime(new Date()); int i = userService.updateUser(user); // if (i > 0) { // 原来的role SysUserRole one = userRoleService.getOne(new LambdaQueryWrapper().eq(SysUserRole::getUserId, dto.getSysUser().getUserId())); one.setRoleId(dto.getRoleId()); //更新新的role userRoleService.updateSysUserRole(one); // } return AjaxResult.success(); } @PostMapping("/shopUserStart") @ApiOperation("账号管理--停用/启用") public AjaxResult shopUserStart(Long userId) { if (userId == null) { return AjaxResult.error("userId不能为空"); } SysUser sysUser = userService.selectUserById(userId); if (sysUser.getStatus().equals("0")) { sysUser.setStatus("1"); } else { sysUser.setStatus("0"); } return toAjax(userService.updateUser(sysUser)); } @Autowired private ISysUserRoleService iSysUserRoleService; @PostMapping("/getUserList") public R> getUserList(@RequestBody List userIds) { List list = userService.list(new LambdaQueryWrapper().in(SysUser::getUserId, userIds)); return R.ok(list); } @PostMapping("/getSysUserFromCompanyId") public R> getSysUserFromCompanyId(@RequestBody CompanySysUserReq companySysUserReq) { LambdaQueryWrapper sysUserLambdaQueryWrapper = new LambdaQueryWrapper<>(); if (companySysUserReq.getUserIds().size() == 0) { List userIds = companySysUserReq.getUserIds(); userIds.add(-1); companySysUserReq.setUserIds(userIds); } sysUserLambdaQueryWrapper.in(SysUser::getUserId, companySysUserReq.getUserIds()); if (companySysUserReq.getName() != null && !"".equals(companySysUserReq.getName())) { sysUserLambdaQueryWrapper.like(SysUser::getUserName, companySysUserReq.getName()).or().like(SysUser::getPhonenumber, companySysUserReq.getName()); } sysUserLambdaQueryWrapper.eq(SysUser::getRoleType,2); List list = userService.list(sysUserLambdaQueryWrapper); ArrayList companyUserListVos = new ArrayList<>(); for (SysUser sysUser : list) { CompanyUserListVo companyUserListVo = new CompanyUserListVo(); companyUserListVo.setId(sysUser.getUserId()); companyUserListVo.setPassword(sysUser.getPassword()); companyUserListVo.setPhoneNumber(sysUser.getPhonenumber()); companyUserListVo.setUserName(sysUser.getUserName()); companyUserListVo.setNickName(sysUser.getNickName()); companyUserListVos.add(companyUserListVo); } return R.ok(companyUserListVos); } @PostMapping("/getSysUserFromCompanyIdOne") public R> getSysUserFromCompanyIdOne(@RequestBody CompanySysUserReq companySysUserReq) { LambdaQueryWrapper sysUserLambdaQueryWrapper = new LambdaQueryWrapper<>(); sysUserLambdaQueryWrapper.eq(SysUser::getRoleType, 2); if (companySysUserReq.getName() != null && !"".equals(companySysUserReq.getName())) { sysUserLambdaQueryWrapper.like(SysUser::getPhonenumber, companySysUserReq.getName()); } List list = userService.list(sysUserLambdaQueryWrapper); ArrayList companyUserListVos = new ArrayList<>(); for (SysUser sysUser : list) { CompanyUserListVo companyUserListVo = new CompanyUserListVo(); companyUserListVo.setId(sysUser.getUserId()); companyUserListVo.setPassword(sysUser.getPassword()); companyUserListVo.setPhoneNumber(sysUser.getPhonenumber()); companyUserListVo.setUserName(sysUser.getUserName()); companyUserListVo.setNickName(sysUser.getNickName()); companyUserListVos.add(companyUserListVo); } return R.ok(companyUserListVos); } @PostMapping("/isAdmin") public R isAdmin(@RequestBody Long userId){ SysUserRole one = sysUserRoleService.getOne(new LambdaQueryWrapper().eq(SysUserRole::getUserId, userId)); Long id =one.getRoleId(); return R.ok(id.intValue()); } @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(); } } @PostMapping("/companySaveSysUser") public R> companySaveSysUser(@RequestBody CompanyAddSysUserDto dto) { List list = dto.getList(); ArrayList userIds = new ArrayList<>(); try { for (CompanyUserListVo companyUserListVo : list) { List ids = userService.getSysUserFromPhone(companyUserListVo.getPhoneNumber()); SysUser sysUser = new SysUser(); if (ids.size() > 0) { sysUser.setRoleType(2); sysUser.setObjectId(dto.getCompanyId()); if(companyUserListVo.getPassword()!=null){ sysUser.setPassword(SecurityUtils.encryptPassword(companyUserListVo.getPassword())); } sysUser.setPhonenumber(companyUserListVo.getPhoneNumber()); sysUser.setUserName(companyUserListVo.getPhoneNumber()); sysUser.setNickName(companyUserListVo.getNickName()); sysUser.setUserId(ids.get(0).longValue()); userService.updateUser(sysUser); userIds.add(ids.get(0).intValue()); } else { sysUser.setRoleType(2); sysUser.setObjectId(dto.getCompanyId()); if(companyUserListVo.getPassword()!=null){ sysUser.setPassword(SecurityUtils.encryptPassword(companyUserListVo.getPassword())); } sysUser.setPhonenumber(companyUserListVo.getPhoneNumber()); sysUser.setUserName(companyUserListVo.getPhoneNumber()); sysUser.setNickName(companyUserListVo.getNickName()); sysUser.setCreateBy(SecurityUtils.getUsername()); sysUser.setCreateTime(new Date()); userService.insertUser(sysUser); SysUserRole sysUserRole = new SysUserRole(); sysUserRole.setRoleId(20l); sysUserRole.setUserId(sysUser.getUserId()); sysUserRoleService.insertSysUserRole(sysUserRole); userIds.add(sysUser.getUserId().intValue()); } } return R.ok(userIds); } catch (Exception e) { e.printStackTrace(); return R.ok(new ArrayList<>()); } } @PostMapping("/addTRepairSysUser") public R> addTRepairSysUser(@RequestBody TRepairShopAddDto dto) { List list = dto.getList(); ArrayList userIds = new ArrayList<>(); try { for (TRepairShopAdd add : list) { List ids = userService.getSysUserFromPhone(add.getPhonenumber()); if (ids.size() > 0) { userIds.add(ids.get(0).intValue()); } else { SysUser sysUser = new SysUser(); sysUser.setPhonenumber(add.getPhonenumber()); sysUser.setCreateTime(new Date()); sysUser.setCreateBy(SecurityUtils.getUsername()); sysUser.setPassword(SecurityUtils.encryptPassword(add.getPwd())); sysUser.setRoleType(4); sysUser.setObjectId(dto.getRepairId()); sysUser.setUserName(add.getPhonenumber()); sysUser.setNickName(add.getPhonenumber()); userService.insertUser(sysUser); userIds.add(sysUser.getUserId().intValue()); } } return R.ok(userIds); } catch (Exception e) { e.printStackTrace(); return R.ok(new ArrayList<>()); } } @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 @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); } /** * 注册用户信息 */ @InnerAuth @PostMapping("/register") public R register(@RequestBody SysUser sysUser) { String username = sysUser.getUserName(); if (!("true".equals(configService.selectConfigByKey("sys.account.registerUser")))) { return R.fail("当前系统没有开启注册功能!"); } if (!userService.checkUserNameUnique(sysUser)) { return R.fail("保存用户'" + username + "'失败,注册账号已存在"); } return R.ok(userService.registerUser(sysUser)); } /** * 获取用户信息 * * @return 用户信息 */ @GetMapping("/getInfo") public AjaxResult getInfo() { SysUser user = userService.selectUserById(SecurityUtils.getUserId()); // 角色集合 Set roles = permissionService.getRolePermission(user); // 权限集合 Set permissions = permissionService.getMenuPermission(user); AjaxResult ajax = AjaxResult.success(); ajax.put("user", user); ajax.put("roles", roles); ajax.put("permissions", permissions); return ajax; } /** * 重置密码 */ @RequiresPermissions("system:user:edit") @Log(title = "用户管理", businessType = BusinessType.UPDATE) @PutMapping("/resetPwd") public AjaxResult resetPwd(@RequestBody SysUser user) { userService.checkUserAllowed(user); userService.checkUserDataScope(user.getUserId()); user.setPassword(SecurityUtils.encryptPassword(user.getPassword())); user.setUpdateBy(SecurityUtils.getUsername()); return toAjax(userService.resetPwd(user)); } /** * 状态修改 */ @RequiresPermissions("system:user:edit") @Log(title = "用户管理", businessType = BusinessType.UPDATE) @PutMapping("/changeStatus") public AjaxResult changeStatus(@RequestBody SysUser user) { userService.checkUserAllowed(user); userService.checkUserDataScope(user.getUserId()); user.setUpdateBy(SecurityUtils.getUsername()); return toAjax(userService.updateUserStatus(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 dtoList * @return */ @PostMapping("/addUserRole") public R addUserRole(@RequestBody List dtoList){ List userRoles = new ArrayList<>(); for (SysUserRoleDTO userRoleDTO : dtoList) { SysUserRole userRole = new SysUserRole(); userRole.setRoleId(userRoleDTO.getRoleId()); userRole.setUserId(userRoleDTO.getUserId()); userRoles.add(userRole); } return R.ok(sysUserRoleService.saveBatch(userRoles)); } /** * 通过手机号查询用户 * @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); } }