Swiftgram/submodules/Postbox/Sources/PeerNotificationSettings.swift

import Foundation

public enum PeerNotificationSettingsBehavior: PostboxCoding {
    enum CodingKeys: String, CodingKey {
        case _case = "_v"
        case toValue
        case atTimestamp
    }

    case none
    case reset(atTimestamp: Int32, toValue: PeerNotificationSettings)
    
    public init(decoder: PostboxDecoder) {
        switch decoder.decodeInt32ForKey("_v", orElse: 0) {
            case 0:
                self = .none
            case 1:
                if let toValue = decoder.decodeObjectForKey("toValue") as? PeerNotificationSettings {
                    self = .reset(atTimestamp: decoder.decodeInt32ForKey("atTimestamp", orElse: 0), toValue: toValue)
                } else {
                    assertionFailure()
                    self = .none
                }
            default:
                assertionFailure()
                self = .none
        }
    }
    
    public func encode(_ encoder: PostboxEncoder) {
        switch self {
            case .none:
                encoder.encodeInt32(0, forKey: "_v")
            case let .reset(atTimestamp, toValue):
                encoder.encodeInt32(1, forKey: "_v")
                encoder.encodeInt32(atTimestamp, forKey: "atTimestamp")
                encoder.encodeObject(toValue, forKey: "toValue")
        }
    }
}

public protocol PeerNotificationSettings: PostboxCoding {
    func isRemovedFromTotalUnreadCount(`default`: Bool) -> Bool

    var behavior: PeerNotificationSettingsBehavior { get }

    func isEqual(to other: PeerNotificationSettings) -> Bool
}

public final class PostboxGlobalNotificationSettings {
    public let defaultIncludePeer: (_ peer: Peer) -> Bool

    public init(
        defaultIncludePeer: @escaping (_ peer: Peer) -> Bool
    ) {
        self.defaultIncludePeer = defaultIncludePeer
    }
}

public func resolvedIsRemovedFromTotalUnreadCount(globalSettings: PostboxGlobalNotificationSettings, peer: Peer, peerSettings: PeerNotificationSettings?) -> Bool {
    let defaultValue = !globalSettings.defaultIncludePeer(peer)
    if let peerSettings = peerSettings {
        return peerSettings.isRemovedFromTotalUnreadCount(default: defaultValue)
    } else {
        return defaultValue
    }
}