io.joshworks.snappy.http.body.Body Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of snappy Show documentation
The newest version!
/*
 * Copyright 2017 Josue Gontijo
 *
 * 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 io.joshworks.snappy.http.body;

import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import io.joshworks.snappy.http.ConnegHandler;
import io.joshworks.snappy.http.JsonNode;
import io.joshworks.snappy.http.MediaType;
import io.joshworks.snappy.parser.Parser;
import io.joshworks.snappy.parser.ParserUtil;
import io.joshworks.snappy.parser.Parsers;
import io.undertow.server.HttpServerExchange;
import io.undertow.util.HeaderMap;
import io.undertow.util.HeaderValues;
import io.undertow.util.Headers;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Type;
import java.util.List;
import java.util.Map;
import java.util.Queue;
import java.util.Scanner;
import java.util.Set;
import java.util.zip.GZIPInputStream;

/**
 * Request body
 * Created by josh on 3/12/17.
 */
public class Body {

    private final InputStream is;
    private final HeaderMap requestHeaders;
    private final MediaType negotiatedConsumeType;

    public Body(HttpServerExchange exchange) {
        this.requestHeaders = exchange.getRequestHeaders();
        this.is = getRequestInputStream(exchange);

        ConnegHandler.NegotiatedMediaType negotiatedMediaType = exchange.getAttachment(ConnegHandler.NEGOTIATED_MEDIA_TYPE);
        this.negotiatedConsumeType = negotiatedMediaType == null ? MediaType.APPLICATION_JSON_TYPE : negotiatedMediaType.consumes;
    }

    public InputStream asByteStream() {
        try {
            return new ByteArrayInputStream(asByteArray());
        } catch (Exception e) {
            throw new BodyReadException(e);
        }
    }

    public byte[] asByteArray() {
        try {
            return ParserUtil.getBytes(this.is);
        } catch (Exception e) {
            throw new BodyReadException(e);
        }
    }

    public String asString() {
        byte[] rawBody;
        try {
            rawBody = ParserUtil.getBytes(is);
            return new String(rawBody);
        } catch (IOException e) {
            throw new BodyReadException(e);
        }
    }

    public JsonNode asJson() {
        try {
            byte[] bytes = ParserUtil.getBytes(this.is);
            String jsonString = new String(bytes, getCharset()).trim();
            return new JsonNode(jsonString);
        } catch (IOException e) {
            throw new BodyReadException(e);
        }
    }

    public Map asMap() {
        Type type = new TypeToken>() {
        }.getType();
        return new Gson().fromJson(asString(), type);
    }

    public  Set asSetOf(Class classType) {
        Type type = new TypeToken>() {
        }.getType();
        return new Gson().fromJson(asString(), type);
    }

    public  List asListOf(Class classType) {
        Type type = new TypeToken>() {
        }.getType();
        return new Gson().fromJson(asString(), type);
    }

    public  Queue asQueueOf(Class classType) {
        Type type = new TypeToken>() {
        }.getType();
        return new Gson().fromJson(asString(), type);
    }

    public  T asObject(Class type) {
        Parser parser = Parsers.getParser(negotiatedConsumeType);

        String data = readData();
        return parser.readValue(data, type);
    }

    public  List asObject(TypeToken typeToken) {
        Parser parser = Parsers.getParser(negotiatedConsumeType);

        String data = readData();
        return parser.readValue(data, typeToken.getType());
    }

    private String readData() {
        try {
            //ref: http://web.archive.org/web/20140531042945/https://weblogs.java.net/blog/pat/archive/2004/10/stupid_scanner_1.html
            Scanner s = new Scanner(is, getCharset()).useDelimiter("\\A");
            return s.hasNext() ? s.next() : "";
        } catch (Exception e) {
            throw new BodyReadException(e);
        }
    }

    private String getCharset() {
        String charset = "UTF-8";
        HeaderValues encodings = requestHeaders.get(Headers.CONTENT_ENCODING);
        if (encodings != null && encodings.isEmpty()) {
            charset = encodings.getFirst();
        }
        return charset;
    }

    private InputStream getRequestInputStream(HttpServerExchange exchange) {
        try {
            if (ParserUtil.isGzipped(exchange.getResponseHeaders().get(Headers.CONTENT_ENCODING))) {
                return new GZIPInputStream(exchange.getInputStream());
            }
            return exchange.getInputStream();
        } catch (Exception e) {
            throw new RuntimeException("Failed to get request InputStream", e);
        }
    }
}