Implemented a lot of stuff for javac, but we ran into 2 major issues still to be implemented:

1. The visit mode of a lombok handler (does not trigger off of annotations, instead sees every field, method, type, and statement), needs to be coded,
2. triggering off of annotations via APT's annotation handling system skips method-local classes. We'll need to recode this via an AST visitor like we need for issue #1

Other than that, triggering off of annotations works swimmingly!

2 files changed, 118 insertions, 0 deletions

diff --git a/src/lombok/javac/apt/PKG.java b/src/lombok/javac/apt/PKG.java
new file mode 100644
index 00000000..2ecf1c7a
--- /dev/null
+++ b/src/lombok/javac/apt/PKG.java
@@ -0,0 +1,56 @@
+package lombok.javac.apt;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+
+import lombok.Lombok;
+
+
+class PKG {
+	private PKG() {}
+	
+	static boolean isInstanceOf(Object o, String className) {
+		if ( o == null ) return false;
+		return isInstanceOf(o.getClass(), className);
+	}
+	
+	static boolean isInstanceOf(Class<?> c, String className) {
+		if ( c == Object.class || c == null ) return false;
+		
+		if ( c.getName().equals(className) ) return true;
+		
+		if ( isInstanceOf(c.getSuperclass(), className) ) return true;
+		for ( Class<?> iface : c.getInterfaces() ) {
+			if ( isInstanceOf(iface, className) ) return true;
+		}
+		
+		return false;
+	}
+	
+	static byte[] readResource(String name) {
+		return readResource(PKG.class.getClassLoader(), name);
+	}
+	
+	static byte[] readResource(ClassLoader loader, String name) {
+		InputStream in = loader.getResourceAsStream(name);
+		if ( in == null ) throw Lombok.sneakyThrow(new IOException("Not found: " + name));
+		
+		try {
+			return readStream(in);
+		} catch (IOException e) {
+			throw Lombok.sneakyThrow(e);
+		}
+	}
+	
+	static byte[] readStream(InputStream in) throws IOException {
+		ByteArrayOutputStream baos = new ByteArrayOutputStream();
+		byte[] b = new byte[65536];
+		while ( true ) {
+			int r = in.read(b);
+			if ( r == -1 ) break;
+			if ( r > 0 ) baos.write(b, 0, r);
+		}
+		return baos.toByteArray();
+	}
+}
diff --git a/src/lombok/javac/apt/Processor.java b/src/lombok/javac/apt/Processor.java
new file mode 100644
index 00000000..14a62367
--- /dev/null
+++ b/src/lombok/javac/apt/Processor.java
@@ -0,0 +1,62 @@
+package lombok.javac.apt;
+
+import java.util.Set;
+
+import javax.annotation.processing.AbstractProcessor;
+import javax.annotation.processing.ProcessingEnvironment;
+import javax.annotation.processing.RoundEnvironment;
+import javax.annotation.processing.SupportedAnnotationTypes;
+import javax.annotation.processing.SupportedSourceVersion;
+import javax.lang.model.SourceVersion;
+import javax.lang.model.element.Element;
+import javax.lang.model.element.TypeElement;
+
+import lombok.javac.HandlerLibrary;
+import lombok.javac.JavacNode;
+
+import com.sun.source.util.Trees;
+import com.sun.tools.javac.processing.JavacProcessingEnvironment;
+
+
+@SupportedAnnotationTypes("*")
+@SupportedSourceVersion(SourceVersion.RELEASE_6)
+public class Processor extends AbstractProcessor {
+	private JavacProcessingEnvironment processingEnv;
+	private HandlerLibrary handlers;
+	private Trees trees;
+	
+	@Override public void init(ProcessingEnvironment processingEnv) {
+		super.init(processingEnv);
+		if ( !(processingEnv instanceof JavacProcessingEnvironment) ) this.processingEnv = null;
+		else {
+			this.processingEnv = (JavacProcessingEnvironment) processingEnv;
+			handlers = HandlerLibrary.load(processingEnv.getMessager());
+		}
+	}
+	
+	@Override public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
+		if ( processingEnv == null ) return false;
+		
+		trees = Trees.instance(processingEnv);
+		
+		for ( TypeElement annotationType : annotations ) {
+			if ( !handlers.hasHandlerFor(annotationType) ) continue;
+			for ( Element element : roundEnv.getElementsAnnotatedWith(annotationType) ) {
+				System.out.println("HIGHER PING: " + element);
+				handlers.handleAnnotation(createNode(element), annotationType);
+			}
+		}
+		
+		for ( Element element : roundEnv.getRootElements() ) {
+			if ( element instanceof TypeElement ) {
+				handlers.handleType((TypeElement)element);
+			}
+		}
+		
+		return false;
+	}
+	
+	private JavacNode createNode(Element element) {
+		return new JavacNode(trees, processingEnv, element);
+	}
+}