Skip to content

Commit 22f61c3

Browse files
Fix improper cast insertion
1 parent 0bddc73 commit 22f61c3

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/main/java/anthonisen/felix/astParsing/visitors/CastInsertionVisitor.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,11 @@
33
import java.util.Map;
44
import java.util.Optional;
55
import com.github.javaparser.ast.expr.MethodCallExpr;
6+
import com.github.javaparser.ast.expr.CastExpr;
7+
import com.github.javaparser.ast.expr.EnclosedExpr;
68
import com.github.javaparser.ast.expr.Expression;
79
import com.github.javaparser.ast.expr.NameExpr;
10+
import com.github.javaparser.ast.type.ClassOrInterfaceType;
811
import com.github.javaparser.ast.visitor.ModifierVisitor;
912
import com.github.javaparser.ast.visitor.Visitable;
1013

@@ -30,8 +33,10 @@ public Visitable visit(MethodCallExpr n, Void arg) {
3033
MethodData data = methodMap.get(n.getNameAsString());
3134
if (data != null && data.shouldCast()) {
3235
String castString = data.castString().replace("*", ref.second);
33-
expr.setName("(" + castString + ") " + ref.first); // TODO use the correct methods to make a
34-
// castexpr
36+
ClassOrInterfaceType castType = new ClassOrInterfaceType(null, castString);
37+
CastExpr cast = new CastExpr(castType, n);
38+
EnclosedExpr enclosedCast = new EnclosedExpr(cast);
39+
return enclosedCast;
3540
}
3641
}
3742
return super.visit(n, arg);

0 commit comments

Comments
 (0)