fixed UI breaking on device rotation (#330)

Co-authored-by: Shresth Pratap Singh <[email protected]>
amahi · Jul 29, 2020 · a24fb30 · a24fb30
1 parent c0ce1ed
commit a24fb30
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/AmahiAnywhere/AmahiAnywhere/Presentation/Walkthrough/WalkthroughViewController.swift b/AmahiAnywhere/AmahiAnywhere/Presentation/Walkthrough/WalkthroughViewController.swift
@@ -31,6 +31,12 @@ class WalkthroughViewController: BaseUIViewController {
         arrowButton.imageView?.contentMode = .scaleAspectFit
     }
 
+    override func viewWillTransition(to size: CGSize, with coordinator: UIViewControllerTransitionCoordinator) {
+        collectionView.frame.size = size
+        collectionView.collectionViewLayout.invalidateLayout()
+        collectionView.scrollToItem(at: IndexPath(row: pageControl.currentPage, section: 0), at: .centeredHorizontally, animated: false)
+    }
+
     @IBAction func skipTapped(_ sender: Any) {
         showLogin()
     }
@@ -87,13 +93,5 @@ extension WalkthroughViewController: UICollectionViewDelegate, UICollectionViewD
             collectionView.backgroundColor = .clear
         }
     }
-
-    override func willTransition(to newCollection: UITraitCollection, with coordinator: UIViewControllerTransitionCoordinator) {
-        collectionView.collectionViewLayout.invalidateLayout()
-
-        DispatchQueue.main.async {
-            self.collectionView.scrollToItem(at: IndexPath(item: self.pageControl.currentPage, section: 0), at: .centeredHorizontally, animated: true)
-        }
-    }
 }