package cn.mb.cloud.auth.security.social.handler;
|
|
import cn.hutool.core.util.StrUtil;
|
import cn.mb.cloud.auth.security.entity.User;
|
import cn.mb.cloud.auth.security.service.IUserService;
|
import cn.mb.cloud.common.api.vo.UserVO;
|
import cn.mb.cloud.common.cache.RedisFastJsonTemplate;
|
import cn.mb.cloud.common.core.constant.CacheConstants;
|
import cn.mb.cloud.common.core.constant.enums.LoginTypeEnum;
|
import cn.mb.cloud.common.core.exception.ValidateCodeException;
|
import cn.mb.cloud.common.core.util.WebUtils;
|
import com.baomidou.mybatisplus.core.toolkit.StringPool;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import lombok.AllArgsConstructor;
|
import lombok.SneakyThrows;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.stereotype.Component;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
/**
|
* @author jason
|
* @date 2018/11/18
|
*/
|
@Slf4j
|
@Component("SMS")
|
@AllArgsConstructor
|
public class SmsLoginHandler extends AbstractLoginHandler {
|
|
private final IUserService userService;
|
private final RedisFastJsonTemplate redisTemplate;
|
|
/**
|
* 验证码登录传入为手机号 不用不处理
|
*
|
* @param mobile
|
* @return
|
*/
|
@Override
|
public String identify(String mobile) {
|
return mobile;
|
}
|
|
|
@Override
|
public Boolean check(String loginStr) {
|
checkCode(loginStr, WebUtils.getRequest());
|
return Boolean.TRUE;
|
}
|
|
/**
|
* 通过mobile 获取用户信息
|
*
|
* @param identify
|
* @return
|
*/
|
@Override
|
public UserVO info(String identify, String threeAvatar, String threeName) {
|
//进行收集号码等
|
User user = userService.getOne(Wrappers.<User>query().lambda()
|
.eq(User::getUsername, identify)
|
.eq(User::getDelFlag, 0));
|
|
if (user == null) {
|
log.info("手机号未注册:{}", identify);
|
return null;
|
}
|
UserVO userVO = new UserVO();
|
BeanUtils.copyProperties(user, userVO);
|
return userVO;
|
}
|
|
/**
|
* 检查code
|
*
|
* @param request
|
*/
|
@SneakyThrows
|
private void checkCode(String loginStr, HttpServletRequest request) {
|
String code = request.getParameter("code");
|
|
if (StrUtil.isBlank(code)) {
|
throw new ValidateCodeException("验证码不能为空");
|
}
|
|
String key = CacheConstants.DEFAULT_CODE_KEY + LoginTypeEnum.SMS.getType() + StringPool.AT + loginStr;
|
if (!redisTemplate.hasKey(key)) {
|
throw new ValidateCodeException("验证码不合法");
|
}
|
|
Object codeObj = redisTemplate.opsForValue().get(key);
|
|
if (codeObj == null) {
|
throw new ValidateCodeException("验证码不合法");
|
}
|
|
String saveCode = codeObj.toString();
|
if (StrUtil.isBlank(saveCode)) {
|
redisTemplate.delete(key);
|
throw new ValidateCodeException("验证码不合法");
|
}
|
|
if (!StrUtil.equals(saveCode, code)) {
|
redisTemplate.delete(key);
|
throw new ValidateCodeException("验证码不合法");
|
}
|
|
redisTemplate.delete(key);
|
}
|
|
}
|
