From 4a63c69d7f60321832a5e363e491f1fe0913c022 Mon Sep 17 00:00:00 2001
From: leha-bot <leha-bot@users.noreply.github.com>
Date: Mon, 17 Sep 2018 04:56:50 +0300
Subject: [PATCH] Enable messages unpinning for channels

Also make History Widget react on pinning Channel message event and show
it in UI.

This commit is based on upstream commit
https://github.com/telegramdesktop/tdesktop/commit/75d8d01b1754e4f9c6ffc601848455bda9c4516d

Related to #114.
Closes #8.
---
 .../history/history_inner_widget.cpp          |  6 ++---
 Telegram/SourceFiles/history/history_item.cpp |  7 +++++
 Telegram/SourceFiles/history/history_item.h   |  1 +
 .../SourceFiles/history/history_widget.cpp    | 26 +++++++++----------
 4 files changed, 23 insertions(+), 17 deletions(-)

diff --git a/Telegram/SourceFiles/history/history_inner_widget.cpp b/Telegram/SourceFiles/history/history_inner_widget.cpp
index 99e1d9165..791f43404 100644
--- a/Telegram/SourceFiles/history/history_inner_widget.cpp
+++ b/Telegram/SourceFiles/history/history_inner_widget.cpp
@@ -1240,9 +1240,9 @@ void HistoryInner::showContextMenu(QContextMenuEvent *e, bool showFromTouch) {
 				_menu->addAction(lang(lng_context_edit_msg), _widget, SLOT(onEditMessage()));
 			}
 			if (item->canPin()) {
-				bool ispinned = (item->history()->peer->asChannel()->pinnedMessageId() == item->id);
-				_menu->addAction(lang(ispinned ? lng_context_unpin_msg : lng_context_pin_msg), _widget,
-				                 ispinned ? SLOT(onUnpinMessage()) : SLOT(onPinMessage()));
+				bool isPinned = item->isPinned();
+				_menu->addAction(lang(isPinned ? lng_context_unpin_msg : lng_context_pin_msg), _widget,
+				                 isPinned ? SLOT(onUnpinMessage()) : SLOT(onPinMessage()));
 			}
 		}
 		if (lnkPhoto) {
diff --git a/Telegram/SourceFiles/history/history_item.cpp b/Telegram/SourceFiles/history/history_item.cpp
index fa6d27748..c87e44402 100644
--- a/Telegram/SourceFiles/history/history_item.cpp
+++ b/Telegram/SourceFiles/history/history_item.cpp
@@ -822,6 +822,13 @@ void HistoryItem::setId(MsgId newId) {
 	}
 }
 
+bool HistoryItem::isPinned() const {
+	if (auto channel = _history->peer->asChannel()) {
+		return (channel->pinnedMessageId() == id);
+	}
+	return false;
+}
+
 bool HistoryItem::canPin() const {
 	if (id < 0 || !toHistoryMessage()) {
 		return false;
diff --git a/Telegram/SourceFiles/history/history_item.h b/Telegram/SourceFiles/history/history_item.h
index 0abb75e66..8610a5907 100644
--- a/Telegram/SourceFiles/history/history_item.h
+++ b/Telegram/SourceFiles/history/history_item.h
@@ -717,6 +717,7 @@ public:
 		return _text.isEmpty();
 	}
 
+	bool isPinned() const;
 	bool canPin() const;
 	bool canForward() const;
 	bool canEdit(const QDateTime &cur) const;
diff --git a/Telegram/SourceFiles/history/history_widget.cpp b/Telegram/SourceFiles/history/history_widget.cpp
index c566c9003..48dcd662b 100644
--- a/Telegram/SourceFiles/history/history_widget.cpp
+++ b/Telegram/SourceFiles/history/history_widget.cpp
@@ -5992,23 +5992,19 @@ void HistoryWidget::onEditMessage() {
 }
 
 void HistoryWidget::onPinMessage() {
-	HistoryItem *to = App::contextItem();
-	if (!to || !to->canPin() || !_peer || !_peer->isMegagroup()) return;
+	auto to = App::contextItem();
+	if (!to || !to->canPin()) return;
 
 	Ui::show(Box<PinMessageBox>(_peer->asChannel(), to->id));
 }
 
 void HistoryWidget::onUnpinMessage() {
-	if (!_peer || !_peer->isMegagroup()) return;
+	if (!_peer || !_peer->isChannel()) return;
 
 	Ui::show(Box<ConfirmBox>(lang(lng_pinned_unpin_sure), lang(lng_pinned_unpin), base::lambda_guarded(this, [this] {
-		                         if (!_peer || !_peer->asChannel()) return;
-
-		                         _peer->asChannel()->clearPinnedMessage();
-		                         if (pinnedMsgVisibilityUpdated()) {
-			                         updateControlsGeometry();
-			                         update();
-		                         }
+		                         auto channel = _peer ? _peer->asChannel() : nullptr;
+		                         if (!channel) return;
+		                         channel->clearPinnedMessage();
 
 		                         Ui::hideLayer();
 		                         MTP::send(MTPchannels_UpdatePinnedMessage(
@@ -6024,8 +6020,10 @@ void HistoryWidget::unpinDone(const MTPUpdates &updates) {
 }
 
 void HistoryWidget::onPinnedHide() {
-	if (!_peer || !_peer->asChannel()) return;
-	if (!_peer->asChannel()->pinnedMessageId()) {
+	auto channel = _peer ? _peer->asChannel() : nullptr;
+	if (!channel) return;
+	auto pinnedId = channel->pinnedMessageId();
+	if (!pinnedId) {
 		if (pinnedMsgVisibilityUpdated()) {
 			updateControlsGeometry();
 			update();
@@ -6033,10 +6031,10 @@ void HistoryWidget::onPinnedHide() {
 		return;
 	}
 
-	if (_peer->asChannel()->canPinMessages()) {
+	if (channel->canPinMessages()) {
 		onUnpinMessage();
 	} else {
-		Global::RefHiddenPinnedMessages().insert(_peer->id, _peer->asChannel()->pinnedMessageId());
+		Global::RefHiddenPinnedMessages().insert(_peer->id, pinnedId);
 		Local::writeUserSettings();
 		if (pinnedMsgVisibilityUpdated()) {
 			updateControlsGeometry();