From c39dd6b5923a7d9f6e7e2ccbc652c894977f2ce9 Mon Sep 17 00:00:00 2001
From: Tanveer Singh <tanveersengh@gmail.com>
Date: Sun, 12 Nov 2023 10:21:14 +0800
Subject: [PATCH] Prevent editing EC for doctors

---
 .../address/logic/commands/EditCommand.java   | 41 +++++++++++--------
 1 file changed, 25 insertions(+), 16 deletions(-)

diff --git a/src/main/java/seedu/address/logic/commands/EditCommand.java b/src/main/java/seedu/address/logic/commands/EditCommand.java
index e686c196983..3cbff2caa37 100644
--- a/src/main/java/seedu/address/logic/commands/EditCommand.java
+++ b/src/main/java/seedu/address/logic/commands/EditCommand.java
@@ -125,24 +125,14 @@ private Person getPersonToEdit(Model model) throws CommandException {
     private Person getEditedPerson(Model model, Person personToEdit) throws CommandException {
         Person editedPerson;
         if (personToEdit instanceof Patient) {
-            if (editPersonDescriptor.getTags().isPresent()
-                    && !editPersonDescriptor.isValidPatientTagList(editPersonDescriptor.getTags().get())) {
-                logger.warning("Invalid tag for patient");
-                throw new CommandException("Please enter a valid patient tag.");
-            }
-            editedPerson = createEditedPatient((Patient) personToEdit, editPersonDescriptor);
+            Patient patientToEdit = (Patient) personToEdit;
+            validatePatient();
+            editedPerson = createEditedPatient(patientToEdit, editPersonDescriptor);
         } else {
             assert personToEdit.isDoctor();
-            if (editPersonDescriptor.getCondition().isPresent() || editPersonDescriptor.getBloodType().isPresent()) {
-                logger.warning("Error thrown - tried to edit condition / bloodtype of doctor");
-                throw new CommandException("Doctors cannot have Condition or BloodType fields.");
-            }
-            if (editPersonDescriptor.getTags().isPresent()
-                    && !editPersonDescriptor.isValidDoctorTagList(editPersonDescriptor.getTags().get())) {
-                logger.warning(editPersonDescriptor.getTags().toString());
-                throw new CommandException("Please enter valid Doctor tags.");
-            }
-            editedPerson = createEditedDoctor((Doctor) personToEdit, editPersonDescriptor);
+            Doctor doctorToEdit = (Doctor) personToEdit;
+            validateDoctor();
+            editedPerson = createEditedDoctor(doctorToEdit, editPersonDescriptor);
         }
         if (!personToEdit.isSamePerson(editedPerson) && model.hasPerson(editedPerson)) {
             logger.warning("Edited Person and orignal person are the same");
@@ -150,6 +140,25 @@ private Person getEditedPerson(Model model, Person personToEdit) throws CommandE
         }
         return editedPerson;
     }
+    private void validatePatient() throws CommandException {
+        if (editPersonDescriptor.getTags().isPresent()
+                && !editPersonDescriptor.isValidPatientTagList(editPersonDescriptor.getTags().get())) {
+            logger.warning("Invalid tag for patient");
+            throw new CommandException("Please enter a valid patient tag.");
+        }
+    }
+    private void validateDoctor() throws CommandException {
+        if (editPersonDescriptor.getCondition().isPresent() || editPersonDescriptor.getBloodType().isPresent()
+                || editPersonDescriptor.getEmergencyContact().isPresent()) {
+            logger.warning("Error thrown - tried to edit condition/blood type/emergency contact of doctor");
+            throw new CommandException("Doctors cannot have Condition, BloodType or Emergency Contact fields.");
+        }
+        if (editPersonDescriptor.getTags().isPresent()
+                && !editPersonDescriptor.isValidDoctorTagList(editPersonDescriptor.getTags().get())) {
+            logger.warning(editPersonDescriptor.getTags().toString());
+            throw new CommandException("Please enter valid Doctor tags.");
+        }
+    }
 
     /**
      * Creates and returns a {@code Person} with the details of {@code personToEdit}