Refactored some parts in AppScreen

Author: amyyong11

src/main/java/screens/app_screen/AppScreen.java

@@ -51,7 +51,7 @@ public AppScreen(String currentUsername, ArrayList<Chat> chats) {
 
         // adding the action listeners for the +private-chat and +group-chat buttons
         addPrivateChat.addActionListener(e -> {
-            PrivateChatfactory privateChatfactory = (PrivateChatfactory) new CommonPrivatechat();
+            PrivateChatfactory privateChatfactory = new CommonPrivatechat();
             ChatInputBoundry inputBoundry = new ChatInteractor(privateChatfactory);
             ChatController controller = new ChatController(inputBoundry);
             new ChatView(controller, true);
@@ -165,32 +165,14 @@ public void updateChatOrder(Chat chat){
 
     }
 
-    /**
-     * Add a new chat to the screen, if the chat already exists (i.e. there exists a chat with the
-     * same ID, do nothing)
-     * @param chat The new chat to be added
-     */
-    public void addNewChat(Chat chat){
-
-        if (!(this.chats.contains(chat))){
-            updateChatOrder(chat);
-            jFrame.remove(this.jScrollPane);
-            createGateway();
-
-            // refresh the screen
-            displayAppScreen();
-
-        }
-    }
-
     /**
      * Update the order of chats that appear on screen if there was a change to conversation history
-     * This should not be called if chatID is not an ID of an existing chat that the current user has
-     * @param chatID The ID of the given chat
+     * This should be called if a new chat was added or if an existing chat has a new message
+     * @param chat The chat with an update
      */
     @Override
-    public void updateScreen(String chatID) {
-        updateChatOrder(getChat(chatID));
+    public void updateScreen(Chat chat) {
+        updateChatOrder(chat);
         jFrame.remove(this.jScrollPane);
         createGateway();
 

src/main/java/use_cases/app_screen_use_case/Refresh.java

@@ -1,5 +1,7 @@
 package use_cases.app_screen_use_case;
 
+import entities.chat.Chat;
+
 public interface Refresh {
-    void updateScreen(String chatID);
+    void updateScreen(Chat chat);
 }