package com.aurea.maven.plugins.util;
   
   import java.io.InputStream;
   import java.io.OutputStream;
   
   import javax.xml.bind.JAXBContext;
   import javax.xml.bind.Marshaller;
   import javax.xml.bind.Unmarshaller;
   import javax.xml.bind.annotation.XmlRootElement;
  
  public class JAXBHelper {
  
  	public static void marshal(Object o, OutputStream os) throws Exception {
  		try {
  			String jp = getJAXBPackage(o.getClass());
  			if (jp != null) {
  				JAXBContext ctxt = JAXBContext.newInstance(getJAXBPackage(o.getClass()));
  				Marshaller m = ctxt.createMarshaller();
  				m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
  				m.marshal(o, os);
  			}
  		} catch (Exception e) {
  			e.printStackTrace();
  			throw (e);
  		}
  	}
  
  	public static Object unmarshal(Class<?> c, InputStream is) throws Exception {
  
  		Object result = null;
  		String jp = getJAXBPackage(c);
  
  		try {
  			if (jp != null) {
  				JAXBContext ctxt = JAXBContext.newInstance(c);
  				Unmarshaller um = ctxt.createUnmarshaller();
  				result = um.unmarshal(is);
  			}
  		} catch (Exception e) {
  			e.printStackTrace();
  			throw (e);
  		}
  
  		return result;
  	}
  
  	@SuppressWarnings("unchecked")
  	private static String getJAXBPackage(@SuppressWarnings("rawtypes") Class clazz) {
  		if (clazz.getAnnotation(XmlRootElement.class) != null) {
  			return clazz.getPackage().getName();
  		} else {
  			if (clazz.getSuperclass() != null) {
  				return getJAXBPackage(clazz.getSuperclass());
  			} else {
  				return null;
  			}
  		}
  	}
  }