Dismiss gallery when currently displayed entry becomes deleted

submodules/GalleryUI/Sources/GalleryController.swift

@@ -463,7 +463,6 @@ public class GalleryController: ViewController, StandalonePresentableController
                     }
             }
         }
-        |> take(1)
         
         let semaphore: DispatchSemaphore?
         if synchronousLoad {
@@ -487,6 +486,23 @@ public class GalleryController: ViewController, StandalonePresentableController
                         strongSelf.configuration = configuration
                         
                         let entries = view.entries
+                        
+                        if let centralEntryStableId = strongSelf.centralEntryStableId {
+                            var found = false
+                            for i in 0 ..< entries.count {
+                                let message = entries[i].message
+                                if message.stableId == centralEntryStableId {
+                                    found = true
+                                    break
+                                }
+                            }
+                            
+                            if !found {
+                                strongSelf.dismiss(forceAway: true)
+                                return
+                            }
+                        }
+                        
                         var centralEntryStableId: UInt32?
                         loop: for i in 0 ..< entries.count {
                             let message = entries[i].message
@@ -515,15 +531,17 @@ public class GalleryController: ViewController, StandalonePresentableController
                             strongSelf.entries = entries.reversed()
                             strongSelf.hasLeftEntries = view.hasLater
                             strongSelf.hasRightEntries = view.hasEarlier
-                            if let centralEntryStableId = centralEntryStableId {
+                            if strongSelf.centralEntryStableId == nil, let centralEntryStableId = centralEntryStableId {
                                 strongSelf.centralEntryStableId = centralEntryStableId
                             }
                         } else {
                             strongSelf.entries = entries
                             strongSelf.hasLeftEntries = view.hasEarlier
                             strongSelf.hasRightEntries = view.hasLater
+                            if strongSelf.centralEntryStableId == nil {
                                 strongSelf.centralEntryStableId = centralEntryStableId
                             }
+                        }
                         if strongSelf.isViewLoaded {
                             var items: [GalleryItem] = []
                             var centralItemIndex: Int?
@@ -547,9 +565,12 @@ public class GalleryController: ViewController, StandalonePresentableController
                             strongSelf.galleryNode.pager.replaceItems(items, centralItemIndex: centralItemIndex)
                             
                             if strongSelf.temporaryDoNotWaitForReady {
+                                if !strongSelf.didSetReady {
                                     strongSelf.didSetReady = true
                                     strongSelf._ready.set(.single(true))
+                                }
                             } else {
+                                if !strongSelf.didSetReady {
                                     let ready = strongSelf.galleryNode.pager.ready() |> timeout(2.0, queue: Queue.mainQueue(), alternate: .single(Void())) |> afterNext { [weak strongSelf] _ in
                                         strongSelf?.didSetReady = true
                                     }
@@ -559,6 +580,7 @@ public class GalleryController: ViewController, StandalonePresentableController
                         }
                     }
                 }
+            }
             var process = false
             let _ = syncResult.modify { processed, _ in
                 if !processed {
@@ -568,7 +590,7 @@ public class GalleryController: ViewController, StandalonePresentableController
                 return (true, nil)
             }
             semaphore?.signal()
-            if process {
+            if process || true {
                 Queue.mainQueue().async {
                     f()
                 }

submodules/GalleryUI/Sources/Items/UniversalVideoGalleryItem.swift

@@ -1482,7 +1482,27 @@ final class UniversalVideoGalleryItemNode: ZoomableContentGalleryItemNode {
             let baseNavigationController = self.baseNavigationController()
             let mediaManager = self.context.sharedContext.mediaManager
             var expandImpl: (() -> Void)?
-            let overlayNode = OverlayUniversalVideoNode(postbox: self.context.account.postbox, audioSession: context.sharedContext.mediaManager.audioSession, manager: context.sharedContext.mediaManager.universalVideoManager, content: item.content, expand: {
+            
+            let shouldBeDismissed: Signal<Bool, NoError>
+            if let contentInfo = item.contentInfo, case let .message(message) = contentInfo {
+                let viewKey = PostboxViewKey.messages(Set([message.id]))
+                shouldBeDismissed = context.account.postbox.combinedView(keys: [viewKey])
+                |> map { views -> Bool in
+                    guard let view = views.views[viewKey] as? MessagesView else {
+                        return false
+                    }
+                    if view.messages.isEmpty {
+                        return true
+                    } else {
+                        return false
+                    }
+                }
+                |> distinctUntilChanged
+            } else {
+                shouldBeDismissed = .single(false)
+            }
+            
+            let overlayNode = OverlayUniversalVideoNode(postbox: self.context.account.postbox, audioSession: context.sharedContext.mediaManager.audioSession, manager: context.sharedContext.mediaManager.universalVideoManager, content: item.content, shouldBeDismissed: shouldBeDismissed, expand: {
                 expandImpl?()
             }, close: { [weak mediaManager] in
                 mediaManager?.setOverlayVideoNode(nil)

submodules/TelegramUniversalVideoContent/Sources/OverlayUniversalVideoNode.swift

@@ -16,6 +16,8 @@ public final class OverlayUniversalVideoNode: OverlayMediaItemNode {
     
     private var validLayoutSize: CGSize?
     
+    private var shouldBeDismissedDisposable: Disposable?
+    
     override public var group: OverlayMediaItemNodeGroup? {
         return OverlayMediaItemNodeGroup(rawValue: 0)
     }
@@ -35,7 +37,7 @@ public final class OverlayUniversalVideoNode: OverlayMediaItemNode {
     public var customClose: (() -> Void)?
     public var controlsAreShowingUpdated: ((Bool) -> Void)?
     
-    public init(postbox: Postbox, audioSession: ManagedAudioSession, manager: UniversalVideoManager, content: UniversalVideoContent, expand: @escaping () -> Void, close: @escaping () -> Void) {
+    public init(postbox: Postbox, audioSession: ManagedAudioSession, manager: UniversalVideoManager, content: UniversalVideoContent, shouldBeDismissed: Signal<Bool, NoError> = .single(false), expand: @escaping () -> Void, close: @escaping () -> Void) {
         self.content = content
         self.defaultExpand = expand
         
@@ -112,6 +114,16 @@ public final class OverlayUniversalVideoNode: OverlayMediaItemNode {
         }
         
         self.videoNode.canAttachContent = true
+        
+        self.shouldBeDismissedDisposable = (shouldBeDismissed
+        |> filter { $0 }
+        |> deliverOnMainQueue).start(next: { [weak self] _ in
+            guard let strongSelf = self else {
+                return
+            }
+            strongSelf.dismiss()
+            closeImpl?()
+        })
     }
     
     override public func didLoad() {