com.develhack.lombok.javac.handlers.feature.DTOHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of develhacked-lombok Show documentation

The library enhancing Java classes at compilation by annotations that contained in the Develhack Core Library.

The newest version!

package com.develhack.lombok.javac.handlers.feature;

import lombok.core.AST.Kind;
import lombok.core.AnnotationValues;
import lombok.core.AnnotationValues.AnnotationValueDecodeFail;
import lombok.core.HandlerPriority;
import lombok.javac.JavacAnnotationHandler;
import lombok.javac.JavacNode;

import org.mangosdk.spi.ProviderFor;

import com.develhack.annotation.feature.Access;
import com.develhack.annotation.feature.Accessible;
import com.develhack.annotation.feature.DTO;
import com.develhack.lombok.eclipse.handlers.feature.AccessibleHandler;
import com.sun.tools.javac.tree.JCTree.JCAnnotation;
import com.sun.tools.javac.tree.JCTree.JCVariableDecl;

@ProviderFor(JavacAnnotationHandler.class)
@HandlerPriority(DTOHandler.PRIORITY)
public class DTOHandler extends AbstractFeatureHandler {

	public static final int PRIORITY = AccessibleHandler.PRIORITY + 1;

	public DTOHandler() {
		super(DTO.class);
	}

	@Override
	public void handle(AnnotationValues annotationValues, JCAnnotation ast, JavacNode annotationNode) {

		super.handle(annotationValues, ast, annotationNode);

		JavacNode typeNode = annotationNode.up();

		if (typeNode.getKind() != Kind.TYPE) {
			annotationNode.addWarning(String.format("@%s is only applicable to the type.", getAnnotationName()));
			return;
		}

		DTO dto;
		try {
			dto = annotationValues.getInstance();
		} catch (AnnotationValueDecodeFail e) {
			return;
		}

		for (JavacNode childe : typeNode.down()) {

			if (childe.getKind() != Kind.FIELD) continue;

			JCVariableDecl field = (JCVariableDecl) childe.get();

			if (isExcludedFrom(field, DTO.class)) continue;
			if (isTransient(field)) continue;

			AnnotationValues accessibleValues = findAnnotationValues(Accessible.class, field.mods.annotations);

			if (accessibleValues != null) {
				Accessible accessible;
				try {
					accessible = accessibleValues.getInstance();
				} catch (AnnotationValueDecodeFail e) {
					continue;
				}
				Access getAccess = accessible.get();
				Access setAccess = accessible.set();

				supplementGetter(field, getAccess);
				supplementSetter(field, setAccess);

			} else {

				supplementGetter(field, dto.get());
				supplementSetter(field, dto.set());
			}
		}

		supplementSuperInterface(java.io.Serializable.class);
		supplementSuppressWaring("serial");
	}
}