com.manydesigns.portofino.pageactions.rest.messagebodywriters.StripesMessageBodyWriter Maven / Gradle / Ivy

Go to download
package com.manydesigns.portofino.pageactions.rest.messagebodywriters;

import com.manydesigns.elements.ElementsThreadLocals;
import net.sourceforge.stripes.action.Resolution;
import net.sourceforge.stripes.controller.StripesRequestWrapper;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import javax.ws.rs.ConstrainedTo;
import javax.ws.rs.RuntimeType;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.MessageBodyWriter;
import javax.ws.rs.ext.Provider;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;

/**
 * @author Angelo Lupo          - [email protected]
 * @author Giampiero Granatella - [email protected]
 * @author Emanuele Poggi       - [email protected]
 * @author Alessio Stalla       - [email protected]
 */
@Provider
@ConstrainedTo(RuntimeType.SERVER)
public class StripesMessageBodyWriter implements MessageBodyWriter {

    public static final String copyright =
            "Copyright (C) 2005-2017 ManyDesigns srl";

    @Override
    public boolean isWriteable(Class type, Type genericType, Annotation[] annotations, MediaType mediaType) {
        return Resolution.class.isAssignableFrom(type);
    }

    @Override
    public long getSize(Resolution resolution, Class type, Type genericType, Annotation[] annotations, MediaType mediaType) {
        return -1;
    }

    @Override
    public void writeTo(
            Resolution resolution, Class type, Type genericType, Annotation[] annotations, final MediaType mediaType,
            final MultivaluedMap httpHeaders, final OutputStream entityStream)
            throws IOException, WebApplicationException {
        // OutputStream and Writer for HttpServletResponseWrapper.
        final ServletOutputStream responseStream = new ServletOutputStream() {
            public boolean isReady() {
                return true;
            }

            /*@Override
            public void setWriteListener(WriteListener writeListener) {}*/

            @Override
            public void write(final int b) throws IOException {
                entityStream.write(b);
            }

            @Override
            public void write(byte[] b, int off, int len) throws IOException {
                entityStream.write(b, off, len);
            }

            @Override
            public void write(byte[] b) throws IOException {
                entityStream.write(b);
            }

            @Override
            public void flush() throws IOException {
                entityStream.flush();
            }

            @Override
            public void close() throws IOException {
                entityStream.close();
            }
        };
        final PrintWriter responseWriter = new PrintWriter(new OutputStreamWriter(entityStream, "UTF-8")); //TODO
        HttpServletRequest request = ElementsThreadLocals.getHttpServletRequest();
        HttpServletResponse response = ElementsThreadLocals.getHttpServletResponse();
        try {
            try {
                request = StripesRequestWrapper.findStripesWrapper(request);
            } catch (IllegalStateException e) {
                request = new StripesRequestWrapper(request);
            }
            resolution.execute(request, new HttpServletResponseWrapper(response) {

                @Override
                public ServletOutputStream getOutputStream() throws IOException {
                    return responseStream;
                }

                @Override
                public PrintWriter getWriter() throws IOException {
                    return responseWriter;
                }

                @Override
                public void addHeader(String name, String value) {
                    httpHeaders.add(name, value);
                }

                @Override
                public void setHeader(String name, String value) {
                    httpHeaders.putSingle(name, value);
                }

                @Override
                public String getContentType() {
                    return mediaType.toString();
                }
            });
            responseWriter.flush();
            responseStream.flush();
        } catch (IOException e) {
            throw e;
        } catch (Exception e) {
            throw new WebApplicationException(e);
        }
    }
}