diff --git a/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java b/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java index 32d5e1d8a6cf1accff5a0336945c5d336aca5eb0..b2a41228c630fbf7d90459bbd2e27d38f57e4151 100644 --- a/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java +++ b/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java @@ -11,9 +11,7 @@ import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.*; -/** - * @author adward - */ + @Slf4j public class AiFileUtil { private AiFileUtil() { diff --git a/src/main/java/com/jeesite/modules/util/vir/SecurityUtils.java b/src/main/java/com/jeesite/modules/util/vir/SecurityUtils.java new file mode 100644 index 0000000000000000000000000000000000000000..b6741ac596c9f4dde52db0ee08a000ca2dde6540 --- /dev/null +++ b/src/main/java/com/jeesite/modules/util/vir/SecurityUtils.java @@ -0,0 +1,41 @@ +package com.vir.ai.admin.utils; + +import cn.hutool.json.JSONObject; +import com.vir.ai.admin.controller.vo.project.JwtUser; +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; +import org.apache.commons.lang3.ArrayUtils; +import org.apache.commons.lang3.StringUtils; +import org.springframework.security.core.Authentication; +import org.springframework.security.core.context.SecurityContextHolder; +import org.springframework.web.context.request.RequestContextHolder; +import org.springframework.web.context.request.ServletRequestAttributes; + +import javax.servlet.http.HttpServletRequest; +import java.io.IOException; + + +public class SecurityUtils { + + public static Authentication getAuthentication() { + return SecurityContextHolder.getContext().getAuthentication(); + } + + + public static String getCurrentUser() { + Authentication authentication = SecurityContextHolder.getContext().getAuthentication(); + if (authentication != null) { + Object principal = authentication.getPrincipal(); + /* if (principal instanceof JwtUser) { + return (JwtUser) principal; + }*/ + JSONObject jsonObject = new JSONObject(principal); + Object claims = jsonObject.get("claims"); + JSONObject claimsJsonObject2 = new JSONObject(claims); + return claimsJsonObject2.get("userId").toString(); + } + return null; + } + + +}