package cn.iocoder.yudao.module.ums.service.member;
|
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import cn.iocoder.yudao.module.system.service.auth.WxcpPasswordSetupGrant;
|
import cn.iocoder.yudao.module.system.service.user.AdminUserService;
|
import cn.iocoder.yudao.module.ums.dal.dataobject.member.UmsMemberDO;
|
import cn.iocoder.yudao.module.ums.dal.mysql.member.UmsMemberMapper;
|
import jakarta.annotation.Resource;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.ums.enums.ErrorCodeConstants.*;
|
|
@Service
|
public class WxcpInitialPasswordService {
|
@Resource private UmsMemberMapper memberMapper;
|
@Resource private AdminUserService userService;
|
@Resource private WxcpPasswordSetupGrant grant;
|
|
@Transactional(rollbackFor = Exception.class)
|
public String setPassword(Long userId, String token, String password) {
|
if (!grant.valid(token, userId)) throw exception(MEMBER_PASSWORD_SETUP_EXPIRED);
|
UmsMemberDO member = memberMapper.selectByUserId(userId);
|
if (member == null || !Boolean.TRUE.equals(member.getPasswordSetupRequired())) {
|
throw exception(MEMBER_PASSWORD_SETUP_NOT_REQUIRED);
|
}
|
int changed = memberMapper.update(null, new LambdaUpdateWrapper<UmsMemberDO>()
|
.eq(UmsMemberDO::getId, member.getId())
|
.eq(UmsMemberDO::getUserId, userId)
|
.eq(UmsMemberDO::getPasswordSetupRequired, true)
|
.set(UmsMemberDO::getPasswordSetupRequired, false));
|
if (changed != 1) throw exception(MEMBER_PASSWORD_SETUP_NOT_REQUIRED);
|
userService.updateUserPassword(userId, password);
|
String username = userService.getUser(userId).getUsername();
|
grant.revoke(token);
|
return username;
|
}
|
}
|
