From 120f10829bf12d6d14b7ddc0ffbf81906b6cb3cc Mon Sep 17 00:00:00 2001
From: Peter <>
Date: Sun, 28 Jul 2019 01:06:10 +0100
Subject: [PATCH] Fix presences on share screen

---
 .../ShareControllerPeerGridItem.swift         | 26 +++++++++++++------
 1 file changed, 18 insertions(+), 8 deletions(-)

diff --git a/submodules/TelegramUI/TelegramUI/ShareControllerPeerGridItem.swift b/submodules/TelegramUI/TelegramUI/ShareControllerPeerGridItem.swift
index 1bb990c5d1..447842c54c 100644
--- a/submodules/TelegramUI/TelegramUI/ShareControllerPeerGridItem.swift
+++ b/submodules/TelegramUI/TelegramUI/ShareControllerPeerGridItem.swift
@@ -111,7 +111,7 @@ final class ShareControllerPeerGridItem: GridItem {
     func node(layout: GridNodeLayout, synchronousLoad: Bool) -> GridItemNode {
         let node = ShareControllerPeerGridItemNode()
         node.controllerInteraction = self.controllerInteraction
-        node.setup(account: self.account, theme: self.theme, strings: self.strings, peer: self.peer, presence: self.presence, search: self.search, synchronousLoad: synchronousLoad)
+        node.setup(account: self.account, theme: self.theme, strings: self.strings, peer: self.peer, presence: self.presence, search: self.search, synchronousLoad: synchronousLoad, force: false)
         return node
     }
     
@@ -121,13 +121,14 @@ final class ShareControllerPeerGridItem: GridItem {
             return
         }
         node.controllerInteraction = self.controllerInteraction
-        node.setup(account: self.account, theme: self.theme, strings: self.strings, peer: self.peer, presence: self.presence, search: self.search, synchronousLoad: false)
+        node.setup(account: self.account, theme: self.theme, strings: self.strings, peer: self.peer, presence: self.presence, search: self.search, synchronousLoad: false, force: false)
     }
 }
 
 final class ShareControllerPeerGridItemNode: GridItemNode {
-    private var currentState: (Account, RenderedPeer, Bool, PeerPresence?)?
+    private var currentState: (Account, PresentationTheme, PresentationStrings, RenderedPeer, Bool, PeerPresence?)?
     private let peerNode: SelectablePeerNode
+    private var presenceManager: PeerPresenceStatusManager?
     
     var controllerInteraction: ShareControllerInteraction?
     
@@ -138,7 +139,7 @@ final class ShareControllerPeerGridItemNode: GridItemNode {
         
         self.peerNode.toggleSelection = { [weak self] in
             if let strongSelf = self {
-                if let (_, peer, search, _) = strongSelf.currentState {
+                if let (_, _, _, peer, search, _) = strongSelf.currentState {
                     if let _ = peer.peers[peer.peerId] {
                         strongSelf.controllerInteraction?.togglePeer(peer, search)
                     }
@@ -146,10 +147,16 @@ final class ShareControllerPeerGridItemNode: GridItemNode {
             }
         }
         self.addSubnode(self.peerNode)
+        self.presenceManager = PeerPresenceStatusManager(update: { [weak self] in
+            guard let strongSelf = self, let currentState = strongSelf.currentState else {
+                return
+            }
+            strongSelf.setup(account: currentState.0, theme: currentState.1, strings: currentState.2, peer: currentState.3, presence: currentState.5, search: currentState.4, synchronousLoad: false, force: true)
+        })
     }
     
-    func setup(account: Account, theme: PresentationTheme, strings: PresentationStrings, peer: RenderedPeer, presence: PeerPresence?, search: Bool, synchronousLoad: Bool) {
-        if self.currentState == nil || self.currentState!.0 !== account || self.currentState!.1 != peer || !arePeerPresencesEqual(self.currentState!.3, presence) {
+    func setup(account: Account, theme: PresentationTheme, strings: PresentationStrings, peer: RenderedPeer, presence: PeerPresence?, search: Bool, synchronousLoad: Bool, force: Bool) {
+        if force || self.currentState == nil || self.currentState!.0 !== account || self.currentState!.3 != peer || !arePeerPresencesEqual(self.currentState!.5, presence) {
             let itemTheme = SelectablePeerNodeTheme(textColor: theme.actionSheet.primaryTextColor, secretTextColor: theme.chatList.secretTitleColor, selectedTextColor: theme.actionSheet.controlAccentColor, checkBackgroundColor: theme.actionSheet.opaqueItemBackgroundColor, checkFillColor: theme.actionSheet.controlAccentColor, checkColor: theme.actionSheet.checkContentColor, avatarPlaceholderColor: theme.list.mediaPlaceholderColor)
             
             let timestamp = Int32(CFAbsoluteTimeGetCurrent() + NSTimeIntervalSince1970)
@@ -163,15 +170,18 @@ final class ShareControllerPeerGridItemNode: GridItemNode {
             
             self.peerNode.theme = itemTheme
             self.peerNode.setup(account: account, theme: theme, strings: strings, peer: peer, online: online, synchronousLoad: synchronousLoad)
-            self.currentState = (account, peer, search, presence)
+            self.currentState = (account, theme, strings, peer, search, presence)
             self.setNeedsLayout()
+            if let presence = presence as? TelegramUserPresence {
+                self.presenceManager?.reset(presence: presence)
+            }
         }
         self.updateSelection(animated: false)
     }
     
     func updateSelection(animated: Bool) {
         var selected = false
-        if let controllerInteraction = self.controllerInteraction, let (_, peer, _, _) = self.currentState {
+        if let controllerInteraction = self.controllerInteraction, let (_, _, _, peer, _, _) = self.currentState {
             selected = controllerInteraction.selectedPeerIds.contains(peer.peerId)
         }