Fix - In calendar is not yet loaded but user calls reload data, save …

…the closure to delayed
patchthecode · Nov 2, 2016 · da0a0ad · da0a0ad
1 parent 11255d3
commit da0a0ad
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/Sources/UserInteractionFunctions.swift b/Sources/UserInteractionFunctions.swift
@@ -222,6 +222,9 @@ extension JTAppleCalendarView {
                            animation: Bool = false,
                            completionHandler: (() -> Void)? = nil) {
         if !calendarIsAlreadyLoaded {
+            if let validCompletionHandler = completionHandler {
+                delayedExecutionClosure.append(validCompletionHandler)
+            }
             return
         }
         reloadData(checkDelegateDataSource: true,
@@ -502,12 +505,11 @@ extension JTAppleCalendarView {
     /// number for a vertical scrolling calendar
     /// - Parameter completionHandler: A completion handler that will be
     ///   executed at the end of the scroll animation
-    public func scrollToDate(
-        _ date: Date,
-        triggerScrollToDateDelegate: Bool = true,
-        animateScroll: Bool = true,
-        preferredScrollPosition: UICollectionViewScrollPosition? = nil,
-        completionHandler: (() -> Void)? = nil) {
+    public func scrollToDate(_ date: Date,
+                             triggerScrollToDateDelegate: Bool = true,
+                             animateScroll: Bool = true,
+                             preferredScrollPosition: UICollectionViewScrollPosition? = nil,
+                             completionHandler: (() -> Void)? = nil) {
             if !calendarIsAlreadyLoaded {
                 delayedExecutionClosure.append {
                     self.scrollToDate(