Add an updateDate(to:_) method

Fixes #2

Author: calda

README.md

@@ -61,7 +61,7 @@ class InterfaceController: WKInterfaceController {
             // ...
         }
 
-        timePickerDataSource.setup(withInitiallySelectedDate: Date())
+        timePickerDataSource.updateDate(to: Date())
     }
 
     @IBAction func hourPickerDidUpdate(_ index: Int) {

Sample App/WatchKit Sample App Extension/InterfaceController.swift

@@ -33,7 +33,7 @@ class InterfaceController: WKInterfaceController {
             self?.selectedTimeLabel.setText(timeFormatter.string(from: selectedTime))
         }
 
-        timePickerDataSource.setup(withInitiallySelectedDate: Date())
+        timePickerDataSource.updateDate(to: Date())
     }
 
     @IBAction func hourPickerDidUpdate(_ index: Int) {

WatchKitTimePicker/TimePickerDataSource.swift

@@ -43,6 +43,8 @@ public class TimePickerDataSource {
         self.hoursPicker = hoursPicker
         self.minutesPicker = minutesPicker
         self.amPmPicker = amPmPicker
+
+        setup()
     }
 
 
@@ -81,8 +83,8 @@ public class TimePickerDataSource {
                 Locale.current.calendar.pmSymbol]
         }
     }()
-    
-    public func setup(withInitiallySelectedDate initiallySelectedDate: Date?) {
+
+    private func setup() {
         hoursPicker?.setItems(hourPickerOptions.map { hourValue in
             let pickerItem = WKPickerItem()
             pickerItem.title = "\(hourValue)"
@@ -116,31 +118,30 @@ public class TimePickerDataSource {
             hoursPicker?.setRelativeWidth(0.5, withAdjustment: 0)
             minutesPicker?.setRelativeWidth(0.5, withAdjustment: 0)
         }
-        
-        // set the initial values of the pickers
-        if let initiallySelectedDate = initiallySelectedDate {
-            let dateComponents = Calendar.current.dateComponents(Set(arrayLiteral: .hour, .minute), from: initiallySelectedDate)
-            let hours = dateComponents.hour!
-            let minutes = dateComponents.minute!
-            
-            hoursPicker?.setSelectedItemIndex(hours)
-            selectedHour = hourPickerOptions[hours]
-            
-            let displayedMinutesPerHour = minutePickerOptions.count / 24
-            let corresponsingMinuteIndex = (minutePickerOptions.firstIndex(where: { $0 >= minutes }) ?? 0) + (displayedMinutesPerHour * hours)
-            minutesPicker?.setSelectedItemIndex(corresponsingMinuteIndex)
-            selectedMinute = minutePickerOptions[corresponsingMinuteIndex]
-            
-            if userHas24HourTimeEnabled {
-                amPm = nil
+    }
+
+    public func updateDate(to date: Date) {
+        let dateComponents = Calendar.current.dateComponents(Set(arrayLiteral: .hour, .minute), from: date)
+        let hours = dateComponents.hour!
+        let minutes = dateComponents.minute!
+
+        hoursPicker?.setSelectedItemIndex(hours)
+        selectedHour = hourPickerOptions[hours]
+
+        let displayedMinutesPerHour = minutePickerOptions.count / 24
+        let corresponsingMinuteIndex = (minutePickerOptions.firstIndex(where: { $0 >= minutes }) ?? 0) + (displayedMinutesPerHour * hours)
+        minutesPicker?.setSelectedItemIndex(corresponsingMinuteIndex)
+        selectedMinute = minutePickerOptions[corresponsingMinuteIndex]
+
+        if userHas24HourTimeEnabled {
+            amPm = nil
+        } else {
+            if hours < 12 {
+                amPmPicker?.setSelectedItemIndex(0)
+                amPm = .am
             } else {
-                if hours < 12 {
-                    amPmPicker?.setSelectedItemIndex(0)
-                    amPm = .am
-                } else {
-                    amPmPicker?.setSelectedItemIndex(1)
-                    amPm = .pm
-                }
+                amPmPicker?.setSelectedItemIndex(1)
+                amPm = .pm
             }
         }
     }