Added admin edit check

CS2103-AY1819S1-F10-1 · Nov 1, 2018 · ef951b6 · ef951b6
1 parent 050885d
commit ef951b6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/seedu/address/logic/commands/SelfEditCommand.java b/src/main/java/seedu/address/logic/commands/SelfEditCommand.java
@@ -35,6 +35,8 @@ public class SelfEditCommand extends Command {
 
     public static final String MESSAGE_EDIT_PERSON_SUCCESS = "Edited Person: %1$s";
     public static final String MESSAGE_NOT_EDITED = "At least one field to edit must be provided.";
+    public static final String ADMIN_EDIT_ERROR = "You're currently logged in as admin, not as an employee!\nAdmin"
+        + " cannot be edit. Perhaps you meant edit instead?";
 
     private EditCommand.EditPersonDescriptor editPersonDescriptor;
 
@@ -52,6 +54,10 @@ public SelfEditCommand(EditCommand.EditPersonDescriptor editPersonDescriptor) {
     public CommandResult runBody(Model model, CommandHistory history) throws CommandException {
         requireNonNull(model);
 
+        if (model.getLoggedInUser().isAdminUser()) {
+            throw new CommandException(ADMIN_EDIT_ERROR);
+        }
+
         Person personToEdit = model.getLoggedInUser().getPerson();
         Person editedPerson = EditCommand.createEditedPerson(personToEdit, editPersonDescriptor);