From 61b054d78784f27be5420474af4ccb047e1668cc Mon Sep 17 00:00:00 2001
From: Petr Jiricka <petr.jiricka@sun.com>
Date: Thu, 17 Dec 2009 12:56:58 +0100
Subject: Fix to javac handler; it now properly detects single package names,
 such as "package foo;"

---
 src/core/lombok/javac/JavacAST.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/core/lombok/javac')

diff --git a/src/core/lombok/javac/JavacAST.java b/src/core/lombok/javac/JavacAST.java
index 9cfe7f34..2218b13f 100644
--- a/src/core/lombok/javac/JavacAST.java
+++ b/src/core/lombok/javac/JavacAST.java
@@ -42,6 +42,7 @@ import com.sun.tools.javac.tree.JCTree.JCClassDecl;
 import com.sun.tools.javac.tree.JCTree.JCCompilationUnit;
 import com.sun.tools.javac.tree.JCTree.JCExpression;
 import com.sun.tools.javac.tree.JCTree.JCFieldAccess;
+import com.sun.tools.javac.tree.JCTree.JCIdent;
 import com.sun.tools.javac.tree.JCTree.JCImport;
 import com.sun.tools.javac.tree.JCTree.JCMethodDecl;
 import com.sun.tools.javac.tree.JCTree.JCStatement;
@@ -87,7 +88,7 @@ public class JavacAST extends AST<JavacAST, JavacNode, JCTree> {
 	}
 	
 	private static String packageDeclaration(JCCompilationUnit cu) {
-		return cu.pid instanceof JCFieldAccess ? cu.pid.toString() : null;
+		return (cu.pid instanceof JCFieldAccess || cu.pid instanceof JCIdent) ? cu.pid.toString() : null;
 	}
 	
 	private static Collection<String> imports(JCCompilationUnit cu) {
-- 
cgit