1 files changed, 32 insertions, 0 deletions
diff --git a/src/utils/lombok/javac/java6/CommentCollectingParser.java b/src/utils/lombok/javac/java6/CommentCollectingParser.java
new file mode 100644
index 00000000..bc33bf71
--- /dev/null
+++ b/src/utils/lombok/javac/java6/CommentCollectingParser.java
@@ -0,0 +1,32 @@
+package lombok.javac.java6;
+
+import java.util.Map;
+
+import lombok.javac.Comment;
+
+import com.sun.tools.javac.parser.EndPosParser;
+import com.sun.tools.javac.parser.Lexer;
+import com.sun.tools.javac.tree.JCTree.JCCompilationUnit;
+import com.sun.tools.javac.util.List;
+
+class CommentCollectingParser extends EndPosParser {
+	
+	private final Map<JCCompilationUnit, List<Comment>> commentsMap;
+	private final Lexer lexer;
+
+	protected CommentCollectingParser(Factory fac, Lexer S, boolean keepDocComments, Map<JCCompilationUnit, List<Comment>> commentsMap) {
+		super(fac, S, keepDocComments);
+		lexer = S;
+		this.commentsMap = commentsMap;
+		
+	}
+	
+	@Override public JCCompilationUnit compilationUnit() {
+		JCCompilationUnit result = super.compilationUnit();
+		if (lexer instanceof CommentCollectingScanner) {
+			List<Comment> comments = ((CommentCollectingScanner)lexer).getComments();
+			commentsMap.put(result, comments);
+		}
+		return result;
+	}
+}
+\ No newline at end of file