io.quarkus.vertx.graphql.runtime.VertxGraphqlRecorder Maven / Gradle / Ivy

Go to download
package io.quarkus.vertx.graphql.runtime;

import java.util.function.Consumer;
import java.util.function.Supplier;

import io.quarkus.runtime.annotations.Recorder;
import io.vertx.core.Handler;
import io.vertx.core.Vertx;
import io.vertx.core.http.HttpHeaders;
import io.vertx.ext.web.Route;
import io.vertx.ext.web.RoutingContext;
import io.vertx.ext.web.handler.graphql.GraphiQLHandler;
import io.vertx.ext.web.handler.graphql.GraphiQLHandlerOptions;

@Recorder
public class VertxGraphqlRecorder {
    public Handler handler(Supplier vertx) {

        GraphiQLHandlerOptions options = new GraphiQLHandlerOptions();
        options.setEnabled(true);

        Handler handler = GraphiQLHandler.create(vertx.get(), options);

        return new Handler() {
            @Override
            public void handle(RoutingContext event) {
                if (event.normalizedPath().length() == (event.currentRoute().getPath().length()
                        + (event.mountPoint() == null ? 0 : event.mountPoint().length() - 1))) {
                    event.response().setStatusCode(302);
                    event.response().headers().set(HttpHeaders.LOCATION, (event.mountPoint() == null ? "" : event.mountPoint())
                            + event.currentRoute().getPath().substring(1) + "/");
                    event.response().end();
                    return;
                }

                handler.handle(event);
            }
        };
    }

    public Consumer routeFunction(Handler bodyHandler) {
        return new Consumer() {
            @Override
            public void accept(Route route) {
                route.handler(bodyHandler);
            }
        };
    }

}