com.crosstreelabs.jaxrs.api.versioned.providers.HierarchicalMapFormUrlEncodedProvider Maven / Gradle / Ivy

Go to download
/*
 * 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 com.crosstreelabs.jaxrs.api.versioned.providers;

import com.crosstreelabs.jaxrs.api.versioned.util.AnnotationUtils;
import com.crosstreelabs.jaxrs.api.versioned.util.EncodingUtils;
import com.crosstreelabs.jaxrs.api.versioned.util.QueryStringUtils;
import com.crosstreelabs.jaxrs.api.versioned.util.StreamUtils;
import com.crosstreelabs.jaxrs.api.versioned.util.StringUtils;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.Consumes;
import javax.ws.rs.Encoded;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.MessageBodyReader;
import javax.ws.rs.ext.MessageBodyWriter;

@Produces("application/x-www-form-urlencoded")
@Consumes("application/x-www-form-urlencoded")
public class HierarchicalMapFormUrlEncodedProvider
        implements MessageBodyReader,
                MessageBodyWriter {
    
    @Context
    private HttpServletRequest request;
    
    @Override
    public boolean isReadable(final Class type,
            final Type genericType,
            final Annotation[] annotations,
            final MediaType mediaType) {
        return Map.class.equals(type) || HashMap.class.equals(type);
    }

    @Override
    public Map readFrom(
            final Class type,
            final Type genericType,
            final Annotation[] annotations,
            final MediaType mediaType,
            final MultivaluedMap httpHeaders,
            final InputStream entityStream) throws IOException {
        return readFrom(request, annotations,entityStream);
    }
    public Map readFrom(
            final HttpServletRequest request,
            final Annotation[] annotations,
            final InputStream entityStream) throws IOException {
        // Request body params
        Map result
                = QueryStringUtils.toMap(StreamUtils.toString(entityStream));
        
        // Query string params
        if (request != null) {
            Map tmp = request.getParameterMap();
            Collection coll = new ArrayList<>();
            for (String key : tmp.keySet()) {
                for (String value : tmp.get(key)) {
                    coll.add(key+"="+value);
                }
            }
            String tmpStr = StringUtils.join(coll,"&");
            result.putAll(QueryStringUtils.toMap(tmpStr));
        }
        
        // Decode if required
        if (AnnotationUtils.find(Encoded.class, annotations) == null) {
            return (Map)EncodingUtils.decode(result, StandardCharsets.UTF_8);
        }
        return result;
    }

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

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

    @Override
    public void writeTo(final Map data,
            final Class type,
            final Type genericType,
            final Annotation[] annotations,
            final MediaType mediaType,
            final MultivaluedMap httpHeaders,
            final OutputStream entityStream) throws IOException {
        
        boolean encoded = AnnotationUtils.find(Encoded.class, annotations) != null;
        entityStream.write(QueryStringUtils.toQueryString(data, encoded, StandardCharsets.UTF_8).getBytes(StandardCharsets.UTF_8));
    }

}