software.coley.cafedude.tree.visitor.writer.DeclarationWriter 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.AttributeConstants;
import software.coley.cafedude.classfile.annotation.Annotation;
import software.coley.cafedude.classfile.annotation.TargetInfo;
import software.coley.cafedude.classfile.annotation.TypeAnnotation;
import software.coley.cafedude.classfile.annotation.TypePath;
import software.coley.cafedude.classfile.attribute.*;
import software.coley.cafedude.tree.visitor.AnnotationVisitor;
import software.coley.cafedude.tree.visitor.DeclarationVisitor;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.ArrayList;
import java.util.List;

/**
 * @author Justus Garbe
 */
public class DeclarationWriter implements DeclarationVisitor {
	protected final List attributes = new ArrayList<>();
	private final List visibleAnnotations = new ArrayList<>();
	private final List invisibleAnnotations = new ArrayList<>();
	private final List visibleTypeAnnotations = new ArrayList<>();
	private final List invisibleTypeAnnotations = new ArrayList<>();
	protected Symbols symbols;

	DeclarationWriter(Symbols symbols) {
		this.symbols = symbols;
	}

	@Nullable
	@Override
	public DeclarationVisitor declarationDelegate() {
		return null;
	}

	@Nullable
	@Override
	public AnnotationVisitor visitAnnotation(@Nonnull String type, boolean visible) {
		return new AnnotationWriter(symbols, values -> {
			Annotation annotation = new Annotation(symbols.newUtf8(type), values);
			if (visible)
				visibleAnnotations.add(annotation);
			else
				invisibleAnnotations.add(annotation);
		});
	}

	@Nullable
	@Override
	public AnnotationVisitor visitTypeAnnotation(String type, TargetInfo target, TypePath path,
												 boolean visible) {
		return new AnnotationWriter(symbols, values -> {
			// copy values over to type annotation
			TypeAnnotation typeAnnotation = new TypeAnnotation(symbols.newUtf8(type), values, target, path);
			if (visible)
				visibleTypeAnnotations.add(typeAnnotation);
			else
				invisibleTypeAnnotations.add(typeAnnotation);
		});
	}

	@Override
	public void visitSignature(@Nonnull String signature) {
		attributes.add(new SignatureAttribute(
				symbols.newUtf8(AttributeConstants.SIGNATURE), symbols.newUtf8(signature)));
	}

	@Override
	public void visitSynthetic(boolean synthetic) {
		if (synthetic) {
			attributes.add(new SyntheticAttribute(
					symbols.newUtf8(AttributeConstants.SYNTHETIC)));
		}
	}

	@Override
	public void visitDeprecated(boolean deprecated) {
		if (deprecated) {
			attributes.add(new DeprecatedAttribute(
					symbols.newUtf8(AttributeConstants.DEPRECATED)));
		}
	}

	void visitDeclarationEnd() {
		if (!visibleAnnotations.isEmpty()) {
			attributes.add(new AnnotationsAttribute(
					symbols.newUtf8(AttributeConstants.RUNTIME_VISIBLE_ANNOTATIONS),
					visibleAnnotations, true));
		}
		if (!invisibleAnnotations.isEmpty()) {
			attributes.add(new AnnotationsAttribute(
					symbols.newUtf8(AttributeConstants.RUNTIME_INVISIBLE_ANNOTATIONS),
					invisibleAnnotations, false));
		}
		if (!visibleTypeAnnotations.isEmpty()) {
			attributes.add(new AnnotationsAttribute(
					symbols.newUtf8(AttributeConstants.RUNTIME_VISIBLE_TYPE_ANNOTATIONS),
					visibleTypeAnnotations, true));
		}
		if (!invisibleTypeAnnotations.isEmpty()) {
			attributes.add(new AnnotationsAttribute(
					symbols.newUtf8(AttributeConstants.RUNTIME_INVISIBLE_TYPE_ANNOTATIONS),
					invisibleTypeAnnotations, false));
		}
	}
}