review: feat: support to parse CtType, CtClass and CtField from JDK element CtPath.toString()

doc/path.md

@@ -20,6 +20,15 @@ path = new CtPathStringBuilder().fromString(".spoon.test.path.testclasses.Foo.fo
 List<CtElement> l = path.evaluateOn(root)
 ```
 
+If the root parameter is omitted, spoon will try to find shadow elements (e.g. jdk classes).
+
+> Querying shadow elements currently only supports types, methods and fields.
+
+```java
+path = new CtPathStringBuilder().fromString("#subPackage[name=java]#subPackage[name=util]#containedType[name=HashSet]");
+List<CtElement> l = path.evaluateOn()
+```
+
 ### Creating AST paths
 
 #### From an existing element

src/main/java/spoon/reflect/declaration/CtClass.java

@@ -49,6 +49,21 @@ public interface CtClass<T> extends CtType<T>, CtStatement, CtSealable {
 	@PropertyGetter(role = CONSTRUCTOR)
 	CtConstructor<T> getConstructor(CtTypeReference<?>... parameterTypes);
 
+	/**
+	 * Returns the constructor of the class that takes the given signature.
+	 * e.g. java.util.HashSet has constructor with no parameters `()`
+	 * e.g. java.util.HashSet has constructor with a int and a float parameters `(int,float)`
+	 * e.g. java.util.HashSet has constructor with a java.util.Collection parameter `(java.util.Collection)`
+	 *
+	 * Derived from {@link #getTypeMembers()}
+	 *
+	 * @param signature the constructor signature
+	 * @return the constructor matching the signature
+	 */
+	@DerivedProperty
+	@PropertyGetter(role = CONSTRUCTOR)
+	CtConstructor<T> getConstructorBySignature(String signature);
+
 	/**
 	 * Returns the constructors of this class. This includes the default
 	 * constructor if this class has no constructors explicitly declared.

src/main/java/spoon/reflect/declaration/CtType.java

@@ -234,6 +234,16 @@ public interface CtType<T> extends CtNamedElement, CtTypeInformation, CtTypeMemb
 	@PropertyGetter(role = METHOD)
 	<R> CtMethod<R> getMethod(String name, CtTypeReference<?>... parameterTypes);
 
+	/**
+	 * Gets a method from its signature.
+	 *
+	 * @param signature the method signature, e.g. "doSomething(int,java.lang.String)"
+	 * @return null if does not exit
+	 *
+	 */
+	@PropertyGetter(role = METHOD)
+	<R> CtMethod<R> getMethodBySignature(String signature);
+
 	/**
 	 * Returns the methods that are directly declared by this class or
 	 * interface.

src/main/java/spoon/reflect/path/CtPath.java

@@ -21,6 +21,12 @@ public interface CtPath {
 	 */
 	<T extends CtElement> List<T> evaluateOn(CtElement... startNode);
 
+	/**
+	 * Search for elements matching this CtPatch from shadow model.
+	 * @return the list of matched elements for this path
+	 */
+	CtElement evaluateOnShadowModel();
+
 	/**
 	 *
 	 * Returns the path that is relative to the given element (subpath from it to the end of the path).

src/main/java/spoon/reflect/path/CtPathStringBuilder.java

@@ -13,8 +13,8 @@
 import spoon.reflect.path.impl.CtNamedPathElement;
 import spoon.reflect.path.impl.CtPathElement;
 import spoon.reflect.path.impl.CtPathImpl;
-import spoon.reflect.path.impl.CtTypedNameElement;
 import spoon.reflect.path.impl.CtRolePathElement;
+import spoon.reflect.path.impl.CtTypedNameElement;
 
 import java.util.ArrayDeque;
 import java.util.Deque;
@@ -145,7 +145,8 @@ private String parseArgumentValue(Tokenizer tokenizer, String argName, CtPathEle
 				}
 			} else if ("]".equals(token) || ";".equals(token)) {
 				//finished reading of argument value
-				pathElement.addArgument(argName, argValue.toString());
+				//[fix bug]:AbstractPathElement.getArguments([constructor with no parameters])
+				pathElement.addArgument(argName, argValue.toString().replace("())", "()"));
 				return token;
 			}
 			argValue.append(token);

src/main/java/spoon/reflect/path/impl/CtPathImpl.java

@@ -7,8 +7,12 @@
  */
 package spoon.reflect.path.impl;
 
+import spoon.reflect.declaration.CtClass;
 import spoon.reflect.declaration.CtElement;
+import spoon.reflect.declaration.CtType;
+import spoon.reflect.factory.TypeFactory;
 import spoon.reflect.path.CtPath;
+import spoon.reflect.path.CtRole;
 
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -36,6 +40,70 @@
 		return (List<T>) filtered;
 	}
 
+	@Override
+	public CtElement evaluateOnShadowModel() {
+		List<String> classRoleNameList = new LinkedList<>();
+		CtType<?> ctType = null;
+		for (CtPathElement element : elements) {
+			if (element instanceof CtRolePathElement) {    // search by CtRolePathElement
+				Collection<String> values = ((CtRolePathElement) element).getArguments().values();
+				String val = null;
+				if (values.iterator().hasNext()) {
+					val = values.iterator().next();
+				}
+				if (val != null) {
+					if (CtRole.SUB_PACKAGE.equals(((CtRolePathElement) element).getRole())
+							|| CtRole.CONTAINED_TYPE.equals(((CtRolePathElement) element).getRole())) {
+						classRoleNameList.add(val);
+					}
+					Class<?> cls = getJdkClass(String.join(".", classRoleNameList));
+					if (cls != null) {
+						if (ctType == null) {
+							ctType = new TypeFactory().get(cls);
+						} else {
+							if (CtRole.METHOD.equals(((CtRolePathElement) element).getRole())) {
+								return ctType.getMethodBySignature(val);
+							}
+							if (CtRole.CONSTRUCTOR.equals(((CtRolePathElement) element).getRole())) {
+								return ((CtClass) ctType).getConstructorBySignature(val);
+							}
+							if (CtRole.FIELD.equals(((CtRolePathElement) element).getRole())) {
+								return ctType.getField(val);
+							}
+						}
+					}
+				}
+			}
+		}
+		return ctType;
+	}
+
+	private Class<?> getJdkClass(String name) {
+		name = name.replaceAll("[\\[\\]]", "");
+		switch (name) {
+			case "byte":
+				return byte.class;
+			case "int":
+				return int.class;
+			case "long":
+				return long.class;
+			case "float":
+				return float.class;
+			case "double":
+				return double.class;
+			case "char":
+				return char.class;
+			case "boolean":
+				return boolean.class;
+			default:
+				try {
+					return Class.forName(name);
+				} catch (ClassNotFoundException e) {
+				}
+		}
+		return null;
+	}
+
 	@Override
 	public CtPath relativePath(CtElement parent) {
 		List<CtElement> roots = new ArrayList<>();

src/main/java/spoon/support/reflect/declaration/CtClassImpl.java

@@ -88,6 +88,20 @@ public CtConstructor<T> getConstructor(CtTypeReference<?>... parameterTypes) {
 		return null;
 	}
 
+	@Override
+	public CtConstructor<T> getConstructorBySignature(String signature) {
+		for (CtTypeMember typeMember : getTypeMembers()) {
+			if (!(typeMember instanceof CtConstructor)) {
+				continue;
+			}
+			CtConstructor<T> c = (CtConstructor<T>) typeMember;
+			if (c.getSignature().replaceAll(c.getDeclaringType().getQualifiedName(), "").equals(signature)) {
+				return c;
+			}
+		}
+		return null;
+	}
+
 	@Override
 	public Set<CtConstructor<T>> getConstructors() {
 		Set<CtConstructor<T>> constructors = new SignatureBasedSortedSet<>();

src/main/java/spoon/support/reflect/declaration/CtTypeImpl.java

@@ -737,6 +737,21 @@ public <R> CtMethod<R> getMethod(String name, CtTypeReference<?>... parameterTyp
 		return null;
 	}
 
+	@Override
+	@SuppressWarnings("unchecked")
+	public <R> CtMethod<R> getMethodBySignature(String signature) {
+		if (signature == null) {
+			return null;
+		}
+		String name = signature.substring(0, signature.indexOf('('));
+		for (CtMethod<?> candidate : getMethodsByName(name)) {
+			if (candidate.getSignature().equals(signature)) {
+				return (CtMethod<R>) candidate;
+			}
+		}
+		return null;
+	}
+
 	protected boolean hasSameParameters(CtExecutable<?> candidate, CtTypeReference<?>... parameterTypes) {
 		if (candidate.getParameters().size() != parameterTypes.length) {
 			return false;

src/test/java/spoon/test/path/PathTest.java

@@ -39,6 +39,7 @@
 import spoon.reflect.declaration.CtType;
 import spoon.reflect.declaration.CtTypeMember;
 import spoon.reflect.factory.Factory;
+import spoon.reflect.factory.TypeFactory;
 import spoon.reflect.path.CtElementPathBuilder;
 import spoon.reflect.path.CtPath;
 import spoon.reflect.path.CtPathBuilder;
@@ -55,6 +56,7 @@
 import static org.junit.jupiter.api.Assertions.assertSame;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.junit.jupiter.api.Assertions.fail;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
 
 /**
  * Created by nicolas on 10/06/2015.
@@ -409,4 +411,31 @@ public void testFieldOfArrayType() {
 		assertEquals(1, result.size());
 		assertSame(argType, result.get(0));
 	}
+
+	@Test
+	public void testGetJdkElementByCtPathString() {
+		CtPath path;
+		// test class
+		path = new CtPathStringBuilder().fromString("#subPackage[name=java]#subPackage[name=util]#containedType[name=HashSet]");
+		assertNotNull(path.evaluateOnShadowModel());
+		CtType class_HashSet = new TypeFactory().get(java.util.HashSet.class);
+		assertEquals(path.evaluateOnShadowModel(), class_HashSet);
+		// if unable to get the method or the field, it will try to return the class.
+		// test method
+		path = new CtPathStringBuilder().fromString("#subPackage[name=java]#subPackage[name=util]#containedType[name=HashSet]#method[signature=contains(java.lang.Object)]");
+		assertNotNull(path.evaluateOnShadowModel());
+		assertEquals(path.evaluateOnShadowModel(), class_HashSet.getMethodBySignature("contains(java.lang.Object)"));
+		// test constructor
+		path = new CtPathStringBuilder().fromString("#subPackage[name=java]#subPackage[name=util]#containedType[name=HashSet]#constructor[signature=()]");
+		assertNotNull(path.evaluateOnShadowModel());
+		assertEquals(path.evaluateOnShadowModel(), ((CtClass) class_HashSet).getConstructorBySignature("()"));
+
+		path = new CtPathStringBuilder().fromString("#subPackage[name=java]#subPackage[name=util]#containedType[name=HashSet]#constructor[signature=(int)]");
+		assertNotNull(path.evaluateOnShadowModel());
+		assertEquals(path.evaluateOnShadowModel(), ((CtClass) class_HashSet).getConstructorBySignature("(int)"));
+		// test field
+		path = new CtPathStringBuilder().fromString("#subPackage[name=java]#subPackage[name=util]#containedType[name=HashSet]#field[name=map]");
+		assertNotNull(path.evaluateOnShadowModel());
+		assertEquals(path.evaluateOnShadowModel(), class_HashSet.getField("map"));
+	}
 }