[AST] Method call in separate values

This commit is contained in:
jeffcheasey88 2023-11-09 10:13:38 +01:00
parent b78ef3213e
commit d402407ae2
2 changed files with 31 additions and 8 deletions

View file

@ -307,7 +307,23 @@ public class JavaParser extends Parser<JavaElement> {
StateTree<JavaElement> value_call = value_name.then((validator) -> validator.validate( StateTree<JavaElement> value_call = value_name.then((validator) -> validator.validate(
(token) -> token.getValue().equals("."), (token) -> token.getValue().equals("."),
(bag, token) -> bag.set(bag.<Token>get().concat(token)))); (bag, token) -> {
Value prev = bag.get("prev");
if(prev == null){
bag.set("prev", new Value(bag.<Token>get()));
}else{
Integer paramters = bag.get("args");
List<Value> list = null;
if(paramters != null){
list = new ArrayList<>();
for(int i = 0; i < paramters; i++) list.add(bag.<Bag>get("arg"+i).get());
bag.remove("args");
}
MethodCallValue methodCall = new MethodCallValue(prev, bag.get(), list);
bag.set("prev", methodCall);
}
bag.set(token);
}));
value_call.end((a,b) -> a); value_call.end((a,b) -> a);
value_call.then(new RedirectStateTree<>(gen, (global, local) -> global.set(global.<Token>get().concat(local.get())))).then(value_name); value_call.then(new RedirectStateTree<>(gen, (global, local) -> global.set(global.<Token>get().concat(local.get())))).then(value_name);
value_call.then(value_name); value_call.then(value_name);
@ -337,9 +353,7 @@ public class JavaParser extends Parser<JavaElement> {
for(int i = 0; i < paramters; i++) list.add(bag.<Bag>get("arg"+i).get()); for(int i = 0; i < paramters; i++) list.add(bag.<Bag>get("arg"+i).get());
} }
MethodCallValue methodCall = new MethodCallValue(bag.get(), list); MethodCallValue methodCall = new MethodCallValue(bag.get("prev"), bag.get(), list);
//take arguments from bag
System.out.println("MethodCall "+bag);
bag.set(methodCall); bag.set(methodCall);
return null; return null;
}); });
@ -1167,7 +1181,7 @@ public class JavaParser extends Parser<JavaElement> {
} }
public static void main(String[] args) throws Exception{ public static void main(String[] args) throws Exception{
File file = new File("C:\\Users\\jeffc\\eclipse-workspace\\"); File file = new File("C:\\Users\\jeffc\\eclipse-workspace\\peer-at-code-backend\\src\\dev\\peerat\\backend\\routes\\users\\ChangePassword.java");
BufferedReader reader = new BufferedReader(new FileReader(file)); BufferedReader reader = new BufferedReader(new FileReader(file));
@ -1180,7 +1194,7 @@ public class JavaParser extends Parser<JavaElement> {
List<Value> values = new ArrayList<>(); List<Value> values = new ArrayList<>();
jFile.findAll((e) -> e instanceof Value, values); jFile.findAll((e) -> e instanceof Value, values);
System.out.println("Find "+values.size()+" values"); System.out.println("Find "+values.size()+" values");
for(Value v : values) System.out.println(v); for(Value v : values) System.out.println(v.getClass().getSimpleName()+" ->"+v);
System.out.println((System.currentTimeMillis()-time)+"ms"); System.out.println((System.currentTimeMillis()-time)+"ms");

View file

@ -152,14 +152,20 @@ public class Value extends JavaElement{
public static class MethodCallValue extends Value{ public static class MethodCallValue extends Value{
private Value base;
private Token token; private Token token;
private List<Value> parameters; private List<Value> parameters;
public MethodCallValue(Token token, List<Value> parameters){ public MethodCallValue(Value base, Token token, List<Value> parameters){
this.base = base;
this.token = token; this.token = token;
this.parameters = parameters; this.parameters = parameters;
} }
private Value base(){
return this.base;
}
public Token getToken(){ public Token getToken(){
return token; return token;
} }
@ -170,7 +176,7 @@ public class Value extends JavaElement{
@Override @Override
public String toString(){ public String toString(){
return token.getValue()+((parameters == null ? "":parameters)); return base+token.getValue()+((parameters == null ? "":parameters));
} }
@Override @Override
@ -180,6 +186,7 @@ public class Value extends JavaElement{
@Override @Override
public <E extends JavaElement> E find(Function<JavaElement, Boolean> finder){ public <E extends JavaElement> E find(Function<JavaElement, Boolean> finder){
if(finder.apply(base)) return (E) base;
if(parameters != null){ if(parameters != null){
for(Value value : parameters){ for(Value value : parameters){
if(finder.apply(value)) return (E) value; if(finder.apply(value)) return (E) value;
@ -190,6 +197,8 @@ public class Value extends JavaElement{
@Override @Override
public <E extends JavaElement> void findAll(Function<JavaElement, Boolean> finder, List<E> list){ public <E extends JavaElement> void findAll(Function<JavaElement, Boolean> finder, List<E> list){
if(finder.apply(base)) list.add((E) base);
base.findAll(finder, list);
if(parameters != null){ if(parameters != null){
for(Value value : parameters){ for(Value value : parameters){
if(finder.apply(value)) list.add((E) value); if(finder.apply(value)) list.add((E) value);