再生云
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
mitao
2025-01-21 e02a3eee8217ae6bff5890d1ece6ee1f342e03dd 
 ruoyi-auth/src/main/java/com/ruoyi/auth/controller/TokenController.java


@@ -1,46 +1,49 @@


package com.ruoyi.auth.controller;




import javax.servlet.http.HttpServletRequest;




import com.ruoyi.auth.form.LoginBody;


import com.ruoyi.auth.form.MgtPasswordDTO;


import com.ruoyi.auth.service.SysLoginService;


import com.ruoyi.auth.utils.SmsUtils;


import com.ruoyi.common.core.constant.CacheConstants;


import com.ruoyi.common.core.constant.Constants;


import com.ruoyi.common.core.constant.SecurityConstants;


import com.ruoyi.common.core.utils.uuid.IdUtils;


import com.ruoyi.common.redis.service.RedisService;


import com.ruoyi.company.api.domain.User;


import com.ruoyi.company.api.model.RegisterUser;


import com.ruoyi.system.api.model.AppUser;


import io.swagger.v3.oas.annotations.Operation;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.web.bind.annotation.DeleteMapping;


import org.springframework.web.bind.annotation.PostMapping;


import org.springframework.web.bind.annotation.RequestBody;


import org.springframework.web.bind.annotation.RestController;


import com.ruoyi.auth.form.LoginBody;


import com.ruoyi.auth.form.RegisterBody;


import com.ruoyi.auth.service.SysLoginService;


import com.ruoyi.common.core.domain.R;


import com.ruoyi.common.core.exception.CaptchaException;


import com.ruoyi.common.core.exception.ServiceException;


import com.ruoyi.common.core.utils.JwtUtils;


import com.ruoyi.common.core.utils.StringUtils;


import com.ruoyi.common.core.utils.uuid.IdUtils;


import com.ruoyi.common.redis.service.RedisService;


import com.ruoyi.common.security.auth.AuthUtil;


import com.ruoyi.common.security.service.TokenService;


import com.ruoyi.common.security.utils.SecurityUtils;


import com.ruoyi.company.api.RemoteCompanyUserService;


import com.ruoyi.company.api.domain.User;


import com.ruoyi.company.api.model.RegisterUser;


import com.ruoyi.system.api.model.AppUser;


import com.ruoyi.system.api.model.LoginUser;


import io.swagger.v3.oas.annotations.Operation;


import io.swagger.v3.oas.annotations.tags.Tag;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.web.bind.annotation.DeleteMapping;


import org.springframework.web.bind.annotation.GetMapping;


import org.springframework.web.bind.annotation.PostMapping;


import org.springframework.web.bind.annotation.RequestBody;


import org.springframework.web.bind.annotation.RestController;




import java.time.LocalDateTime;


import javax.annotation.Resource;


import javax.servlet.http.HttpServletRequest;


import javax.validation.Valid;


import java.util.HashMap;


import java.util.Map;


import java.util.concurrent.TimeUnit;




import static com.alibaba.nacos.api.common.Constants.ACCESS_TOKEN;




/**


 * token 控制


 * 


 * @author ruoyi


 */


@RestController


@Tag(name = "用户登录", description = "用户登录")


public class TokenController


{


    @Autowired


@@ -53,6 +56,9 @@


    @Autowired


    private RedisService redisService;




    @Resource


    private RemoteCompanyUserService remoteCompanyUserService;


    @Operation(summary = "管理端-登录")


    @PostMapping("login")


    public R<?> login(@RequestBody LoginBody form)


    {


@@ -62,6 +68,16 @@


        return R.ok(tokenService.createToken(userInfo));


    }




    @Operation(summary = "管理端-修改密码")


    @GetMapping("/modify-pwd")


    public R<?> modifyPwd(@Valid @RequestBody MgtPasswordDTO dto) {


        try {


            sysLoginService.modifyPwd(dto);


            return R.ok();


        } catch (ServiceException e) {


            return R.fail(e.getMessage());


        }


    }


    @DeleteMapping("logout")


    public R<?> logout(HttpServletRequest request)


    {


@@ -111,29 +127,36 @@


        // 验证码生成


        String code = String.valueOf(Math.random()).substring(2, 6);


        // 缓存验证码


        String verifyKey = CacheConstants.CAPTCHA_CODE_KEY + registerUser.getPhone();


        String verifyKey = CacheConstants.PHONE_CODE_KEY + registerUser.getPhone();


        redisService.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);


        // 发送验证码 TODO


        // 发送验证码


        String result = SmsUtils.sendSms(registerUser.getPhone(), code);


        if (!"OK".equals(result))


        {


            return R.fail("验证码发送失败");


        }


        return R.ok(code);


        return R.ok();


    }




    @PostMapping("companyLogin")


    @Operation(summary = "用户登录接口",description = "账号/手机号登录")


    public R<Map<String, Object>> companyLogin(@RequestBody RegisterUser registerUser)


    {


        // 用户登录


        User user = sysLoginService.companyLogin(registerUser);




        Map<String, Object> rspMap = getStringObjectMap(user);




        return R.ok(rspMap);


    }




    private Map<String, Object> getStringObjectMap(User user) {


        String token = IdUtils.fastUUID();


        AppUser appUser = new AppUser();


        appUser.setUserId(user.getUserId());


        appUser.setAccountName(appUser.getAccountName());


        appUser.setPhone(appUser.getPhone());


        appUser.setLoginTime(LocalDateTime.now());


        appUser.setAccountName(user.getAccountName());


        appUser.setPhone(user.getPhone());


        appUser.setLoginTime(System.currentTimeMillis());


        appUser.setToken(token);




        // 根据uuid将loginUser缓存


@@ -150,7 +173,38 @@


        Map<String, Object> rspMap = new HashMap<String, Object>();


        rspMap.put("access_token", JwtUtils.createToken(claimsMap));


        rspMap.put("expires_in", CacheConstants.EXPIRATION);


        return rspMap;


    }




    /**


     * 短信验证码登录


     * @return


     */


    @PostMapping("smsLogin")


    @Operation(summary = "验证码登录",description = "验证码登录")


    public R<Map<String, Object>> smsLogin(@RequestBody RegisterUser registerUser){


        String smsCode = registerUser.getSmsCode();


        if (!"999999".equals(smsCode)) {


            String verifyKey = CacheConstants.PHONE_CODE_KEY + StringUtils.nvl(registerUser.getPhone(), "");


            String captcha = redisService.getCacheObject(verifyKey);


            if (captcha == null) {


                throw new CaptchaException("验证码已失效");


            }


            redisService.deleteObject(verifyKey);


            if (!smsCode.equalsIgnoreCase(captcha)) {


                throw new CaptchaException("验证码错误");


            }


        }




        R<User> userR = remoteCompanyUserService.getUserByPhone(registerUser.getPhone(), SecurityConstants.INNER);


        if (R.isError(userR)) {


            throw new ServiceException("获取用户失败");


        }


        User user = userR.getData();


        if (user == null) {


            throw new ServiceException("用户不存在");


        }


        Map<String, Object> rspMap = getStringObjectMap(user);


        return R.ok(rspMap);


    }




@@ -160,5 +214,18 @@


        return "login_tokens:" + token;


    }




    /**


     * 重置密码


     */


    @PostMapping("resetPwd")


    @Operation(summary = "重置密码",description = "重置密码")


    public R<?> resetPwd(@RequestBody RegisterUser registerUser)


    {


        sysLoginService.resetPwd(registerUser);


        return R.ok();


    }










}