132 lines
5.1 KiB
Java
132 lines
5.1 KiB
Java
package com.youlai.system.service.impl;
|
||
|
||
import cn.hutool.captcha.AbstractCaptcha;
|
||
import cn.hutool.captcha.CaptchaUtil;
|
||
import cn.hutool.captcha.generator.CodeGenerator;
|
||
import cn.hutool.core.util.IdUtil;
|
||
import cn.hutool.core.util.StrUtil;
|
||
import com.youlai.system.common.constant.CacheConstants;
|
||
import com.youlai.system.core.security.jwt.JwtTokenProvider;
|
||
import com.youlai.system.model.dto.CaptchaResult;
|
||
import com.youlai.system.model.dto.LoginResult;
|
||
import com.youlai.system.plugin.captcha.CaptchaProperties;
|
||
import com.youlai.system.service.AuthService;
|
||
import io.jsonwebtoken.Claims;
|
||
import jakarta.servlet.http.HttpServletRequest;
|
||
import lombok.RequiredArgsConstructor;
|
||
import org.springframework.data.redis.core.StringRedisTemplate;
|
||
import org.springframework.security.authentication.AuthenticationManager;
|
||
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
|
||
import org.springframework.security.core.Authentication;
|
||
import org.springframework.security.core.context.SecurityContextHolder;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.web.context.request.RequestContextHolder;
|
||
import org.springframework.web.context.request.ServletRequestAttributes;
|
||
|
||
import java.awt.*;
|
||
import java.util.Date;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
/**
|
||
* 认证服务实现类
|
||
*
|
||
* @author haoxr
|
||
* @since 2.4.0
|
||
*/
|
||
@Service
|
||
@RequiredArgsConstructor
|
||
public class AuthServiceImpl implements AuthService {
|
||
|
||
private final AuthenticationManager authenticationManager;
|
||
private final StringRedisTemplate redisTemplate;
|
||
private final JwtTokenProvider jwtTokenProvider;
|
||
private final CodeGenerator codeGenerator;
|
||
private final Font captchaFont;
|
||
private final CaptchaProperties captchaProperties;
|
||
|
||
/**
|
||
* 登录
|
||
*
|
||
* @param username 用户名
|
||
* @param password 密码
|
||
* @return 登录结果
|
||
*/
|
||
@Override
|
||
public LoginResult login(String username, String password) {
|
||
UsernamePasswordAuthenticationToken authenticationToken =
|
||
new UsernamePasswordAuthenticationToken(username.toLowerCase().trim(), password);
|
||
Authentication authentication = authenticationManager.authenticate(authenticationToken);
|
||
String accessToken = jwtTokenProvider.createToken(authentication);
|
||
return LoginResult.builder()
|
||
.tokenType("Bearer")
|
||
.accessToken(accessToken)
|
||
.build();
|
||
}
|
||
|
||
/**
|
||
* 注销
|
||
*/
|
||
@Override
|
||
public void logout() {
|
||
HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
|
||
String token = jwtTokenProvider.resolveToken(request);
|
||
if (StrUtil.isNotBlank(token)) {
|
||
Claims claims = jwtTokenProvider.getTokenClaims(token);
|
||
String jti = claims.get("jti", String.class);
|
||
Date expiration = claims.getExpiration();
|
||
if (expiration != null) {
|
||
long ttl = expiration.getTime() - System.currentTimeMillis();
|
||
redisTemplate.opsForValue().set(CacheConstants.BLACKLIST_TOKEN_PREFIX + jti, null, ttl, TimeUnit.MILLISECONDS);
|
||
} else {
|
||
redisTemplate.opsForValue().set(CacheConstants.BLACKLIST_TOKEN_PREFIX + jti, null);
|
||
}
|
||
}
|
||
SecurityContextHolder.clearContext();
|
||
}
|
||
|
||
/**
|
||
* 获取验证码
|
||
*
|
||
* @return 验证码
|
||
*/
|
||
@Override
|
||
public CaptchaResult getCaptcha() {
|
||
|
||
String type = captchaProperties.getType();
|
||
int width = captchaProperties.getWidth();
|
||
int height = captchaProperties.getHeight();
|
||
int interfereCount = captchaProperties.getInterfereCount();
|
||
int codeLength = captchaProperties.getCode().getLength();
|
||
|
||
AbstractCaptcha captcha;
|
||
if ("circle".equalsIgnoreCase(type)) {
|
||
captcha = CaptchaUtil.createCircleCaptcha(width, height, codeLength, interfereCount);
|
||
} else if ("gif".equalsIgnoreCase(type)) {
|
||
captcha = CaptchaUtil.createGifCaptcha(width, height, codeLength);
|
||
} else if ("line".equalsIgnoreCase(type)) {
|
||
captcha = CaptchaUtil.createLineCaptcha(width, height, codeLength, interfereCount);
|
||
} else if ("shear".equalsIgnoreCase(type)) {
|
||
captcha = CaptchaUtil.createShearCaptcha(width, height, codeLength, interfereCount);
|
||
} else {
|
||
throw new IllegalArgumentException("Invalid captcha type: " + type);
|
||
}
|
||
captcha.setGenerator(codeGenerator);
|
||
captcha.setTextAlpha(captchaProperties.getTextAlpha());
|
||
captcha.setFont(captchaFont);
|
||
|
||
String captchaCode = captcha.getCode();
|
||
String imageBase64Data = captcha.getImageBase64Data();
|
||
|
||
// 验证码文本缓存至Redis，用于登录校验
|
||
String captchaKey = IdUtil.fastSimpleUUID();
|
||
redisTemplate.opsForValue().set(CacheConstants.CAPTCHA_CODE_PREFIX + captchaKey,captchaCode,
|
||
captchaProperties.getExpireSeconds(), TimeUnit.SECONDS);
|
||
|
||
return CaptchaResult.builder()
|
||
.captchaKey(captchaKey)
|
||
.captchaBase64(imageBase64Data)
|
||
.build();
|
||
}
|
||
|
||
}
|