org.tkit.quarkus.jpa.deployment.JPABuild Maven / Gradle / Ivy

Go to download
package org.tkit.quarkus.jpa.deployment;

import java.util.List;
import java.util.Map;

import jakarta.persistence.Entity;
import jakarta.persistence.EntityManager;
import jakarta.persistence.Id;

import org.jboss.jandex.*;
import org.tkit.quarkus.jpa.daos.AbstractDAO;

import io.quarkus.arc.deployment.UnremovableBeanBuildItem;
import io.quarkus.deployment.annotations.BuildProducer;
import io.quarkus.deployment.annotations.BuildStep;
import io.quarkus.deployment.builditem.BytecodeTransformerBuildItem;
import io.quarkus.deployment.builditem.CombinedIndexBuildItem;
import io.quarkus.deployment.builditem.FeatureBuildItem;

/**
 * The JPA build extension.
 */
public class JPABuild {

    /**
     * The extension name.
     */
    private static final String FEATURE_NAME = "tkit-jpa";

    /**
     * The abstract entity service class.
     */
    private static final DotName DOT_NAME_REPOSITORY = DotName.createSimple(AbstractDAO.class.getName());

    /**
     * The entity class.
     */
    private static final DotName ENTITY = DotName.createSimple(Entity.class.getName());
    private static final DotName ID = DotName.createSimple(Id.class.getName());
    private static final DotName DOT_NAME_ENTITY_MANAGER = DotName.createSimple(EntityManager.class.getName());

    /**
     * The name of the entity annotation attribute name.
     */
    private static final String ATTRIBUTE_NAME = "name";

    /**
     * The extension name.
     *
     * @return the feature build item.
     */
    @BuildStep
    FeatureBuildItem createFeatureItem() {
        return new FeatureBuildItem(FEATURE_NAME);
    }

    @BuildStep
    UnremovableBeanBuildItem ensureBeanLookupAvailable() {
        return new UnremovableBeanBuildItem(new UnremovableBeanBuildItem.BeanTypeExclusion(DOT_NAME_ENTITY_MANAGER));
    }

    /**
     * Update entity dao services to have entity class name and entity name.
     *
     * @param index the index.
     * @param transformers the transformer
     */
    @BuildStep
    void build(CombinedIndexBuildItem index,
            BuildProducer transformers) {

        IndexView view = index.getIndex();
        for (ClassInfo classInfo : view.getAllKnownSubclasses(DOT_NAME_REPOSITORY)) {
            if (classInfo.superClassType().kind() == Type.Kind.PARAMETERIZED_TYPE) {
                Type entity = classInfo.superClassType().asParameterizedType().arguments().get(0);
                ClassInfo ec = view.getClassByName(entity.name());
                String name = entity.name().withoutPackagePrefix();

                Map> am = ec.annotationsMap();
                if (am != null) {
                    List annotations = am.get(ENTITY);
                    if (annotations != null && !annotations.isEmpty()) {
                        AnnotationInstance annotationInstance = annotations.get(0);
                        for (AnnotationValue a : annotationInstance.values()) {
                            if (ATTRIBUTE_NAME.equals(a.name())) {
                                name = a.asString();
                            }
                        }
                    }
                }

                String idAttributeName = getIdAttributeName(view, ec);
                transformers.produce(new BytecodeTransformerBuildItem(classInfo.name().toString(),
                        new EntityServiceBuilderEnhancer(name, entity.name().toString(), idAttributeName)));
            }
        }
    }

    private String getIdAttributeName(IndexView view, ClassInfo ec) {
        for (FieldInfo f : ec.fields()) {
            if (f.hasAnnotation(ID)) {
                return f.name();
            }
        }
        if (ec.superName() != null) {
            return getIdAttributeName(view, view.getClassByName(ec.superName()));
        }
        return "";
    }

}