fix: crash

AyuGram · Feb 20, 2024 · e580640 · e580640
1 parent 3a9b470
commit e580640
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/Telegram/SourceFiles/data/data_session.cpp b/Telegram/SourceFiles/data/data_session.cpp
@@ -4263,8 +4263,9 @@ void Session::registerItemView(not_null<ViewElement*> view) {
 }
 
 void Session::unregisterItemView(not_null<ViewElement*> view) {
-	if (!_heavyViewParts.contains(view)) {
-		return; // AyuGram: fix crash when using `saveDeletedMessages`
+	// Expects(!_heavyViewParts.contains(view));
+	if (_heavyViewParts.contains(view)) {
+		view->unloadHeavyPart(); // AyuGram: fix crash when using `saveDeletedMessages`
 	}
 
 	_shownSpoilers.remove(view);

diff --git a/Telegram/SourceFiles/history/history_widget.cpp b/Telegram/SourceFiles/history/history_widget.cpp
@@ -3273,6 +3273,7 @@ void HistoryWidget::messagesReceived(
 		not_null<PeerData*> peer,
 		const MTPmessages_Messages &messages,
 		int requestId) {
+	// Expects(_history != nullptr);
 	if (!_history) {
 		return; // AyuGram: fix crash when using `saveDeletedMessages`
 	}

diff --git a/Telegram/SourceFiles/history/view/history_view_top_bar_widget.cpp b/Telegram/SourceFiles/history/view/history_view_top_bar_widget.cpp
@@ -788,8 +788,8 @@ void TopBarWidget::backClicked() {
 		auto settings = &AyuSettings::getInstance();
 		if (settings->hideAllChatsFolder) {
 			const auto filters = &_controller->session().data().chatsFilters();
-			const auto lookup_id = filters->lookupId(_controller->session().premium() ? 0 : 1);
-			_controller->setActiveChatsFilter(lookup_id);
+			const auto lookupId = filters->lookupId(_controller->session().premium() ? 0 : 1);
+			_controller->setActiveChatsFilter(lookupId);
 		} else {
 			_controller->closeFolder();
 		}