Add setBudget Command

Author: yuanxing-y

src/main/java/seedu/address/logic/commands/SetBudgetCommand.java

@@ -0,0 +1,26 @@
+package seedu.address.logic.commands;
+
+import seedu.address.logic.commands.exceptions.CommandException;
+import seedu.address.model.Model;
+
+import static seedu.address.commons.util.CollectionUtil.requireAllNonNull;
+
+public class SetBudgetCommand extends Command {
+    public static final String COMMAND_WORD = "setBudget";
+    public static final String MESSAGE_SET_BUDGET_SUCCESS = "Successful! Te budget is now S$ %.2f.\n";
+    public static final String MESSAGE_SET_BUDGET_FAIL = "Set Budget failed, please enter a valid budget.\n";
+
+    private final double budget;
+
+    public SetBudgetCommand(double budget) {
+        requireAllNonNull(budget);
+        this.budget = budget;
+    }
+
+    @Override
+    public CommandResult execute(Model model) throws CommandException {
+        model.setExpenseBookBudget(budget);
+        String msg = String.format(MESSAGE_SET_BUDGET_SUCCESS, budget);
+        return new CommandResult(msg);
+    }
+}

src/main/java/seedu/address/logic/parser/ExpenseBookParser.java

@@ -6,10 +6,7 @@
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
-import seedu.address.logic.commands.Command;
-import seedu.address.logic.commands.DescriptionCommand;
-import seedu.address.logic.commands.HelpCommand;
-import seedu.address.logic.commands.ShowBudgetCommand;
+import seedu.address.logic.commands.*;
 import seedu.address.logic.parser.exceptions.ParseException;
 
 /**
@@ -49,7 +46,9 @@ public Command parseCommand(String userInput) throws ParseException {
                     .COMMAND_WORD:
                 return new ShowBudgetCommandParser().parse(arguments);
 
-
+            case SetBudgetCommand
+                    .COMMAND_WORD:
+                return new SetBudgetCommandParser().parse(arguments);
 
             default:
                 throw new ParseException(MESSAGE_UNKNOWN_COMMAND);

src/main/java/seedu/address/logic/parser/SetBudgetCommandParser.java

@@ -0,0 +1,22 @@
+package seedu.address.logic.parser;
+
+import seedu.address.commons.core.index.Index;
+import seedu.address.logic.commands.DeleteCommand;
+import seedu.address.logic.commands.SetBudgetCommand;
+import seedu.address.logic.parser.exceptions.ParseException;
+
+import static seedu.address.commons.core.Messages.MESSAGE_INVALID_COMMAND_FORMAT;
+
+public class SetBudgetCommandParser implements Parser<SetBudgetCommand> {
+    @Override
+    public SetBudgetCommand parse(String userInput) throws ParseException {
+        try {
+            //needs oop later
+            double budget = Double.parseDouble(userInput.split(" ")[1]);
+            return new SetBudgetCommand(budget);
+        } catch (Exception pe) {
+            throw new ParseException(
+                    String.format(MESSAGE_INVALID_COMMAND_FORMAT, SetBudgetCommand.MESSAGE_SET_BUDGET_FAIL), pe);
+        }
+    }
+}

src/main/java/seedu/address/model/ExpenseModelManager.java

@@ -174,4 +174,10 @@ public double getExpenseBookRemaining() {
         return expenseBook.getRemainingBudget();
     }
 
+    @Override
+    public void setExpenseBookBudget(double budget) {
+        expenseBook.setBudget(budget);
+    }
+
+
 }

src/main/java/seedu/address/model/Model.java

@@ -100,4 +100,6 @@ public interface Model {
     double getExpenseBookBudget();
 
     double getExpenseBookRemaining();
+
+    void setExpenseBookBudget(double budget);
 }

src/main/java/seedu/address/model/ModelManager.java

@@ -186,6 +186,12 @@ public double getExpenseBookRemaining() {
         return 0;
     }
 
+    //dummy
+    @Override
+    public void setExpenseBookBudget(double budget) {
+
+    }
+
     @Override
     public boolean equals(Object obj) {
         // short circuit if same object