Work on HandleData, as well as generalizing features in the the PKG class and updating HandleGetter/Setter to call into it.

1 files changed, 12 insertions, 0 deletions

diff --git a/src/lombok/javac/HandlerLibrary.java b/src/lombok/javac/HandlerLibrary.java
index f9404668..ab5388d3 100644
--- a/src/lombok/javac/HandlerLibrary.java
+++ b/src/lombok/javac/HandlerLibrary.java
@@ -13,6 +13,7 @@ import javax.annotation.processing.Messager;
 import javax.lang.model.element.TypeElement;
 import javax.tools.Diagnostic;
 
+import lombok.core.PrintAST;
 import lombok.core.SpiLoadUtil;
 import lombok.core.TypeLibrary;
 import lombok.core.TypeResolver;
@@ -27,6 +28,7 @@ public class HandlerLibrary {
 	private final Map<String, AnnotationHandlerContainer<?>> annotationHandlers = new HashMap<String, AnnotationHandlerContainer<?>>();
 	private final Collection<JavacASTVisitor> visitorHandlers = new ArrayList<JavacASTVisitor>();
 	private final Messager messager;
+	private boolean skipPrintAST = true;
 	
 	public HandlerLibrary(Messager messager) {
 		this.messager = messager;
@@ -111,6 +113,8 @@ public class HandlerLibrary {
 		String rawType = annotation.annotationType.toString();
 		boolean handled = false;
 		for ( String fqn : resolver.findTypeMatches(node, rawType) ) {
+			boolean isPrintAST = fqn.equals(PrintAST.class.getName());
+			if ( isPrintAST == skipPrintAST ) continue;
 			AnnotationHandlerContainer<?> container = annotationHandlers.get(fqn);
 			if ( container == null ) continue;
 			
@@ -137,4 +141,12 @@ public class HandlerLibrary {
 	public boolean hasHandlerFor(TypeElement annotationType) {
 		return annotationHandlers.containsKey(annotationType.getQualifiedName().toString());
 	}
+	
+	public void skipPrintAST() {
+		skipPrintAST = true;
+	}
+	
+	public void skipAllButPrintAST() {
+		skipPrintAST = false;
+	}
 }