From f064692e5765007f3110a19a4ecf6fd24193167f Mon Sep 17 00:00:00 2001
From: John Preston <johnprestonmail@gmail.com>
Date: Fri, 23 Oct 2020 15:28:20 +0300
Subject: [PATCH] Close media viewer when photo message is deleted.

---
 Telegram/SourceFiles/media/view/media_view_overlay_widget.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Telegram/SourceFiles/media/view/media_view_overlay_widget.cpp b/Telegram/SourceFiles/media/view/media_view_overlay_widget.cpp
index a4ed8e4496..10f12c9eed 100644
--- a/Telegram/SourceFiles/media/view/media_view_overlay_widget.cpp
+++ b/Telegram/SourceFiles/media/view/media_view_overlay_widget.cpp
@@ -3676,7 +3676,8 @@ void OverlayWidget::setSession(not_null<Main::Session*> session) {
 
 	session->data().itemRemoved(
 	) | rpl::filter([=](not_null<const HistoryItem*> item) {
-		return (_document != nullptr) && (item->fullId() == _msgid);
+		return (_document != nullptr || _photo != nullptr)
+			&& (item->fullId() == _msgid);
 	}) | rpl::start_with_next([=] {
 		close();
 	}, _sessionLifetime);