From 4054a250df414bb2c756d12b133aaaa8cd6bbce3 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Tue, 20 Oct 2015 19:00:49 +0100
Subject: Java: read formal method arguments

---
 src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java')

diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
index b4ddad9..7466f07 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
@@ -5,6 +5,9 @@ import java.util.function.Function;
 
 import com.google.common.collect.ImmutableList;
 
+import static org.zwobble.couscous.ast.FormalArgumentNode.formalArg;
+import static org.zwobble.couscous.ast.VariableDeclaration.var;
+
 public class ClassNodeBuilder {
     private final TypeName name;
     private final ImmutableList.Builder<FieldDeclarationNode> fields;
@@ -85,6 +88,11 @@ public class ClassNodeBuilder {
             return ThisReferenceNode.thisReference(name);
         }
         
+        public MethodBuilder<T> argument(String id, String name, TypeName type) {
+            arguments.add(formalArg(var(id, name, type)));
+            return this;
+        }
+        
         public MethodBuilder<T> argument(FormalArgumentNode argument) {
             arguments.add(argument);
             return this;
-- 
cgit v1.2.3