package at.gv.egiz.eaaf.modules.pvp2.impl.verification;

import org.opensaml.messaging.context.MessageContext;
import org.opensaml.messaging.handler.AbstractMessageHandler;
import org.opensaml.messaging.handler.MessageHandlerException;
import org.opensaml.saml.common.SAMLObject;
import org.opensaml.saml.common.messaging.context.SAMLMessageInfoContext;
import org.opensaml.saml.common.messaging.context.SAMLPeerEntityContext;

import lombok.extern.slf4j.Slf4j;

@Slf4j
public class EaafMessageContextInitializationHandler extends AbstractMessageHandler<SAMLObject> {

  @Override
  protected void doInvoke(MessageContext<SAMLObject> messageContext) throws MessageHandlerException {
    log.trace("Injecting sub-context to SAML2 message ... ");
    messageContext.addSubcontext(new SAMLPeerEntityContext());
    messageContext.addSubcontext(new SAMLMessageInfoContext());

  }

}