io.automatiko.engine.quarkus.function.deployment.devconsole.DevConsoleProcessor Maven / Gradle / Ivy

Go to download
package io.automatiko.engine.quarkus.function.deployment.devconsole;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Optional;

import org.eclipse.microprofile.config.ConfigProvider;
import org.eclipse.microprofile.openapi.models.media.Schema;
import org.jboss.jandex.AnnotationInstance;
import org.jboss.jandex.AnnotationTarget.Kind;
import org.jboss.jandex.MethodInfo;

import com.fasterxml.jackson.databind.ObjectMapper;

import io.automatiko.engine.function.dev.WorkflowFunctionInfo;
import io.automatiko.engine.quarkus.function.deployment.AutomatikoFunctionProcessor;
import io.automatiko.engine.quarkus.function.deployment.ExampleGenerator;
import io.quarkus.deployment.IsDevelopment;
import io.quarkus.deployment.annotations.BuildStep;
import io.quarkus.deployment.builditem.CombinedIndexBuildItem;
import io.quarkus.devui.spi.page.CardPageBuildItem;
import io.quarkus.devui.spi.page.Page;
import io.smallrye.openapi.runtime.io.schema.SchemaFactory;
import io.smallrye.openapi.runtime.scanner.spi.AnnotationScannerContext;

public class DevConsoleProcessor {

    @BuildStep(onlyIf = IsDevelopment.class)
    CardPageBuildItem create(CombinedIndexBuildItem index) throws Exception {
        CardPageBuildItem cardPageBuildItem = new CardPageBuildItem();
        workaroundMultiModuleDevMode(cardPageBuildItem);
        List info = getInfo(index);

        cardPageBuildItem.addPage(Page.webComponentPageBuilder()
                .icon("font-awesome-solid:diagram-project")
                .componentLink("qwc-automatiko-functions.js")
                .title("Functions")
                .staticLabel(String.valueOf(info.size())));

        cardPageBuildItem.addBuildTimeData("functions", info);

        return cardPageBuildItem;
    }

    protected List getInfo(CombinedIndexBuildItem index) {
        Optional host = Optional
                .of(ConfigProvider.getConfig().getOptionalValue("quarkus.http.host", String.class).orElse("localhost"));
        Optional port = Optional
                .of(ConfigProvider.getConfig().getOptionalValue("quarkus.http.port", Integer.class).orElse(8080));

        String path = ConfigProvider.getConfig().getOptionalValue("quarkus.http.root-path", String.class).orElse("");

        List infos = new ArrayList();
        Collection functions = index.getIndex()
                .getAnnotations(AutomatikoFunctionProcessor.createDotName("io.quarkus.funqy.Funq"));
        ObjectMapper mapper = new ObjectMapper();
        ExampleGenerator generator = new ExampleGenerator();
        AnnotationScannerContext ctx = AutomatikoFunctionProcessor.buildAnnotationScannerContext(index.getIndex());

        for (AnnotationInstance f : functions) {
            if (f.target().kind().equals(Kind.METHOD)) {
                MethodInfo mi = f.target().asMethod();
                // create function trigger descriptor for every found function

                StringBuilder curlGet = new StringBuilder("curl -X GET http://").append(host.get()).append(":")
                        .append(port.get()).append(path.endsWith("/") ? "" : path)
                        .append("/").append(mi.name()).append("?");

                StringBuilder curlPost = new StringBuilder("curl -X POST http://").append(host.get()).append(":")
                        .append(port.get()).append(path.endsWith("/") ? "" : path)
                        .append("/").append(mi.name()).append(" ");

                SchemaFactory.typeToSchema(ctx,
                        mi.parameters().get(0).type(), null, Collections.emptyList());
                Schema fSchema = ctx.getOpenApi().getComponents().getSchemas()
                        .get(mi.parameters().get(0).type().name().local());

                Map example = generator.generate(fSchema, ctx.getOpenApi());
                try {
                    String putInstructions = mapper.writerWithDefaultPrettyPrinter().writeValueAsString(example);

                    StringBuilder getInstructions = new StringBuilder();
                    AutomatikoFunctionProcessor.flatMap(null, example).entrySet()
                            .forEach(e -> getInstructions.append(e.getKey() + "=" + e.getValue() + "&"));

                    curlGet.append(getInstructions.deleteCharAt(getInstructions.length() - 1).toString());

                    curlPost.append("-d \"")
                            .append(putInstructions.toString().replaceAll("\"", "\\\\\\\"").replaceAll("\\s+", ""))
                            .append("\"");
                    infos.add(
                            new WorkflowFunctionInfo(mi.name(), path.endsWith("/") ? path + mi.name() : path + "/" + mi.name(),
                                    getInstructions.deleteCharAt(getInstructions.length() - 1).toString(),
                                    curlGet.toString(),
                                    putInstructions,
                                    curlPost.toString()));
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }

        return infos;
    }

    private void workaroundMultiModuleDevMode(CardPageBuildItem cardPageBuildItem) {
        Class c = CardPageBuildItem.class;

        while (c.getSuperclass() != null) {
            try {
                c = c.getSuperclass();
                Field f = c.getDeclaredField("extensionIdentifier");
                f.setAccessible(true);
                f.set(cardPageBuildItem, "io.automatiko.quarkus.automatiko-function");
            } catch (Exception e) {

            }
        }
    }
}