Add static method builder to ClassNodeBuilder

1 files changed, 12 insertions, 3 deletions

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())