diff options
Diffstat (limited to 'src/core/lombok/eclipse/EclipseAST.java')
| -rw-r--r-- | src/core/lombok/eclipse/EclipseAST.java | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/src/core/lombok/eclipse/EclipseAST.java b/src/core/lombok/eclipse/EclipseAST.java index 370b40fc..0f04f749 100644 --- a/src/core/lombok/eclipse/EclipseAST.java +++ b/src/core/lombok/eclipse/EclipseAST.java @@ -28,6 +28,8 @@ import org.eclipse.jdt.internal.compiler.CompilationResult; import java.util.Collection; import java.util.Collections; import java.util.List; +import java.util.regex.Matcher; +import java.util.regex.Pattern; import lombok.Lombok; import lombok.core.AST; @@ -45,6 +47,9 @@ import org.eclipse.jdt.internal.compiler.ast.Initializer; import org.eclipse.jdt.internal.compiler.ast.LocalDeclaration; import org.eclipse.jdt.internal.compiler.ast.Statement; import org.eclipse.jdt.internal.compiler.ast.TypeDeclaration; +import org.eclipse.jdt.internal.core.BasicCompilationUnit; +import org.eclipse.jdt.internal.core.JavaElement; +import org.eclipse.jdt.internal.core.JavaProject; /** * Wraps around Eclipse's internal AST view to add useful features as well as the ability to visit parents from children, @@ -177,6 +182,30 @@ public class EclipseAST extends AST<EclipseAST, EclipseNode, ASTNode> { return ast.compilationResult.fileName == null ? null : new String(ast.compilationResult.fileName); } + private static final Pattern PROJECT_NAME_FROM_FILEPATH = Pattern.compile("^/([^/]+)/(.*)$"); + /** + * Returns the JavaProject object (eclipse's abstraction of the project) associated with the source file that is represented by this AST. + */ + public JavaProject getProject() { + CompilationUnitDeclaration cud = (CompilationUnitDeclaration) top().get(); + + if (cud.compilationResult().getCompilationUnit() instanceof JavaElement) { + JavaElement icu = (JavaElement) cud.compilationResult.compilationUnit; + return (JavaProject) icu.getJavaProject(); + } + + char[] fn = cud.compilationResult().getFileName(); + + Matcher m = PROJECT_NAME_FROM_FILEPATH.matcher(new String(fn)); + if (m.matches()) { + String projName = m.group(1); + String path = m.group(2); + return EclipseProjectSearcher.getProject(projName); + } + + return null; + } + /** * Call this method to move an EclipseAST generated for a diet parse to rebuild itself for the full parse - * with filled in method bodies and such. Also propagates problems and errors, which in diet parse |