package com.buckosoft.PicMan.service.support;

 import java.io.IOException;

 import java.io.InputStream;

 import java.io.InputStreamReader;

 import java.io.OutputStream;

 import java.io.OutputStreamWriter;

 import java.lang.annotation.Annotation;

 import java.lang.reflect.Type;

 import javax.ws.rs.WebApplicationException;

 import javax.ws.rs.core.MediaType;

 import javax.ws.rs.core.MultivaluedMap;

 import javax.ws.rs.ext.MessageBodyReader;

 import javax.ws.rs.ext.MessageBodyWriter;

 import com.thoughtworks.xstream.XStream;

 import com.thoughtworks.xstream.io.xml.CompactWriter;

 import com.thoughtworks.xstream.io.xml.StaxDriver;

 public class XMLProvider implements MessageBodyReader<Object>, MessageBodyWriter<Object> {

         private static final XStream xstream = new XStream(new StaxDriver());

         //private static final Set<Class<?>> processed = new HashSet<Class<?>>();

         private static final String DEFAULT_ENCODING = "utf-8";

         @Override

         public boolean isWriteable(Class<?> type, Type genericType,

                         Annotation[] annotations, MediaType mediaType) {

                 return true;

         }

         @Override

         public boolean isReadable(Class<?> type, Type genericType,

                         Annotation[] annotations, MediaType mediaType) {

                 return true;

         }

         protected static String getCharsetAsString(MediaType m) {

                 if (m == null) {

                         return DEFAULT_ENCODING;

                 }

                 String result = m.getParameters().get("charset");

                 return (result == null) ? DEFAULT_ENCODING : result;

         }

         protected XStream getXStream(Class<?> type) {

                 return xstream;

         }

         @Override

         public long getSize(Object t, Class<?> type, Type genericType,

                         Annotation[] annotations, MediaType mediaType) {

                 return -1;

         }

         @Override

         public void writeTo(Object t, Class<?> type, Type genericType,

                         Annotation[] annotations, MediaType mediaType,

                         MultivaluedMap<String, Object> httpHeaders,

                         OutputStream entityStream) throws IOException,

                         WebApplicationException {                

                 String encoding = getCharsetAsString(mediaType);

                 XStream xStream = getXStream(t.getClass());

                 xStream.marshal(t, new CompactWriter(new OutputStreamWriter(entityStream, encoding)));

         }

         @Override

         public Object readFrom(Class<Object> type, Type genericType,

                         Annotation[] annotations, MediaType mediaType,

                         MultivaluedMap<String, String> httpHeaders, InputStream entityStream)

                         throws IOException, WebApplicationException {

                 String encoding = getCharsetAsString(mediaType);

                 XStream xStream = getXStream(type);

                 return xStream.fromXML(new InputStreamReader(entityStream, encoding));

         }

 }