From 9d99a3beada432b446c09ebee3be5cc24db3cf13 Mon Sep 17 00:00:00 2001
From: Harry Linrui XU <xulr0820@hotmail.com>
Date: Wed, 19 Apr 2023 22:16:58 +0200
Subject: [PATCH] Created class for storing budget names

---
 .../demo/data/Budget/BudgetRegister.java      | 35 +++++++++++++++++++
 .../Budget/FileHandlingBudgetArchive.java     |  2 ++
 2 files changed, 37 insertions(+)
 create mode 100644 src/main/java/no/ntnu/idatt1002/demo/data/Budget/BudgetRegister.java
 create mode 100644 src/main/java/no/ntnu/idatt1002/demo/data/Budget/FileHandlingBudgetArchive.java

diff --git a/src/main/java/no/ntnu/idatt1002/demo/data/Budget/BudgetRegister.java b/src/main/java/no/ntnu/idatt1002/demo/data/Budget/BudgetRegister.java
new file mode 100644
index 00000000..0ab516a2
--- /dev/null
+++ b/src/main/java/no/ntnu/idatt1002/demo/data/Budget/BudgetRegister.java
@@ -0,0 +1,35 @@
+package no.ntnu.idatt1002.demo.data.Budget;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class BudgetRegister {
+  private final List<String> budgetNames;
+
+  public BudgetRegister() {
+    budgetNames = new ArrayList<>();
+  }
+
+  public List<String> getBudgetNames() {
+    return this.budgetNames;
+  }
+
+  public void addBudgetName(String name) {
+    if (name == null) throw new IllegalArgumentException("Name cannot be null");
+    if (name.isBlank()) throw new IllegalArgumentException("Name cannot be blank");
+    budgetNames.add(name);
+  }
+
+  public void removeBudgetName(String name) {
+    if (name == null) throw new IllegalArgumentException("Name cannot be null");
+    if (name.isBlank()) throw new IllegalArgumentException("Name cannot be blank");
+    budgetNames.add(name);
+  }
+
+  @Override
+  public String toString() {
+    StringBuilder namesToString = new StringBuilder();
+    budgetNames.forEach(name -> namesToString.append(name).append("\n"));
+    return namesToString.toString();
+  }
+}
diff --git a/src/main/java/no/ntnu/idatt1002/demo/data/Budget/FileHandlingBudgetArchive.java b/src/main/java/no/ntnu/idatt1002/demo/data/Budget/FileHandlingBudgetArchive.java
new file mode 100644
index 00000000..75e0102f
--- /dev/null
+++ b/src/main/java/no/ntnu/idatt1002/demo/data/Budget/FileHandlingBudgetArchive.java
@@ -0,0 +1,2 @@
+package no.ntnu.idatt1002.demo.data.Budget;public class FileHandlingBudgetArchive {
+}
-- 
GitLab