cloud.agileframework.spring.util.SecurityUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-util Show documentation

Spring extension Toolset

There is a newer version: 2.2.0

package cloud.agileframework.spring.util;

import com.google.common.collect.Lists;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.User;
import org.springframework.security.core.userdetails.UserDetails;

import javax.servlet.http.HttpServletRequest;

/**
 * @author 佟盟
 * 日期 2020-10-21 12:04
 * 描述 TODO
 * @version 1.0
 * @since 1.0
 */
public class SecurityUtil {
    private static final String AGILE_SECURITY = "$AGILE_SECURITY";
    private static final String AGILE_SECURITY_USER_NAME = "$AGILE_SECURITY_USER_NAME";
    private static final String ANONYMOUS_USERNAME = "anonymous";

    public static UserDetails currentUser() {
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication == null) {
            authentication = (Authentication) ServletUtil.getCurrentRequest().getAttribute(AGILE_SECURITY);
        }
        if (authentication != null && authentication.getPrincipal() instanceof UserDetails) {

            return (UserDetails) authentication.getPrincipal();
        }
        return new User(ANONYMOUS_USERNAME, "", Lists.newArrayList());
    }

    public static String currentUsername() {
        Object username = ServletUtil.getCurrentRequest().getAttribute(AGILE_SECURITY_USER_NAME);
        return username == null ? ANONYMOUS_USERNAME : (String) username;
    }

    public static void setCurrentUser(HttpServletRequest request, Authentication currentAuthentication) {
        request.setAttribute(AGILE_SECURITY, currentAuthentication);
        request.setAttribute(AGILE_SECURITY_USER_NAME, ((UserDetails) currentAuthentication.getPrincipal()).getUsername());
    }
}