io.vertx.up.uca.rs.hunt.Outcome Maven / Gradle / Ivy

Go to download
package io.vertx.up.uca.rs.hunt;

import io.vertx.core.http.HttpHeaders;
import io.vertx.core.http.HttpServerResponse;
import io.vertx.up.commune.Envelop;
import io.vertx.up.eon.Strings;

import javax.ws.rs.core.MediaType;
import java.util.Objects;
import java.util.Set;

/**
 * Content-Type
 * Accept
 */
final class Outcome {

    static void media(final HttpServerResponse response, final Set produces) {
        /*
         * @Produces means that server generate response to client
         */
        if (produces.isEmpty()) {
            /*
             * Default situation
             */
            response.putHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON);
        } else {
            if (produces.contains(MediaType.WILDCARD_TYPE)) {
                /*
                 * Default situation
                 */
                response.putHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON);
            } else {
                /*
                 * MediaType extract from first set
                 */
                final MediaType type = produces.iterator().next();
                if (Objects.isNull(type)) {
                    /*
                     * Default situation
                     */
                    response.putHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON);
                } else {
                    /*
                     * Type + SLASH + SubType
                     */
                    final String content = type.getType() + Strings.SLASH + type.getSubtype();
                    response.putHeader(HttpHeaders.CONTENT_TYPE, content);
                }
            }
        }
    }

    static void out(final HttpServerResponse response, final Envelop envelop, final Set produces) {
        /*
         * Set response data for current request / event
         * The mime type has been set before this step ( mime ( HttpServerResponse, Set )
         */
        if (!response.ended()) {
            final String headerStr = response.headers().get(HttpHeaders.CONTENT_TYPE);
            if (Objects.isNull(headerStr)) {
                /*
                 * Default String mode
                 */
                response.end(envelop.toString());
            } else {
                /*
                 * Extract the data `MediaType` from response header
                 */
                final MediaType type = MediaType.valueOf(headerStr);
                if (MediaType.WILDCARD_TYPE.equals(type)) {
                    /*
                     * Default String mode
                     */
                    response.end(envelop.outString());
                } else {
                    /*
                     * Download stream here for file download here
                     */
                    if (MediaType.APPLICATION_OCTET_STREAM_TYPE.equals(type)) {
                        /*
                         * Buffer only ( Buffer output )
                         */
                        response.end(envelop.outBuffer());
                    } else {
                        /*
                         * Other situation for uniform response
                         */
                        response.end(envelop.outString());
                    }
                }
            }
        }
        response.closed();
    }
}