Util -> Find method

src/be/jeffcheasey88/peeratcode/parser/java/Class.java

@@ -107,4 +107,15 @@ public class Class extends JavaElement{
 		}
 		System.out.println(start+"}");
 	}
+
+	@Override
+	public JavaElement find(java.util.function.Function<JavaElement, Boolean> search, java.util.function.Function<JavaElement, Boolean> deep){
+		if(search.apply(this)) return this;
+		if(!deep.apply(this)) return null;
+		for(JavaElement element : this.childs){
+			JavaElement result = element.find(search, deep);
+			if(result != null) return result;
+		}
+		return null;
+	}
 }

src/be/jeffcheasey88/peeratcode/parser/java/Function.java

@@ -103,4 +103,5 @@ public class Function extends OperationContainer{
 		for(JavaElement child : getChilds()) child.show(tab+1);
 		System.out.println(start+"}");
 	}
+	
 }

src/be/jeffcheasey88/peeratcode/parser/java/JavaElement.java

@@ -1,9 +1,13 @@
 package be.jeffcheasey88.peeratcode.parser.java;
 
+import java.util.function.Function;
+
 public abstract class JavaElement {
 
 	public abstract int parse(String content, CleanerPool global, CleanerPool local) throws Exception;
 	
+	public abstract JavaElement find(Function<JavaElement, Boolean> search, Function<JavaElement, Boolean> deep);
+	
 	//Only for development
 	public abstract void show(int tab);
 }

src/be/jeffcheasey88/peeratcode/parser/java/JavaParser.java

@@ -8,6 +8,7 @@ import java.io.FileWriter;
 import java.lang.reflect.Modifier;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.function.Function;
 
 import be.jeffcheasey88.peeratcode.parser.java.CleanerPool.Cleaner;
 
@@ -109,6 +110,10 @@ public class JavaParser{
 		content = content.substring(index);
 	}
 	
+	public JavaElement find(Function<JavaElement, Boolean> search, Function<JavaElement, Boolean> deep){
+		return this.clazz.find(search, deep);
+	}
+	
 	public Package getPackage(){
 		return this.pack;
 	}

src/be/jeffcheasey88/peeratcode/parser/java/Variable.java

@@ -131,6 +131,14 @@ public class Variable extends JavaElement{
 		System.out.println(start+Modifier.toString(modifier)+(modifier > 0 ? " ":"")+type+" "+name+(value == null ? ";":" = "+value+";"));
 	}
 	
+	@Override
+	public JavaElement find(java.util.function.Function<JavaElement, Boolean> search, java.util.function.Function<JavaElement, Boolean> deep){
+		if(search.apply(this)) return this;
+		if(!deep.apply(this)) return null;
+		//Value of the variable
+		return null;
+	}
+	
 	public static class Value extends Variable{
 		
 		private String value;

src/be/jeffcheasey88/peeratcode/parser/java/operations/AssigmentOperation.java

@@ -33,5 +33,11 @@ public class AssigmentOperation extends JavaElement{
 		System.out.println(start+variable+" = "+value+";");
 	}
 	
-
+	@Override
+	public JavaElement find(java.util.function.Function<JavaElement, Boolean> search, java.util.function.Function<JavaElement, Boolean> deep){
+		if(search.apply(this)) return this;
+		if(!deep.apply(this)) return null;
+		//value
+		return null;
+	}
 }

src/be/jeffcheasey88/peeratcode/parser/java/operations/LoopAffectOperation.java

@@ -29,6 +29,11 @@ public class LoopAffectOperation extends JavaElement{
 		System.out.println(start+"loop affect??;");
 	}
 	
+	@Override
+	public JavaElement find(java.util.function.Function<JavaElement, Boolean> search, java.util.function.Function<JavaElement, Boolean> deep){
+		return search.apply(this) ? this : null;
+	}
+
 	public static class ContinueOperation extends LoopAffectOperation{
 		
 		private static Pattern PATTERN = Pattern.compile("^(\\s*continue\\s*;).*$");

src/be/jeffcheasey88/peeratcode/parser/java/operations/MethodCallOperation.java

@@ -33,5 +33,8 @@ public class MethodCallOperation extends JavaElement{
 		System.out.println(start+value+";");
 	}
 	
-	
+	@Override
+	public JavaElement find(java.util.function.Function<JavaElement, Boolean> search, java.util.function.Function<JavaElement, Boolean> deep){
+		return search.apply(this) ? this : null;
+	}
 }

src/be/jeffcheasey88/peeratcode/parser/java/operations/OperationContainer.java

@@ -21,15 +21,12 @@ public abstract class OperationContainer extends JavaElement{
 	@Override
 	public int parse(String content, CleanerPool global, CleanerPool local) throws Exception{
 		System.out.println("OperationContainer.parse -> "+content);
-//		content = local.unzipOne(content, (s,p) -> s);
-
 		while(!(content.replaceAll("\\s+", "").isEmpty())) content = internalParse(content, global, local);
 		return 0;
 	}
 	
 	public int parseOne(String content, CleanerPool global, CleanerPool local) throws Exception{
 		System.out.println("OperationContainer.parseOne -> "+content);
-//		content = local.unzip(content, (s,p) -> s);
 		String modify = internalParse(content, global, local);
 		return content.length()-modify.length();
 	}
@@ -68,4 +65,15 @@ public abstract class OperationContainer extends JavaElement{
 	public List<JavaElement> getChilds(){
 		return this.childs;
 	}
+	
+	@Override
+	public JavaElement find(java.util.function.Function<JavaElement, Boolean> search, java.util.function.Function<JavaElement, Boolean> deep){
+		if(search.apply(this)) return this;
+		if(!deep.apply(this)) return null;
+		for(JavaElement element : this.childs){
+			JavaElement result = element.find(search, deep);
+			if(result != null) return result;
+		}
+		return null;
+	}
 }

src/be/jeffcheasey88/peeratcode/parser/java/operations/ReturnOperation.java

@@ -47,4 +47,13 @@ public class ReturnOperation extends JavaElement{
 		System.out.println(start+"return "+toChange+";");
 	}
 	
+	@Override
+	public JavaElement find(java.util.function.Function<JavaElement, Boolean> search, java.util.function.Function<JavaElement, Boolean> deep){
+		if(search.apply(this)) return this;
+		if(!deep.apply(this)) return null;
+		JavaElement result = value.find(search, deep);
+		if(result != null) return result;
+		return null;
+	}
+
 }

test/be/jeffcheasey88/peeratcode/parser/java/SearchTest.java

@@ -0,0 +1,59 @@
+package be.jeffcheasey88.peeratcode.parser.java;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.Reader;
+
+import javax.lang.model.element.Element;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.TestInstance;
+import org.junit.jupiter.api.TestInstance.Lifecycle;
+
+import be.jeffcheasey88.peeratcode.parser.java.operations.ConditionalOperation.ForOperation;
+import be.jeffcheasey88.peeratcode.parser.java.operations.ConditionalOperation.WhileOperation;
+import be.jeffcheasey88.peeratcode.parser.java.operations.DoOperation;
+import be.jeffcheasey88.peeratcode.parser.java.operations.LoopAffectOperation.BreakOperation;
+import be.jeffcheasey88.peeratcode.parser.java.operations.LoopAffectOperation.ContinueOperation;
+import be.jeffcheasey88.peeratcode.parser.java.operations.MethodCallOperation;
+import be.jeffcheasey88.peeratcode.parser.java.operations.SynchronizedOperation;
+
+@TestInstance(Lifecycle.PER_CLASS)
+class SearchTest{
+	
+	JavaParser parse(String code) throws Exception{
+		BufferedReader reader = new BufferedReader(new Reader(){public int read(char[] cbuf, int off, int len) throws IOException{return 0;}public void close() throws IOException {}}) {
+			private boolean read = false;
+			@Override
+			public String readLine() throws IOException{
+				if(read) return null;
+				read = true;
+				return code;
+			}
+		};
+		
+		JavaParser parser = new JavaParser(reader);
+		parser.parse();
+		return parser;
+	}
+	
+	@Test
+	void methodCallSearch(){
+		try {
+			JavaParser parser = parse("package be.jeffcheasey88.peeratcode.parser.java; class Test{ void function(){ int i = 0; if(i == 0){ this.none(); } } }");
+			
+			JavaElement element = parser.find(
+					(e) -> e instanceof MethodCallOperation,
+					(e) -> true);
+			
+			assertNotNull(element);
+		} catch (Exception e) {
+			e.printStackTrace();
+		}
+		
+	}
+	
+}