Organize into sub-packages

1 files changed, 58 insertions, 0 deletions

diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/model/Cupboard.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/model/Cupboard.java
new file mode 100644
index 0000000..fedf04e
--- /dev/null
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/model/Cupboard.java
@@ -0,0 +1,58 @@
+package com.ufund.api.ufundapi.model;
+
+import java.util.ArrayList;
+
+public class Cupboard {
+    ArrayList<Need> needs;
+
+    public Cupboard() {
+        needs = new ArrayList<>();
+    }
+
+    public void addNeed(Need need) {
+        needs.add(need);
+    }
+
+    public void createNeed(String name, double max, Need.GoalType type) {
+        int id = needs.size();
+        Need need = new Need(name, id, max, type);
+        addNeed(need);
+    }
+
+    public void updateID(int id) {
+        for (int i = 0; i < needs.size(); i++) {
+            needs.get(i).setID(i);
+        }
+    }
+
+    public void removeNeed(Need need) {
+        needs.remove(need);
+    }
+
+    public void updateNeed(Need need) {
+        for (int i = 0; i < needs.size(); i++) {
+            if (needs.get(i).getID() == need.getID()) {
+                needs.set(i, need);
+                return;
+            }
+        }
+    }
+
+    public void removeNeed(String name) {
+        for (int i = 0; i < needs.size(); i++) {
+            if (needs.get(i).getName().equals(name)) {
+                needs.remove(i);
+                return;
+            }
+        }
+    }
+
+    public Need[] getNeeds() {
+        return needs.toArray(new Need[0]);
+    }
+
+    public Need getNeed(int index) {
+        return needs.get(index);
+    }
+
+}