@@ -39,13 +39,18 @@ public boolean isStorageAvailable(){
39
39
*/
40
40
private Event loadEvent (String [] tokens ){
41
41
42
- assert tokens .length == 3 : "Token length should be 3" ;
43
-
42
+ assert tokens .length == 3 || tokens .length == 4 : "Token length should be 3 or 4" ;
43
+ if (tokens .length == 3 ) {
44
+ String name = tokens [0 ].trim ();
45
+ LocalDateTime from = LocalDateTime .parse (tokens [1 ].trim ());
46
+ LocalDateTime to = LocalDateTime .parse (tokens [2 ].trim ());
47
+ return new Event (name , from , to );
48
+ }
44
49
String name = tokens [0 ].trim ();
45
- LocalDateTime from = LocalDateTime .parse (tokens [1 ].trim ());
46
- LocalDateTime to = LocalDateTime . parse (tokens [2 ].trim ());
47
-
48
- return new Event (name , from , to );
50
+ LocalDateTime by = LocalDateTime .parse (tokens [1 ].trim ());
51
+ int goal = Integer . parseInt (tokens [2 ].trim ());
52
+ int completed = Integer . parseInt ( tokens [ 3 ]. trim ());
53
+ return new Goal (name , by , goal , completed );
49
54
}
50
55
51
56
/**
@@ -83,8 +88,13 @@ public void saveEvents(ArrayList<Event> eventList) {
83
88
FileWriter fw = new FileWriter (path );
84
89
85
90
for (Event event : eventList ) {
86
- fw .write (String .format ("%s | %s | %s \r \n " ,
87
- event .getName (), event .getFrom (), event .getTo ()));
91
+ if (event .getClass () == Goal .class ) {
92
+ fw .write (String .format ("%s | %s | %s | %s \r \n " ,
93
+ event .getName (), event .getTo (), ((Goal ) event ).getGoal (), ((Goal ) event ).getCompleted ()));
94
+ } else {
95
+ fw .write (String .format ("%s | %s | %s \r \n " ,
96
+ event .getName (), event .getFrom (), event .getTo ()));
97
+ }
88
98
}
89
99
fw .close ();
90
100
} catch (IOException e ){
0 commit comments