com.github.javaparser.resolution.Navigator Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of javaparser-core Show documentation
Show all versions of javaparser-core Show documentation
The core parser functionality. This may be all you need.
/*
* Copyright (C) 2015-2016 Federico Tomassetti
* Copyright (C) 2017-2023 The JavaParser Team.
*
* This file is part of JavaParser.
*
* JavaParser can be used either under the terms of
* a) the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* b) the terms of the Apache License
*
* You should have received a copy of both licenses in LICENCE.LGPL and
* LICENCE.APACHE. Please refer to those files for details.
*
* JavaParser is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*/
package com.github.javaparser.resolution;
import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.ast.Node;
import com.github.javaparser.ast.body.*;
import com.github.javaparser.ast.expr.MethodCallExpr;
import com.github.javaparser.ast.expr.NameExpr;
import com.github.javaparser.ast.expr.SimpleName;
import com.github.javaparser.ast.stmt.ReturnStmt;
import com.github.javaparser.ast.stmt.SwitchStmt;
import java.util.Optional;
import java.util.function.Predicate;
/**
* This class can be used to easily retrieve nodes from a JavaParser AST.
*
* Note that methods with the prefix `demand` indicate that if the search value is not found, an exception will be thrown.
*
* @author Federico Tomassetti
*/
public final class Navigator {
private Navigator() {
// prevent instantiation
}
public static ClassOrInterfaceDeclaration demandClass(CompilationUnit cu, String qualifiedName) {
ClassOrInterfaceDeclaration cd = demandClassOrInterface(cu, qualifiedName);
if (cd.isInterface()) {
throw new IllegalStateException("Type is not a class");
}
return cd;
}
public static ClassOrInterfaceDeclaration demandClassOrInterface(CompilationUnit compilationUnit, String qualifiedName) {
return findType(compilationUnit, qualifiedName)
.map(res -> res.toClassOrInterfaceDeclaration().orElseThrow(() -> new IllegalStateException("Type is not a class or an interface, it is " + res.getClass().getCanonicalName())))
.orElseThrow(() -> new IllegalStateException("No type named '" + qualifiedName + "'found"));
}
/**
* Returns the {@code (i+1)}'th constructor of the given type declaration, in textual order. The constructor that
* appears first has the index 0, the second one the index 1, and so on.
*
* @param td The type declaration to search in. Note that only classes and enums have constructors.
* @param index The index of the desired constructor.
* @return The desired ConstructorDeclaration if it was found, else an exception is thrown.
*/
public static ConstructorDeclaration demandConstructor(TypeDeclaration td, int index) {
// TODO: Refactor to use `td.findAll(ConstructorDeclaration.class);` - potential difference re: searching only immediate children?
ConstructorDeclaration found = null;
int i = 0;
for (BodyDeclaration bd : td.getMembers()) {
if (bd instanceof ConstructorDeclaration) {
ConstructorDeclaration cd = (ConstructorDeclaration) bd;
if (i == index) {
found = cd;
break;
}
i++;
}
}
if (found == null) {
throw new IllegalStateException("No constructor with index " + index);
}
return found;
}
public static EnumDeclaration demandEnum(CompilationUnit cu, String qualifiedName) {
Optional> res = findType(cu, qualifiedName);
if (!res.isPresent()) {
throw new IllegalStateException("No type found");
}
if (!(res.get() instanceof EnumDeclaration)) {
throw new IllegalStateException("Type is not an enum");
}
return (EnumDeclaration) res.get();
}
public static VariableDeclarator demandField(ClassOrInterfaceDeclaration cd, String name) {
for (BodyDeclaration bd : cd.getMembers()) {
if (bd instanceof FieldDeclaration) {
FieldDeclaration fd = (FieldDeclaration) bd;
for (VariableDeclarator vd : fd.getVariables()) {
if (vd.getName().getId().equals(name)) {
return vd;
}
}
}
}
throw new IllegalStateException("No field with given name");
}
public static ClassOrInterfaceDeclaration demandInterface(CompilationUnit cu, String qualifiedName) {
ClassOrInterfaceDeclaration cd = demandClassOrInterface(cu, qualifiedName);
if (!cd.isInterface()) {
throw new IllegalStateException("Type is not an interface");
}
return cd;
}
public static MethodDeclaration demandMethod(TypeDeclaration cd, String name) {
MethodDeclaration found = null;
for (BodyDeclaration bd : cd.getMembers()) {
if (bd instanceof MethodDeclaration) {
MethodDeclaration md = (MethodDeclaration) bd;
if (md.getNameAsString().equals(name)) {
if (found != null) {
throw new IllegalStateException("Ambiguous getName");
}
found = md;
}
}
}
if (found == null) {
throw new IllegalStateException("No method called " + name);
}
return found;
}
public static N demandNodeOfGivenClass(Node node, Class clazz) {
return node.findFirst(clazz).orElseThrow(IllegalArgumentException::new);
}
public static Node demandParentNode(Node node) {
return node.getParentNode().orElseThrow(() -> new IllegalStateException("Parent not found, the node does not appear to be inserted in a correct AST"));
}
/**
* Traverses the parent chain starting at {@code node} and returns the
* first Node that returns make {@code isAcceptedParentNode} evaluate to
* {@code true}.
*/
public static Node demandParentNode(Node node, Predicate isAcceptedParentNode) {
Node parent = node;
do {
parent = demandParentNode(parent);
} while (!isAcceptedParentNode.test(parent));
return parent;
}
public static ReturnStmt demandReturnStmt(MethodDeclaration method) {
return demandNodeOfGivenClass(method, ReturnStmt.class);
}
public static SwitchStmt demandSwitch(Node node) {
return findSwitchHelper(node).orElseThrow(IllegalArgumentException::new);
}
public static Optional demandVariableDeclaration(Node node, String name) {
return node.findFirst(VariableDeclarator.class, n -> n.getNameAsString().equals(name));
}
public static Optional findMethodCall(Node node, String methodName) {
return node.findFirst(MethodCallExpr.class, n -> n.getNameAsString().equals(methodName));
}
public static Optional findNameExpression(Node node, String name) {
return node.findFirst(NameExpr.class, n -> n.getNameAsString().equals(name));
}
/**
* @deprecated Use {@link #demandNodeOfGivenClass(Node, Class)}
*/
@Deprecated
public static N findNodeOfGivenClass(Node node, Class clazz) {
return demandNodeOfGivenClass(node, clazz);
}
/**
* @deprecated Use {@link #demandReturnStmt(MethodDeclaration)}
*/
@Deprecated
public static ReturnStmt findReturnStmt(MethodDeclaration method) {
return demandReturnStmt(method);
}
public static Optional findSimpleName(Node node, String name) {
return node.findFirst(SimpleName.class, n -> n.asString().equals(name));
}
/**
* @deprecated Use {@link #demandSwitch(Node)}
*/
@Deprecated
public static SwitchStmt findSwitch(Node node) {
return demandSwitch(node);
}
private static Optional findSwitchHelper(Node node) {
if (node instanceof SwitchStmt) {
return Optional.of((SwitchStmt) node);
}
return node.findFirst(SwitchStmt.class);
}
/**
* Looks among the type declared in the Compilation Unit for one having the specified name.
* The name can be qualified with respect to the compilation unit. For example, if the compilation
* unit is in package a.b; and it contains two top level classes named C and D, with class E being defined inside D
* then the qualifiedName that can be resolved are "C", "D", and "D.E".
*/
public static Optional> findType(CompilationUnit cu, String qualifiedName) {
if (cu.getTypes().isEmpty()) {
return Optional.empty();
}
final String typeName = getOuterTypeName(qualifiedName);
Optional> type = cu.getTypes().stream().filter((t) -> t.getName().getId().equals(typeName)).findFirst();
final String innerTypeName = getInnerTypeName(qualifiedName);
if (type.isPresent() && !innerTypeName.isEmpty()) {
return findType(type.get(), innerTypeName);
}
return type;
}
/**
* Looks among the type declared in the TypeDeclaration for one having the specified name.
* The name can be qualified with respect to the TypeDeclaration. For example, if the class declaration defines
* class D and class D contains an internal class named E then the qualifiedName that can be resolved are "D", and
* "D.E".
*/
public static Optional> findType(TypeDeclaration td, String qualifiedName) {
final String typeName = getOuterTypeName(qualifiedName);
Optional> type = Optional.empty();
for (Node n : td.getMembers()) {
if (n instanceof TypeDeclaration && ((TypeDeclaration) n).getName().getId().equals(typeName)) {
type = Optional.of((TypeDeclaration) n);
break;
}
}
final String innerTypeName = getInnerTypeName(qualifiedName);
if (type.isPresent() && !innerTypeName.isEmpty()) {
return findType(type.get(), innerTypeName);
}
return type;
}
private static String getInnerTypeName(String qualifiedName) {
if (qualifiedName.contains(".")) {
return qualifiedName.split("\\.", 2)[1];
}
return "";
}
private static String getOuterTypeName(String qualifiedName) {
return qualifiedName.split("\\.", 2)[0];
}
/**
* @deprecated Use {@link #demandParentNode(Node)}
*/
@Deprecated
public static Node requireParentNode(Node node) {
return demandParentNode(node);
}
}