org.jboss.cdi.lang.model.tck.LangModelUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cdi-tck-lang-model Show documentation
CDI Lang Model TCK Tests
There is a newer version: 4.1.0
package org.jboss.cdi.lang.model.tck;

import jakarta.enterprise.lang.model.declarations.ClassInfo;
import jakarta.enterprise.lang.model.declarations.FieldInfo;
import jakarta.enterprise.lang.model.declarations.MethodInfo;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

class LangModelUtils {
    static ClassInfo classOfField(ClassInfo clazz, String fieldName) {
        return singleField(clazz, fieldName).type().asClass().declaration();
    }

    static FieldInfo singleField(ClassInfo clazz, String name) {
        FieldInfo result = null;
        for (FieldInfo field : clazz.fields()) {
            if (name.equals(field.name())) {
                if (result == null) {
                    result = field;
                } else {
                    throw new IllegalStateException("More than 1 declaration of field '" + name + "' on '" + clazz.simpleName() + "'");
                }
            }
        }
        if (result == null) {
            throw new IllegalStateException("No declaration of field '" + name + "' on '" + clazz.simpleName() + "'");
        }
        return result;
    }

    static FieldInfo singleDeclaredField(ClassInfo clazz, String name) {
        FieldInfo result = null;
        for (FieldInfo field : clazz.fields()) {
            if (!field.declaringClass().equals(clazz)) {
                continue;
            }

            if (name.equals(field.name())) {
                if (result == null) {
                    result = field;
                } else {
                    throw new IllegalStateException("More than 1 declaration of field '" + name + "' on '" + clazz.simpleName() + "'");
                }
            }
        }
        if (result == null) {
            throw new IllegalStateException("No declaration of field '" + name + "' on '" + clazz.simpleName() + "'");
        }
        return result;
    }

    static Collection collectFields(ClassInfo clazz, String name) {
        List result = new ArrayList<>();
        for (FieldInfo field : clazz.fields()) {
            if (field.name().equals(name)) {
                result.add(field);
            }
        }
        return result;
    }

    static MethodInfo singleMethod(ClassInfo clazz, String name) {
        MethodInfo result = null;
        for (MethodInfo method : clazz.methods()) {
            if (name.equals(method.name())) {
                if (result == null) {
                    result = method;
                } else {
                    throw new IllegalStateException("More than 1 declaration of method '" + name + "' on '" + clazz.simpleName() + "'");
                }
            }
        }
        if (result == null) {
            throw new IllegalStateException("No declaration of method '" + name + "' on '" + clazz.simpleName() + "'");
        }
        return result;
    }

    static MethodInfo singleDeclaredMethod(ClassInfo clazz, String name) {
        MethodInfo result = null;
        for (MethodInfo method : clazz.methods()) {
            if (!method.declaringClass().equals(clazz)) {
                continue;
            }

            if (name.equals(method.name())) {
                if (result == null) {
                    result = method;
                } else {
                    throw new IllegalStateException("More than 1 declaration of method '" + name + "' on '" + clazz.simpleName() + "'");
                }
            }
        }
        if (result == null) {
            throw new IllegalStateException("No declaration of method '" + name + "' on '" + clazz.simpleName() + "'");
        }
        return result;
    }

    static Collection collectMethods(ClassInfo clazz, String name) {
        List result = new ArrayList<>();
        for (MethodInfo method : clazz.methods()) {
            if (method.name().equals(name)) {
                result.add(method);
            }
        }
        return result;
    }
}