132 lines
5.7 KiB
Java
132 lines
5.7 KiB
Java
package com.ruoyi.promo.controller;
|
|
|
|
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
|
|
import com.ruoyi.admin.mapper.PromoAdminMapper;
|
|
import com.ruoyi.admin.service.TtUserService;
|
|
import com.ruoyi.common.annotation.Anonymous;
|
|
import com.ruoyi.common.core.controller.BaseController;
|
|
import com.ruoyi.common.core.domain.R;
|
|
import com.ruoyi.common.core.redis.RedisCache;
|
|
import com.ruoyi.common.utils.StringUtils;
|
|
import com.ruoyi.domain.common.constant.UserType;
|
|
import com.ruoyi.domain.entity.sys.TtUser;
|
|
import com.ruoyi.system.service.ISysConfigService;
|
|
import com.ruoyi.thirdparty.common.model.ApiSmsBody;
|
|
import com.ruoyi.thirdparty.common.service.ApiSmsService;
|
|
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.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;
|
|
|
|
/**
|
|
* 推广后台短信验证码控制器
|
|
* POST /promo/sms/getVerifyCode - 获取短信验证码（需要滑块验证）
|
|
*/
|
|
@Api(tags = "推广后台短信")
|
|
@Slf4j
|
|
@RestController
|
|
@RequestMapping("/promo/sms")
|
|
public class PromoSmsController extends BaseController {
|
|
|
|
private final ISysConfigService sysConfigService;
|
|
private final ApiSmsService apiSmsService;
|
|
private final PromoAdminMapper promoAdminMapper;
|
|
|
|
@Autowired
|
|
private RedisCache redisCache;
|
|
|
|
@Autowired
|
|
private TtUserService userService;
|
|
|
|
public PromoSmsController(ISysConfigService sysConfigService,
|
|
ApiSmsService apiSmsService,
|
|
PromoAdminMapper promoAdminMapper) {
|
|
this.sysConfigService = sysConfigService;
|
|
this.apiSmsService = apiSmsService;
|
|
this.promoAdminMapper = promoAdminMapper;
|
|
}
|
|
|
|
/**
|
|
* 获取短信验证码（推广后台版本）
|
|
* 需要先完成/promo/captcha/check 滑块验证
|
|
*/
|
|
@ApiOperation("获取短信验证码")
|
|
@Anonymous
|
|
@PostMapping("/getVerifyCode")
|
|
public R<Object> getVerifyCode(@RequestBody ApiSmsBody smsBody) {
|
|
String websiteMaintenance = sysConfigService.selectConfigByKey("websiteMaintenance");
|
|
if ("1".equals(websiteMaintenance)) {
|
|
return R.fail("网站维护中......");
|
|
}
|
|
|
|
String phoneNumber = smsBody.getPhoneNumber();
|
|
String captchaId = smsBody.getCaptchaId();
|
|
if (!"1".equals(smsBody.getType()) && !"2".equals(smsBody.getType())) {
|
|
return R.fail("验证码类型错误");
|
|
}
|
|
if ("2".equals(smsBody.getType())) {
|
|
TtUser user = new LambdaQueryChainWrapper<>(userService.getBaseMapper())
|
|
.eq(TtUser::getPhoneNumber, phoneNumber)
|
|
.eq(TtUser::getDelFlag, "0")
|
|
.eq(TtUser::getStatus, "0")
|
|
.one();
|
|
boolean hasPromoRole = user != null
|
|
&& (UserType.ANCHOR.getCode().equals(user.getUserType())
|
|
|| UserType.BUSINESS_USER.getCode().equals(user.getUserType())
|
|
|| UserType.CARD_MERCHANT.getCode().equals(user.getUserType()));
|
|
// 超级管理员账号（promo_admin.username / phone_number）也可以是手机号，同样允许短信登录
|
|
boolean isSuperAdmin = !hasPromoRole && promoAdminMapper.findByAccount(phoneNumber) != null;
|
|
// 手机号没注册过：带有效招商邀请码时允许发码（登录即自动注册为主播）
|
|
boolean canAutoRegister = !hasPromoRole && !isSuperAdmin
|
|
&& StringUtils.isNotEmpty(smsBody.getParentInvitationCode())
|
|
&& findBusinessByInvitationCode(smsBody.getParentInvitationCode()) != null;
|
|
if (!hasPromoRole && !isSuperAdmin && !canAutoRegister) {
|
|
return R.fail("该手机号无推广后台登录权限");
|
|
}
|
|
}
|
|
|
|
// 校验推广后台滑块验证码
|
|
if (StringUtils.isEmpty(captchaId)) {
|
|
return R.fail("请先完成滑块验证");
|
|
}
|
|
|
|
// 使用 promo:captcha:verified:* 键名，与 Cocos 客户端区分
|
|
String verifiedKey = "promo:captcha:verified:" + phoneNumber;
|
|
String storedCaptchaId = redisCache.getCacheObject(verifiedKey);
|
|
if (StringUtils.isEmpty(storedCaptchaId) || !storedCaptchaId.equals(captchaId)) {
|
|
return R.fail("滑块验证已过期或无效，请重新验证");
|
|
}
|
|
|
|
String msg = apiSmsService.getVerifyCode(smsBody);
|
|
if (StringUtils.isNotEmpty(msg)) {
|
|
// 短信发送失败（次数/余额不足），返回特定错误码
|
|
if (msg.contains("次数") || msg.contains("余额") || msg.contains("不足")) {
|
|
return R.fail(999, "短信发送次数不足");
|
|
}
|
|
return R.fail(msg);
|
|
}
|
|
|
|
log.info("[推广后台短信发送成功] phone={}", phoneNumber);
|
|
return R.ok("获取验证码成功，请注意短信通知！");
|
|
}
|
|
|
|
/**
|
|
* 按邀请码找招商账号（必须是 04 招商）
|
|
*/
|
|
private TtUser findBusinessByInvitationCode(String invitationCode) {
|
|
TtUser parent = new LambdaQueryChainWrapper<>(userService.getBaseMapper())
|
|
.eq(TtUser::getInvitationCode, invitationCode.trim().toUpperCase())
|
|
.eq(TtUser::getDelFlag, "0")
|
|
.eq(TtUser::getStatus, "0")
|
|
.one();
|
|
if (parent == null || !UserType.BUSINESS_USER.getCode().equals(parent.getUserType())) {
|
|
return null;
|
|
}
|
|
return parent;
|
|
}
|
|
}
|