package wox.serial;

import java.io.*;
import org.litesoft.core.typeutils.Objects;
import java.util.*;

import org.jdom.*;
import org.jdom.output.*;

/**
 * Version: 1.0
 * Author: Carlos R. Jaimez Gonzalez
 * Author: Simon M. Lucas
 * Site: http://woxserializer.sourceforge.net/
 *
 * Version: 1.5
 * Author: Steven M Lewis
 *
 * Version: 2.0
 * Author: George Smith
 * SVN: http://svn.xp-dev.com/svn/WOX2/
 * Note: XML form for vs 2 is more compact and therefor incompatible with vs 1
 */

public class XMLwriterImpl implements XMLwriter
{
    private Writer mWriter;
    private boolean mPrettyPrint;
    private Stack<Element> mElements = new Stack<Element>();
    private boolean mTopElementStartTagSealed;

    public XMLwriterImpl( Writer pWriter, boolean pPrettyPrint )
    {
        Util.assertRequired( "Writer", mWriter = pWriter );
        mPrettyPrint = pPrettyPrint;
    }

    public void writeStartElement( String pName )
    {
        Util.assertXmlName( "Name", pName );
        assertXmlNotWritten();
        Element zElement = new Element( pName );
        mElements.push( zElement );
        mTopElementStartTagSealed = false;
    }

    public void writeAttributeString( String pName, String pValue )
    {
        Util.assertXmlName( "Name", pName );
        Util.assertRequired( "Value", pValue );
        assertTopElementNotSealed( "writeAttributeString" );
        mElements.peek().setAttribute( pName, XMLutils.encodeText( pValue ) );
    }

    public void writeElementText( String pElementText )
    {
        Util.assertRequired( "ElementText", pElementText );
        assertTopElementNotSealed( "writeElementText" );
        mElements.peek().setText( XMLutils.encodeText( pElementText ) );
        mTopElementStartTagSealed = true;
    }

    public void writeEndElement()
    {
        assertIsTopElement();
        mTopElementStartTagSealed = true;
        Element zElement = mElements.pop();
        if ( !mElements.isEmpty() )
        {
            mElements.peek().addContent( zElement );
            return;
        }
        Format zFormat = mPrettyPrint ? Format.getPrettyFormat() : Format.getCompactFormat();
        XMLOutputter out = new XMLOutputter( zFormat.setOmitDeclaration( false ) );
        try
        {
            out.output( zElement, mWriter );
            Closeable zCloseable = mWriter;
            mWriter = null;
            zCloseable.close();
        }
        catch ( IOException e )
        {
            throw new IllegalStateException( e );
        }
        finally
        {
            dispose();
        }
    }

    public void dispose()
    {
        if ( mWriter != null )
        {
            mElements.clear();
            try
            {
                mWriter.close();
            }
            catch ( IOException e )
            {
                // Whatever
            }
            mWriter = null;
        }
    }

    @Override
    protected void finalize()
            throws Throwable
    {
        dispose();
        super.finalize();
    }

    private void assertXmlNotWritten()
    {
        if ( mWriter == null )
        {
            throw new IllegalStateException( "Root Element already closed and Written" );
        }
    }

    private void assertIsTopElement()
    {
        assertXmlNotWritten();
        if ( mElements.isEmpty() )
        {
            throw new IllegalStateException( "No Root Element Written" );
        }
    }

    private void assertTopElementNotSealed( String pWhat )
    {
        assertIsTopElement();
        if ( mTopElementStartTagSealed )
        {
            throw new IllegalStateException( "May not '" + pWhat + "', as the current Element has already been sealed" );
        }
    }
}