org.restheart.exchange.GraphQLRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of restheart-commons Show documentation
RESTHeart Commons - Common classes for core components and plugins.
There is a newer version: 8.1.7
/*-
 * ========================LICENSE_START=================================
 * restheart-commons
 * %%
 * Copyright (C) 2019 - 2024 SoftInstigate
 * %%
 * 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.
 * =========================LICENSE_END==================================
 */
package org.restheart.exchange;

import java.io.IOException;

import org.restheart.utils.ChannelReader;

import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.google.gson.JsonSyntaxException;

import io.undertow.server.HttpServerExchange;
import io.undertow.util.Headers;

public class GraphQLRequest extends ServiceRequest {

    private static final String GRAPHQL_CONTENT_TYPE = "application/graphql";
    private static final String QUERY_FIELD = "query";
    private static final String OPERATION_NAME_FIELD = "operationName";
    private static final String VARIABLES_FIELD = "variables";

    private final String appUri;


    private GraphQLRequest(HttpServerExchange exchange, String appUri) {
        super(exchange);
        this.appUri = appUri;
    }

    public static GraphQLRequest init(HttpServerExchange exchange, String appUri) throws IOException, JsonSyntaxException, BadRequestException {
        var method = exchange.getRequestMethod();

        if (!method.equalToString("OPTIONS") && !method.equalToString("POST")) {
            throw new BadRequestException("Method not allowed", 405);
        }

        return new GraphQLRequest(exchange, appUri);
    }

    public static GraphQLRequest of(HttpServerExchange exchange) {
        return of(exchange, GraphQLRequest.class);
    }

    @Override
    public JsonElement parseContent() throws IOException, BadRequestException {
        var exchange = getExchange();
        var rawBody = ChannelReader.readString(exchange);

        if (isContentTypeGraphQL(exchange)) {
            return _parseContentGraphQL(rawBody);
        } else if (isContentTypeJson(exchange)) {
            return _parseContentJson(rawBody);
        } else if (!exchange.getRequestMethod().equalToString("OPTIONS")) {
            var error = "Bad request: " + Headers.CONTENT_TYPE + " must be either " + GRAPHQL_CONTENT_TYPE + " or application/json";
            throw new BadRequestException(error);
        }

        return null;
    }

    private JsonElement _parseContentJson(String rawBody) throws JsonSyntaxException, BadRequestException {
        return JsonParser.parseString(rawBody);
    }

    private JsonElement _parseContentGraphQL(String rawBody) {
        var jsonObject = new JsonObject();
        jsonObject.addProperty(QUERY_FIELD, rawBody);

        return jsonObject;
    }

    public String getQuery() {
        var _content = this.getContent();
        if (_content != null
            && _content.isJsonObject()
            && _content.getAsJsonObject().has(QUERY_FIELD)
            && _content.getAsJsonObject().get(QUERY_FIELD).isJsonPrimitive()
            && _content.getAsJsonObject().getAsJsonPrimitive(QUERY_FIELD).isString()) {
            return _content.getAsJsonObject().getAsJsonPrimitive(QUERY_FIELD).getAsString();
        } else {
            return null;
        }
    }

    public String getOperationName() {
        var _content = this.getContent();
        if (_content != null
            && _content.isJsonObject()
            && _content.getAsJsonObject().has(OPERATION_NAME_FIELD)
            && _content.getAsJsonObject().get(OPERATION_NAME_FIELD).isJsonPrimitive()
            && _content.getAsJsonObject().getAsJsonPrimitive(OPERATION_NAME_FIELD).isString()) {
            return _content.getAsJsonObject().getAsJsonPrimitive(OPERATION_NAME_FIELD).getAsString();
        } else {
            return null;
        }
    }

    public JsonObject getVariables() {
        var _content = this.getContent();
        if (_content != null
            && _content.isJsonObject()
            && _content.getAsJsonObject().has(VARIABLES_FIELD)
            && _content.getAsJsonObject().get(VARIABLES_FIELD).isJsonObject()) {
            return _content.getAsJsonObject().getAsJsonObject(VARIABLES_FIELD);
        } else {
            return null;
        }
    }

    public String getGraphQLAppURI() {
       return this.appUri;
    }

    public boolean hasVariables() {
        var _content = this.getContent();

        return _content != null
            &&  _content.isJsonObject()
            && _content.getAsJsonObject().has(VARIABLES_FIELD)
            && _content.getAsJsonObject().get(VARIABLES_FIELD).isJsonObject();
    }

    private static boolean isContentTypeGraphQL(HttpServerExchange exchange){
        var contentType = getContentType(exchange);
        return GRAPHQL_CONTENT_TYPE.equals(contentType) || (contentType != null && contentType.startsWith(GRAPHQL_CONTENT_TYPE));
    }
}