io.smallrye.mutiny.vertx.codegen.lang.ConstantCodeWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertx-mutiny-generator Show documentation

The newest version!

package io.smallrye.mutiny.vertx.codegen.lang;


import java.io.PrintWriter;
import java.util.Map;

import io.vertx.codegen.ClassModel;
import io.vertx.codegen.ConstantInfo;
import io.vertx.codegen.MethodInfo;
import io.vertx.codegen.doc.Doc;
import io.vertx.codegen.doc.Token;
import io.vertx.codegen.type.TypeInfo;

import static io.smallrye.mutiny.vertx.codegen.lang.CodeGenHelper.genConvReturn;

public class ConstantCodeWriter implements CodeWriter {

    private final Map> methodTypeArgMap;

    public ConstantCodeWriter(Map> methodTypeArgMap) {
        this.methodTypeArgMap = methodTypeArgMap;
    }

    @Override
    public void generate(ClassModel model, PrintWriter writer) {
        for (ConstantInfo constant : model.getConstants()) {
            genConstant(model, constant, writer);
        }
    }

    private void genConstant(ClassModel model, ConstantInfo constant, PrintWriter writer) {
        Doc doc = constant.getDoc();
        if (doc != null) {
            writer.println("  /**");
            Token.toHtml(doc.getTokens(), "   *", CodeGenHelper::renderLinkToHtml, "\n", writer);
            writer.println("   */");
        }
        writer.print(model.isConcrete() ? "  public static final" : "");
        writer.format(" %s %s = %s;%n",
                genTypeName(constant.getType()),
                constant.getName(),
                genConvReturn(methodTypeArgMap, constant.getType(), null, model.getType().getName() + "." + constant.getName()));
    }

}