Use slap for setLength method under add command

CS2103AUG2016-W09-C3 · Nov 6, 2016 · ec98333 · ec98333
1 parent d2456b0
commit ec98333
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/main/java/todoit/taskbook/logic/commands/AddCommand.java b/src/main/java/todoit/taskbook/logic/commands/AddCommand.java
@@ -70,11 +70,8 @@ private String setLength(String length, String datetime, String endDatetime) thr
                 LocalDateTime startDate = DateParser.parseDate(datetime);
                 LocalDateTime endDate = DateParser.parseDate(endDatetime);
                 long hourDifference = ChronoUnit.HOURS.between(startDate, endDate);
-                if(hourDifference >= 24){
-                    return Long.toString(TimeUnit.HOURS.toDays(hourDifference)) + "d";
-                } else {
-                    return hourDifference + "h";
-                }
+                String finalTimeDifference = getDifferenceString(hourDifference);
+                return finalTimeDifference;
             }
         } else {
             if(!endDatetime.equals("-1")){
@@ -83,6 +80,14 @@ private String setLength(String length, String datetime, String endDatetime) thr
             return length;
         }
     }
+
+    private String getDifferenceString(long hourDifference){
+        if(hourDifference >= 24){
+            return Long.toString(TimeUnit.HOURS.toDays(hourDifference)) + "d";
+        } else {
+            return hourDifference + "h";
+        }
+    }
 
     public AddCommand(String name, String priority, String information, String doneFlag, Set<String> tagsFromArgs) 
             throws IllegalValueException {