mirror of
https://gitee.com/dromara/RuoYi-Vue-Plus.git
synced 2026-09-19 09:55:16 +08:00
252 lines
8.8 KiB
Java
252 lines
8.8 KiB
Java
package com.ruoyi.winery.component;
|
|
|
|
import cn.binarywang.wx.miniapp.api.WxMaService;
|
|
import cn.binarywang.wx.miniapp.bean.WxMaJscode2SessionResult;
|
|
import cn.binarywang.wx.miniapp.bean.WxMaPhoneNumberInfo;
|
|
import cn.hutool.core.util.IdUtil;
|
|
import cn.hutool.core.util.StrUtil;
|
|
import cn.hutool.json.JSONObject;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
import com.github.binarywang.wxpay.service.WxPayService;
|
|
import com.github.binarywang.wxpay.util.SignUtils;
|
|
import com.ruoyi.common.constant.Constants;
|
|
import com.ruoyi.common.constant.UserConstants;
|
|
import com.ruoyi.common.core.domain.AjaxResult;
|
|
import com.ruoyi.common.core.domain.entity.SysUser;
|
|
import com.ruoyi.common.core.domain.model.LoginUser;
|
|
import com.ruoyi.common.core.redis.RedisCache;
|
|
import com.ruoyi.common.exception.CustomException;
|
|
import com.ruoyi.common.exception.user.UserPasswordNotMatchException;
|
|
import com.ruoyi.common.utils.MessageUtils;
|
|
import com.ruoyi.common.utils.SecurityUtils;
|
|
import com.ruoyi.common.utils.StringUtils;
|
|
import com.ruoyi.framework.manager.AsyncManager;
|
|
import com.ruoyi.framework.manager.factory.AsyncFactory;
|
|
import com.ruoyi.framework.web.service.TokenService;
|
|
import com.ruoyi.system.service.ISysUserService;
|
|
import com.ruoyi.winery.config.wx.WxMiniProperties;
|
|
import com.ruoyi.winery.domain.winery.WineryMauser;
|
|
import com.ruoyi.winery.service.IWineryMauserService;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import me.chanjar.weixin.common.error.WxErrorException;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.security.authentication.AuthenticationManager;
|
|
import org.springframework.security.authentication.BadCredentialsException;
|
|
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
|
|
import org.springframework.security.core.Authentication;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import javax.annotation.Resource;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
import static com.ruoyi.winery.define.MiniDefine.*;
|
|
|
|
|
|
/**
|
|
* @author tottimctj
|
|
* @since 2020-12-15
|
|
*/
|
|
|
|
@Component
|
|
@Slf4j
|
|
public class MiniComponent {
|
|
|
|
@Autowired
|
|
IWineryMauserService wineryMauserService;
|
|
|
|
@Autowired
|
|
WxMaService wxMaService;
|
|
|
|
@Autowired
|
|
private RedisCache redisCache;
|
|
|
|
|
|
@Autowired
|
|
private WxMiniProperties wxMiniProperties;
|
|
|
|
@Autowired
|
|
private WxPayService wxPayService;
|
|
|
|
|
|
@Autowired
|
|
private TokenService tokenService;
|
|
|
|
@Resource
|
|
private AuthenticationManager authenticationManager;
|
|
|
|
@Autowired
|
|
private ISysUserService userService;
|
|
|
|
@Autowired
|
|
private IWineryMauserService iWineryMauserService;
|
|
|
|
|
|
public WxMaJscode2SessionResult login(String code, Long deptId) throws WxErrorException {
|
|
|
|
WxMaJscode2SessionResult sessionInfo = wxMaService.getUserService().getSessionInfo(code);
|
|
|
|
WineryMauser user = wineryMauserService.getOne(
|
|
new LambdaQueryWrapper<WineryMauser>()
|
|
.eq(WineryMauser::getDeptId, deptId)
|
|
.eq(WineryMauser::getOpenId, sessionInfo.getOpenid())
|
|
);
|
|
String key = sessionInfo.getOpenid();
|
|
redisCache.setCacheObject(key, sessionInfo.getSessionKey(), 7200, TimeUnit.SECONDS);
|
|
if (user == null) {
|
|
user = new WineryMauser();
|
|
user.setOpenId(sessionInfo.getOpenid());
|
|
user.setDeptId(deptId);
|
|
log.info("新增user:{}", user);
|
|
}
|
|
|
|
if (StrUtil.isNotBlank(sessionInfo.getUnionid())) {
|
|
user.setUnionId(sessionInfo.getUnionid());
|
|
}
|
|
|
|
user.setStatus(0);
|
|
wineryMauserService.saveOrUpdate(user);
|
|
|
|
return sessionInfo;
|
|
}
|
|
|
|
public AjaxResult registration(String openid, String mobile, String nickName, Long deptId, String avatar) {
|
|
|
|
SysUser user = new SysUser();
|
|
String userName = MINI_USER_SYMBOL + openid + "-" + deptId;
|
|
user.setUserName(userName);
|
|
user.setPhonenumber(mobile);
|
|
user.setNickName(nickName);
|
|
user.setDeptId(deptId);
|
|
user.setAvatar(avatar);
|
|
user.setPassword(MINI_DEFUALT_PASSWORD);
|
|
user.setRoleIds(new Long[]{MINI_DEFUALT_ROLEID});
|
|
user.setPostIds(new Long[]{MINI_DEFUALT_POSTID});
|
|
|
|
if (UserConstants.NOT_UNIQUE.equals(userService.checkUserNameUnique(user.getUserName()))) {
|
|
return AjaxResult.error("新增用户" + user.getUserName() + "失败，登录账号已存在");
|
|
}
|
|
|
|
//
|
|
// else 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.setCreateBy(MINI_MANAGE_USER);
|
|
user.setPassword(SecurityUtils.encryptPassword(user.getPassword()));
|
|
|
|
|
|
// 创建查找小程序用户
|
|
WineryMauser wineryMauser = iWineryMauserService.getOne(
|
|
new LambdaQueryWrapper<WineryMauser>()
|
|
.eq(WineryMauser::getOpenId, openid)
|
|
.eq(WineryMauser::getDeptId, user.getDeptId()));
|
|
|
|
if (wineryMauser == null) {
|
|
wineryMauser = new WineryMauser(user);
|
|
}
|
|
|
|
if (userService.insertUser(user) > 0 && iWineryMauserService.saveOrUpdate(wineryMauser)) {
|
|
AjaxResult ajax = AjaxResult.success();
|
|
ajax.put(Constants.TOKEN, loginByMini(userName));
|
|
return ajax;
|
|
} else {
|
|
return AjaxResult.error();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 小程序登录
|
|
*
|
|
* @param userName
|
|
* @return
|
|
*/
|
|
public String loginByMini(String userName) {
|
|
// 用户验证
|
|
Authentication authentication = null;
|
|
try {
|
|
// 该方法会去调用UserDetailsServiceImpl.loadUserByUsername
|
|
authentication = authenticationManager
|
|
.authenticate(new UsernamePasswordAuthenticationToken(userName, MINI_DEFUALT_PASSWORD));
|
|
} catch (Exception e) {
|
|
if (e instanceof BadCredentialsException) {
|
|
AsyncManager.me().execute(AsyncFactory.recordLogininfor(userName, Constants.LOGIN_FAIL, MessageUtils.message("user.password.not.match")));
|
|
throw new UserPasswordNotMatchException();
|
|
} else {
|
|
AsyncManager.me().execute(AsyncFactory.recordLogininfor(userName, Constants.LOGIN_FAIL, e.getMessage()));
|
|
throw new CustomException(e.getMessage());
|
|
}
|
|
}
|
|
AsyncManager.me().execute(AsyncFactory.recordLogininfor(userName, Constants.LOGIN_SUCCESS, MessageUtils.message("user.login.success")));
|
|
LoginUser loginUser = (LoginUser) authentication.getPrincipal();
|
|
// 生成token
|
|
return tokenService.createToken(loginUser);
|
|
|
|
}
|
|
|
|
/**
|
|
* 获取手机号
|
|
*
|
|
* @param json 微信加密信息
|
|
* @return
|
|
*/
|
|
public String getMobile(JSONObject json) {
|
|
|
|
String openid = json.getStr("openid");
|
|
String deptId = json.getStr("deptId");
|
|
// 创建查找小程序用户
|
|
WineryMauser user = iWineryMauserService.getOne(
|
|
new LambdaQueryWrapper<WineryMauser>()
|
|
.eq(WineryMauser::getOpenId, openid)
|
|
.eq(WineryMauser::getDeptId, deptId));
|
|
|
|
JSONObject detail = json.getJSONObject("detail");
|
|
String encryptedData = detail.getStr("encryptedData");
|
|
String iv = detail.getStr("iv");
|
|
|
|
|
|
String sessionKey = redisCache.getCacheObject(openid);
|
|
|
|
if (StrUtil.isBlank(sessionKey)) {
|
|
return null;
|
|
}
|
|
|
|
WxMaPhoneNumberInfo mobile = wxMaService.getUserService().getPhoneNoInfo(sessionKey, encryptedData, iv);
|
|
|
|
user.setMobile(mobile.getPhoneNumber());
|
|
|
|
wineryMauserService.saveOrUpdate(user);
|
|
|
|
return mobile.getPhoneNumber();
|
|
}
|
|
|
|
public JSONObject getRealAuthJson(String openid) {
|
|
|
|
|
|
JSONObject json = new JSONObject();
|
|
json.set("api_version", "1.0");
|
|
json.set("mch_id", wxMiniProperties.getMchId());
|
|
json.set("appid", wxMiniProperties.getAppId());
|
|
json.set("scope", "pay_identity");
|
|
json.set("response_type", "code");
|
|
// json.set("openid", "okT7b4rOqy7FkJ_DxuBhyRPo0sGA");
|
|
json.set("openid", openid);
|
|
json.set("sign_type", "HMAC-SHA256");
|
|
json.set("nonce_str", IdUtil.fastSimpleUUID());
|
|
|
|
Map<String, String> params = json.toBean(HashMap.class);
|
|
|
|
json.set("sign", SignUtils.createSign(params, "HMAC-SHA256", wxMiniProperties.getMchKey(), (String[]) null));
|
|
|
|
|
|
return json;
|
|
}
|
|
|
|
}
|