Swiftgram/Postbox/UnreadMessageCountsView.swift

import Foundation

public enum UnreadMessageCountsTotalItem {
    case raw
    case filtered
}

public enum UnreadMessageCountsTotalCategory {
    case messages
    case chats
}

public enum UnreadMessageCountsItem: Equatable {
    case total(UnreadMessageCountsTotalItem, UnreadMessageCountsTotalCategory)
    case peer(PeerId)
    case group(PeerGroupId)
}

private enum MutableUnreadMessageCountsItemEntry {
    case total(ChatListTotalUnreadState)
    case peer(PeerId, Int32)
    case group(PeerGroupId, ChatListGroupReferenceUnreadCounters)
}

public enum UnreadMessageCountsItemEntry {
    case total(ChatListTotalUnreadState)
    case peer(PeerId, Int32)
    case group(PeerGroupId, Int32)
}

final class MutableUnreadMessageCountsView: MutablePostboxView {
    fileprivate var entries: [MutableUnreadMessageCountsItemEntry]
    
    init(postbox: Postbox, items: [UnreadMessageCountsItem]) {
        self.entries = items.map { item in
            switch item {
                case .total:
                    return .total(postbox.messageHistoryMetadataTable.getChatListTotalUnreadState())
                case let .peer(peerId):
                    var count: Int32 = 0
                    if let combinedState = postbox.readStateTable.getCombinedState(peerId) {
                        count = combinedState.count
                    }
                    return .peer(peerId, count)
                case let .group(groupId):
                    return .group(groupId, ChatListGroupReferenceUnreadCounters(postbox: postbox, groupId: groupId))
            }
        }
    }
    
    func replay(postbox: Postbox, transaction: PostboxTransaction) -> Bool {
        var updated = false
        
        if transaction.currentUpdatedTotalUnreadState != nil || !transaction.alteredInitialPeerCombinedReadStates.isEmpty {
            for i in 0 ..< self.entries.count {
                switch self.entries[i] {
                    case let .total(state):
                        if transaction.currentUpdatedTotalUnreadState != nil {
                            let updatedState = postbox.messageHistoryMetadataTable.getChatListTotalUnreadState()
                            if updatedState != state {
                                self.entries[i] = .total(updatedState)
                                updated = true
                            }
                        }
                    case let .peer(peerId, _):
                        if transaction.alteredInitialPeerCombinedReadStates[peerId] != nil {
                            var updatedCount: Int32 = 0
                            if let combinedState = postbox.readStateTable.getCombinedState(peerId) {
                                updatedCount = combinedState.count
                            }
                            self.entries[i] = .peer(peerId, updatedCount)
                            updated = true
                        }
                    case let .group(_, counters):
                        if counters.replay(postbox: postbox, transaction: transaction) {
                            updated = true
                        }
                }
            }
        }
        
        return updated
    }
    
    func immutableView() -> PostboxView {
        return UnreadMessageCountsView(self)
    }
}

public final class UnreadMessageCountsView: PostboxView {
    public let entries: [UnreadMessageCountsItemEntry]
    
    init(_ view: MutableUnreadMessageCountsView) {
        self.entries = view.entries.map { entry in
            switch entry {
                case let .total(count):
                    return .total(count)
                case let .peer(peerId, count):
                    return .peer(peerId, count)
                case let .group(groupId, counters):
                    let (unread, mutedUnread) = counters.getCounters()
                    return .group(groupId, unread + mutedUnread)
            }
        }
    }
    
    public func count(for item: UnreadMessageCountsItem) -> Int32? {
        for entry in self.entries {
            switch entry {
                case let .total(state):
                    if case let .total(value, category) = item {
                        switch value {
                            case .raw:
                                switch category {
                                    case .messages:
                                        return state.absoluteCounters.messageCount
                                    case .chats:
                                        return state.absoluteCounters.chatCount
                                }
                            case .filtered:
                                switch category {
                                    case .messages:
                                        return state.filteredCounters.messageCount
                                    case .chats:
                                        return state.filteredCounters.chatCount
                                }
                        }
                    }
                case let .peer(peerId, count):
                    if case .peer(peerId) = item {
                        return count
                    }
                case let .group(groupId, count):
                    if case .group(groupId) = item {
                        return count
                    }
            }
        }
        return nil
    }
}