package org.codehaus.xfire.soap;

import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;

import org.codehaus.xfire.MessageContext;
import org.codehaus.xfire.exchange.AbstractMessage;
import org.codehaus.xfire.exchange.InMessage;
import org.codehaus.xfire.exchange.MessageSerializer;
import org.codehaus.xfire.exchange.OutMessage;
import org.codehaus.xfire.fault.XFireFault;
import org.codehaus.yom.Element;
import org.codehaus.yom.Elements;
import org.codehaus.yom.stax.StaxSerializer;

public class SoapSerializer
    implements MessageSerializer
{
    public static final String SERIALIZE_PROLOG = "xfire.serializeProlog";
    private MessageSerializer serializer;

    public SoapSerializer(MessageSerializer serializer)
    {
        this.serializer = serializer;
    }

    /**
     * Sends a message wrapped in a SOAP Envelope and Body.
     * 
     * @param message
     * @param writer
     * @param context
     * @throws XFireFault
     */

    public void writeMessage(OutMessage message, XMLStreamWriter writer, MessageContext context)
        throws XFireFault
    {
        try
        {
            QName env = message.getSoapVersion().getEnvelope();

            Boolean serializeObj = (Boolean) context.getProperty(SERIALIZE_PROLOG);
            boolean serializeProlog = (serializeObj != null) ? serializeObj.booleanValue() : true;
            if (serializeProlog)
                writer.writeStartDocument(message.getEncoding(), "1.0");
            
            writer.setPrefix(env.getPrefix(), env.getNamespaceURI());
            writer.writeStartElement(env.getPrefix(),
                                     env.getLocalPart(),
                                     env.getNamespaceURI());
            writer.writeNamespace(env.getPrefix(), env.getNamespaceURI());

            if (message.getHeader() != null && message.getHeader().getChildCount() > 0)
            {
                QName header = message.getSoapVersion().getHeader();
                writer.writeStartElement(header.getPrefix(),
                                         header.getLocalPart(),
                                         header.getNamespaceURI());
                
                writeHeaders(message, writer);
                
                writer.writeEndElement();
            }
            
            QName body = message.getSoapVersion().getBody();
            writer.writeStartElement(body.getPrefix(),
                                     body.getLocalPart(),
                                     body.getNamespaceURI());
    
            serializer.writeMessage(message, writer, context);
    
            writer.writeEndElement();
            writer.writeEndElement();
            
            if (serializeProlog)
                writer.writeEndDocument();
    
            writer.close();
        }
        catch (XMLStreamException e)
        {
            throw new XFireFault("Couldn't write message.", e, XFireFault.RECEIVER);
        }
    }

    protected void writeHeaders(AbstractMessage msg, XMLStreamWriter writer)
            throws XMLStreamException
    {
        StaxSerializer ser = new StaxSerializer();

        Elements elements = msg.getHeader().getChildElements();
        for (int i = 0; i < elements.size(); i++)
        {
            Element e = elements.get(i);
            
            ser.writeElement(e, writer);
        }
    }

    public void readMessage(InMessage message, MessageContext context)
        throws XFireFault
    {
        throw new UnsupportedOperationException();
    }
}

@@ -829,24 +829,109 @@
 zer;%0A    %7D%0A%0A
+    public MessageSerializer getSerializer()%0A    %7B%0A        return serializer;%0A    %7D%0A%0A
     /**%0A    

Make it possible to get the inner serializer.