package com.ruoyi.web.controller.api;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.ruoyi.common.constant.Constants;
|
import com.ruoyi.common.core.domain.AjaxResult;
|
import com.ruoyi.common.core.domain.R;
|
import com.ruoyi.common.core.domain.entity.SysRole;
|
import com.ruoyi.common.core.domain.entity.SysUser;
|
import com.ruoyi.common.core.domain.model.LoginBody;
|
import com.ruoyi.common.core.domain.model.LoginUser;
|
import com.ruoyi.common.redis.service.RedisService;
|
import com.ruoyi.common.utils.NumberUtil;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.ruoyi.framework.web.service.SysLoginService;
|
import com.ruoyi.framework.web.service.TokenService;
|
import com.ruoyi.system.model.TbUser;
|
import com.ruoyi.system.service.TbUserService;
|
import com.ruoyi.system.utils.wx.body.resp.Code2SessionRespBody;
|
import com.ruoyi.system.utils.wx.body.resq.Code2SessionResqBody;
|
import com.ruoyi.system.utils.wx.model.WeixinProperties;
|
import com.ruoyi.system.utils.wx.pojo.AppletUserEncrypteData;
|
import com.ruoyi.system.utils.wx.tools.WxAppletTools;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.client.RestTemplate;
|
|
import javax.annotation.Resource;
|
import javax.validation.Valid;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* <p>
|
* 微信小程序登录 前端控制器
|
* </p>
|
*
|
* @author xiaochen
|
* @since 2024-08-06
|
*/
|
@Slf4j
|
@RestController
|
@RequestMapping("/wxLogin")
|
@Api(tags = "登录模块")
|
public class WxLoginController {
|
|
@Autowired
|
private WeixinProperties wxConfig;
|
@Autowired
|
private RestTemplate wxRestTemplate;
|
@Resource
|
private RedisService redisService;
|
|
@Resource
|
private TokenService tokenService;
|
|
|
@Autowired
|
private TbUserService tbUserService;
|
|
|
|
/**
|
* 账号密码登录
|
*
|
* @param loginBody 登录信息
|
* @return 结果
|
*/
|
@ApiOperation(value = "手机验证码登录",notes = "手机验证码登录")
|
@PostMapping("/login")
|
public AjaxResult login(@Valid @RequestBody LoginBody loginBody)
|
{
|
AjaxResult ajax = AjaxResult.success();
|
// 生成令牌
|
Object cacheObject = redisService.getCacheObject("login_" + loginBody.getPhone());
|
if(cacheObject==null || !cacheObject.toString().equals(loginBody.getCode())){
|
return AjaxResult.error("验证码错误");
|
}
|
TbUser user = tbUserService.getOne(new LambdaQueryWrapper<TbUser>().eq(TbUser::getPhone, loginBody.getPhone()).ne(TbUser::getStatus,3).eq(TbUser::getIsDelete, 0));
|
if(user!=null && user.getStatus()==2){
|
return AjaxResult.error("登录失败，当前账号已被冻结");
|
}
|
if(user==null){
|
TbUser tbUser = new TbUser();
|
tbUser.setPhone(loginBody.getPhone());
|
tbUser.setStatus(1);
|
tbUser.setUserName(loginBody.getPhone());
|
tbUser.setInviteId(loginBody.getInviteId());
|
tbUserService.save(tbUser);
|
user = tbUser;
|
}
|
LoginUser loginUser = new LoginUser();
|
loginUser.setUserId(Long.valueOf(user.getId()));
|
loginUser.setUser(new SysUser());
|
ajax.put(Constants.TOKEN, tokenService.createToken(loginUser));
|
ajax.put("username",user.getUserName());
|
ajax.put("avatar",user.getAvatar());
|
return ajax;
|
}
|
|
@ApiOperation(value = "通过code获得openid,获取用户信息",tags = {"微信小程序登录"})
|
@PostMapping("/openIdByJsCode")
|
public R<Map<String, Object>> openIdByJsCode(@RequestBody AppletUserEncrypteData data) {
|
log.info("<<<<<<<<换取openid开始<<<<<<<<:{}", data.getCode());
|
WxAppletTools appletTools = new WxAppletTools(wxRestTemplate, wxConfig, redisService);
|
Code2SessionRespBody body = appletTools.getOpenIdByJscode2session(new Code2SessionResqBody().build(data.getCode()));
|
String openid = body.getOpenid();
|
String sessionKey = body.getSessionKey();
|
return R.ok();
|
}
|
|
|
|
@ApiOperation(value = "发送验证码",tags = {"发送验证码"})
|
@PostMapping("/sendCode")
|
public R<?> sendCode(String phone) {
|
if (StringUtils.isBlank(phone)) {
|
return R.fail("手机号不能为空");
|
}
|
String code = NumberUtil.getRandomInteger(6);
|
redisService.setCacheObject("login_"+phone, code, 5L, TimeUnit.MINUTES);
|
// 发送验证码
|
|
|
return R.ok();
|
}
|
|
|
@ApiOperation(value = "获取协议",tags = {"获取协议 1用户 2隐私"})
|
@PostMapping("/getAgreement")
|
public R<?> getAgreement(Integer type) {
|
|
return R.ok();
|
}
|
|
|
|
|
|
}
|
