package com.tourguide.modules.user.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.tourguide.common.exception.BusinessException;
|
import com.tourguide.modules.user.domain.ClientType;
|
import com.tourguide.modules.user.domain.UserRole;
|
import com.tourguide.modules.user.dto.AuthResponse;
|
import com.tourguide.modules.user.dto.GuideUserResponse;
|
import com.tourguide.modules.user.dto.LoginRequest;
|
import com.tourguide.modules.user.dto.MiniRegisterRequest;
|
import com.tourguide.modules.user.dto.MiniUserResponse;
|
import com.tourguide.modules.user.dto.UserResponse;
|
import com.tourguide.modules.user.entity.AdminUser;
|
import com.tourguide.modules.user.entity.GuideUser;
|
import com.tourguide.modules.user.entity.MiniUser;
|
import com.tourguide.modules.user.mapper.MiniUserMapper;
|
import com.tourguide.security.JwtTokenService;
|
import com.tourguide.security.TokenPayload;
|
import com.tourguide.security.TokenSessionService;
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.nio.charset.StandardCharsets;
|
import java.util.UUID;
|
|
/** 统一编排登录流程，但后台账号与小程序账号始终从不同数据表读取。 */
|
@Service
|
public class AuthService {
|
private final AdminUserService adminUserService;
|
private final MiniUserService miniUserService;
|
private final GuideUserService guideUserService;
|
private final MiniUserMapper miniUserMapper;
|
private final PasswordEncoder passwordEncoder;
|
private final JwtTokenService jwtTokenService;
|
private final TokenSessionService tokenSessionService;
|
|
public AuthService(AdminUserService adminUserService, MiniUserService miniUserService, GuideUserService guideUserService,
|
MiniUserMapper miniUserMapper,
|
PasswordEncoder passwordEncoder, JwtTokenService jwtTokenService, TokenSessionService tokenSessionService) {
|
this.adminUserService = adminUserService;
|
this.miniUserService = miniUserService;
|
this.guideUserService = guideUserService;
|
this.miniUserMapper = miniUserMapper;
|
this.passwordEncoder = passwordEncoder;
|
this.jwtTokenService = jwtTokenService;
|
this.tokenSessionService = tokenSessionService;
|
}
|
|
public AuthResponse<UserResponse> loginAdmin(LoginRequest request) {
|
AdminUser user = adminUserService.findByUsername(request.getUsername());
|
if (user == null || !passwordEncoder.matches(request.getPassword(), user.getPassword())) {
|
throw new BusinessException(401, "用户名或密码错误");
|
}
|
String token = issueToken(user.getId(), user.getUsername(), user.getRoleCode(), ClientType.ADMIN);
|
return new AuthResponse<UserResponse>(token, jwtTokenService.getExpirationSeconds(), UserResponse.from(user));
|
}
|
|
public AuthResponse<MiniUserResponse> loginMini(LoginRequest request) {
|
MiniUser user = miniUserService.findByUsername(request.getUsername());
|
if (user == null || user.getStatus() == null || user.getStatus() != 1 || !passwordEncoder.matches(request.getPassword(), user.getPassword())) {
|
throw new BusinessException(401, "用户名或密码错误");
|
}
|
String token = issueToken(user.getId(), user.getUsername(), UserRole.MINI_USER.name(), ClientType.MINI);
|
return new AuthResponse<MiniUserResponse>(token, jwtTokenService.getExpirationSeconds(), MiniUserResponse.from(user));
|
}
|
|
@Transactional
|
public AuthResponse<MiniUserResponse> loginMiniWechat(String openid, String mobile) {
|
MiniUser user = miniUserMapper.selectOne(new LambdaQueryWrapper<MiniUser>()
|
.eq(MiniUser::getOpenid, openid)
|
.last("LIMIT 1"));
|
if (user == null) {
|
user = new MiniUser();
|
user.setUsername(buildWechatUsername(openid));
|
user.setPassword(passwordEncoder.encode(UUID.randomUUID().toString()));
|
user.setNickname("微信用户");
|
user.setMobile(mobile);
|
user.setOpenid(openid);
|
user.setStatus(1);
|
user.setDeleted(0);
|
miniUserMapper.insert(user);
|
} else {
|
if (user.getStatus() == null || user.getStatus() != 1) {
|
throw new BusinessException(401, "小程序用户已停用");
|
}
|
if (mobile != null && !mobile.equals(user.getMobile())) {
|
user.setMobile(mobile);
|
miniUserMapper.updateById(user);
|
}
|
}
|
String token = issueToken(user.getId(), user.getUsername(), UserRole.MINI_USER.name(), ClientType.MINI);
|
return new AuthResponse<MiniUserResponse>(token, jwtTokenService.getExpirationSeconds(), MiniUserResponse.from(user));
|
}
|
|
public AuthResponse<GuideUserResponse> loginGuide(LoginRequest request) {
|
GuideUser user = guideUserService.findByUsername(request.getUsername());
|
if (!isGuideLoginable(user) || !passwordEncoder.matches(request.getPassword(), user.getPassword())) {
|
throw new BusinessException(401, "用户名或密码错误");
|
}
|
return issueGuideToken(user);
|
}
|
|
public AuthResponse<GuideUserResponse> loginGuideByPhone(String phone) {
|
return issueGuideToken(findLoginableGuideByPhone(phone));
|
}
|
|
public void ensureGuidePhoneCanLogin(String phone) {
|
findLoginableGuideByPhone(phone);
|
}
|
|
private GuideUser findLoginableGuideByPhone(String phone) {
|
GuideUser user = guideUserService.findByPhone(phone);
|
if (!isGuideLoginable(user)) {
|
throw new BusinessException(401, "手机号未注册或账号已停用");
|
}
|
return user;
|
}
|
|
private boolean isGuideLoginable(GuideUser user) {
|
return user != null
|
&& user.getStatus() != null
|
&& (user.getStatus() == GuideUserService.STATUS_ENABLED
|
|| user.getStatus() == GuideUserService.STATUS_PENDING);
|
}
|
|
private AuthResponse<GuideUserResponse> issueGuideToken(GuideUser user) {
|
String token = issueToken(user.getId(), user.getUsername(), UserRole.GUIDE_USER.name(), ClientType.GUIDE);
|
return new AuthResponse<GuideUserResponse>(token, jwtTokenService.getExpirationSeconds(), GuideUserResponse.from(user));
|
}
|
|
@Transactional
|
public AuthResponse<MiniUserResponse> registerMini(MiniRegisterRequest request) {
|
if (miniUserMapper.selectCount(new LambdaQueryWrapper<MiniUser>().eq(MiniUser::getUsername, request.getUsername())) > 0) {
|
throw new BusinessException(409, "小程序用户名已存在");
|
}
|
MiniUser user = new MiniUser();
|
user.setUsername(request.getUsername());
|
user.setPassword(passwordEncoder.encode(request.getPassword()));
|
user.setNickname(request.getNickname());
|
user.setMobile(request.getMobile());
|
user.setStatus(1);
|
user.setDeleted(0);
|
miniUserMapper.insert(user);
|
String token = issueToken(user.getId(), user.getUsername(), UserRole.MINI_USER.name(), ClientType.MINI);
|
return new AuthResponse<MiniUserResponse>(token, jwtTokenService.getExpirationSeconds(), MiniUserResponse.from(user));
|
}
|
|
public void logout(String token, ClientType expectedClientType) {
|
TokenPayload payload = jwtTokenService.parse(token);
|
if (payload.getClientType() != expectedClientType) { throw new BusinessException(401, "Token端类型不匹配"); }
|
tokenSessionService.remove(payload);
|
}
|
|
private String issueToken(Long userId, String username, String authority, ClientType clientType) {
|
String token = jwtTokenService.createToken(userId, username, authority, clientType);
|
tokenSessionService.save(jwtTokenService.parse(token), jwtTokenService.getExpirationSeconds());
|
return token;
|
}
|
|
private String buildWechatUsername(String openid) {
|
return "wx_" + UUID.nameUUIDFromBytes(openid.getBytes(StandardCharsets.UTF_8))
|
.toString().replace("-", "").substring(0, 29);
|
}
|
}
|
