com.payneteasy.superfly.security.validator.CompoundAuthenticationValidator Maven / Gradle / Ivy

Go to download
package com.payneteasy.superfly.security.validator;

import java.util.HashSet;
import java.util.Set;

import org.springframework.security.core.Authentication;
import org.springframework.security.core.AuthenticationException;

import com.payneteasy.superfly.security.authentication.CompoundAuthentication;
import com.payneteasy.superfly.security.exception.PreconditionsException;

/**
 * Validator which validates that {@link CompoundAuthentication} contains
 * all the required authentications.
 *
 * @author Roman Puchkovskiy
 */
public class CompoundAuthenticationValidator implements AuthenticationValidator {

    private Class[] requiredClasses = new Class[]{};

    public void setRequiredClasses(Class[] classes) {
        this.requiredClasses = classes;
    }

    public void validate(Authentication auth) throws AuthenticationException {
        if (auth instanceof CompoundAuthentication) {
            CompoundAuthentication compound = (CompoundAuthentication) auth;
            validateCompound(compound);
        } else {
            throw new IllegalArgumentException(auth.getClass().getName() + " is not instance of a CompoundAuthentication");
        }
    }

    protected void validateCompound(CompoundAuthentication compound) {
        Authentication[] auths = compound.getReadyAuthentications();
        Set> classes = new HashSet>();
        for (Authentication auth : auths) {
            classes.add(auth.getClass());
        }
        for (Class clazz : requiredClasses) {
            if (!classes.contains(clazz)) {
                throw new PreconditionsException("Unexpected authentication class before authentication");
            }
        }
    }

}