ru.mynewtons.starter.oauth2.config.security.audit.UsernameAuditorAware Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-boot-oauth2-starter Show documentation

MyNewtons OAuth2 starter

There is a newer version: 1.0.2

Show newest version

package ru.mynewtons.starter.oauth2.config.security.audit;

import lombok.extern.slf4j.Slf4j;
import org.springframework.data.domain.AuditorAware;
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;

@Slf4j
public class UsernameAuditorAware implements AuditorAware {

    @Override
    public String getCurrentAuditor() {
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();

        if (authentication == null || !authentication.isAuthenticated()) {
            return null;
        }

        Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
        if (principal instanceof String) {
            return (String) principal;
        }
        if (principal instanceof User) {
            User user = (User) principal;
            return user.getUsername();
        }
        if (principal instanceof UserDetails) {
            UserDetails userDetails = (UserDetails) principal;
            return  userDetails.getUsername();
        }
        if (principal instanceof Authentication) {
            Authentication auth = (Authentication) principal;
            return auth.getName();
        }
        log.warn(" Object principal is not String/org.springframework.security.core.userdetails.User/Authentication");
        return null;
    }
}