Merge pull request #21 from RIT-SWEN-262/league-model

League model

4 files changed, 84 insertions, 2 deletions

diff --git a/src/main/java/design/controller/userinput/Menu.java b/src/main/java/design/controller/userinput/Menu.java
index ab633e1..01e80a9 100644
--- a/src/main/java/design/controller/userinput/Menu.java
+++ b/src/main/java/design/controller/userinput/Menu.java
@@ -23,7 +23,7 @@ public abstract class Menu {
         try {
             int i = Integer.parseInt(line) - 1;
             menuOptions.get(i).onAction();
-        } catch (IndexOutOfBoundsException ex) {
+        } catch (IndexOutOfBoundsException | NumberFormatException ex) {
             System.err.printf("Invalid option \"%s\"\n", line);
             present();
         }
diff --git a/src/main/java/design/controller/userinput/menus/ImportExportMenu.java b/src/main/java/design/controller/userinput/menus/ImportExportMenu.java
new file mode 100644
index 0000000..21ffa6c
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/ImportExportMenu.java
@@ -0,0 +1,81 @@
+package design.controller.userinput.menus;
+
+import design.controller.userinput.Menu;
+import design.controller.userinput.MenuOption;
+import design.persistence.*;
+import design.persistence.importexport.DataHandler;
+import design.persistence.importexport.DataSource;
+import design.persistence.importexport.JSONHandler;
+import design.persistence.importexport.XMLHandler;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.List;
+import java.util.Scanner;
+
+public class ImportExportMenu extends Menu {
+
+    @Override
+    public String getTitle() {
+        return "import export menu";
+    }
+
+   @Override
+    public List<MenuOption> getMenuOptions() {
+        List<MenuOption> opts = new java.util.ArrayList<>();
+
+
+        opts.add(new MenuOption("return to main menu", () -> new MainMenu().present()));
+
+        opts.add(new MenuOption("import leagues...", () -> promptForPath(true, false)));
+        opts.add(new MenuOption("export leagues...", () -> promptForPath(false, false)));
+        opts.add(new MenuOption("import profiles...", () -> promptForPath(true, true)));
+        opts.add(new MenuOption("export profiles...", () -> promptForPath(false, true)));
+
+        return opts;
+    }
+
+    private void promptForPath(boolean isImporting, boolean isPersonalProfile)
+    {
+        System.out.print("Enter file path: ");
+        Scanner sc = new Scanner(System.in);
+        String filePath = sc.nextLine();
+        File file = new File(filePath);
+
+        DataSource source;
+        if (isPersonalProfile) {
+            source = PersonalDatabase.instance();
+        } else {
+            source = LeagueDatabase.instance();
+        }
+
+        DataHandler handler;
+        String ext = getExtension(filePath);
+        switch (ext) {
+            case "json" -> handler = new JSONHandler(source);
+            case "xml" -> handler = new XMLHandler(source);
+            default -> {
+                System.out.println("Unsupported file type: " + ext);
+                this.present();
+                return;
+            }
+        }
+
+        try {
+            if (isImporting) {
+                handler.importData(file);
+            } else {
+                handler.exportData(file);
+            }
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+
+        this.present();
+    }
+
+    private static String getExtension(String fileName) {
+        int i = fileName.lastIndexOf('.');
+        return (i >= 0) ? fileName.substring(i + 1).toLowerCase() : "unknown";
+    }
+}
diff --git a/src/main/java/design/controller/userinput/menus/MainMenu.java b/src/main/java/design/controller/userinput/menus/MainMenu.java
index 4331e56..3f520e4 100644
--- a/src/main/java/design/controller/userinput/menus/MainMenu.java
+++ b/src/main/java/design/controller/userinput/menus/MainMenu.java
@@ -27,6 +27,7 @@ public class MainMenu extends Menu {
             options.add(new MenuOption("statistics...", () -> new StatisticsMenu().present()));
             options.add(new MenuOption("log round...", () -> new HolePlayMenu().present()));
             options.add(new MenuOption("league play...", () -> new SelectLeague().present()));
+            options.add(new MenuOption("manage data...", () -> new ImportExportMenu().present()));
         }
         return options;
     }
diff --git a/src/main/java/design/controller/userinput/menus/SelectLeague.java b/src/main/java/design/controller/userinput/menus/SelectLeague.java
index 08f6c91..e1675e9 100644
--- a/src/main/java/design/controller/userinput/menus/SelectLeague.java
+++ b/src/main/java/design/controller/userinput/menus/SelectLeague.java
@@ -26,7 +26,7 @@ public class SelectLeague extends Menu {
     public List<MenuOption> getMenuOptions() {
         List<MenuOption> options = new ArrayList<>();
         for (League l : LeagueDatabase.instance().getLeagues()) {
-            options.add(new MenuOption(l.getName(), () -> {
+            options.add(new MenuOption(l.toString(), () -> {
                 if (l instanceof ScrambleLeague sl && sl.locateTeam(golfer) == null) {
                     Scanner sc = new Scanner(System.in);
                     System.out.print("You are not a member of a team for this league. Would you like to create one (Y/n): ");