Merge pull request #125 from Vavinan/97-close-insights-window-before-…

…exit

Resolve: Close insights window before exit from the program

src/main/java/budgetbuddy/BudgetBuddy.java

@@ -8,6 +8,7 @@
 import budgetbuddy.exceptions.InvalidEditTransactionData;
 import budgetbuddy.exceptions.InvalidIndexException;
 import budgetbuddy.exceptions.InvalidTransactionTypeException;
+import budgetbuddy.insights.Insight;
 import budgetbuddy.parser.Parser;
 import budgetbuddy.storage.DataStorage;
 import budgetbuddy.transaction.TransactionList;
@@ -42,12 +43,13 @@ public void run() {
 
 
         boolean isRunning = true;
-
+        
         while (isRunning) {
             String input = in.nextLine();
             try {
                 switch (input.split(" ")[0]) {
                 case "bye":
+                    Insight.closeInsightFrames();
                     UserInterface.printGoodBye();
                     isRunning = false;
                     break;

src/main/java/budgetbuddy/insights/Insight.java

@@ -14,6 +14,7 @@
 import javax.swing.JPanel;
 import java.awt.BorderLayout;
 import java.util.ArrayList;
+import java.awt.Window;
 
 import static java.lang.Math.abs;
 
@@ -102,4 +103,18 @@ private static int indexOf(Category[] array, Category target) {
         return -1;
     }
 
+    //@@author Vavinan
+    public static void closeInsightFrames() {
+        // Close any open insight frames here
+        for (Window window : Window.getWindows()) {
+            if (window instanceof JFrame) {
+                String title = ((JFrame) window).getTitle();
+                if (title != null && (title.equals("Income Insights") || title.equals("Expense Insights"))) {
+                    window.dispose();
+                }
+            }
+        }
+    }
+    //@@author 
+
 }