package org.jboss.axis.message;
import org.jboss.axis.AxisFault;
import org.jboss.axis.Constants;
import org.jboss.axis.MessageContext;
import org.jboss.axis.encoding.DeserializationContext;
import org.jboss.axis.soap.SOAPConstants;
import org.jboss.axis.utils.Messages;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import javax.xml.namespace.QName;
public class EnvelopeBuilder extends SOAPHandler
{
private SOAPEnvelopeAxisImpl envelope;
private SOAPConstants soapConstants = SOAPConstants.SOAP11_CONSTANTS;
private boolean gotHeader = false;
private boolean gotBody = false;
public EnvelopeBuilder(String messageType, SOAPConstants soapConstants)
{
envelope = new SOAPEnvelopeAxisImpl(false, soapConstants);
envelope.setMessageType(messageType);
myElement = envelope;
}
public EnvelopeBuilder(SOAPEnvelopeAxisImpl env, String messageType)
{
envelope = env;
envelope.setMessageType(messageType);
myElement = envelope;
}
public SOAPEnvelopeAxisImpl getEnvelope()
{
return envelope;
}
public void startElement(String namespace, String localName,
String prefix, Attributes attributes,
DeserializationContext context)
throws SAXException
{
if (!localName.equals(Constants.ELEM_ENVELOPE))
throw new SAXException(Messages.getMessage("badTag00", localName));
MessageContext msgContext = context.getMessageContext();
SOAPConstants singleVersion = null;
if (msgContext != null)
{
singleVersion = (SOAPConstants)msgContext.getProperty(Constants.MC_SINGLE_SOAP_VERSION);
}
if (namespace.equals(Constants.URI_SOAP11_ENV))
{
soapConstants = SOAPConstants.SOAP11_CONSTANTS;
}
else if (namespace.equals(Constants.URI_SOAP12_ENV))
{
soapConstants = SOAPConstants.SOAP12_CONSTANTS;
}
else
{
soapConstants = null;
}
if ((soapConstants == null) ||
(singleVersion != null && soapConstants != singleVersion))
{
soapConstants = SOAPConstants.SOAP11_CONSTANTS;
if (singleVersion == null) singleVersion = soapConstants;
try
{
AxisFault fault = new AxisFault(soapConstants.getVerMismatchFaultCodeQName(),
null, Messages.getMessage("versionMissmatch00"), null, null, null);
SOAPHeaderElementAxisImpl newHeader = new
SOAPHeaderElementAxisImpl(soapConstants.getEnvelopeURI(),
Constants.ELEM_UPGRADE);
SOAPElementAxisImpl innerHeader = new
SOAPElementAxisImpl(soapConstants.getEnvelopeURI(),
Constants.ELEM_SUPPORTEDENVELOPE);
innerHeader.addAttribute(null, Constants.ATTR_QNAME,
new QName(singleVersion.getEnvelopeURI(), Constants.ELEM_ENVELOPE));
newHeader.addChildElement(innerHeader);
fault.addHeader(newHeader);
throw new SAXException(fault);
}
catch (javax.xml.soap.SOAPException e)
{
throw new SAXException(e);
}
}
if (context.getMessageContext() != null)
context.getMessageContext().setSOAPConstants(soapConstants);
if (soapConstants == SOAPConstants.SOAP12_CONSTANTS &&
attributes.getValue(Constants.URI_SOAP12_ENV, Constants.ATTR_ENCODING_STYLE) != null)
{
AxisFault fault = new AxisFault(Constants.FAULT_SOAP12_SENDER,
null, Messages.getMessage("noEncodingStyleAttrAppear", "Envelope"), null, null, null);
throw new SAXException(fault);
}
envelope.setPrefix(prefix);
envelope.setNamespaceURI(namespace);
envelope.setNSMappings(context.getCurrentNSMappings());
envelope.setSoapConstants(soapConstants);
context.pushNewElement(envelope);
}
public SOAPHandler onStartChild(String namespace,
String localName,
String prefix,
Attributes attributes,
DeserializationContext context)
throws SAXException
{
QName thisQName = new QName(namespace, localName);
if (thisQName.equals(soapConstants.getHeaderQName()))
{
if (gotHeader)
throw new SAXException(Messages.getMessage("only1Header00"));
gotHeader = true;
return new HeaderBuilder(envelope);
}
if (thisQName.equals(soapConstants.getBodyQName()))
{
if (gotBody)
throw new SAXException(Messages.getMessage("only1Body00"));
gotBody = true;
return new BodyBuilder(envelope);
}
if (!gotBody)
throw new SAXException(Messages.getMessage("noCustomElems00"));
if (soapConstants == SOAPConstants.SOAP12_CONSTANTS)
{
throw new SAXException(Messages.getMessage("noElemAfterBody12"));
}
try
{
SOAPElementAxisImpl element = new SOAPElementAxisImpl(namespace, localName, prefix,
attributes, context);
if (element.getFixupDeserializer() != null)
return (SOAPHandler)element.getFixupDeserializer();
}
catch (AxisFault axisFault)
{
throw new SAXException(axisFault);
}
return null;
}
public void onEndChild(String namespace, String localName,
DeserializationContext context)
{
}
public void endElement(String namespace, String localName,
DeserializationContext context)
throws SAXException
{
envelope.setDirty(false);
}
}