From 90dae189f638148a7b72add782023088cf22e847 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 2 Oct 2015 19:11:03 +0100
Subject: Use the name of the class node when adding classes to project

---
 .../org/zwobble/couscous/MapBackedProject.java     | 26 ++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

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

diff --git a/src/main/java/org/zwobble/couscous/MapBackedProject.java b/src/main/java/org/zwobble/couscous/MapBackedProject.java
index 66b3681..f8e457f 100644
--- a/src/main/java/org/zwobble/couscous/MapBackedProject.java
+++ b/src/main/java/org/zwobble/couscous/MapBackedProject.java
@@ -4,12 +4,30 @@ import java.util.Map;
 
 import org.zwobble.couscous.values.ConcreteType;
 
-import lombok.Builder;
-import lombok.Singular;
+import com.google.common.collect.ImmutableMap;
 
-@Builder
 public class MapBackedProject implements Project {
-    @Singular(value="addClass")
+    public static Builder builder() {
+        return new Builder();
+    }
+    
+    public static class Builder {
+        private final ImmutableMap.Builder<String, ConcreteType<?>> classes;
+        
+        private Builder() {
+            classes = ImmutableMap.builder();
+        }
+        
+        public Builder addClass(ConcreteType<?> clazz) {
+            classes.put(clazz.getName(), clazz);
+            return this;
+        }
+        
+        public Project build() {
+            return new MapBackedProject(classes.build());
+        }
+    }
+    
     private Map<String, ConcreteType<?>> classes;
 
     public MapBackedProject(Map<String, ConcreteType<?>> classes) {
-- 
cgit v1.2.3