Swiftgram/submodules/TelegramCore/Sources/TelegramEngine/Peers/RecentlySearchedPeerIds.swift

import Foundation
import Postbox
import SwiftSignalKit


func _internal_addRecentlySearchedPeer(postbox: Postbox, peerId: PeerId) -> Signal<Void, NoError> {
    return postbox.transaction { transaction -> Void in
        if let entry = CodableEntry(RecentPeerItem(rating: 0.0)) {
            transaction.addOrMoveToFirstPositionOrderedItemListItem(collectionId: Namespaces.OrderedItemList.RecentlySearchedPeerIds, item: OrderedItemListEntry(id: RecentPeerItemId(peerId).rawValue, contents: entry), removeTailIfCountExceeds: 20)
        }
    }
}

func _internal_removeRecentlySearchedPeer(postbox: Postbox, peerId: PeerId) -> Signal<Void, NoError> {
    return postbox.transaction { transaction -> Void in
        transaction.removeOrderedItemListItem(collectionId: Namespaces.OrderedItemList.RecentlySearchedPeerIds, itemId: RecentPeerItemId(peerId).rawValue)
    }
}

func _internal_clearRecentlySearchedPeers(postbox: Postbox) -> Signal<Void, NoError> {
    return postbox.transaction { transaction -> Void in
        transaction.replaceOrderedItemListItems(collectionId: Namespaces.OrderedItemList.RecentlySearchedPeerIds, items: [])
    }
}

public struct RecentlySearchedPeerSubpeerSummary: Equatable {
    public let count: Int
    
    public init(count: Int) {
        self.count = count
    }
}

public struct RecentlySearchedPeer: Equatable {
    public let peer: RenderedPeer
    public let presence: TelegramUserPresence?
    public let notificationSettings: TelegramPeerNotificationSettings?
    public let unreadCount: Int32
    public let subpeerSummary: RecentlySearchedPeerSubpeerSummary?
    
    public init(peer: RenderedPeer, presence: TelegramUserPresence?, notificationSettings: TelegramPeerNotificationSettings?, unreadCount: Int32, subpeerSummary: RecentlySearchedPeerSubpeerSummary?) {
        self.peer = peer
        self.presence = presence
        self.notificationSettings = notificationSettings
        self.unreadCount = unreadCount
        self.subpeerSummary = subpeerSummary
    }
}

public func _internal_recentlySearchedPeers(postbox: Postbox) -> Signal<[RecentlySearchedPeer], NoError> {
    return postbox.combinedView(keys: [.orderedItemList(id: Namespaces.OrderedItemList.RecentlySearchedPeerIds)])
    |> mapToSignal { view -> Signal<[RecentlySearchedPeer], NoError> in
        var peerIds: [PeerId] = []
        if let view = view.views[.orderedItemList(id: Namespaces.OrderedItemList.RecentlySearchedPeerIds)] as? OrderedItemListView {
            for item in view.items {
                let peerId = RecentPeerItemId(item.id).peerId
                peerIds.append(peerId)
            }
        }
        var keys: [PostboxViewKey] = []
        let unreadCountsKey: PostboxViewKey = .unreadCounts(items: peerIds.map { UnreadMessageCountsItem.peer(id: $0, handleThreads: true) })
        keys.append(unreadCountsKey)
        keys.append(contentsOf: peerIds.map({ .peer(peerId: $0, components: .all) }))
        
        return postbox.combinedView(keys: keys)
        |> mapToSignal { view -> Signal<[RecentlySearchedPeer], NoError> in
            var result: [RecentlySearchedPeer] = []
            var unreadCounts: [PeerId: Int32] = [:]
            if let unreadCountsView = view.views[unreadCountsKey] as? UnreadMessageCountsView {
                for entry in unreadCountsView.entries {
                    if case let .peer(peerId, state) = entry {
                        unreadCounts[peerId] = state?.count ?? 0
                    }
                }
            }
            
            var migratedPeerIds: [EnginePeer.Id: EnginePeer.Id] = [:]
            for peerId in peerIds {
                if let peerView = view.views[.peer(peerId: peerId, components: .all)] as? PeerView {
                    var presence: TelegramUserPresence?
                    var unreadCount = unreadCounts[peerId] ?? 0
                    if let peer = peerView.peers[peerId] {
                        if let associatedPeerId = peer.associatedPeerId {
                            presence = peerView.peerPresences[associatedPeerId] as? TelegramUserPresence
                        } else {
                            presence = peerView.peerPresences[peerId] as? TelegramUserPresence
                        }
                        
                        if let channel = peer as? TelegramChannel {
                            if case .member = channel.participationStatus {
                            } else {
                                unreadCount = 0
                            }
                        }
                        
                        if let group = peer as? TelegramGroup, let migrationReference = group.migrationReference {
                            migratedPeerIds = [group.id: migrationReference.peerId]
                        }
                    }
                    
                    var subpeerSummary: RecentlySearchedPeerSubpeerSummary?
                    if let cachedData = peerView.cachedData as? CachedChannelData {
                        let count: Int32 = cachedData.participantsSummary.memberCount ?? 0
                        subpeerSummary = RecentlySearchedPeerSubpeerSummary(count: Int(count))
                    }
                    
                    result.append(RecentlySearchedPeer(peer: RenderedPeer(peerId: peerId, peers: SimpleDictionary(peerView.peers), associatedMedia: peerView.media), presence: presence, notificationSettings: peerView.notificationSettings as? TelegramPeerNotificationSettings, unreadCount: unreadCount, subpeerSummary: subpeerSummary))
                }
            }
            
            if !migratedPeerIds.isEmpty {
                return postbox.transaction { transaction -> Signal<[RecentlySearchedPeer], NoError> in
                    for (previousPeerId, updatedPeerId) in migratedPeerIds {
                        transaction.removeOrderedItemListItem(collectionId: Namespaces.OrderedItemList.RecentlySearchedPeerIds, itemId: RecentPeerItemId(previousPeerId).rawValue)
                        if let entry = CodableEntry(RecentPeerItem(rating: 0.0)) {
                            transaction.addOrMoveToFirstPositionOrderedItemListItem(collectionId: Namespaces.OrderedItemList.RecentlySearchedPeerIds, item: OrderedItemListEntry(id: RecentPeerItemId(updatedPeerId).rawValue, contents: entry), removeTailIfCountExceeds: 20)
                        }
                    }
                    return .complete()
                }
                |> switchToLatest
            } else {
                return .single(result)
            }
        }
    }
}