/*
* $Id: ObjectToFeed.java 20321 2010-11-24 15:21:24Z dfeist $
* --------------------------------------------------------------------------------------
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
*
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.module.atom.transformers;
import org.mule.api.transformer.TransformerException;
import org.mule.transformer.AbstractDiscoverableTransformer;
import org.mule.transformer.types.DataTypeFactory;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.StringReader;
import org.apache.abdera.Abdera;
import org.apache.abdera.model.Document;
import org.apache.abdera.model.Element;
import org.apache.abdera.model.Feed;
import org.apache.abdera.parser.Parser;
/**
* <code>ObjectToInputStream</code> converts serilaizable object to a input stream but
* treats <code>java.lang.String</code> differently by converting to bytes using
* the <code>String.getBytes()</code> method.
*/
public class ObjectToFeed extends AbstractDiscoverableTransformer
{
public ObjectToFeed()
{
this.registerSourceType(DataTypeFactory.BYTE_ARRAY);
this.registerSourceType(DataTypeFactory.INPUT_STREAM);
this.registerSourceType(DataTypeFactory.STRING);
setReturnDataType(DataTypeFactory.create(Feed.class));
}
@Override
public Object doTransform(Object src, String outputEncoding) throws TransformerException
{
try
{
Parser parser = Abdera.getInstance().getParser();
Document<Element> doc;
if (src instanceof InputStream)
{
doc = parser.parse((InputStream) src, outputEncoding);
}
else if (src instanceof byte[])
{
doc = parser.parse(new ByteArrayInputStream((byte[]) src), outputEncoding);
}
else
{
doc = parser.parse(new StringReader((String) src));
}
//we only need to check for the registered source types
return doc.getRoot();
}
catch (Exception e)
{
throw new TransformerException(this, e);
}
}
}