software.coley.cafedude.tree.visitor.writer.AnnotationDefaultWriter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of cafedude-tree Show documentation

Tree module for CafeDude, containing a simplified intermediate model for classes

The newest version!

package software.coley.cafedude.tree.visitor.writer;

import software.coley.cafedude.classfile.annotation.Annotation;
import software.coley.cafedude.classfile.annotation.AnnotationElementValue;
import software.coley.cafedude.classfile.annotation.ArrayElementValue;
import software.coley.cafedude.classfile.annotation.ElementValue;
import software.coley.cafedude.tree.Constant;
import software.coley.cafedude.tree.visitor.AnnotationArrayVisitor;
import software.coley.cafedude.tree.visitor.AnnotationDefaultVisitor;
import software.coley.cafedude.tree.visitor.AnnotationVisitor;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.function.Consumer;

/**
 * Annotation visitor implementation to write back the {@link ElementValue default value}
 * portion of an {@link Annotation}
 *
 * @author Justus Garbe
 */
public class AnnotationDefaultWriter implements AnnotationDefaultVisitor {
	private final Symbols symbols;
	private final Consumer callback;

	AnnotationDefaultWriter(Symbols symbols, Consumer callback) {
		this.symbols = symbols;
		this.callback = callback;
	}

	@Override
	public void visitDefaultValue(@Nonnull Constant value) {
		callback.accept(symbols.newElementValue(value));
	}

	@Nullable
	@Override
	public AnnotationVisitor visitDefaultAnnotation(@Nonnull String type) {
		return new AnnotationWriter(symbols, ev -> {
			Annotation annotation = new Annotation(symbols.newUtf8(type), ev);
			callback.accept(new AnnotationElementValue('@', annotation));
		});
	}

	@Nullable
	@Override
	public AnnotationArrayVisitor visitDefaultArray() {
		return new AnnotationArrayWriter(symbols, elements -> {
			callback.accept(new ArrayElementValue('[', elements));
		});
	}
}