package cn.mb.cloud.auth.security.controller;
|
|
import cn.hutool.core.util.RandomUtil;
|
import cn.hutool.crypto.digest.BCrypt;
|
import cn.mb.cloud.auth.security.entity.User;
|
import cn.mb.cloud.auth.security.service.IUserService;
|
import cn.mb.cloud.auth.security.util.SmsUtil;
|
import cn.mb.cloud.auth.security.util.StringUtil;
|
import cn.mb.cloud.auth.security.util.VerifyCode;
|
import cn.mb.cloud.common.cache.RedisFastJsonTemplate;
|
import cn.mb.cloud.common.core.constant.CacheConstants;
|
import cn.mb.cloud.common.core.constant.enums.ErrorCodeConstants;
|
import cn.mb.cloud.common.core.constant.enums.LoginTypeEnum;
|
import cn.mb.cloud.common.core.exception.BusinessException;
|
import cn.mb.cloud.common.core.exception.ValidateCodeException;
|
import cn.mb.cloud.common.core.util.ResponseData;
|
import com.aliyuncs.exceptions.ClientException;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.StringPool;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.tencentcloudapi.sms.v20210111.models.SendSmsResponse;
|
import com.tencentcloudapi.sms.v20210111.models.SendStatus;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiImplicitParam;
|
import io.swagger.annotations.ApiImplicitParams;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.AllArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.validation.constraints.NotBlank;
|
import java.time.LocalDateTime;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* 系统社交登录账号表
|
*
|
* @author jason
|
* @date 2018-08-16 21:30:41
|
*/
|
@Slf4j
|
@RestController
|
@RequestMapping("/social")
|
@AllArgsConstructor
|
@Api(value = "三方账号管理模块", tags = "三方账号管理模块")
|
public class SysSocialDetailsController {
|
|
|
|
private final RedisFastJsonTemplate redisFastJsonTemplate;
|
|
|
private final IUserService userService;
|
|
|
@ResponseBody
|
@PostMapping("/register")
|
@ApiOperation(value = "小程序注册")
|
@ApiImplicitParams({
|
@ApiImplicitParam(value = "手机号", name = "phone", required = true, dataType = "string"),
|
@ApiImplicitParam(value = "验证码", name = "code", dataType = "string"),
|
@ApiImplicitParam(value = "密码", name = "password", dataType = "string"),
|
@ApiImplicitParam(value = "昵称（微信）", name = "nickName", dataType = "string"),
|
@ApiImplicitParam(value = "头像（微信）", name = "avatarUrl", dataType = "string"),
|
@ApiImplicitParam(value = "性别（微信）", name = "gender", dataType = "Integer"),
|
@ApiImplicitParam(value = "来源（传名称）", name = "source", dataType = "string"),
|
@ApiImplicitParam(value = "openId", name = "openId", dataType = "string"),
|
})
|
public ResponseData phoneRegister(String phone, String code,String password,String source,
|
String nickName,String avatarUrl,Integer gender,String openId){
|
try {
|
//验证验证码
|
if (StringUtils.isEmpty(phone)) {
|
return ResponseData.fail("手机号不能为空");
|
}
|
if(StringUtils.isEmpty(openId)){
|
//验证验证码
|
ResponseData responseData = validateSms(1, phone, code);
|
String dataCode = responseData.getCode();
|
if(!dataCode.equals(ErrorCodeConstants.SUCCESS.getValue())){
|
return responseData;
|
}
|
}
|
//验证手机号是否注册
|
User user = userService.getOne(new QueryWrapper<User>()
|
.eq("username",phone)
|
.eq("del_flag",0));
|
if(user == null){
|
user = new User();
|
if(StringUtils.isEmpty(password)){
|
password = phone;
|
}
|
//手机号注册
|
String pswMd5 = BCrypt.hashpw(password, BCrypt.gensalt());
|
user.setUsername(phone);
|
user.setPassword(pswMd5);
|
user.setSource(source);
|
user.setName(nickName);
|
user.setWechatName(nickName);
|
user.setOpenId(openId);
|
if(StringUtils.isEmpty(nickName)){
|
user.setName(phone.replaceAll("(\\d{3})\\d{4}(\\d{4})", "$1****$2"));
|
}
|
user.setAvatar(avatarUrl);
|
user.setGender(gender);
|
user.setCreateDate(LocalDateTime.now());
|
userService.save(user);
|
}else {
|
user.setWechatName(nickName);
|
user.setOpenId(openId);
|
userService.updateById(user);
|
}
|
return ResponseData.success(user.getId());
|
}catch (Exception e){
|
e.printStackTrace();
|
return ResponseData.fail();
|
}
|
}
|
|
@ResponseBody
|
@PostMapping("/forgetPassword")
|
@ApiOperation(value = "忘记密码")
|
@ApiImplicitParams({
|
@ApiImplicitParam(value = "手机号", name = "phone", required = true, dataType = "string"),
|
@ApiImplicitParam(value = "验证码", name = "code", required = true, dataType = "string"),
|
@ApiImplicitParam(value = "密码", name = "password", required = true, dataType = "string"),
|
})
|
public ResponseData forgetPassword(String phone,String code,String password){
|
try {
|
if (StringUtils.isEmpty(phone)) {
|
return ResponseData.fail("手机号不能为空");
|
}
|
if (StringUtils.isEmpty(code)) {
|
return ResponseData.fail("验证码不能为空");
|
}
|
if (StringUtils.isEmpty(password)) {
|
return ResponseData.fail("密码不能为空");
|
}
|
//验证验证码
|
ResponseData responseData = validateSms(2, phone, code);
|
String dataCode = responseData.getCode();
|
if(!dataCode.equals(ErrorCodeConstants.SUCCESS.getValue())){
|
return responseData;
|
}
|
//修改密码
|
String pswMd5 = BCrypt.hashpw(password, BCrypt.gensalt());
|
return userService.forgetPassword(phone,pswMd5);
|
}catch (Exception e){
|
e.printStackTrace();
|
return ResponseData.fail();
|
}
|
}
|
|
@GetMapping("/sms/captcha")
|
@ApiOperation(value = "获取验证码")
|
@ApiImplicitParams({
|
@ApiImplicitParam(paramType = "query", name = "type", value = "类型(1=注册，2=忘记密码 )", required = true, dataType = "int"),
|
@ApiImplicitParam(paramType = "query", name = "mobile", value = "手机号", required = true, dataType = "String"),
|
})
|
public ResponseData sendSmsCode(@NotBlank(message = "手机号码不能为空") String mobile, @NotBlank(message = "类型不能为空") Byte type)
|
throws BusinessException, ValidateCodeException, ClientException {
|
if (StringUtils.isEmpty(mobile)) {
|
return ResponseData.fail("请先绑定手机号码");
|
}
|
User member = userService.getOne(Wrappers.<User>query()
|
.lambda().eq(User::getUsername, mobile).eq(User::getDelFlag, 0));
|
|
if (member == null && type == 2) {
|
log.info("手机号未注册:{}", mobile);
|
return ResponseData.fail("手机号未注册");
|
}
|
StringUtil.checkNull(mobile, "手机账号不能为空");
|
//判格式
|
StringUtil.checkFormat(mobile, "请输入格式正确的11位号码");
|
Object codeObj = redisFastJsonTemplate.opsForValue()
|
.get(CacheConstants.DEFAULT_CODE_KEY + type + StringPool.AT + mobile);
|
|
if (codeObj != null) {
|
log.info("手机号验证码未过期:{}，{}", mobile, codeObj);
|
//return ResponseData.fail("验证码发送过于频繁");
|
}
|
String code = RandomUtil.randomNumbers(Integer.parseInt(VerifyCode.CODE_SIZE));
|
log.info("手机号生成验证码成功:{},{}", mobile, code);
|
|
SendSmsResponse response = null;
|
try {
|
String[] mobileSet = {"+86"+mobile};
|
String[] templateParamSet={code+""};
|
response = SmsUtil.sendSms(mobileSet, templateParamSet);
|
} catch (com.netflix.client.ClientException e) {
|
e.printStackTrace();
|
}
|
|
// 测试环境不发短信
|
//SendSmsResponse response = new SendSmsResponse();
|
SendStatus sendStatus = response.getSendStatusSet()[0];
|
if (sendStatus.getCode().equals("Ok")) {
|
redisFastJsonTemplate.opsForValue().set(
|
CacheConstants.DEFAULT_CODE_KEY + type + StringPool.AT + mobile, code,
|
5 * 60, TimeUnit.SECONDS);
|
} else {
|
// 判断存在时销毁
|
redisFastJsonTemplate.delete(CacheConstants.DEFAULT_CODE_KEY + type + StringPool.AT + mobile);
|
}
|
return ResponseData.success(code);
|
}
|
|
@PostMapping("/sms/captcha/validate")
|
@ApiOperation(value = "验证短信")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "type", value = "类型(1=注册，2=忘记密码 )", required = true, dataType = "int", paramType = "query"),
|
@ApiImplicitParam(name = "phone", value = "手机号", required = true, dataType = "String", paramType = "query"),
|
@ApiImplicitParam(name = "code", value = "验证码", required = true, dataType = "String", paramType = "query"),
|
})
|
public ResponseData validateSms(Integer type, String phone, String code) {
|
if (type==null) {
|
return ResponseData.fail("类型不能为空");
|
}
|
if (StringUtils.isEmpty(phone)) {
|
return ResponseData.fail("手机号不能为空");
|
}
|
if (StringUtils.isEmpty(code)) {
|
return ResponseData.fail("验证码不能为空");
|
}
|
return userService.verifyCode(type, phone, code);
|
|
}
|
|
}
|
