Swiftgram/submodules/TelegramCore/Sources/TelegramEngine/Messages/Message.swift

import Postbox

public final class EngineMessage: Equatable {
    public typealias Id = MessageId
    public typealias StableId = UInt32
    public typealias Index = MessageIndex
    public typealias Tags = MessageTags
    public typealias Attribute = MessageAttribute
    public typealias GroupInfo = MessageGroupInfo
    public typealias Flags = MessageFlags
    public typealias GlobalTags = GlobalMessageTags
    public typealias LocalTags = LocalMessageTags
    public typealias ForwardInfo = MessageForwardInfo
    public typealias CustomTag = MemoryBuffer
    
    public enum InputTag: Hashable {
        case tag(Tags)
        case custom(CustomTag)
    }
    
    private let impl: Message
    
    public var stableId: UInt32 {
        return self.impl.stableId
    }
    
    public var stableVersion: UInt32 {
        return self.impl.stableVersion
    }
    
    public var id: Id {
        return self.impl.id
    }
    public var globallyUniqueId: Int64? {
        return self.impl.globallyUniqueId
    }
    public var groupingKey: Int64? {
        return self.impl.groupingKey
    }
    public var groupInfo: GroupInfo? {
        return self.impl.groupInfo
    }
    public var threadId: Int64? {
        return self.impl.threadId
    }
    public var timestamp: Int32 {
        return self.impl.timestamp
    }
    public var flags: Flags {
        return self.impl.flags
    }
    public var tags: Tags {
        return self.impl.tags
    }
    public var globalTags: GlobalTags {
        return self.impl.globalTags
    }
    public var localTags: LocalTags {
        return self.impl.localTags
    }
    public var forwardInfo: ForwardInfo? {
        return self.impl.forwardInfo
    }
    public var author: EnginePeer? {
        return self.impl.author.flatMap(EnginePeer.init)
    }
    public var text: String {
        return self.impl.text
    }
    public var attributes: [Attribute] {
        return self.impl.attributes
    }
    public var media: [Media] {
        return self.impl.media
    }
    public var peers: SimpleDictionary<EnginePeer.Id, Peer> {
        return self.impl.peers
    }
    public var associatedMessages: SimpleDictionary<EngineMessage.Id, Message> {
        return self.impl.associatedMessages
    }
    public var associatedMessageIds: [EngineMessage.Id] {
        return self.impl.associatedMessageIds
    }
    public var associatedMedia: [MediaId: Media] {
        return self.impl.associatedMedia
    }
    public var associatedThreadInfo: Message.AssociatedThreadInfo? {
        return self.impl.associatedThreadInfo
    }
    public var associatedStories: [StoryId: CodableEntry] {
        return self.impl.associatedStories
    }
    
    public var index: MessageIndex {
        return self.impl.index
    }

    public init(
        stableId: UInt32,
        stableVersion: UInt32,
        id: EngineMessage.Id,
        globallyUniqueId: Int64?,
        groupingKey: Int64?,
        groupInfo: EngineMessage.GroupInfo?,
        threadId: Int64?,
        timestamp: Int32,
        flags: EngineMessage.Flags,
        tags: EngineMessage.Tags,
        globalTags: EngineMessage.GlobalTags,
        localTags: EngineMessage.LocalTags,
        customTags: [EngineMessage.CustomTag],
        forwardInfo: EngineMessage.ForwardInfo?,
        author: EnginePeer?,
        text: String,
        attributes: [Attribute],
        media: [EngineMedia],
        peers: [EnginePeer.Id: EnginePeer],
        associatedMessages: [EngineMessage.Id: EngineMessage],
        associatedMessageIds: [EngineMessage.Id],
        associatedMedia: [MediaId: Media],
        associatedThreadInfo: Message.AssociatedThreadInfo?,
        associatedStories: [StoryId: CodableEntry]
    ) {
        var mappedPeers: [PeerId: Peer] = [:]
        for (id, peer) in peers {
            mappedPeers[id] = peer._asPeer()
        }

        var mappedAssociatedMessages: [MessageId: Message] = [:]
        for (id, message) in associatedMessages {
            mappedAssociatedMessages[id] = message._asMessage()
        }

        self.impl = Message(
            stableId: stableId,
            stableVersion: stableVersion,
            id: id,
            globallyUniqueId: globallyUniqueId,
            groupingKey: groupingKey,
            groupInfo: groupInfo,
            threadId: threadId,
            timestamp: timestamp,
            flags: flags,
            tags: tags,
            globalTags: globalTags,
            localTags: localTags,
            customTags: customTags,
            forwardInfo: forwardInfo,
            author: author?._asPeer(),
            text: text,
            attributes: attributes,
            media: media.map { $0._asMedia() },
            peers: SimpleDictionary(mappedPeers),
            associatedMessages: SimpleDictionary(mappedAssociatedMessages),
            associatedMessageIds: associatedMessageIds,
            associatedMedia: associatedMedia,
            associatedThreadInfo: associatedThreadInfo,
            associatedStories: associatedStories
        )
    }

    public init(_ impl: Message) {
        self.impl = impl
    }

    public func _asMessage() -> Message {
        return self.impl
    }
    
    public static func ==(lhs: EngineMessage, rhs: EngineMessage) -> Bool {
        if lhs.id != rhs.id {
            return false
        }
        if lhs.globallyUniqueId != rhs.globallyUniqueId {
            return false
        }
        if lhs.groupingKey != rhs.groupingKey {
            return false
        }
        if lhs.groupInfo != rhs.groupInfo {
            return false
        }
        if lhs.threadId != rhs.threadId {
            return false
        }
        if lhs.timestamp != rhs.timestamp {
            return false
        }
        if lhs.flags != rhs.flags {
            return false
        }
        if lhs.tags != rhs.tags {
            return false
        }
        if lhs.globalTags != rhs.globalTags {
            return false
        }
        if lhs.localTags != rhs.localTags {
            return false
        }
        if lhs.forwardInfo != rhs.forwardInfo {
            return false
        }
        if lhs.author != rhs.author {
            return false
        }
        if lhs.text != rhs.text {
            return false
        }
        if !areMediaArraysEqual(lhs.media, rhs.media) {
            return false
        }
        if lhs.associatedThreadInfo != rhs.associatedThreadInfo {
            return false
        }
        if lhs.attributes.count != rhs.attributes.count {
            return false
        }
        if lhs.stableVersion != rhs.stableVersion {
            return false
        }
        return true
    }
}