From 9d845a8deeb622736d54693ceb8efc9ea50497ca Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Tue, 6 Oct 2015 16:55:10 +0100
Subject: Add static method builder to ClassNodeBuilder

---
 .../java/org/zwobble/couscous/ast/ClassNodeBuilder.java   | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

(limited to 'src/main/java/org/zwobble')

diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
index ce8c9a0..2a0d967 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
@@ -34,19 +34,28 @@ public class ClassNodeBuilder {
                 builder.arguments.build(),
                 builder.statements.build()));
     }
-    
+
     public ClassNodeBuilder method(MethodNode method) {
         this.methods.add(method);
         return this;
     }
     
+    public ClassNodeBuilder staticMethod(String name, Function<MethodBuilder<MethodNode>, MethodBuilder<MethodNode>> build) {
+        return method(name, true, build);
+    }
+    
     public ClassNodeBuilder method(String name, Function<MethodBuilder<MethodNode>, MethodBuilder<MethodNode>> build) {
-        this.methods.add(build.apply(methodBuilder(name)).build());
+        return method(name, false, build);
+    }
+    
+    public ClassNodeBuilder method(String name, boolean isStatic, Function<MethodBuilder<MethodNode>, MethodBuilder<MethodNode>> build) {
+        this.methods.add(build.apply(methodBuilder(name, isStatic)).build());
         return this;
     }
 
-    private MethodBuilder<MethodNode> methodBuilder(String name) {
+    private MethodBuilder<MethodNode> methodBuilder(String name, boolean isStatic) {
         return new MethodBuilder<MethodNode>(builder -> MethodNode.builder()
+            .isStatic(isStatic)
             .name(name)
             .arguments(builder.arguments.build())
             .body(builder.statements.build())
-- 
cgit v1.2.3