package wox.serial;

import java.io.*;

/**
 * The Easy class is used to serialize/de-serialize objects to/from XML.
 *
 * 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 Easy
{
    private final WoxHandlers mWoxHandlers = new WoxHandlers();

    public void add( WoxFieldNameMapper pWoxMapper )
    {
        mWoxHandlers.addWoxFieldNameMapper( pWoxMapper );
    }

    public void add( WoxTypeMapper pWoxMapper )
    {
        mWoxHandlers.addTypeMapper( pWoxMapper );
    }

    public void add( WoxTypeHandler pWoxHandler )
    {
        mWoxHandlers.addTypeHandler( pWoxHandler );
    }

    public void add( WoxGenericsHandler pWoxHandler )
    {
        mWoxHandlers.addGenericsHandler( pWoxHandler );
    }

    public void add( WoxIsAHandler pWoxHandler )
    {
        mWoxHandlers.addIsAHandler( pWoxHandler );
    }

    /**
     * Format pObject to XML
     *
     * @param pObject - object to serialize
     *
     * @return xml form
     */
    public String format( Object pObject )
    {
        return format( pObject, false );
    }

    /**
     * Format pObject to XML
     *
     * @param pObject      - object to serialize
     * @param pPrettyPrint - if true pretty print the xml - NOTE this may cause problems in the "wire" protocol
     *
     * @return xml form
     */
    public String format( Object pObject, boolean pPrettyPrint )
    {
        try
        {
            StringWriter zWriter = new StringWriter();
            buildXML( zWriter, pObject, pPrettyPrint, false );
            return zWriter.toString().replace( "\r\n", "\n" ).trim();
        }
        catch ( IOException e )
        {
            throw new IllegalStateException( e ); // With StringWriter, should not be possible
        }
    }

    /**
     * Format an object's type data to XML
     *
     * @param pObject - object to serialize the type of
     */
    public String formatType( Object pObject )
            throws IOException
    {
        try
        {
            StringWriter zWriter = new StringWriter();
            buildXML( zWriter, pObject, true, true );
            return zWriter.toString().replace( "\r\n", "\n" ).trim();
        }
        catch ( IOException e )
        {
            throw new IllegalStateException( e ); // With StringWriter, should not be possible
        }
    }

    /**
     * Parse an object from a String
     *
     * @param pXMLData - !null
     */
    public Object parse( String pXMLData )
    {
        Reader zReader = new StringReader( pXMLData );
        try
        {
            return parseFromReader( zReader );
        }
        catch ( IOException e )
        {
            throw new IllegalStateException( e ); // With StringReader, should not be possible
        }
    }

    /**
     * Internal code to do the work of formatting
     *
     * @param pWriter      - !null writer
     * @param pObject      - null object to serialize
     * @param pPrettyPrint - if true pretty print the xml - NOTE this may cause in the "wire" protocol
     * @param pJustType    - if true, then No Data, just type info on object and its fields
     */
    protected void buildXML( Writer pWriter, Object pObject, boolean pPrettyPrint, boolean pJustType )
            throws IOException
    {
        XMLwriter zWriter = new XMLwriterImpl( pWriter, pPrettyPrint );
        try
        {
            WoxWriter woxWriter = new SimpleWriter( mWoxHandlers );
            woxWriter.write( pObject, null, zWriter, pJustType );
        }
        catch ( RuntimeException e )
        {
            e.printStackTrace();
            throw e;
        }
        finally
        {
            zWriter.dispose();
        }
    }

    /**
     * Internal code to do the parsing
     *
     * @param pReader - !null reader
     */
    protected Object parseFromReader( Reader pReader )
            throws IOException
    {
        XMLreader zReader = new XMLreaderImpl( pReader );
        try
        {
            //this creates the WOX reader
            SimpleReader woxReader = new SimpleReader( mWoxHandlers );
            if ( zReader.readStartElement() )
            {
                //reads the object from the XML file. We pass the xmlReader positioned in the first node!
                return woxReader.read( zReader );
            }
        }
        catch ( RuntimeException e )
        {
            e.printStackTrace();
            throw e;
        }
        finally
        {
            zReader.dispose();
        }
        return null;
    }
}