org.stjs.generator.check.declaration.ClassEnumWithoutMembersCheck Maven / Gradle / Ivy

package org.stjs.generator.check.declaration;

import javax.lang.model.element.Element;
import javax.lang.model.element.ElementKind;
import javax.lang.model.element.TypeElement;

import org.stjs.generator.GenerationContext;
import org.stjs.generator.check.CheckContributor;
import org.stjs.generator.check.CheckVisitor;
import org.stjs.generator.javac.InternalUtils;
import org.stjs.generator.javac.TreeUtils;

import com.sun.source.tree.ClassTree;
import com.sun.source.tree.Tree;
import com.sun.source.tree.VariableTree;

/**
 * this class checks that no member is added to an enum declaration (other than of course the enum constants)
 * 
 * @author acraciun
 */
public class ClassEnumWithoutMembersCheck implements CheckContributor {

	private void checkMember(Tree member, GenerationContext context) {
		if (InternalUtils.isSyntheticConstructor(member)) {
			return;
		}
		boolean ok = false;
		if (member instanceof VariableTree) {
			Element memberElement = TreeUtils.elementFromDeclaration((VariableTree) member);
			ok = memberElement.getKind() == ElementKind.ENUM_CONSTANT;
		}
		if (!ok) {
			context.addError(member, "Enums with fields or methods are not supported");
		}
	}

	@Override
	public Void visit(CheckVisitor visitor, ClassTree tree, GenerationContext context) {
		TypeElement typeElement = TreeUtils.elementFromDeclaration(tree);
		if (typeElement.getKind() != ElementKind.ENUM) {
			return null;
		}
		for (Tree member : tree.getMembers()) {
			checkMember(member, context);
		}
		return null;
	}
}