prepare / use appropriate previews for the home screen / favorites

duckduckgo · Feb 1, 2024 · 229de25 · 229de25
1 parent 103dcd3
commit 229de25
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 5 deletions.
diff --git a/Core/UIViewExtension.swift b/Core/UIViewExtension.swift
@@ -73,4 +73,17 @@ extension UIView {
             view.removeFromSuperview()
         }
     }
+
+    @MainActor
+    public func createImageSnapshot(inBounds bounds: CGRect? = nil) -> UIImage? {
+        let bounds = bounds ?? self.frame
+        let size = bounds.size
+        UIGraphicsBeginImageContextWithOptions(size, false, UIScreen.main.scale)
+        UIGraphicsGetCurrentContext()?.translateBy(x: -bounds.origin.x, y: -bounds.origin.y)
+        drawHierarchy(in: frame, afterScreenUpdates: true)
+        let image = UIGraphicsGetImageFromCurrentImageContext()
+        UIGraphicsEndImageContext()
+        return image
+    }
+
 }
diff --git a/DuckDuckGo/MainViewController.swift b/DuckDuckGo/MainViewController.swift
@@ -326,16 +326,34 @@ class MainViewController: UIViewController {
             self?.newTab()
         } onSwipeStarted: { [weak self] in
             self?.hideKeyboard()
+            self?.updatePreviewForCurrentTab()
+        }
+    }
+
+    func updatePreviewForCurrentTab() {
+        assert(Thread.isMainThread)
+
+        if !viewCoordinator.logoContainer.isHidden,
+           self.tabManager.current()?.link == nil,
+           let tab = self.tabManager.model.currentTab {
 
-            guard let self,
-                    let currentTab = self.tabManager.current(),
-                    currentTab.link != nil else { return }
-
+            // Home screen with logo
+            if let image = viewCoordinator.logoContainer.createImageSnapshot(inBounds: viewCoordinator.contentContainer.frame) {
+                previewsSource.update(preview: image, forTab: tab)
+            }
+
+        } else if let currentTab = self.tabManager.current(), currentTab.link != nil {
+            // Web view
             currentTab.preparePreview(completion: { image in
                 guard let image else { return }
                 self.previewsSource.update(preview: image,
                                            forTab: currentTab.tabModel)
             })
+        } else if let tab = self.tabManager.model.currentTab {
+            // Favorites, etc
+            if let image = viewCoordinator.contentContainer.createImageSnapshot() {
+                previewsSource.update(preview: image, forTab: tab)
+            }
         }
     }
 

diff --git a/DuckDuckGo/SwipeTabsCoordinator.swift b/DuckDuckGo/SwipeTabsCoordinator.swift
@@ -163,7 +163,7 @@ extension SwipeTabsCoordinator: UICollectionViewDelegate {
 
     private func prepareCurrentView() {
 
-        if tabsModel.currentTab?.link == nil {
+        if !coordinator.logoContainer.isHidden {
             currentView = coordinator.logoContainer
         } else {
             currentView = coordinator.contentContainer.subviews.last