From a9f334fde93a0b07820962bff0b602ef0be731cf Mon Sep 17 00:00:00 2001
From: Roel Spilker <r.spilker@gmail.com>
Date: Tue, 1 Nov 2011 00:13:22 +0100
Subject: Fixed issue 284 now also for java7, introduced more stubbing

---
 .../javac/java7/CommentCollectingParser.java       | 33 ++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 src/utils/lombok/javac/java7/CommentCollectingParser.java

(limited to 'src/utils/lombok/javac/java7/CommentCollectingParser.java')

diff --git a/src/utils/lombok/javac/java7/CommentCollectingParser.java b/src/utils/lombok/javac/java7/CommentCollectingParser.java
new file mode 100644
index 00000000..7c3cdc71
--- /dev/null
+++ b/src/utils/lombok/javac/java7/CommentCollectingParser.java
@@ -0,0 +1,33 @@
+package lombok.javac.java7;
+
+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.parser.ParserFactory;
+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(ParserFactory fac, Lexer S,
+			boolean keepDocComments, boolean keepLineMap, Map<JCCompilationUnit, List<Comment>> commentsMap) {
+		super(fac, S, keepDocComments, keepLineMap);
+		lexer = S;
+		this.commentsMap = commentsMap;
+	}
+	
+	public JCCompilationUnit parseCompilationUnit() {
+		JCCompilationUnit result = super.parseCompilationUnit();
+		if (lexer instanceof CommentCollectingScanner) {
+			List<Comment> comments = ((CommentCollectingScanner)lexer).getComments();
+			commentsMap.put(result, comments);
+		}
+		return result;
+	}
+}
\ No newline at end of file
-- 
cgit