From cd01fdd7d16db518d7eb1af23ea2be8acfc4554d Mon Sep 17 00:00:00 2001
From: ChenKangg <cckcyc@gmail.com>
Date: Fri, 5 Apr 2024 19:06:30 +0800
Subject: [PATCH] update filesave with location and entertainment

---
 .../seedu/omnitravel/storage/FileSave.java    | 22 +++++++++----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/src/main/java/seedu/omnitravel/storage/FileSave.java b/src/main/java/seedu/omnitravel/storage/FileSave.java
index bf19e16995..cfb2e675aa 100644
--- a/src/main/java/seedu/omnitravel/storage/FileSave.java
+++ b/src/main/java/seedu/omnitravel/storage/FileSave.java
@@ -1,9 +1,5 @@
 package seedu.omnitravel.storage;
-import seedu.omnitravel.travelactivitytypes.TravelActivityList;
-import seedu.omnitravel.travelactivitytypes.Accommodation;
-import seedu.omnitravel.travelactivitytypes.Food;
-import seedu.omnitravel.travelactivitytypes.Landmark;
-import seedu.omnitravel.travelactivitytypes.TravelActivity;
+import seedu.omnitravel.travelactivitytypes.*;
 
 import java.io.FileWriter;
 import java.io.FileNotFoundException;
@@ -32,21 +28,25 @@ public void loadFileContents(TravelActivityList list) throws FileNotFoundExcepti
             LocalDate date = LocalDate.parse(line[3]);
             String duration = line[4];
             String tag = line.length >= 6 ? line[5].trim() : "";
-            String expense = line.length >= 7 ? line[6].trim() : "";
+            String location = line.length >= 7 ? line[6].trim() : "";
+            String expense = line.length >= 8 ? line[7].trim() : "";
             TravelActivity activity;
             switch (type) {
             case "accommodation":
-                activity = new Accommodation(description, date, duration, tag, expense);
+                activity = new Accommodation(description, date, duration, tag, location, expense);
                 break;
             case "food":
-                activity = new Food(description, date, duration, tag, expense);
+                activity = new Food(description, date, duration, tag, location, expense);
                 break;
             case "landmark":
-                activity = new Landmark(description, date, duration, tag, expense);
+                activity = new Landmark(description, date, duration, tag, location, expense);
                 break;
-            case "general":
-                activity = new TravelActivity(description, date, duration, tag, expense);
+            case "entertainment":
+                activity = new Entertainment(description, date, duration, tag, location, expense);
                 break;
+            case "general":
+            activity = new TravelActivity(description, date, duration, tag, location, expense);
+            break;
             default:
                 throw new FileNotFoundException("File is corrupted or has invalid format");
             }