uk.gov.ida.saml.hub.transformers.inbound.SamlStatusToIdaStatusCodeMapper Maven / Gradle / Ivy

Go to download
package uk.gov.ida.saml.hub.transformers.inbound;

import org.opensaml.saml.saml2.core.Status;
import org.opensaml.saml.saml2.core.StatusCode;
import org.opensaml.saml.saml2.core.StatusDetail;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import uk.gov.ida.saml.core.extensions.StatusValue;
import uk.gov.ida.saml.hub.domain.IdpIdaStatus;

import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Optional;

import static java.util.Optional.ofNullable;
import static java.util.stream.Collectors.toList;

public class SamlStatusToIdaStatusCodeMapper extends SamlStatusToAuthenticationStatusCodeMapper {

    private static final Logger LOG = LoggerFactory.getLogger(SamlStatusToIdaStatusCodeMapper.class);

    private final Map statusMappings;

    public SamlStatusToIdaStatusCodeMapper() {
        this.statusMappings = SamlStatusToIdpIdaStatusMappingsFactory.getSamlToIdpIdaStatusMappings();
    }

    @Override
    public Optional map(Status samlStatus) {
        final String statusCodeValue = getStatusCodeValue(samlStatus);
        final Optional subStatusCodeValue = getSubStatusCodeValue(samlStatus);
        final List statusDetailValues = getStatusDetailValues(samlStatus);

        Optional result =  statusMappings.keySet().stream()
                .filter(k -> k.matches(statusCodeValue, subStatusCodeValue, statusDetailValues))
                .findFirst()
                .map(statusMappings::get);

        if (result.isEmpty()){
            LOG.info(String.format("Could not map SAML Status: %s, %s, %s", statusCodeValue, subStatusCodeValue, String.join(",", statusDetailValues)));
        }

        return result;
    }

    private Optional getSubStatusCodeValue(final Status status) {
        return ofNullable(status.getStatusCode().getStatusCode()).map(StatusCode::getValue);
    }

    private List getStatusDetailValues(Status samlStatus) {
        Optional statusDetail = ofNullable(samlStatus.getStatusDetail());

        return statusDetail.map(x -> x.getUnknownXMLObjects().stream()
                        .filter(child -> child.getElementQName().getLocalPart().equals(StatusValue.DEFAULT_ELEMENT_LOCAL_NAME))
                        .map(statusDetailVal -> statusDetailVal.getDOM().getFirstChild().getTextContent()).collect(toList())
        ).orElseGet(Collections::emptyList);
    }
}