org.baswell.routes.ResponseTypeMapper.java Source code

Introduction

Here is the source code for org.baswell.routes.ResponseTypeMapper.java

Source

/*
 * Copyright 2015 Corey Baswell
 *
 * 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.baswell.routes;

import javax.xml.bind.annotation.XmlRootElement;
import java.io.InputStream;
import java.lang.reflect.Method;
import java.util.Set;

import static org.baswell.routes.RoutesMethods.*;
import static org.baswell.routes.Pair.*;

class ResponseTypeMapper {
    static ResponseType mapResponseType(Method method, RouteConfiguration routeConfiguration) {
        Class returnType = method.getReturnType();
        if ((returnType == void.class) || (returnType == Void.class)) {
            return ResponseType.VOID;
        } else if (returnType.isArray() && (returnType.getComponentType() == byte.class)) {
            return ResponseType.BYTES_CONTENT;
        } else if (returnType == InputStream.class) {
            return ResponseType.STREAM_CONTENT;
        } else if ((returnType == String.class) && !routeConfiguration.returnedStringIsContent) {
            return ResponseType.FORWARD_DISPATCH;
        } else {
            return ResponseType.STRING_CONTENT;
        }
    }

    static Pair<ResponseStringWriteStrategy, String> mapResponseStringWriteStrategy(Method method,
            Set<MediaType> respondToMedia, String contentType, AvailableLibraries availableLibraries) {
        return mapResponseStringWriteStrategy(method.getReturnType(), respondToMedia, contentType,
                availableLibraries);
    }

    static Pair<ResponseStringWriteStrategy, String> mapResponseStringWriteStrategy(Object returnedObject,
            Set<MediaType> respondToMedia, String contentType, AvailableLibraries availableLibraries) {
        return mapResponseStringWriteStrategy(returnedObject.getClass(), respondToMedia, contentType,
                availableLibraries);
    }

    static Pair<ResponseStringWriteStrategy, String> mapResponseStringWriteStrategy(Class returnType,
            Set<MediaType> respondToMedia, String contentType, AvailableLibraries availableLibraries) {
        String returnClassName = returnType.getCanonicalName();
        String returnTypePackage = returnType.getPackage().getName();

        MediaType mediaType = contentType == null ? null : MediaType.findFromMimeType(contentType);
        if ((mediaType == null) && (respondToMedia != null) && (respondToMedia.size() == 1)) {
            mediaType = respondToMedia.iterator().next();
        }

        if (returnTypePackage.startsWith("org.json")) {
            return pair(ResponseStringWriteStrategy.TO_STRING, MIMETypes.JSON);
        } else if (returnTypePackage.startsWith("org.w3c.dom")) {
            return pair(ResponseStringWriteStrategy.W3C_NODE, MIMETypes.XML);
        } else if (returnClassName.equals("org.jdom2.Document")) {
            return pair(ResponseStringWriteStrategy.JDOM2_DOCUMENT, MIMETypes.XML);
        } else if (returnClassName.equals("org.jdom2.Element")) {
            return pair(ResponseStringWriteStrategy.JDOM2_ELEMENT, MIMETypes.XML);
        } else if (returnType.getAnnotation(XmlRootElement.class) != null) {
            return pair(ResponseStringWriteStrategy.JAXB, MIMETypes.XML);
        } else if (classImplementsInterface(returnType, CharSequence.class)) {
            return pair(ResponseStringWriteStrategy.TO_STRING, contentType);
        } else if ((((mediaType != null) && (mediaType == MediaType.JSON))) && availableLibraries.gsonAvailable()) {
            return pair(ResponseStringWriteStrategy.GSON, MIMETypes.JSON);
        } else if ((((mediaType != null) && (mediaType == MediaType.JSON)))
                && availableLibraries.jacksonAvailable()) {
            return pair(ResponseStringWriteStrategy.JACKSON, MIMETypes.JSON);
        } else {
            return null;
        }
    }
}