From 5116bf791a6360627e6683d818623b49e25900a5 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 10 Oct 2015 14:16:07 +0100
Subject: Move project classes into correct package

---
 .../couscous/interpreter/MapBackedProject.java     | 56 ++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 src/main/java/org/zwobble/couscous/interpreter/MapBackedProject.java

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

diff --git a/src/main/java/org/zwobble/couscous/interpreter/MapBackedProject.java b/src/main/java/org/zwobble/couscous/interpreter/MapBackedProject.java
new file mode 100644
index 0000000..7206b1a
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/interpreter/MapBackedProject.java
@@ -0,0 +1,56 @@
+package org.zwobble.couscous.interpreter;
+
+import java.util.Map;
+
+import org.zwobble.couscous.ast.TypeName;
+import org.zwobble.couscous.interpreter.values.ConcreteType;
+
+import com.google.common.collect.ImmutableMap;
+
+import lombok.val;
+
+public class MapBackedProject implements Project {
+    public static Builder builder() {
+        return new Builder();
+    }
+    
+    public static class Builder {
+        private final ImmutableMap.Builder<TypeName, ConcreteType> classes;
+        
+        private Builder() {
+            classes = ImmutableMap.builder();
+        }
+        
+        public Builder addClass(ConcreteType clazz) {
+            classes.put(clazz.getName(), clazz);
+            return this;
+        }
+        
+        public Builder addClasses(Iterable<ConcreteType> classes) {
+            for (val clazz : classes) {
+                addClass(clazz);
+            }
+            return this;
+        }
+        
+        public Project build() {
+            return new MapBackedProject(classes.build());
+        }
+    }
+    
+    private Map<TypeName, ConcreteType> classes;
+
+    public MapBackedProject(Map<TypeName, ConcreteType> classes) {
+        this.classes = classes;
+    }
+    
+    @Override
+    public ConcreteType findClass(TypeName name) {
+        if (classes.containsKey(name)) {
+            return classes.get(name);            
+        } else {
+            throw new IllegalArgumentException("Cannot find class: " + name);
+        }
+        
+    }
+}
-- 
cgit v1.2.3