alzq-onest-system/src/main/java/com/ruoyi/system/controller/SysProfileController.java

package com.ruoyi.system.controller;

import com.ruoyi.common.core.constant.UserConstants;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.utils.PassUtil;
import com.ruoyi.common.core.utils.StringUtils;
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.service.TokenService;
import com.ruoyi.common.security.utils.SecurityUtils;
import com.ruoyi.system.api.RemoteFileService;
import com.ruoyi.system.api.domain.SysFile;
import com.ruoyi.system.api.domain.SysUser;
import com.ruoyi.system.api.model.LoginUser;
import com.ruoyi.system.domain.dto.AppResetPwd;
import com.ruoyi.system.service.ISysConfigService;
import com.ruoyi.system.service.ISysDeptService;
import com.ruoyi.system.service.ISysUserService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.util.Arrays;

/**
 * 个人信息 业务处理
 *
 * @author ruoyi
 */
@Api(tags = "个人信息控制器")
@RestController
@RequestMapping("/user/profile")
public class SysProfileController extends BaseController {
    private final ISysUserService userService;

    private final TokenService tokenService;


    private final RemoteFileService remoteFileService;

    @Resource
    private ISysDeptService deptService;

    @Resource
    private ISysConfigService configService;

    public SysProfileController(ISysUserService userService, TokenService tokenService, RemoteFileService remoteFileService) {
        this.userService = userService;
        this.tokenService = tokenService;
        this.remoteFileService = remoteFileService;
    }

    /**
     * 获取个人信息
     */
    @ApiOperation(value = "获取用户个人信息", notes = "获取用户个人信息")
    @GetMapping
    public AjaxResult profile() {
        Long[] deptIds = {314391L, 314392L, 314393L, 314394L};
        String username = SecurityUtils.getUsername();
        Long deptId = SecurityUtils.getLoginUser().getSysUser().getDeptId();
        SysUser user = userService.selectUserByUserName(username);
        /**图片前缀**/
        String uri = configService.selectConfigByKey("fileUrl");
        if (StringUtils.isNotEmpty(user.getAvatar())) {
            user.setAvatar(uri + user.getAvatar());
        }
        AjaxResult ajax = AjaxResult.success(user);
        ajax.put("roleGroup", userService.selectUserRoleGroup(username));
        ajax.put("postGroup", userService.selectUserPostGroup(username));
        boolean isItSelf = Arrays.asList(deptIds).contains(deptId);
        if (isItSelf) {
            ajax.put("mapDeptId", deptId);
        } else if (deptService.thisDeptIdIsParentIdChild(deptId, 314391L) > 0) {//敖勒召其镇
            ajax.put("mapDeptId", 314391L);
        } else if (deptService.thisDeptIdIsParentIdChild(deptId, 314392L) > 0) {//城川镇
            ajax.put("mapDeptId", 314392L);
        } else if (deptService.thisDeptIdIsParentIdChild(deptId, 314393L) > 0) {//昂素镇
            ajax.put("mapDeptId", 314393L);
        } else if (deptService.thisDeptIdIsParentIdChild(deptId, 314394L) > 0) {//上海庙镇
            ajax.put("mapDeptId", 314394L);
        } {//鄂托克前旗
            ajax.put("mapDeptId", 365L);
        }
        return ajax;
    }

    /**
     * 修改用户
     */
    @ApiOperation(value = "修改用户个人信息", notes = "修改用户个人信息")
    @Log(title = "个人信息", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult updateProfile(@RequestBody SysUser user) {
        //截取头像
        if (StringUtils.isNotEmpty(user.getAvatar())) {
            int index = user.getAvatar().lastIndexOf("group");
            if (index != -1) {
                user.setAvatar(user.getAvatar().substring(index));
            }
        }
        LoginUser loginUser = SecurityUtils.getLoginUser();
        SysUser sysUser = loginUser.getSysUser();
        user.setUserName(sysUser.getUserName());
        if (StringUtils.isNotEmpty(user.getPhonenumber())
                && UserConstants.NOT_UNIQUE.equals(userService.checkPhoneUnique(user))) {
            return AjaxResult.error("修改用户'" + user.getUserName() + "'失败，手机号码已存在");
        } else if (StringUtils.isNotEmpty(user.getEmail())
                && UserConstants.NOT_UNIQUE.equals(userService.checkEmailUnique(user))) {
            return AjaxResult.error("修改用户'" + user.getUserName() + "'失败，邮箱账号已存在");
        }
        user.setUserId(sysUser.getUserId());
        user.setPassword(null);
        if (userService.updateUserProfile(user) > 0) {
            // 更新缓存用户信息
            loginUser.getSysUser().setNickName(user.getNickName());
            loginUser.getSysUser().setPhonenumber(user.getPhonenumber());
            loginUser.getSysUser().setEmail(user.getEmail());
            loginUser.getSysUser().setSex(user.getSex());
            tokenService.setLoginUser(loginUser);
            return AjaxResult.success();
        }
        return AjaxResult.error("修改个人信息异常，请联系管理员");
    }

    /**
     * 重置密码
     */
    @ApiOperation(value = "重置密码", notes = "重置密码")
    @Log(title = "个人信息", businessType = BusinessType.UPDATE)
    @PutMapping("/updatePwd")
    public AjaxResult updatePwd(String oldPassword, String newPassword) {
        String username = SecurityUtils.getUsername();
        SysUser user = userService.selectUserByUserName(username);
        String password = user.getPassword();
        if (!SecurityUtils.matchesPassword(oldPassword, password)) {
            return AjaxResult.error("修改密码失败，旧密码错误");
        }
        if (SecurityUtils.matchesPassword(newPassword, password)) {
            return AjaxResult.error("新密码不能与旧密码相同");
        }
        if (PassUtil.isStringPwd(newPassword).equals("0")) {
            return AjaxResult.error("密码必须大于等于8位，包含特殊字符，数字，大小写字母");
        }
        if (userService.resetUserPwd(username, SecurityUtils.encryptPassword(newPassword)) > 0) {
            // 更新缓存用户密码
            LoginUser loginUser = SecurityUtils.getLoginUser();
            loginUser.getSysUser().setPassword(SecurityUtils.encryptPassword(newPassword));
            tokenService.setLoginUser(loginUser);
            return AjaxResult.success();
        }
        return AjaxResult.error("修改密码异常，请联系管理员");
    }

    /**
     * 头像上传
     */
    @ApiOperation(value = "头像上传", notes = "头像上传")
    @Log(title = "用户头像", businessType = BusinessType.UPDATE)
    @PostMapping("/avatar")
    public AjaxResult avatar(@RequestParam("avatarfile") MultipartFile file) {
        if (!file.isEmpty()) {
            LoginUser loginUser = SecurityUtils.getLoginUser();
            R<SysFile> fileResult = remoteFileService.upload(file);
            if (StringUtils.isNull(fileResult) || StringUtils.isNull(fileResult.getData())) {
                return AjaxResult.error("文件服务异常，请联系管理员");
            }
            String url = fileResult.getData().getUrl();
            if (userService.updateUserAvatar(loginUser.getUsername(), url)) {
                AjaxResult ajax = AjaxResult.success();
                /**图片前缀**/
                String uri = configService.selectConfigByKey("fileUrl");
                ajax.put("imgUrl", uri + url);
                // 更新缓存用户头像
                loginUser.getSysUser().setAvatar(url);
                tokenService.setLoginUser(loginUser);
                return ajax;
            }
        }
        return AjaxResult.error("上传图片异常，请联系管理员");
    }


    /**
     * app重置密码
     */
    @ApiOperation(value = "app重置密码", notes = "app重置密码")
    @Log(title = "个人信息", businessType = BusinessType.UPDATE)
    @PutMapping("/updateAppPwd")
    public AjaxResult updateAppPwd(@RequestBody AppResetPwd appResetPwd) {
        SysUser user = userService.selectUserByUserName(appResetPwd.getUn());
        String password = user.getPassword();
        if (!SecurityUtils.matchesPassword(appResetPwd.getOp(), password)) {
            return AjaxResult.error("修改密码失败，旧密码错误");
        }
        if (SecurityUtils.matchesPassword(appResetPwd.getNp(), password)) {
            return AjaxResult.error("新密码不能与旧密码相同");
        }
        if (PassUtil.isStringPwd(appResetPwd.getNp()).equals("0")) {
            return AjaxResult.error("密码必须大于等于8位，包含特殊字符，数字，大小写字母");
        }
        if (userService.resetUserPwd(appResetPwd.getUn(), SecurityUtils.encryptPassword(appResetPwd.getNp())) > 0) {
            // 更新缓存用户密码
            LoginUser loginUser = SecurityUtils.getLoginUser();
            loginUser.getSysUser().setPassword(SecurityUtils.encryptPassword(appResetPwd.getNp()));
            tokenService.setLoginUser(loginUser);
            return AjaxResult.success();
        }
        return AjaxResult.error("修改密码异常，请联系管理员");
    }

}