Fix to javac handler; it now properly detects single package names, such as "package foo;"

This commit is contained in:
Petr Jiricka authored and Reinier Zwitserloot committed 2009-12-17 12:56:58 +01:00
1 parent 31d7d8ec92
commit 61b054d787
1 file changed
+2 -1
+2 -1
View File
@@ -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) {