Scroll to the current session instead of just scrolling to the top

tryswift · Mar 6, 2018 · 4863c93 · 4863c93
2 parents 89f026b + a66a66e
commit 4863c93
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/trySwift/RootTabBar/RootTabBarController.swift b/trySwift/RootTabBar/RootTabBarController.swift
@@ -33,7 +33,13 @@ class RootTabBarController: UITabBarController, UITabBarControllerDelegate {
           scrollableToTop.scrollAfterTabTap()
         // or we find the topmost scroll view and scroll it to the top
         } else {
-          firstController.view.findScrollSubview()?.setContentOffset(.zero, animated: true)
+          if let firstController = navigationController.viewControllers.first {
+            if let scrollableToTop = firstController as? ScrollableToTop {
+              scrollableToTop.scrollAfterTabTap()
+            } else {
+              firstController.view.findScrollSubview()?.setContentOffset(.zero, animated: true)
+            }
+          }
         }
       }
     }