io.yawp.repository.models.ObjectModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yawp-core Show documentation
YAWP! core
There is a newer version: 2.08alpha
package io.yawp.repository.models;

import io.yawp.commons.utils.ReflectionUtils;
import io.yawp.commons.utils.kind.KindResolver;
import io.yawp.repository.IdRef;
import io.yawp.repository.annotations.Id;
import io.yawp.repository.annotations.ParentId;

import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.List;

public class ObjectModel {

    private Class clazz;

    public ObjectModel(Class clazz) {
        this.clazz = clazz;
    }

    public String getKind() {
        return KindResolver.getKindFromClass(clazz);
    }

    public Class getClazz() {
        return clazz;
    }

    public Field getIdField() {
        return ReflectionUtils.getFieldWithAnnotation(clazz, Id.class);
    }

    public Field getParentField() {
        return ReflectionUtils.getFieldWithAnnotation(clazz, ParentId.class);
    }

    public String getIdFieldName() {
        return getIdField().getName();
    }

    public boolean isIdRef(Field field) {
        return IdRef.class.isAssignableFrom(field.getType());
    }

    public Class getParentClazz() {
        Field parentField = getParentField();
        if (parentField == null) {
            return null;
        }
        return ReflectionUtils.getIdRefEndpointClazz(parentField);
    }

    public boolean hasParent() {
        return getParentField() != null;
    }

    public ObjectModel getParentModel() {
        return new ObjectModel(getParentClazz());
    }

    public Class getAncestorClazz(int ancestor) {
        Class parentClazz = clazz;
        for (int i = 0; i <= ancestor; i++) {
            ObjectModel model = new ObjectModel(parentClazz);
            parentClazz = model.getParentClazz();
        }
        return parentClazz;
    }

    public int getAncestorNumber(Class ancestorClazz) {
        if (clazz.equals(ancestorClazz)) {
            return -1;
        }

        Class parentClazz = getParentClazz();
        int ancestorNumber = 0;

        while (parentClazz != null && !parentClazz.equals(ancestorClazz)) {
            ObjectModel model = new ObjectModel(parentClazz);
            parentClazz = model.getParentClazz();
            ancestorNumber++;
        }

        if (parentClazz == null) {
            return -1;
        }

        return ancestorNumber;
    }

    public boolean isAncestor(Class ancestorClazz) {
        return getAncestorNumber(ancestorClazz) != -1;
    }

    public List getFieldModels() {
        List fields = ReflectionUtils.getFieldsRecursively(clazz);

        List fieldModels = new ArrayList<>();

        for (Field field : fields) {
            fieldModels.add(new FieldModel(field));
        }

        return fieldModels;
    }

    public FieldModel getFieldModel(String fieldName) {
        return new FieldModel(ReflectionUtils.getFieldRecursively(clazz, fieldName));
    }

    public boolean isIdShuffled() {
        return getIdField().getAnnotation(Id.class).shuffle();
    }

    @SuppressWarnings("unchecked")
    public  T createInstance() {
        try {
            Constructor defaultConstructor = (Constructor) clazz.getDeclaredConstructor(new Class[]{});
            defaultConstructor.setAccessible(true);
            return defaultConstructor.newInstance();

        } catch (InvocationTargetException e) {
            throw new RuntimeException("An exception was thrown when calling the default constructor of the class " + clazz.getSimpleName()
                    + ": ", e);
        } catch (NoSuchMethodException e) {
            throw new RuntimeException("The class " + clazz.getSimpleName()
                    + " must have a default constructor and cannot be an non-static inner class.", e);
        } catch (InstantiationException e) {
            throw new RuntimeException("The class " + clazz.getSimpleName() + " must cannot be abstract.", e);
        } catch (IllegalArgumentException | IllegalAccessException | SecurityException e) {
            throw new RuntimeException("Unexpected error: ", e);
        }
    }
}