Java: read formal method arguments

1 files changed, 13 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
index b8a3b62..22d42ab 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
@@ -22,6 +22,7 @@ import org.eclipse.jdt.core.dom.Modifier;
 import org.eclipse.jdt.core.dom.NumberLiteral;
 import org.eclipse.jdt.core.dom.ReturnStatement;
 import org.eclipse.jdt.core.dom.SimpleName;
+import org.eclipse.jdt.core.dom.SingleVariableDeclaration;
 import org.eclipse.jdt.core.dom.Statement;
 import org.eclipse.jdt.core.dom.StringLiteral;
 import org.eclipse.jdt.core.dom.ThisExpression;
@@ -102,6 +103,14 @@ public class JavaReader {
             method.getName().getIdentifier(),
             true,
             builder -> {
+                for (Object parameterObject : method.parameters()) {
+                    val parameter = (SingleVariableDeclaration)parameterObject;
+                    builder.argument(
+                        parameter.resolveBinding().getKey(),
+                        parameter.getName().getIdentifier(),
+                        typeOf(parameter.resolveBinding()));
+                }
+                
                 for (Object statement : method.getBody().statements()) {
                     for (StatementNode intermediateStatement : readStatement((Statement)statement)) {
                         builder.statement(intermediateStatement);                        
@@ -282,6 +291,10 @@ public class JavaReader {
         return typeOf(expression.resolveTypeBinding());
     }
 
+    private static TypeName typeOf(IVariableBinding variableBinding) {
+        return typeOf(variableBinding.getType());
+    }
+
     private static TypeName typeOf(ITypeBinding typeBinding) {
         return TypeName.of(typeBinding.getQualifiedName());
     }