io.micronaut.ast.groovy.visitor.GroovyEnumElement Maven / Gradle / Ivy
/*
* Copyright 2017-2020 original authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.micronaut.ast.groovy.visitor;
import io.micronaut.core.annotation.Internal;
import io.micronaut.inject.ast.ClassElement;
import io.micronaut.inject.ast.EnumConstantElement;
import io.micronaut.inject.ast.EnumElement;
import io.micronaut.inject.ast.annotation.ElementAnnotationMetadataFactory;
import org.codehaus.groovy.ast.FieldNode;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
/**
* Implementation of {@link EnumElement} for Groovy.
*
* @author graemerocher
* @since 1.0
*/
@Internal
class GroovyEnumElement extends GroovyClassElement implements EnumElement {
protected List enumConstants;
protected List values;
/**
* @param visitorContext The visitor context
* @param nativeElement The native element
* @param annotationMetadataFactory The annotation metadata factory
*/
GroovyEnumElement(GroovyVisitorContext visitorContext,
GroovyNativeElement nativeElement,
ElementAnnotationMetadataFactory annotationMetadataFactory) {
this(visitorContext, nativeElement, annotationMetadataFactory, 0);
}
/**
* @param visitorContext The visitor context
* @param nativeElement The native element
* @param annotationMetadataFactory The annotation metadata
* @param arrayDimensions The number of array dimensions factory
*/
GroovyEnumElement(GroovyVisitorContext visitorContext,
GroovyNativeElement nativeElement,
ElementAnnotationMetadataFactory annotationMetadataFactory,
int arrayDimensions) {
super(visitorContext, nativeElement, annotationMetadataFactory, null, arrayDimensions);
}
@Override
public List values() {
if (values != null) {
return values;
}
initEnum();
return values;
}
@Override
public List elements() {
if (enumConstants != null) {
return enumConstants;
}
initEnum();
return enumConstants;
}
private void initEnum() {
values = new ArrayList<>();
enumConstants = new ArrayList<>();
for (FieldNode field : classNode.getFields()) {
if (field.getName().equals("MAX_VALUE") || field.getName().equals("MIN_VALUE")) {
continue;
}
if (field.isEnum()) {
values.add(field.getName());
enumConstants.add(new GroovyEnumConstantElement(this, visitorContext, field, elementAnnotationMetadataFactory));
}
}
values = Collections.unmodifiableList(values);
enumConstants = Collections.unmodifiableList(enumConstants);
}
@Override
public ClassElement withArrayDimensions(int arrayDimensions) {
return new GroovyEnumElement(visitorContext, getNativeType(), elementAnnotationMetadataFactory, arrayDimensions);
}
}