io.irain.reactor.security.authentication.AuthenticationManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of shore-security Show documentation

shore security starter

There is a newer version: 1.0.3

package io.irain.reactor.security.authentication;

import io.irain.reactor.security.domain.TokenAuthentication;
import io.irain.reactor.security.grant.AuthenticationGrantManager;
import lombok.RequiredArgsConstructor;
import org.springframework.security.authentication.ReactiveAuthenticationManager;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Component;
import reactor.core.publisher.Mono;


/**
 * @author youta
 **/
@Component
@RequiredArgsConstructor
public class AuthenticationManager implements ReactiveAuthenticationManager {

    private final AuthenticationGrantManager authenticationGrantManager;

    @Override
    public Mono authenticate(Authentication authentication) {
        // 已经通过验证，直接返回
        if (authentication.isAuthenticated())
            return Mono.just(authentication);
        return Mono.justOrEmpty(authentication)
                .cast(TokenAuthentication.class)
                .flatMap(this::getTokenAuthentication);

    }

    /**
     * 获取token认证
     *
     * @param tokenAuthentication {@link TokenAuthentication}
     * @return {@link TokenAuthentication}
     */
    private Mono getTokenAuthentication(TokenAuthentication tokenAuthentication) {
        return this.authenticationGrantManager.grant(tokenAuthentication.getLoginUser().getType(),
                        authenticationGrant -> authenticationGrant.userDetails(tokenAuthentication.getLoginUser()))
                .flatMap(userDetails -> {
                    TokenAuthentication authentications = new TokenAuthentication(userDetails, tokenAuthentication);
                    authentications.setAuthenticated(true);
                    SecurityContextHolder.getContext().setAuthentication(authentications);
                    return Mono.just(authentications);
                });
    }

}