update xml adapted task to store time

CS2103AUG2016-W11-C2 · Oct 15, 2016 · 1781e7b · 1781e7b
1 parent 6c430a4
commit 1781e7b
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/main/java/seedu/agendum/storage/XmlAdaptedTask.java b/src/main/java/seedu/agendum/storage/XmlAdaptedTask.java
@@ -6,18 +6,27 @@
 import seedu.agendum.model.task.*;
 
 import javax.xml.bind.annotation.XmlElement;
+
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Optional;
 
 /**
  * JAXB-friendly version of the Task.
  */
 public class XmlAdaptedTask {
 
+    private DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm");
     @XmlElement(required = true)
     private String name;
     @XmlElement(required = true)
     private String isCompleted;
+    @XmlElement(required = false)
+    private String startDateTime;
+    @XmlElement(required = false)
+    private String endDateTime;
 
     @XmlElement
     private List<XmlAdaptedTag> tagged = new ArrayList<>();
@@ -37,6 +46,12 @@ public XmlAdaptedTask(ReadOnlyTask source) {
         name = source.getName().fullName;
         isCompleted = Boolean.toString(source.isCompleted());
         tagged = new ArrayList<>();
+        if (source.getStartDateTime().isPresent()) {
+            startDateTime = source.getStartDateTime().get().format(formatter);
+        }
+        if (source.getEndDateTime().isPresent()) {
+            endDateTime = source.getEndDateTime().get().format(formatter);
+        }
         for (Tag tag : source.getTags()) {
             tagged.add(new XmlAdaptedTag(tag));
         }
@@ -60,6 +75,12 @@ public Task toModelType() throws IllegalValueException {
         if (markedAsCompleted) {
             newTask.markAsCompleted();
         }
+        if (startDateTime != null) {
+            newTask.setStartDateTime(Optional.ofNullable(LocalDateTime.parse(this.startDateTime, formatter)));
+        }
+        if (endDateTime != null) {
+            newTask.setEndDateTime(Optional.ofNullable(LocalDateTime.parse(this.endDateTime, formatter)));
+        }
         return newTask;
     }
 }