org.opendatakit.odktables.entity.serialization.SimpleHTMLMessageWriter.java Source code

Introduction

Here is the source code for org.opendatakit.odktables.entity.serialization.SimpleHTMLMessageWriter.java
Source

/*
 * Copyright (C) 2012-2013 University of Washington
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package org.opendatakit.odktables.entity.serialization;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.nio.charset.Charset;

import javax.ws.rs.Produces;
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 org.opendatakit.aggregate.odktables.rest.ApiConstants;

import com.fasterxml.jackson.dataformat.xml.XmlMapper;

@Produces({ "text/*" })
@Provider
public class SimpleHTMLMessageWriter<T> implements MessageBodyWriter<T> {

    private static final XmlMapper mapper = new XmlMapper();
    private static final String DEFAULT_ENCODING = "utf-8";

    @Override
    public boolean isWriteable(Class<?> type, Type genericType, Annotation annotations[], MediaType mediaType) {
        return mediaType.getType().equals("text") && !mediaType.getSubtype().equals("xml");
    }

    @Override
    public void writeTo(T o, Class<?> aClass, Type type, Annotation[] annotations, MediaType mediaType,
            MultivaluedMap<String, Object> map, OutputStream rawStream)
            throws IOException, WebApplicationException {
        String encoding = getCharsetAsString(mediaType);
        try {
            if (!encoding.equalsIgnoreCase(DEFAULT_ENCODING)) {
                throw new IllegalArgumentException("charset for the response is not utf-8");
            }
            // write it to a byte array
            ByteArrayOutputStream bas = new ByteArrayOutputStream(8192);
            OutputStreamWriter w = new OutputStreamWriter(bas, Charset.forName(ApiConstants.UTF8_ENCODE));
            w.write("<html><head></head><body>");
            w.write(mapper.writeValueAsString(o));
            w.write("</body></html>");
            w.flush();
            w.close();
            // get the array
            byte[] bytes = bas.toByteArray();
            map.putSingle(ApiConstants.OPEN_DATA_KIT_VERSION_HEADER, ApiConstants.OPEN_DATA_KIT_VERSION);
            map.putSingle("Access-Control-Allow-Origin", "*");
            map.putSingle("Access-Control-Allow-Credentials", "true");

            rawStream.write(bytes);
            rawStream.flush();
            rawStream.close();

        } catch (Exception e) {
            throw new IOException(e);
        }
    }

    @Override
    public long getSize(T arg0, Class<?> arg1, Type arg2, Annotation[] arg3, MediaType arg4) {
        return -1;
    }

    protected static String getCharsetAsString(MediaType m) {
        if (m == null) {
            return DEFAULT_ENCODING;
        }
        String result = m.getParameters().get("charset");
        return (result == null) ? DEFAULT_ENCODING : result;
    }
}