diff --git a/src/main/java/org/codedifferently/cbtyson/menus/MainMenu.java b/src/main/java/org/codedifferently/cbtyson/menus/MainMenu.java index 31567a1..f372ade 100644 --- a/src/main/java/org/codedifferently/cbtyson/menus/MainMenu.java +++ b/src/main/java/org/codedifferently/cbtyson/menus/MainMenu.java @@ -40,8 +40,12 @@ public void promptMainMenu() { case 2: break; case 3: + ViewMenu viewMenu = new ViewMenu(); + viewMenu.viewGroups(); break; case 4: + RemoveMenu removeMenu = new RemoveMenu(); + removeMenu.removeStudentFromGroup(); break; case 5: System.out.println("Have a Nice Day!"); diff --git a/src/main/java/org/codedifferently/cbtyson/menus/RemoveMenu.java b/src/main/java/org/codedifferently/cbtyson/menus/RemoveMenu.java index 63cbfb8..24d91a5 100644 --- a/src/main/java/org/codedifferently/cbtyson/menus/RemoveMenu.java +++ b/src/main/java/org/codedifferently/cbtyson/menus/RemoveMenu.java @@ -1,4 +1,44 @@ package org.codedifferently.cbtyson.menus; +import org.codedifferently.cbtyson.data.Group; +import org.codedifferently.cbtyson.data.GroupList; +import org.codedifferently.cbtyson.data.Student; +import org.codedifferently.cbtyson.helpers.InputHandler; + public class RemoveMenu { -} + + public void removeStudentFromGroup() { + + System.out.println("Enter Group ID:"); + String groupID = InputHandler.handleStringInput(); + + System.out.println("Enter Student ID:"); + String studentID = InputHandler.handleStringInput(); + + for (Group group : GroupList.GetGroups()) { + + if (group.getGroupID().equals(groupID)) { + + Student studentToRemove = null; + + for (Student student : group.getStudentList()) { + if (student.getStudentID().equals(studentID)) { + studentToRemove = student; + break; + } + } + + if (studentToRemove != null) { + group.getStudentList().remove(studentToRemove); + System.out.println("Student removed successfully!"); + } else { + System.out.println("Student not found in this group."); + } + + return; + } + } + + System.out.println("Group not found."); + } +} \ No newline at end of file diff --git a/src/main/java/org/codedifferently/cbtyson/menus/ViewMenu.java b/src/main/java/org/codedifferently/cbtyson/menus/ViewMenu.java index 4db08ae..f5e198a 100644 --- a/src/main/java/org/codedifferently/cbtyson/menus/ViewMenu.java +++ b/src/main/java/org/codedifferently/cbtyson/menus/ViewMenu.java @@ -1,4 +1,31 @@ package org.codedifferently.cbtyson.menus; +import org.codedifferently.cbtyson.data.Group; +import org.codedifferently.cbtyson.data.GroupList; +import org.codedifferently.cbtyson.data.Student; + public class ViewMenu { -} + + public void viewGroups() { + + if (GroupList.GetGroups().isEmpty()) { + System.out.println("No groups available."); + return; + } + + for (Group group : GroupList.GetGroups()) { + + System.out.println("=================================="); + System.out.println("Group Name: " + group.getName()); + System.out.println("Group ID: " + group.getGroupID()); + System.out.println("Students:"); + + for (Student student : group.getStudentList()) { + System.out.println(student.getFirstName() + " " + student.getLastName() + " | ID: " + student.getStudentID() + " | GPA: " + student.getGpa()); + } + + System.out.println("=================================="); + System.out.println(); + } + } +} \ No newline at end of file