import Foundation struct MessageHistoryAnchorIndex { let index: MessageIndex let exact: Bool } enum IntermediateMessageHistoryEntry { case Message(IntermediateMessage) case Hole(MessageHistoryHole) var index: MessageIndex { switch self { case let .Message(message): return MessageIndex(id: message.id, timestamp: message.timestamp) case let .Hole(hole): return hole.maxIndex } } } enum RenderedMessageHistoryEntry { case RenderedMessage(Message) case Hole(MessageHistoryHole) var index: MessageIndex { switch self { case let .RenderedMessage(message): return MessageIndex(id: message.id, timestamp: message.timestamp) case let .Hole(hole): return hole.maxIndex } } } private struct MessageDataFlags: OptionSet { var rawValue: Int8 init() { self.rawValue = 0 } init(rawValue: Int8) { self.rawValue = rawValue } static let hasGloballyUniqueId = MessageDataFlags(rawValue: 1 << 0) static let hasGlobalTags = MessageDataFlags(rawValue: 1 << 1) } final class MessageHistoryTable: Table { static func tableSpec(_ id: Int32) -> ValueBoxTable { return ValueBoxTable(id: id, keyType: .binary) } let messageHistoryIndexTable: MessageHistoryIndexTable let messageMediaTable: MessageMediaTable let historyMetadataTable: MessageHistoryMetadataTable let globallyUniqueMessageIdsTable: MessageGloballyUniqueIdTable let unsentTable: MessageHistoryUnsentTable let tagsTable: MessageHistoryTagsTable let globalTagsTable: GlobalMessageHistoryTagsTable let readStateTable: MessageHistoryReadStateTable let synchronizeReadStateTable: MessageHistorySynchronizeReadStateTable let textIndexTable: MessageHistoryTextIndexTable init(valueBox: ValueBox, table: ValueBoxTable, messageHistoryIndexTable: MessageHistoryIndexTable, messageMediaTable: MessageMediaTable, historyMetadataTable: MessageHistoryMetadataTable, globallyUniqueMessageIdsTable: MessageGloballyUniqueIdTable, unsentTable: MessageHistoryUnsentTable, tagsTable: MessageHistoryTagsTable, globalTagsTable: GlobalMessageHistoryTagsTable, readStateTable: MessageHistoryReadStateTable, synchronizeReadStateTable: MessageHistorySynchronizeReadStateTable, textIndexTable: MessageHistoryTextIndexTable) { self.messageHistoryIndexTable = messageHistoryIndexTable self.messageMediaTable = messageMediaTable self.historyMetadataTable = historyMetadataTable self.globallyUniqueMessageIdsTable = globallyUniqueMessageIdsTable self.unsentTable = unsentTable self.tagsTable = tagsTable self.globalTagsTable = globalTagsTable self.readStateTable = readStateTable self.synchronizeReadStateTable = synchronizeReadStateTable self.textIndexTable = textIndexTable super.init(valueBox: valueBox, table: table) } private func key(_ index: MessageIndex, key: ValueBoxKey = ValueBoxKey(length: 8 + 4 + 4 + 4)) -> ValueBoxKey { key.setInt64(0, value: index.id.peerId.toInt64()) key.setInt32(8, value: index.timestamp) key.setInt32(8 + 4, value: index.id.namespace) key.setInt32(8 + 4 + 4, value: index.id.id) return key } private func lowerBound(_ peerId: PeerId) -> ValueBoxKey { let key = ValueBoxKey(length: 8) key.setInt64(0, value: peerId.toInt64()) return key } private func upperBound(_ peerId: PeerId) -> ValueBoxKey { let key = ValueBoxKey(length: 8) key.setInt64(0, value: peerId.toInt64()) return key.successor } private func messagesGroupedByPeerId(_ messages: [StoreMessage]) -> [PeerId: [StoreMessage]] { var dict: [PeerId: [StoreMessage]] = [:] for message in messages { let peerId = message.id.peerId if dict[peerId] == nil { dict[peerId] = [message] } else { dict[peerId]!.append(message) } } return dict } private func messageIdsByPeerId(_ ids: [MessageId]) -> [PeerId: [MessageId]] { var dict: [PeerId: [MessageId]] = [:] for id in ids { let peerId = id.peerId if dict[peerId] == nil { dict[peerId] = [id] } else { dict[peerId]!.append(id) } } return dict } private func processIndexOperationsCommitAccumulatedRemoveIndices(remove: Bool, peerId: PeerId, accumulatedRemoveIndices: inout [(MessageIndex, Bool)], updatedCombinedState: inout CombinedPeerReadState?, invalidateReadState: inout Bool, unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], outputOperations: inout [MessageHistoryOperation], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { if !accumulatedRemoveIndices.isEmpty { let (combinedState, invalidate) = self.readStateTable.deleteMessages(peerId, indices: accumulatedRemoveIndices.map { $0.0 }, incomingStatsInIndices: { peerId, namespace, indices in return self.incomingMessageStatsInIndices(peerId, namespace: namespace, indices: indices) }) if let combinedState = combinedState { updatedCombinedState = combinedState } if invalidate { invalidateReadState = true } if remove { var indicesWithMetadata: [(MessageIndex, Bool, MessageTags)] = [] var globalIndicesWithMetadata: [(GlobalMessageTags, MessageIndex)] = [] if accumulatedRemoveIndices.count == 1 && accumulatedRemoveIndices[0].0.timestamp == Int32.max { assert(true) } for index in accumulatedRemoveIndices { let tagsAndGlobalTags = self.justRemove(index.0, unsentMessageOperations: &unsentMessageOperations) if let (tags, globalTags) = tagsAndGlobalTags { indicesWithMetadata.append((index.0, index.1, tags)) if !globalTags.isEmpty { globalIndicesWithMetadata.append(globalTags, index.0) } } else { indicesWithMetadata.append((index.0, index.1, MessageTags())) } } assert(accumulatedRemoveIndices.count == indicesWithMetadata.count) outputOperations.append(.Remove(indicesWithMetadata)) if !globalIndicesWithMetadata.isEmpty { globalTagsOperations.append(.remove(globalIndicesWithMetadata)) } } else { var indicesWithMetadata: [(MessageIndex, Bool, MessageTags)] = [] var globalIndicesWithMetadata: [(GlobalMessageTags, MessageIndex)] = [] for index in accumulatedRemoveIndices { if let (tags, globalTags) = self.tagsForIndex(index.0) { indicesWithMetadata.append((index.0, index.1, tags)) if !globalTags.isEmpty { globalIndicesWithMetadata.append(globalTags, index.0) } } } assert(accumulatedRemoveIndices.count == indicesWithMetadata.count) outputOperations.append(.Remove(indicesWithMetadata)) if !globalIndicesWithMetadata.isEmpty { globalTagsOperations.append(.remove(globalIndicesWithMetadata)) } } accumulatedRemoveIndices.removeAll() } } private func processIndexOperations(_ peerId: PeerId, operations: [MessageHistoryIndexOperation], processedOperationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { let sharedKey = self.key(MessageIndex(id: MessageId(peerId: PeerId(namespace: 0, id: 0), namespace: 0, id: 0), timestamp: 0)) let sharedBuffer = WriteBuffer() let sharedEncoder = Encoder() var outputOperations: [MessageHistoryOperation] = [] var accumulatedRemoveIndices: [(MessageIndex, Bool)] = [] var accumulatedAddedIncomingMessageIndices = Set() var updatedCombinedState: CombinedPeerReadState? var invalidateReadState = false let commitAccumulatedAddedIndices: () -> Void = { if !accumulatedAddedIncomingMessageIndices.isEmpty { let (combinedState, invalidate) = self.readStateTable.addIncomingMessages(peerId, indices: accumulatedAddedIncomingMessageIndices) if let combinedState = combinedState { updatedCombinedState = combinedState } if invalidate { invalidateReadState = true } accumulatedAddedIncomingMessageIndices.removeAll() } } for operation in operations { switch operation { case let .InsertHole(hole): processIndexOperationsCommitAccumulatedRemoveIndices(remove: true, peerId: peerId, accumulatedRemoveIndices: &accumulatedRemoveIndices, updatedCombinedState: &updatedCombinedState, invalidateReadState: &invalidateReadState, unsentMessageOperations: &unsentMessageOperations, outputOperations: &outputOperations, globalTagsOperations: &globalTagsOperations) commitAccumulatedAddedIndices() self.justInsertHole(hole) outputOperations.append(.InsertHole(hole)) let tags = self.messageHistoryIndexTable.seedConfiguration.existingMessageTags.rawValue & hole.tags for i in 0 ..< 32 { let currentTags = tags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = MessageTags(rawValue: 1 << UInt32(i)) self.tagsTable.add(tag, index: hole.maxIndex) } } case let .InsertMessage(storeMessage): processIndexOperationsCommitAccumulatedRemoveIndices(remove: true, peerId: peerId, accumulatedRemoveIndices: &accumulatedRemoveIndices, updatedCombinedState: &updatedCombinedState, invalidateReadState: &invalidateReadState, unsentMessageOperations: &unsentMessageOperations, outputOperations: &outputOperations, globalTagsOperations: &globalTagsOperations) let message = self.justInsertMessage(storeMessage, sharedKey: sharedKey, sharedBuffer: sharedBuffer, sharedEncoder: sharedEncoder) outputOperations.append(.InsertMessage(message)) if message.flags.contains(.Unsent) && !message.flags.contains(.Failed) { self.unsentTable.add(message.id, operations: &unsentMessageOperations) } let tags = message.tags.rawValue if tags != 0 { for i in 0 ..< 32 { let currentTags = tags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = MessageTags(rawValue: 1 << UInt32(i)) self.tagsTable.add(tag, index: MessageIndex(message)) } } } let globalTags = message.globalTags.rawValue if globalTags != 0 { for i in 0 ..< 32 { let currentTags = globalTags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = GlobalMessageTags(rawValue: 1 << UInt32(i)) if self.globalTagsTable.addMessage(tag, index: MessageIndex(message)) { globalTagsOperations.append(.insertMessage(tag, message)) } } } } if message.flags.contains(.Incoming) { accumulatedAddedIncomingMessageIndices.insert(MessageIndex(message)) } case let .Remove(index, isMessage): commitAccumulatedAddedIndices() accumulatedRemoveIndices.append((index, isMessage)) case let .Update(index, storeMessage): commitAccumulatedAddedIndices() processIndexOperationsCommitAccumulatedRemoveIndices(remove: true, peerId: peerId, accumulatedRemoveIndices: &accumulatedRemoveIndices, updatedCombinedState: &updatedCombinedState, invalidateReadState: &invalidateReadState, unsentMessageOperations: &unsentMessageOperations, outputOperations: &outputOperations, globalTagsOperations: &globalTagsOperations) if let (message, previousTags) = self.justUpdate(index, message: storeMessage, sharedKey: sharedKey, sharedBuffer: sharedBuffer, sharedEncoder: sharedEncoder, unsentMessageOperations: &unsentMessageOperations) { outputOperations.append(.Remove([(index, true, previousTags)])) outputOperations.append(.InsertMessage(message)) if message.flags.contains(.Incoming) { if index != MessageIndex(message) { accumulatedRemoveIndices.append((index, true)) accumulatedAddedIncomingMessageIndices.insert(MessageIndex(message)) } } } case let .UpdateTimestamp(index, timestamp): commitAccumulatedAddedIndices() processIndexOperationsCommitAccumulatedRemoveIndices(remove: true, peerId: peerId, accumulatedRemoveIndices: &accumulatedRemoveIndices, updatedCombinedState: &updatedCombinedState, invalidateReadState: &invalidateReadState, unsentMessageOperations: &unsentMessageOperations, outputOperations: &outputOperations, globalTagsOperations: &globalTagsOperations) let tagsAndGlobalTags = self.justUpdateTimestamp(index, timestamp: timestamp) outputOperations.append(.UpdateTimestamp(index, timestamp)) if let (_, globalTags) = tagsAndGlobalTags { if !globalTags.isEmpty { globalTagsOperations.append(.updateTimestamp(globalTags, index, timestamp)) } } } } commitAccumulatedAddedIndices() processIndexOperationsCommitAccumulatedRemoveIndices(remove: true, peerId: peerId, accumulatedRemoveIndices: &accumulatedRemoveIndices, updatedCombinedState: &updatedCombinedState, invalidateReadState: &invalidateReadState, unsentMessageOperations: &unsentMessageOperations, outputOperations: &outputOperations, globalTagsOperations: &globalTagsOperations) if let updatedCombinedState = updatedCombinedState { outputOperations.append(.UpdateReadState(updatedCombinedState)) } if invalidateReadState { self.synchronizeReadStateTable.set(peerId, operation: .Validate, operations: &updatedPeerReadStateOperations) } if processedOperationsByPeerId[peerId] == nil { processedOperationsByPeerId[peerId] = outputOperations } else { processedOperationsByPeerId[peerId]!.append(contentsOf: outputOperations) } } private func internalStoreMessages(_ messages: [StoreMessage]) -> [InternalStoreMessage] { var internalStoreMessages: [InternalStoreMessage] = [] for message in messages { switch message.id { case let .Id(id): internalStoreMessages.append(InternalStoreMessage(id: id, timestamp: message.timestamp, globallyUniqueId: message.globallyUniqueId, flags: message.flags, tags: message.tags, globalTags: message.globalTags, forwardInfo: message.forwardInfo, authorId: message.authorId, text: message.text, attributes: message.attributes, media: message.media)) case let .Partial(peerId, namespace): let id = self.historyMetadataTable.getNextMessageIdAndIncrement(peerId, namespace: namespace) internalStoreMessages.append(InternalStoreMessage(id: id, timestamp: message.timestamp, globallyUniqueId: message.globallyUniqueId, flags: message.flags, tags: message.tags, globalTags: message.globalTags, forwardInfo: message.forwardInfo, authorId: message.authorId, text: message.text, attributes: message.attributes, media: message.media)) } } return internalStoreMessages } func addMessages(_ messages: [StoreMessage], location: AddMessagesLocation, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) -> [Int64: MessageId] { let messagesByPeerId = self.messagesGroupedByPeerId(messages) var globallyUniqueIdToMessageId: [Int64: MessageId] = [:] var globalTagsInitialized = Set() for (peerId, peerMessages) in messagesByPeerId { var operations: [MessageHistoryIndexOperation] = [] let internalPeerMessages = self.internalStoreMessages(peerMessages) for message in internalPeerMessages { if let globallyUniqueId = message.globallyUniqueId { globallyUniqueIdToMessageId[globallyUniqueId] = message.id } if !message.globalTags.isEmpty { for tag in message.globalTags { if !globalTagsInitialized.contains(tag) { self.globalTagsTable.ensureInitialized(tag) globalTagsInitialized.insert(tag) } } } } self.messageHistoryIndexTable.addMessages(internalPeerMessages, location: location, operations: &operations) self.processIndexOperations(peerId, operations: operations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } return globallyUniqueIdToMessageId } func addHoles(_ messageIds: [MessageId], operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { for (peerId, messageIds) in self.messageIdsByPeerId(messageIds) { var operations: [MessageHistoryIndexOperation] = [] for id in messageIds { self.messageHistoryIndexTable.addHole(id, operations: &operations) } self.processIndexOperations(peerId, operations: operations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } } func removeMessages(_ messageIds: [MessageId], operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { for (peerId, messageIds) in self.messageIdsByPeerId(messageIds) { var operations: [MessageHistoryIndexOperation] = [] for id in messageIds { self.messageHistoryIndexTable.removeMessage(id, operations: &operations) } self.processIndexOperations(peerId, operations: operations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } } func clearHistory(peerId: PeerId, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { let indices = self.allIndices(peerId) self.removeMessages(indices.map { $0.id }, operationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } func removeAllMessagesWithAuthor(peerId: PeerId, authorId: PeerId, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { let indices = self.allIndicesWithAuthor(peerId, authorId: authorId) self.removeMessages(indices.map { $0.id }, operationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } func fillHole(_ id: MessageId, fillType: HoleFill, tagMask: MessageTags?, messages: [StoreMessage], operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { var operations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.fillHole(id, fillType: fillType, tagMask: tagMask, messages: self.internalStoreMessages(messages), operations: &operations) self.processIndexOperations(id.peerId, operations: operations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } func fillMultipleHoles(mainHoleId: MessageId, fillType: HoleFill, tagMask: MessageTags?, messages: [StoreMessage], operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { var operations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.fillMultipleHoles(mainHoleId: mainHoleId, fillType: fillType, tagMask: tagMask, messages: self.internalStoreMessages(messages), operations: &operations) self.processIndexOperations(mainHoleId.peerId, operations: operations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } func updateMessage(_ id: MessageId, message: StoreMessage, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { var operations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.updateMessage(id, message: self.internalStoreMessages([message]).first!, operations: &operations) self.processIndexOperations(id.peerId, operations: operations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } func updateMessageTimestamp(_ id: MessageId, timestamp: Int32, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { var operations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.updateTimestamp(id, timestamp: timestamp, operations: &operations) self.processIndexOperations(id.peerId, operations: operations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } func updateMedia(_ id: MediaId, media: Media?, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], updatedMedia: inout [MediaId: Media?]) { if let previousMedia = self.messageMediaTable.get(id, embedded: { index, id in return self.embeddedMediaForIndex(index, id: id) }) { if let media = media { if !previousMedia.isEqual(media) { self.messageMediaTable.update(id, media: media, messageHistoryTable: self, operationsByPeerId: &operationsByPeerId) updatedMedia[id] = media } } else { updatedMedia[id] = nil if case let .Embedded(index) = self.messageMediaTable.removeReference(id) { self.updateEmbeddedMedia(index, operationsByPeerId: &operationsByPeerId, update: { previousMedia in var updated: [Media] = [] for previous in previousMedia { if previous.id != id { updated.append(previous) } } return updated }) } } } } func resetIncomingReadStates(_ states: [PeerId: [MessageId.Namespace: PeerReadState]], operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?]) { for (peerId, namespaces) in states { if let combinedState = self.readStateTable.resetStates(peerId, namespaces: namespaces) { if operationsByPeerId[peerId] == nil { operationsByPeerId[peerId] = [.UpdateReadState(combinedState)] } else { operationsByPeerId[peerId]!.append(.UpdateReadState(combinedState)) } } self.synchronizeReadStateTable.set(peerId, operation: nil, operations: &updatedPeerReadStateOperations) } } func applyIncomingReadMaxId(_ messageId: MessageId, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?]) { var topMessageId: (MessageId.Id, Bool)? if let topEntry = self.messageHistoryIndexTable.top(messageId.peerId, namespace: messageId.namespace), case let .Message(index) = topEntry { if let message = self.getMessage(index) { topMessageId = (index.id.id, !message.flags.contains(.Incoming)) } else { topMessageId = (index.id.id, false) } } let (combinedState, invalidated) = self.readStateTable.applyIncomingMaxReadId(messageId, incomingStatsInRange: { namespace, fromId, toId in return self.messageHistoryIndexTable.incomingMessageCountInRange(messageId.peerId, namespace: namespace, minId: fromId, maxId: toId) }, topMessageId: topMessageId) if let combinedState = combinedState { if operationsByPeerId[messageId.peerId] == nil { operationsByPeerId[messageId.peerId] = [.UpdateReadState(combinedState)] } else { operationsByPeerId[messageId.peerId]!.append(.UpdateReadState(combinedState)) } } if invalidated { self.synchronizeReadStateTable.set(messageId.peerId, operation: .Validate, operations: &updatedPeerReadStateOperations) } } func applyOutgoingReadMaxId(_ messageId: MessageId, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?]) { let (combinedState, invalidated) = self.readStateTable.applyOutgoingMaxReadId(messageId) if let combinedState = combinedState { if operationsByPeerId[messageId.peerId] == nil { operationsByPeerId[messageId.peerId] = [.UpdateReadState(combinedState)] } else { operationsByPeerId[messageId.peerId]!.append(.UpdateReadState(combinedState)) } } if invalidated { self.synchronizeReadStateTable.set(messageId.peerId, operation: .Validate, operations: &updatedPeerReadStateOperations) } } func applyOutgoingReadMaxIndex(_ messageIndex: MessageIndex, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?]) -> [MessageId] { let (combinedState, invalidated, messageIds) = self.readStateTable.applyOutgoingMaxReadIndex(messageIndex, outgoingIndexStatsInRange: { fromIndex, toIndex in return self.outgoingMessageCountInRange(messageIndex.id.peerId, namespace: messageIndex.id.namespace, fromIndex: fromIndex, toIndex: toIndex) }) if let combinedState = combinedState { if operationsByPeerId[messageIndex.id.peerId] == nil { operationsByPeerId[messageIndex.id.peerId] = [.UpdateReadState(combinedState)] } else { operationsByPeerId[messageIndex.id.peerId]!.append(.UpdateReadState(combinedState)) } } if invalidated { self.synchronizeReadStateTable.set(messageIndex.id.peerId, operation: .Validate, operations: &updatedPeerReadStateOperations) } return messageIds } func applyInteractiveMaxReadIndex(_ messageIndex: MessageIndex, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?]) -> [MessageId] { var topMessageId: (MessageId.Id, Bool)? if let topEntry = self.messageHistoryIndexTable.top(messageIndex.id.peerId, namespace: messageIndex.id.namespace), case let .Message(index) = topEntry { if let message = self.getMessage(index) { topMessageId = (index.id.id, !message.flags.contains(.Incoming)) } else { topMessageId = (index.id.id, false) } } let (combinedState, result, messageIds) = self.readStateTable.applyInteractiveMaxReadIndex(messageIndex, incomingStatsInRange: { namespace, fromId, toId in return self.messageHistoryIndexTable.incomingMessageCountInRange(messageIndex.id.peerId, namespace: namespace, minId: fromId, maxId: toId) }, incomingIndexStatsInRange: { fromIndex, toIndex in return self.incomingMessageCountInRange(messageIndex.id.peerId, namespace: messageIndex.id.namespace, fromIndex: fromIndex, toIndex: toIndex) }, topMessageId: topMessageId, topMessageIndexByNamespace: { namespace in if let topEntry = self.messageHistoryIndexTable.top(messageIndex.id.peerId, namespace: namespace), case let .Message(index) = topEntry { return index } else { return nil } }) if let combinedState = combinedState { if operationsByPeerId[messageIndex.id.peerId] == nil { operationsByPeerId[messageIndex.id.peerId] = [.UpdateReadState(combinedState)] } else { operationsByPeerId[messageIndex.id.peerId]!.append(.UpdateReadState(combinedState)) } } switch result { case let .Push(thenSync): self.synchronizeReadStateTable.set(messageIndex.id.peerId, operation: .Push(state: self.readStateTable.getCombinedState(messageIndex.id.peerId), thenSync: thenSync), operations: &updatedPeerReadStateOperations) case .None: break } return messageIds } func topMessage(_ peerId: PeerId) -> IntermediateMessage? { var currentKey = self.upperBound(peerId) while true { var entry: IntermediateMessageHistoryEntry? self.valueBox.range(self.table, start: currentKey, end: self.lowerBound(peerId), values: { key, value in entry = self.readIntermediateEntry(key, value: value) return true }, limit: 1) if let entry = entry { switch entry { case .Hole: currentKey = self.key(entry.index).predecessor case let .Message(message): return message } } else { break } } return nil } func getMessage(_ index: MessageIndex) -> IntermediateMessage? { let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { let entry = self.readIntermediateEntry(key, value: value) if case let .Message(message) = entry { return message } } else if let tableIndex = self.messageHistoryIndexTable.get(index.id) { if case let .Message(updatedIndex) = tableIndex { let key = self.key(updatedIndex) if let value = self.valueBox.get(self.table, key: key) { let entry = self.readIntermediateEntry(key, value: value) if case let .Message(message) = entry { return message } } } } return nil } func offsetPendingMessagesTimestamps(lowerBound: MessageId, timestamp: Int32, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) { for messageId in self.unsentTable.get() { if messageId.peerId == lowerBound.peerId && messageId.namespace == lowerBound.namespace && messageId.id > lowerBound.id { self.updateMessageTimestamp(messageId, timestamp: timestamp, operationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } } } private func justInsertMessage(_ message: InternalStoreMessage, sharedKey: ValueBoxKey, sharedBuffer: WriteBuffer, sharedEncoder: Encoder) -> IntermediateMessage { sharedBuffer.reset() var type: Int8 = 0 sharedBuffer.write(&type, offset: 0, length: 1) var stableId: UInt32 = self.historyMetadataTable.getNextStableMessageIndexId() sharedBuffer.write(&stableId, offset: 0, length: 4) var stableVersion: UInt32 = 0 sharedBuffer.write(&stableVersion, offset: 0, length: 4) var dataFlags: MessageDataFlags = [] if message.globallyUniqueId != nil { dataFlags.insert(.hasGloballyUniqueId) } if !message.globalTags.isEmpty { dataFlags.insert(.hasGlobalTags) } sharedBuffer.write(&dataFlags, offset: 0, length: 1) if let globallyUniqueId = message.globallyUniqueId { var globallyUniqueIdValue = globallyUniqueId sharedBuffer.write(&globallyUniqueIdValue, offset: 0, length: 8) self.globallyUniqueMessageIdsTable.set(peerId: message.id.peerId, globallyUniqueId: globallyUniqueId, id: message.id) } if !message.globalTags.isEmpty { var globalTagsValue: UInt32 = message.globalTags.rawValue sharedBuffer.write(&globalTagsValue, offset: 0, length: 4) } if self.messageHistoryIndexTable.seedConfiguration.peerNamespacesRequiringMessageTextIndex.contains(message.id.peerId.namespace) { self.textIndexTable.add(messageId: message.id, text: message.text, tags: message.tags) } var flags = MessageFlags(message.flags) sharedBuffer.write(&flags.rawValue, offset: 0, length: 4) var tags = message.tags sharedBuffer.write(&tags.rawValue, offset: 0, length: 4) var intermediateForwardInfo: IntermediateMessageForwardInfo? if let forwardInfo = message.forwardInfo { intermediateForwardInfo = IntermediateMessageForwardInfo(forwardInfo) var forwardInfoFlags: Int8 = 1 if forwardInfo.sourceId != nil { forwardInfoFlags |= 2 } if forwardInfo.sourceMessageId != nil { forwardInfoFlags |= 4 } if forwardInfo.authorSignature != nil { forwardInfoFlags |= 8 } sharedBuffer.write(&forwardInfoFlags, offset: 0, length: 1) var forwardAuthorId: Int64 = forwardInfo.authorId.toInt64() var forwardDate: Int32 = forwardInfo.date sharedBuffer.write(&forwardAuthorId, offset: 0, length: 8) sharedBuffer.write(&forwardDate, offset: 0, length: 4) if let sourceId = forwardInfo.sourceId { var sourceIdValue: Int64 = sourceId.toInt64() sharedBuffer.write(&sourceIdValue, offset: 0, length: 8) } if let sourceMessageId = forwardInfo.sourceMessageId { var sourceMessageIdPeerId: Int64 = sourceMessageId.peerId.toInt64() var sourceMessageIdNamespace: Int32 = sourceMessageId.namespace var sourceMessageIdId: Int32 = sourceMessageId.id sharedBuffer.write(&sourceMessageIdPeerId, offset: 0, length: 8) sharedBuffer.write(&sourceMessageIdNamespace, offset: 0, length: 4) sharedBuffer.write(&sourceMessageIdId, offset: 0, length: 4) } if let authorSignature = forwardInfo.authorSignature { if let data = authorSignature.data(using: .utf8, allowLossyConversion: true) { var length: Int32 = Int32(data.count) sharedBuffer.write(&length, offset: 0, length: 4) sharedBuffer.write(data) } else { var length: Int32 = 0 sharedBuffer.write(&length, offset: 0, length: 4) } } } else { var forwardInfoFlags: Int8 = 0 sharedBuffer.write(&forwardInfoFlags, offset: 0, length: 1) } if let authorId = message.authorId { var varAuthorId: Int64 = authorId.toInt64() var hasAuthor: Int8 = 1 sharedBuffer.write(&hasAuthor, offset: 0, length: 1) sharedBuffer.write(&varAuthorId, offset: 0, length: 8) } else { var hasAuthor: Int8 = 0 sharedBuffer.write(&hasAuthor, offset: 0, length: 1) } if let data = message.text.data(using: .utf8, allowLossyConversion: true) { var length: Int32 = Int32(data.count) sharedBuffer.write(&length, offset: 0, length: 4) sharedBuffer.write(data) } else { var length: Int32 = 0 sharedBuffer.write(&length, offset: 0, length: 4) } let attributesBuffer = WriteBuffer() var attributeCount: Int32 = Int32(message.attributes.count) attributesBuffer.write(&attributeCount, offset: 0, length: 4) for attribute in message.attributes { sharedEncoder.reset() sharedEncoder.encodeRootObject(attribute) let attributeBuffer = sharedEncoder.memoryBuffer() var attributeBufferLength = Int32(attributeBuffer.length) attributesBuffer.write(&attributeBufferLength, offset: 0, length: 4) attributesBuffer.write(attributeBuffer.memory, offset: 0, length: attributeBuffer.length) } sharedBuffer.write(attributesBuffer.memory, offset: 0, length: attributesBuffer.length) var embeddedMedia: [Media] = [] var referencedMedia: [MediaId] = [] for media in message.media { if let mediaId = media.id { let mediaInsertResult = self.messageMediaTable.set(media, index: MessageIndex(message), messageHistoryTable: self) switch mediaInsertResult { case let .Embed(media): embeddedMedia.append(media) case .Reference: referencedMedia.append(mediaId) } } else { embeddedMedia.append(media) } } let embeddedMediaBuffer = WriteBuffer() var embeddedMediaCount: Int32 = Int32(embeddedMedia.count) embeddedMediaBuffer.write(&embeddedMediaCount, offset: 0, length: 4) for media in embeddedMedia { sharedEncoder.reset() sharedEncoder.encodeRootObject(media) let mediaBuffer = sharedEncoder.memoryBuffer() var mediaBufferLength = Int32(mediaBuffer.length) embeddedMediaBuffer.write(&mediaBufferLength, offset: 0, length: 4) embeddedMediaBuffer.write(mediaBuffer.memory, offset: 0, length: mediaBuffer.length) } sharedBuffer.write(embeddedMediaBuffer.memory, offset: 0, length: embeddedMediaBuffer.length) var referencedMediaCount: Int32 = Int32(referencedMedia.count) sharedBuffer.write(&referencedMediaCount, offset: 0, length: 4) for mediaId in referencedMedia { var idNamespace: Int32 = mediaId.namespace var idId: Int64 = mediaId.id sharedBuffer.write(&idNamespace, offset: 0, length: 4) sharedBuffer.write(&idId, offset: 0, length: 8) } self.valueBox.set(self.table, key: self.key(MessageIndex(message), key: sharedKey), value: sharedBuffer) return IntermediateMessage(stableId: stableId, stableVersion: stableVersion, id: message.id, globallyUniqueId: message.globallyUniqueId, timestamp: message.timestamp, flags: flags, tags: message.tags, globalTags: message.globalTags, forwardInfo: intermediateForwardInfo, authorId: message.authorId, text: message.text, attributesData: attributesBuffer.makeReadBufferAndReset(), embeddedMediaData: embeddedMediaBuffer.makeReadBufferAndReset(), referencedMedia: referencedMedia) } private func justInsertHole(_ hole: MessageHistoryHole, sharedBuffer: WriteBuffer = WriteBuffer()) { sharedBuffer.reset() var type: Int8 = 1 sharedBuffer.write(&type, offset: 0, length: 1) var stableId: UInt32 = hole.stableId sharedBuffer.write(&stableId, offset: 0, length: 4) var minId: Int32 = hole.min sharedBuffer.write(&minId, offset: 0, length: 4) var tags: UInt32 = hole.tags sharedBuffer.write(&tags, offset: 0, length: 4) withExtendedLifetime(sharedBuffer, { self.valueBox.set(self.table, key: self.key(hole.maxIndex), value: sharedBuffer.readBufferNoCopy()) }) } private func tagsForIndex(_ index: MessageIndex) -> (MessageTags, GlobalMessageTags)? { let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { switch self.readIntermediateEntry(key, value: value) { case let .Message(message): return (message.tags, message.globalTags) case let .Hole(hole): return (MessageTags(rawValue: hole.tags), GlobalMessageTags()) } } else { return nil } } private func justRemove(_ index: MessageIndex, unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation]) -> (MessageTags, GlobalMessageTags)? { let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { let resultTags: MessageTags let resultGlobalTags: GlobalMessageTags switch self.readIntermediateEntry(key, value: value) { case let .Message(message): let embeddedMediaData = message.embeddedMediaData if embeddedMediaData.length > 4 { var embeddedMediaCount: Int32 = 0 embeddedMediaData.read(&embeddedMediaCount, offset: 0, length: 4) for _ in 0 ..< embeddedMediaCount { var mediaLength: Int32 = 0 embeddedMediaData.read(&mediaLength, offset: 0, length: 4) if let media = Decoder(buffer: MemoryBuffer(memory: embeddedMediaData.memory + embeddedMediaData.offset, capacity: Int(mediaLength), length: Int(mediaLength), freeWhenDone: false)).decodeRootObject() as? Media { self.messageMediaTable.removeEmbeddedMedia(media) } embeddedMediaData.skip(Int(mediaLength)) } } if message.flags.contains(.Unsent) && !message.flags.contains(.Failed) { self.unsentTable.remove(index.id, operations: &unsentMessageOperations) } if let globallyUniqueId = message.globallyUniqueId { self.globallyUniqueMessageIdsTable.remove(peerId: message.id.peerId, globallyUniqueId: globallyUniqueId) } let tags = message.tags.rawValue if tags != 0 { for i in 0 ..< 32 { let currentTags = tags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = MessageTags(rawValue: 1 << UInt32(i)) self.tagsTable.remove(tag, index: index) } } } let globalTags = message.globalTags.rawValue if globalTags != 0 { for i in 0 ..< 32 { let currentTags = globalTags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = GlobalMessageTags(rawValue: 1 << UInt32(i)) self.globalTagsTable.remove(tag, index: index) } } } for mediaId in message.referencedMedia { let _ = self.messageMediaTable.removeReference(mediaId) } if self.messageHistoryIndexTable.seedConfiguration.peerNamespacesRequiringMessageTextIndex.contains(message.id.peerId.namespace) { self.textIndexTable.remove(messageId: message.id) } resultTags = message.tags resultGlobalTags = message.globalTags case let .Hole(hole): let tags = self.messageHistoryIndexTable.seedConfiguration.existingMessageTags.rawValue & hole.tags if tags != 0 { for i in 0 ..< 32 { let currentTags = tags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = MessageTags(rawValue: 1 << UInt32(i)) self.tagsTable.remove(tag, index: hole.maxIndex) } } } resultTags = MessageTags(rawValue: hole.tags) resultGlobalTags = GlobalMessageTags() } self.valueBox.remove(self.table, key: key) return (resultTags, resultGlobalTags) } else { return nil } } func embeddedMediaForIndex(_ index: MessageIndex, id: MediaId) -> Media? { if let message = self.getMessage(index), message.embeddedMediaData.length > 4 { var embeddedMediaCount: Int32 = 0 message.embeddedMediaData.read(&embeddedMediaCount, offset: 0, length: 4) for _ in 0 ..< embeddedMediaCount { var mediaLength: Int32 = 0 message.embeddedMediaData.read(&mediaLength, offset: 0, length: 4) if let readMedia = Decoder(buffer: MemoryBuffer(memory: message.embeddedMediaData.memory + message.embeddedMediaData.offset, capacity: Int(mediaLength), length: Int(mediaLength), freeWhenDone: false)).decodeRootObject() as? Media { if let readMediaId = readMedia.id, readMediaId == id { return readMedia } } message.embeddedMediaData.skip(Int(mediaLength)) } } return nil } func updateEmbeddedMedia(_ index: MessageIndex, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], update: ([Media]) -> [Media]) { if let message = self.getMessage(index) { var embeddedMediaCount: Int32 = 0 message.embeddedMediaData.read(&embeddedMediaCount, offset: 0, length: 4) var previousMedia: [Media] = [] for _ in 0 ..< embeddedMediaCount { var mediaLength: Int32 = 0 message.embeddedMediaData.read(&mediaLength, offset: 0, length: 4) if let readMedia = Decoder(buffer: MemoryBuffer(memory: message.embeddedMediaData.memory + message.embeddedMediaData.offset, capacity: Int(mediaLength), length: Int(mediaLength), freeWhenDone: false)).decodeRootObject() as? Media { previousMedia.append(readMedia) } message.embeddedMediaData.skip(Int(mediaLength)) } let updatedMedia = update(previousMedia) var updated = false if updatedMedia.count != previousMedia.count { updated = true } else { outer: for i in 0 ..< previousMedia.count { if !previousMedia[i].isEqual(updatedMedia[i]) { updated = true break outer } } } if updated { var updatedEmbeddedMediaCount: Int32 = Int32(updatedMedia.count) let updatedEmbeddedMediaBuffer = WriteBuffer() updatedEmbeddedMediaBuffer.write(&updatedEmbeddedMediaCount, offset: 0, length: 4) let encoder = Encoder() for media in updatedMedia { encoder.reset() encoder.encodeRootObject(media) withExtendedLifetime(encoder, { let encodedBuffer = encoder.readBufferNoCopy() var encodedLength: Int32 = Int32(encodedBuffer.length) updatedEmbeddedMediaBuffer.write(&encodedLength, offset: 0, length: 4) updatedEmbeddedMediaBuffer.write(encodedBuffer.memory, offset: 0, length: encodedBuffer.length) }) } withExtendedLifetime(updatedEmbeddedMediaBuffer, { self.storeIntermediateMessage(IntermediateMessage(stableId: message.stableId, stableVersion: message.stableVersion, id: message.id, globallyUniqueId: message.globallyUniqueId, timestamp: message.timestamp, flags: message.flags, tags: message.tags, globalTags: message.globalTags, forwardInfo: message.forwardInfo, authorId: message.authorId, text: message.text, attributesData: message.attributesData, embeddedMediaData: updatedEmbeddedMediaBuffer.readBufferNoCopy(), referencedMedia: message.referencedMedia), sharedKey: self.key(index)) }) let operation: MessageHistoryOperation = .UpdateEmbeddedMedia(index, updatedEmbeddedMediaBuffer.makeReadBufferAndReset()) if operationsByPeerId[index.id.peerId] == nil { operationsByPeerId[index.id.peerId] = [operation] } else { operationsByPeerId[index.id.peerId]!.append(operation) } } } } func updateEmbeddedMedia(_ index: MessageIndex, mediaId: MediaId, media: Media?, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]]) { self.updateEmbeddedMedia(index, operationsByPeerId: &operationsByPeerId, update: { previousMedia in var updatedMedia: [Media] = [] for previous in previousMedia { if previous.id == mediaId { if let media = media { updatedMedia.append(media) } } else { updatedMedia.append(previous) } } return updatedMedia }) } private func justUpdate(_ index: MessageIndex, message: InternalStoreMessage, sharedKey: ValueBoxKey, sharedBuffer: WriteBuffer, sharedEncoder: Encoder, unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation]) -> (IntermediateMessage, MessageTags)? { if let previousMessage = self.getMessage(index) { var previousEmbeddedMediaWithIds: [(MediaId, Media)] = [] if previousMessage.embeddedMediaData.length > 4 { var embeddedMediaCount: Int32 = 0 let previousEmbeddedMediaData = previousMessage.embeddedMediaData previousEmbeddedMediaData.read(&embeddedMediaCount, offset: 0, length: 4) for _ in 0 ..< embeddedMediaCount { var mediaLength: Int32 = 0 previousEmbeddedMediaData.read(&mediaLength, offset: 0, length: 4) if let media = Decoder(buffer: MemoryBuffer(memory: previousEmbeddedMediaData.memory + previousEmbeddedMediaData.offset, capacity: Int(mediaLength), length: Int(mediaLength), freeWhenDone: false)).decodeRootObject() as? Media { if let mediaId = media.id { previousEmbeddedMediaWithIds.append((mediaId, media)) } } previousEmbeddedMediaData.skip(Int(mediaLength)) } } var previousMediaIds = Set() for (mediaId, _) in previousEmbeddedMediaWithIds { previousMediaIds.insert(mediaId) } for mediaId in previousMessage.referencedMedia { previousMediaIds.insert(mediaId) } var updatedMediaIds = Set() for media in message.media { if let mediaId = media.id { updatedMediaIds.insert(mediaId) } } if previousMediaIds != updatedMediaIds || index != MessageIndex(message) { for (_, media) in previousEmbeddedMediaWithIds { self.messageMediaTable.removeEmbeddedMedia(media) } for mediaId in previousMessage.referencedMedia { let _ = self.messageMediaTable.removeReference(mediaId) } } self.valueBox.remove(self.table, key: self.key(index)) if !previousMessage.tags.isEmpty { self.tagsTable.remove(previousMessage.tags, index: index) } if !message.tags.isEmpty { self.tagsTable.add(message.tags, index: MessageIndex(message)) } if previousMessage.globalTags != message.globalTags || (MessageIndex(previousMessage) != index && (!previousMessage.globalTags.isEmpty || !message.globalTags.isEmpty)) { if !previousMessage.globalTags.isEmpty { for tag in previousMessage.globalTags { self.globalTagsTable.remove(tag, index: index) } } if !message.globalTags.isEmpty { for tag in message.globalTags { let _ = self.globalTagsTable.addMessage(tag, index: MessageIndex(message)) } } } if message.globallyUniqueId != previousMessage.globallyUniqueId { if let globallyUniqueId = previousMessage.globallyUniqueId { self.globallyUniqueMessageIdsTable.remove(peerId: message.id.peerId, globallyUniqueId: globallyUniqueId) } if let globallyUniqueId = message.globallyUniqueId { self.globallyUniqueMessageIdsTable.set(peerId: message.id.peerId, globallyUniqueId: globallyUniqueId, id: message.id) } } else if let globallyUniqueId = message.globallyUniqueId, previousMessage.id != message.id { self.globallyUniqueMessageIdsTable.set(peerId: message.id.peerId, globallyUniqueId: globallyUniqueId, id: message.id) } if !message.globalTags.isEmpty || !previousMessage.globalTags.isEmpty { assertionFailure("implement global tags") if index != MessageIndex(message) { } else if message.globalTags != previousMessage.globalTags { } } switch (previousMessage.flags.contains(.Unsent) && !previousMessage.flags.contains(.Failed), message.flags.contains(.Unsent) && !message.flags.contains(.Failed)) { case (true, false): self.unsentTable.remove(index.id, operations: &unsentMessageOperations) case (false, true): self.unsentTable.add(message.id, operations: &unsentMessageOperations) case (true, true): if index != MessageIndex(message) { self.unsentTable.remove(index.id, operations: &unsentMessageOperations) self.unsentTable.add(message.id, operations: &unsentMessageOperations) } case (false, false): break } if self.messageHistoryIndexTable.seedConfiguration.peerNamespacesRequiringMessageTextIndex.contains(message.id.peerId.namespace) { if previousMessage.id != message.id || previousMessage.text != message.text || previousMessage.tags != message.tags { self.textIndexTable.remove(messageId: previousMessage.id) self.textIndexTable.add(messageId: message.id, text: message.text, tags: message.tags) } } sharedBuffer.reset() var type: Int8 = 0 sharedBuffer.write(&type, offset: 0, length: 1) var stableId: UInt32 = previousMessage.stableId sharedBuffer.write(&stableId, offset: 0, length: 4) var stableVersion: UInt32 = previousMessage.stableVersion + 1 sharedBuffer.write(&stableVersion, offset: 0, length: 4) let globallyUniqueId: Int64? = previousMessage.globallyUniqueId var hasGloballyUniqueId: Int8 = globallyUniqueId != nil ? 1 : 0 sharedBuffer.write(&hasGloballyUniqueId, offset: 0, length: 1) if let globallyUniqueId = globallyUniqueId { var globallyUniqueIdValue = globallyUniqueId sharedBuffer.write(&globallyUniqueIdValue, offset: 0, length: 8) } var flags = MessageFlags(message.flags) sharedBuffer.write(&flags.rawValue, offset: 0, length: 4) var tags = message.tags sharedBuffer.write(&tags.rawValue, offset: 0, length: 4) var intermediateForwardInfo: IntermediateMessageForwardInfo? if let forwardInfo = message.forwardInfo { intermediateForwardInfo = IntermediateMessageForwardInfo(forwardInfo) var forwardInfoFlags: Int8 = 1 if forwardInfo.sourceId != nil { forwardInfoFlags |= 2 } if forwardInfo.sourceMessageId != nil { forwardInfoFlags |= 4 } if forwardInfo.authorSignature != nil { forwardInfoFlags |= 8 } sharedBuffer.write(&forwardInfoFlags, offset: 0, length: 1) var forwardAuthorId: Int64 = forwardInfo.authorId.toInt64() var forwardDate: Int32 = forwardInfo.date sharedBuffer.write(&forwardAuthorId, offset: 0, length: 8) sharedBuffer.write(&forwardDate, offset: 0, length: 4) if let sourceId = forwardInfo.sourceId { var sourceIdValue: Int64 = sourceId.toInt64() sharedBuffer.write(&sourceIdValue, offset: 0, length: 8) } if let sourceMessageId = forwardInfo.sourceMessageId { var sourceMessageIdPeerId: Int64 = sourceMessageId.peerId.toInt64() var sourceMessageIdNamespace: Int32 = sourceMessageId.namespace var sourceMessageIdId: Int32 = sourceMessageId.id sharedBuffer.write(&sourceMessageIdPeerId, offset: 0, length: 8) sharedBuffer.write(&sourceMessageIdNamespace, offset: 0, length: 4) sharedBuffer.write(&sourceMessageIdId, offset: 0, length: 4) } if let authorSignature = forwardInfo.authorSignature { if let data = authorSignature.data(using: .utf8, allowLossyConversion: true) { var length: Int32 = Int32(data.count) sharedBuffer.write(&length, offset: 0, length: 4) sharedBuffer.write(data) } else { var length: Int32 = 0 sharedBuffer.write(&length, offset: 0, length: 4) } } } else { var forwardInfoFlags: Int8 = 0 sharedBuffer.write(&forwardInfoFlags, offset: 0, length: 1) } if let authorId = message.authorId { var varAuthorId: Int64 = authorId.toInt64() var hasAuthor: Int8 = 1 sharedBuffer.write(&hasAuthor, offset: 0, length: 1) sharedBuffer.write(&varAuthorId, offset: 0, length: 8) } else { var hasAuthor: Int8 = 0 sharedBuffer.write(&hasAuthor, offset: 0, length: 1) } let data = message.text.data(using: .utf8, allowLossyConversion: true)! var length: Int32 = Int32(data.count) sharedBuffer.write(&length, offset: 0, length: 4) sharedBuffer.write(data) let attributesBuffer = WriteBuffer() var attributeCount: Int32 = Int32(message.attributes.count) attributesBuffer.write(&attributeCount, offset: 0, length: 4) for attribute in message.attributes { sharedEncoder.reset() sharedEncoder.encodeRootObject(attribute) let attributeBuffer = sharedEncoder.memoryBuffer() var attributeBufferLength = Int32(attributeBuffer.length) attributesBuffer.write(&attributeBufferLength, offset: 0, length: 4) attributesBuffer.write(attributeBuffer.memory, offset: 0, length: attributeBuffer.length) } sharedBuffer.write(attributesBuffer.memory, offset: 0, length: attributesBuffer.length) var embeddedMedia: [Media] = [] var referencedMedia: [MediaId] = [] for media in message.media { if let mediaId = media.id { let mediaInsertResult = self.messageMediaTable.set(media, index: MessageIndex(message), messageHistoryTable: self) switch mediaInsertResult { case let .Embed(media): embeddedMedia.append(media) case .Reference: referencedMedia.append(mediaId) } } else { embeddedMedia.append(media) } } let embeddedMediaBuffer = WriteBuffer() var embeddedMediaCount: Int32 = Int32(embeddedMedia.count) embeddedMediaBuffer.write(&embeddedMediaCount, offset: 0, length: 4) for media in embeddedMedia { sharedEncoder.reset() sharedEncoder.encodeRootObject(media) let mediaBuffer = sharedEncoder.memoryBuffer() var mediaBufferLength = Int32(mediaBuffer.length) embeddedMediaBuffer.write(&mediaBufferLength, offset: 0, length: 4) embeddedMediaBuffer.write(mediaBuffer.memory, offset: 0, length: mediaBuffer.length) } sharedBuffer.write(embeddedMediaBuffer.memory, offset: 0, length: embeddedMediaBuffer.length) var referencedMediaCount: Int32 = Int32(referencedMedia.count) sharedBuffer.write(&referencedMediaCount, offset: 0, length: 4) for mediaId in referencedMedia { var idNamespace: Int32 = mediaId.namespace var idId: Int64 = mediaId.id sharedBuffer.write(&idNamespace, offset: 0, length: 4) sharedBuffer.write(&idId, offset: 0, length: 8) } self.valueBox.set(self.table, key: self.key(MessageIndex(message), key: sharedKey), value: sharedBuffer) return (IntermediateMessage(stableId: stableId, stableVersion: stableVersion, id: message.id, globallyUniqueId: message.globallyUniqueId, timestamp: message.timestamp, flags: flags, tags: tags, globalTags: message.globalTags, forwardInfo: intermediateForwardInfo, authorId: message.authorId, text: message.text, attributesData: attributesBuffer.makeReadBufferAndReset(), embeddedMediaData: embeddedMediaBuffer.makeReadBufferAndReset(), referencedMedia: referencedMedia), previousMessage.tags) } else { return nil } } private func justUpdateTimestamp(_ index: MessageIndex, timestamp: Int32) -> (MessageTags, GlobalMessageTags)? { if let previousMessage = self.getMessage(index) { self.valueBox.remove(self.table, key: self.key(index)) let updatedMessage = IntermediateMessage(stableId: previousMessage.stableId, stableVersion: previousMessage.stableVersion + 1, id: previousMessage.id, globallyUniqueId: previousMessage.globallyUniqueId, timestamp: timestamp, flags: previousMessage.flags, tags: previousMessage.tags, globalTags: previousMessage.globalTags, forwardInfo: previousMessage.forwardInfo, authorId: previousMessage.authorId, text: previousMessage.text, attributesData: previousMessage.attributesData, embeddedMediaData: previousMessage.embeddedMediaData, referencedMedia: previousMessage.referencedMedia) self.storeIntermediateMessage(updatedMessage, sharedKey: self.key(index)) let tags = previousMessage.tags.rawValue if tags != 0 { for i in 0 ..< 32 { let currentTags = tags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = MessageTags(rawValue: 1 << UInt32(i)) self.tagsTable.remove(tag, index: index) self.tagsTable.add(tag, index: MessageIndex(id: index.id, timestamp: timestamp)) } } } let globalTags = previousMessage.globalTags.rawValue if globalTags != 0 { for i in 0 ..< 32 { let currentTags = globalTags >> UInt32(i) if currentTags == 0 { break } if (currentTags & 1) != 0 { let tag = GlobalMessageTags(rawValue: 1 << UInt32(i)) self.globalTagsTable.remove(tag, index: index) let _ = self.globalTagsTable.addMessage(tag, index: MessageIndex(id: index.id, timestamp: timestamp)) } } } return (previousMessage.tags, previousMessage.globalTags) } else { return nil } } func unembedMedia(_ index: MessageIndex, id: MediaId) -> Media? { if let message = self.getMessage(index), message.embeddedMediaData.length > 4 { var embeddedMediaCount: Int32 = 0 message.embeddedMediaData.read(&embeddedMediaCount, offset: 0, length: 4) let updatedEmbeddedMediaBuffer = WriteBuffer() var updatedEmbeddedMediaCount = embeddedMediaCount - 1 updatedEmbeddedMediaBuffer.write(&updatedEmbeddedMediaCount, offset: 0, length: 4) var extractedMedia: Media? for _ in 0 ..< embeddedMediaCount { let mediaOffset = message.embeddedMediaData.offset var mediaLength: Int32 = 0 var copyMedia = true message.embeddedMediaData.read(&mediaLength, offset: 0, length: 4) if let media = Decoder(buffer: MemoryBuffer(memory: message.embeddedMediaData.memory + message.embeddedMediaData.offset, capacity: Int(mediaLength), length: Int(mediaLength), freeWhenDone: false)).decodeRootObject() as? Media { if let mediaId = media.id, mediaId == id { copyMedia = false extractedMedia = media } } if copyMedia { updatedEmbeddedMediaBuffer.write(message.embeddedMediaData.memory.advanced(by: mediaOffset), offset: 0, length: message.embeddedMediaData.offset - mediaOffset) } } if let extractedMedia = extractedMedia { var updatedReferencedMedia = message.referencedMedia updatedReferencedMedia.append(extractedMedia.id!) withExtendedLifetime(updatedEmbeddedMediaBuffer, { self.storeIntermediateMessage(IntermediateMessage(stableId: message.stableId, stableVersion: message.stableVersion, id: message.id, globallyUniqueId: message.globallyUniqueId, timestamp: message.timestamp, flags: message.flags, tags: message.tags, globalTags: message.globalTags, forwardInfo: message.forwardInfo, authorId: message.authorId, text: message.text, attributesData: message.attributesData, embeddedMediaData: updatedEmbeddedMediaBuffer.readBufferNoCopy(), referencedMedia: updatedReferencedMedia), sharedKey: self.key(index)) }) return extractedMedia } } return nil } func storeIntermediateMessage(_ message: IntermediateMessage, sharedKey: ValueBoxKey, sharedBuffer: WriteBuffer = WriteBuffer()) { sharedBuffer.reset() var type: Int8 = 0 sharedBuffer.write(&type, offset: 0, length: 1) var stableId: UInt32 = message.stableId sharedBuffer.write(&stableId, offset: 0, length: 4) var stableVersion: UInt32 = message.stableVersion sharedBuffer.write(&stableVersion, offset: 0, length: 4) var hasGloballyUniqueId: Int8 = message.globallyUniqueId != nil ? 1 : 0 sharedBuffer.write(&hasGloballyUniqueId, offset: 0, length: 1) if let globallyUniqueId = message.globallyUniqueId { var globallyUniqueIdValue = globallyUniqueId sharedBuffer.write(&globallyUniqueIdValue, offset: 0, length: 8) } var flagsValue: UInt32 = message.flags.rawValue sharedBuffer.write(&flagsValue, offset: 0, length: 4) var tagsValue: UInt32 = message.tags.rawValue sharedBuffer.write(&tagsValue, offset: 0, length: 4) if let forwardInfo = message.forwardInfo { var forwardInfoFlags: Int8 = 1 if forwardInfo.sourceId != nil { forwardInfoFlags |= 2 } if forwardInfo.sourceMessageId != nil { forwardInfoFlags |= 4 } if forwardInfo.authorSignature != nil { forwardInfoFlags |= 8 } sharedBuffer.write(&forwardInfoFlags, offset: 0, length: 1) var forwardAuthorId: Int64 = forwardInfo.authorId.toInt64() var forwardDate: Int32 = forwardInfo.date sharedBuffer.write(&forwardAuthorId, offset: 0, length: 8) sharedBuffer.write(&forwardDate, offset: 0, length: 4) if let sourceId = forwardInfo.sourceId { var sourceIdValue: Int64 = sourceId.toInt64() sharedBuffer.write(&sourceIdValue, offset: 0, length: 8) } if let sourceMessageId = forwardInfo.sourceMessageId { var sourceMessageIdPeerId: Int64 = sourceMessageId.peerId.toInt64() var sourceMessageIdNamespace: Int32 = sourceMessageId.namespace var sourceMessageIdId: Int32 = sourceMessageId.id sharedBuffer.write(&sourceMessageIdPeerId, offset: 0, length: 8) sharedBuffer.write(&sourceMessageIdNamespace, offset: 0, length: 4) sharedBuffer.write(&sourceMessageIdId, offset: 0, length: 4) } if let authorSignature = forwardInfo.authorSignature { if let data = authorSignature.data(using: .utf8, allowLossyConversion: true) { var length: Int32 = Int32(data.count) sharedBuffer.write(&length, offset: 0, length: 4) sharedBuffer.write(data) } else { var length: Int32 = 0 sharedBuffer.write(&length, offset: 0, length: 4) } } } else { var forwardInfoFlags: Int8 = 0 sharedBuffer.write(&forwardInfoFlags, offset: 0, length: 1) } if let authorId = message.authorId { var varAuthorId: Int64 = authorId.toInt64() var hasAuthor: Int8 = 1 sharedBuffer.write(&hasAuthor, offset: 0, length: 1) sharedBuffer.write(&varAuthorId, offset: 0, length: 8) } else { var hasAuthor: Int8 = 0 sharedBuffer.write(&hasAuthor, offset: 0, length: 1) } let data = message.text.data(using: .utf8, allowLossyConversion: true)! var length: Int32 = Int32(data.count) sharedBuffer.write(&length, offset: 0, length: 4) sharedBuffer.write(data) sharedBuffer.write(message.attributesData.memory, offset: 0, length: message.attributesData.length) sharedBuffer.write(message.embeddedMediaData.memory, offset: 0, length: message.embeddedMediaData.length) var referencedMediaCount: Int32 = Int32(message.referencedMedia.count) sharedBuffer.write(&referencedMediaCount, offset: 0, length: 4) for mediaId in message.referencedMedia { var idNamespace: Int32 = mediaId.namespace var idId: Int64 = mediaId.id sharedBuffer.write(&idNamespace, offset: 0, length: 4) sharedBuffer.write(&idId, offset: 0, length: 8) } self.valueBox.set(self.table, key: self.key(MessageIndex(id: message.id, timestamp: message.timestamp), key: sharedKey), value: sharedBuffer) } private func extractIntermediateEntryAuthor(value: ReadBuffer) -> PeerId? { var type: Int8 = 0 value.read(&type, offset: 0, length: 1) if type == 0 { value.skip(4) // stableId value.skip(4) // stableVersion var hasGloballyUniqueId: Int8 = 0 value.read(&hasGloballyUniqueId, offset: 0, length: 1) if hasGloballyUniqueId != 0 { value.skip(8) // globallyUniqueId } value.skip(4) // flags value.skip(4) // tags var forwardInfoFlags: Int8 = 0 value.read(&forwardInfoFlags, offset: 0, length: 1) if forwardInfoFlags != 0 { value.skip(8) // forwardAuthorId value.skip(4) // forwardDate if (forwardInfoFlags & 2) != 0 { value.skip(8) // forwardSourceIdValue } if (forwardInfoFlags & 4) != 0 { value.skip(8) // forwardSourceMessagePeerId value.skip(4) // forwardSourceMessageNamespace value.skip(4) // forwardSourceMessageIdId } } var hasAuthor: Int8 = 0 value.read(&hasAuthor, offset: 0, length: 1) if hasAuthor == 1 { var varAuthorId: Int64 = 0 value.read(&varAuthorId, offset: 0, length: 8) return PeerId(varAuthorId) } } return nil } private func readIntermediateEntry(_ key: ValueBoxKey, value: ReadBuffer) -> IntermediateMessageHistoryEntry { let index = MessageIndex(id: MessageId(peerId: PeerId(key.getInt64(0)), namespace: key.getInt32(8 + 4), id: key.getInt32(8 + 4 + 4)), timestamp: key.getInt32(8)) var type: Int8 = 0 value.read(&type, offset: 0, length: 1) if type == 0 { var stableId: UInt32 = 0 value.read(&stableId, offset: 0, length: 4) var stableVersion: UInt32 = 0 value.read(&stableVersion, offset: 0, length: 4) var dataFlagsValue: Int8 = 0 value.read(&dataFlagsValue, offset: 0, length: 1) let dataFlags = MessageDataFlags(rawValue: dataFlagsValue) var globallyUniqueId: Int64? if dataFlags.contains(.hasGloballyUniqueId) { var globallyUniqueIdValue: Int64 = 0 value.read(&globallyUniqueIdValue, offset: 0, length: 8) globallyUniqueId = globallyUniqueIdValue } var globalTags: GlobalMessageTags = [] if dataFlags.contains(.hasGlobalTags) { var globalTagsValue: UInt32 = 0 value.read(&globalTagsValue, offset: 0, length: 4) globalTags = GlobalMessageTags(rawValue: globalTagsValue) } var flagsValue: UInt32 = 0 value.read(&flagsValue, offset: 0, length: 4) let flags = MessageFlags(rawValue: flagsValue) var tagsValue: UInt32 = 0 value.read(&tagsValue, offset: 0, length: 4) let tags = MessageTags(rawValue: tagsValue) var forwardInfoFlags: Int8 = 0 value.read(&forwardInfoFlags, offset: 0, length: 1) var forwardInfo: IntermediateMessageForwardInfo? if forwardInfoFlags != 0 { var forwardAuthorId: Int64 = 0 var forwardDate: Int32 = 0 var forwardSourceId: PeerId? var forwardSourceMessageId: MessageId? var authorSignature: String? = nil value.read(&forwardAuthorId, offset: 0, length: 8) value.read(&forwardDate, offset: 0, length: 4) if (forwardInfoFlags & 2) != 0 { var forwardSourceIdValue: Int64 = 0 value.read(&forwardSourceIdValue, offset: 0, length: 8) forwardSourceId = PeerId(forwardSourceIdValue) } if (forwardInfoFlags & 4) != 0 { var forwardSourceMessagePeerId: Int64 = 0 var forwardSourceMessageNamespace: Int32 = 0 var forwardSourceMessageIdId: Int32 = 0 value.read(&forwardSourceMessagePeerId, offset: 0, length: 8) value.read(&forwardSourceMessageNamespace, offset: 0, length: 4) value.read(&forwardSourceMessageIdId, offset: 0, length: 4) forwardSourceMessageId = MessageId(peerId: PeerId(forwardSourceMessagePeerId), namespace: forwardSourceMessageNamespace, id: forwardSourceMessageIdId) } if (forwardInfoFlags & 8) != 0 { var signatureLength: Int32 = 0 value.read(&signatureLength, offset: 0, length: 4) authorSignature = String(data: Data(bytes: value.memory.assumingMemoryBound(to: UInt8.self).advanced(by: value.offset), count: Int(signatureLength)), encoding: .utf8) value.skip(Int(signatureLength)) } forwardInfo = IntermediateMessageForwardInfo(authorId: PeerId(forwardAuthorId), sourceId: forwardSourceId, sourceMessageId: forwardSourceMessageId, date: forwardDate, authorSignature: authorSignature) } var hasAuthor: Int8 = 0 value.read(&hasAuthor, offset: 0, length: 1) var authorId: PeerId? if hasAuthor == 1 { var varAuthorId: Int64 = 0 value.read(&varAuthorId, offset: 0, length: 8) authorId = PeerId(varAuthorId) } var textLength: Int32 = 0 value.read(&textLength, offset: 0, length: 4) let text = String(data: Data(bytes: value.memory.assumingMemoryBound(to: UInt8.self).advanced(by: value.offset), count: Int(textLength)), encoding: .utf8) ?? "" value.skip(Int(textLength)) let attributesOffset = value.offset var attributeCount: Int32 = 0 value.read(&attributeCount, offset: 0, length: 4) for _ in 0 ..< attributeCount { var attributeLength: Int32 = 0 value.read(&attributeLength, offset: 0, length: 4) value.skip(Int(attributeLength)) } let attributesLength = value.offset - attributesOffset let attributesBytes = malloc(attributesLength)! memcpy(attributesBytes, value.memory + attributesOffset, attributesLength) let attributesData = ReadBuffer(memory: attributesBytes, length: attributesLength, freeWhenDone: true) let embeddedMediaOffset = value.offset var embeddedMediaCount: Int32 = 0 value.read(&embeddedMediaCount, offset: 0, length: 4) for _ in 0 ..< embeddedMediaCount { var mediaLength: Int32 = 0 value.read(&mediaLength, offset: 0, length: 4) value.skip(Int(mediaLength)) } let embeddedMediaLength = value.offset - embeddedMediaOffset let embeddedMediaBytes = malloc(embeddedMediaLength)! memcpy(embeddedMediaBytes, value.memory + embeddedMediaOffset, embeddedMediaLength) let embeddedMediaData = ReadBuffer(memory: embeddedMediaBytes, length: embeddedMediaLength, freeWhenDone: true) var referencedMediaIds: [MediaId] = [] var referencedMediaIdsCount: Int32 = 0 value.read(&referencedMediaIdsCount, offset: 0, length: 4) for _ in 0 ..< referencedMediaIdsCount { var idNamespace: Int32 = 0 var idId: Int64 = 0 value.read(&idNamespace, offset: 0, length: 4) value.read(&idId, offset: 0, length: 8) referencedMediaIds.append(MediaId(namespace: idNamespace, id: idId)) } return .Message(IntermediateMessage(stableId: stableId, stableVersion: stableVersion, id: index.id, globallyUniqueId: globallyUniqueId, timestamp: index.timestamp, flags: flags, tags: tags, globalTags: globalTags, forwardInfo: forwardInfo, authorId: authorId, text: text, attributesData: attributesData, embeddedMediaData: embeddedMediaData, referencedMedia: referencedMediaIds)) } else { var stableId: UInt32 = 0 value.read(&stableId, offset: 0, length: 4) var minId: Int32 = 0 value.read(&minId, offset: 0, length: 4) var tags: UInt32 = 0 value.read(&tags, offset: 0, length: 4) return .Hole(MessageHistoryHole(stableId: stableId, maxIndex: index, min: minId, tags: tags)) } } func renderMessage(_ message: IntermediateMessage, peerTable: PeerTable, addAssociatedMessages: Bool = true) -> Message { var parsedAttributes: [MessageAttribute] = [] var parsedMedia: [Media] = [] let attributesData = message.attributesData.sharedBufferNoCopy() if attributesData.length > 4 { var attributeCount: Int32 = 0 attributesData.read(&attributeCount, offset: 0, length: 4) for _ in 0 ..< attributeCount { var attributeLength: Int32 = 0 attributesData.read(&attributeLength, offset: 0, length: 4) if let attribute = Decoder(buffer: MemoryBuffer(memory: attributesData.memory + attributesData.offset, capacity: Int(attributeLength), length: Int(attributeLength), freeWhenDone: false)).decodeRootObject() as? MessageAttribute { parsedAttributes.append(attribute) } attributesData.skip(Int(attributeLength)) } } let embeddedMediaData = message.embeddedMediaData.sharedBufferNoCopy() if embeddedMediaData.length > 4 { var embeddedMediaCount: Int32 = 0 embeddedMediaData.read(&embeddedMediaCount, offset: 0, length: 4) for _ in 0 ..< embeddedMediaCount { var mediaLength: Int32 = 0 embeddedMediaData.read(&mediaLength, offset: 0, length: 4) if let media = Decoder(buffer: MemoryBuffer(memory: embeddedMediaData.memory + embeddedMediaData.offset, capacity: Int(mediaLength), length: Int(mediaLength), freeWhenDone: false)).decodeRootObject() as? Media { parsedMedia.append(media) } embeddedMediaData.skip(Int(mediaLength)) } } for mediaId in message.referencedMedia { if let media = self.messageMediaTable.get(mediaId, embedded: { _ in return nil }) { parsedMedia.append(media) } } var forwardInfo: MessageForwardInfo? if let internalForwardInfo = message.forwardInfo, let forwardAuthor = peerTable.get(internalForwardInfo.authorId) { var source: Peer? if let sourceId = internalForwardInfo.sourceId { source = peerTable.get(sourceId) } forwardInfo = MessageForwardInfo(author: forwardAuthor, source: source, sourceMessageId: internalForwardInfo.sourceMessageId, date: internalForwardInfo.date, authorSignature: internalForwardInfo.authorSignature) } var author: Peer? var peers = SimpleDictionary() if let authorId = message.authorId { author = peerTable.get(authorId) } if let chatPeer = peerTable.get(message.id.peerId) { peers[chatPeer.id] = chatPeer if let associatedPeerId = chatPeer.associatedPeerId { if let peer = peerTable.get(associatedPeerId) { peers[peer.id] = peer } } } for media in parsedMedia { for peerId in media.peerIds { if let peer = peerTable.get(peerId) { peers[peer.id] = peer } } } var associatedMessageIds: [MessageId] = [] var associatedMessages = SimpleDictionary() for attribute in parsedAttributes { for peerId in attribute.associatedPeerIds { if let peer = peerTable.get(peerId) { peers[peer.id] = peer } } associatedMessageIds.append(contentsOf: attribute.associatedMessageIds) for messageId in attribute.associatedMessageIds { if let entry = self.messageHistoryIndexTable.get(messageId) { if case let .Message(index) = entry { if let message = self.getMessage(index) { associatedMessages[messageId] = self.renderMessage(message, peerTable: peerTable, addAssociatedMessages: false) } } } } } return Message(stableId: message.stableId, stableVersion: message.stableVersion, id: message.id, globallyUniqueId: message.globallyUniqueId, timestamp: message.timestamp, flags: message.flags, tags: message.tags, globalTags: message.globalTags, forwardInfo: forwardInfo, author: author, text: message.text, attributes: parsedAttributes, media: parsedMedia, peers: peers, associatedMessages: associatedMessages, associatedMessageIds: associatedMessageIds) } func entriesAround(_ index: MessageIndex, count: Int, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) -> (entries: [IntermediateMessageHistoryEntry], lower: IntermediateMessageHistoryEntry?, upper: IntermediateMessageHistoryEntry?) { var indexOperations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.ensureInitialized(index.id.peerId, operations: &indexOperations) if !indexOperations.isEmpty { self.processIndexOperations(index.id.peerId, operations: indexOperations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } var lowerEntries: [IntermediateMessageHistoryEntry] = [] var upperEntries: [IntermediateMessageHistoryEntry] = [] var lower: IntermediateMessageHistoryEntry? var upper: IntermediateMessageHistoryEntry? self.valueBox.range(self.table, start: self.key(index), end: self.lowerBound(index.id.peerId), values: { key, value in lowerEntries.append(self.readIntermediateEntry(key, value: value)) return true }, limit: count / 2 + 1) if lowerEntries.count >= count / 2 + 1 { lower = lowerEntries.last lowerEntries.removeLast() } self.valueBox.range(self.table, start: self.key(index).predecessor, end: self.upperBound(index.id.peerId), values: { key, value in upperEntries.append(self.readIntermediateEntry(key, value: value)) return true }, limit: count - lowerEntries.count + 1) if upperEntries.count >= count - lowerEntries.count + 1 { upper = upperEntries.last upperEntries.removeLast() } if lowerEntries.count != 0 && lowerEntries.count + upperEntries.count < count { var additionalLowerEntries: [IntermediateMessageHistoryEntry] = [] self.valueBox.range(self.table, start: self.key(lowerEntries.last!.index), end: self.lowerBound(index.id.peerId), values: { key, value in additionalLowerEntries.append(self.readIntermediateEntry(key, value: value)) return true }, limit: count - lowerEntries.count - upperEntries.count + 1) if additionalLowerEntries.count >= count - lowerEntries.count + upperEntries.count + 1 { lower = additionalLowerEntries.last additionalLowerEntries.removeLast() } lowerEntries.append(contentsOf: additionalLowerEntries) } var entries: [IntermediateMessageHistoryEntry] = [] entries.append(contentsOf: lowerEntries.reversed()) entries.append(contentsOf: upperEntries) return (entries: entries, lower: lower, upper: upper) } func entriesAround(_ tagMask: MessageTags, index: MessageIndex, count: Int, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) -> (entries: [IntermediateMessageHistoryEntry], lower: IntermediateMessageHistoryEntry?, upper: IntermediateMessageHistoryEntry?) { var indexOperations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.ensureInitialized(index.id.peerId, operations: &indexOperations) if !indexOperations.isEmpty { self.processIndexOperations(index.id.peerId, operations: indexOperations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } let (indices, lower, upper) = self.tagsTable.indicesAround(tagMask, index: index, count: count) var entries: [IntermediateMessageHistoryEntry] = [] for index in indices { let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { entries.append(readIntermediateEntry(key, value: value)) } else { assertionFailure() } } var lowerEntry: IntermediateMessageHistoryEntry? var upperEntry: IntermediateMessageHistoryEntry? if let lowerIndex = lower { let key = self.key(lowerIndex) if let value = self.valueBox.get(self.table, key: key) { lowerEntry = readIntermediateEntry(key, value: value) } else { assertionFailure() } } if let upperIndex = upper { let key = self.key(upperIndex) if let value = self.valueBox.get(self.table, key: key) { upperEntry = readIntermediateEntry(key, value: value) } else { assertionFailure() } } return (entries, lowerEntry, upperEntry) } func earlierEntries(_ peerId: PeerId, index: MessageIndex?, count: Int, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) -> [IntermediateMessageHistoryEntry] { var indexOperations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.ensureInitialized(peerId, operations: &indexOperations) if !indexOperations.isEmpty { self.processIndexOperations(peerId, operations: indexOperations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } var entries: [IntermediateMessageHistoryEntry] = [] let key: ValueBoxKey if let index = index { key = self.key(index) } else { key = self.upperBound(peerId) } self.valueBox.range(self.table, start: key, end: self.lowerBound(peerId), values: { key, value in entries.append(self.readIntermediateEntry(key, value: value)) return true }, limit: count) return entries } func earlierEntries(_ tagMask: MessageTags, peerId: PeerId, index: MessageIndex?, count: Int, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) -> [IntermediateMessageHistoryEntry] { var indexOperations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.ensureInitialized(peerId, operations: &indexOperations) if !indexOperations.isEmpty { self.processIndexOperations(peerId, operations: indexOperations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } let indices = self.tagsTable.earlierIndices(tagMask, peerId: peerId, index: index, count: count) var entries: [IntermediateMessageHistoryEntry] = [] for index in indices { let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { entries.append(readIntermediateEntry(key, value: value)) } else { assertionFailure() } } return entries } func laterEntries(_ peerId: PeerId, index: MessageIndex?, count: Int, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) -> [IntermediateMessageHistoryEntry] { var indexOperations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.ensureInitialized(peerId, operations: &indexOperations) if !indexOperations.isEmpty { self.processIndexOperations(peerId, operations: indexOperations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } var entries: [IntermediateMessageHistoryEntry] = [] let key: ValueBoxKey if let index = index { key = self.key(index) } else { key = self.lowerBound(peerId) } self.valueBox.range(self.table, start: key, end: self.upperBound(peerId), values: { key, value in entries.append(self.readIntermediateEntry(key, value: value)) return true }, limit: count) return entries } func laterEntries(_ tagMask: MessageTags, peerId: PeerId, index: MessageIndex?, count: Int, operationsByPeerId: inout [PeerId: [MessageHistoryOperation]], unsentMessageOperations: inout [IntermediateMessageHistoryUnsentOperation], updatedPeerReadStateOperations: inout [PeerId: PeerReadStateSynchronizationOperation?], globalTagsOperations: inout [GlobalMessageHistoryTagsOperation]) -> [IntermediateMessageHistoryEntry] { var indexOperations: [MessageHistoryIndexOperation] = [] self.messageHistoryIndexTable.ensureInitialized(peerId, operations: &indexOperations) if !indexOperations.isEmpty { self.processIndexOperations(peerId, operations: indexOperations, processedOperationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations) } let indices = self.tagsTable.laterIndices(tagMask, peerId: peerId, index: index, count: count) var entries: [IntermediateMessageHistoryEntry] = [] for index in indices { let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { entries.append(readIntermediateEntry(key, value: value)) } else { assertionFailure() } } return entries } private func globalTagsIntermediateEntry(_ entry: IntermediateMessageHistoryEntry) -> IntermediateGlobalMessageTagsEntry? { switch entry { case let .Message(message): return .message(message) case .Hole: return nil } } func entriesAround(globalTagMask: GlobalMessageTags, index: MessageIndex, count: Int) -> (entries: [IntermediateGlobalMessageTagsEntry], lower: MessageIndex?, upper: MessageIndex?) { self.globalTagsTable.ensureInitialized(globalTagMask) let (globalEntries, lower, upper) = self.globalTagsTable.entriesAround(globalTagMask, index: index, count: count) var entries: [IntermediateGlobalMessageTagsEntry] = [] for entry in globalEntries { switch entry { case let .hole(index): entries.append(.hole(index)) case let .message(index): let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { if let entry = self.globalTagsIntermediateEntry(readIntermediateEntry(key, value: value)) { entries.append(entry) } else { assertionFailure() } } else { assertionFailure() } } } return (entries, lower?.index, upper?.index) } func earlierEntries(globalTagMask: GlobalMessageTags, index: MessageIndex?, count: Int) -> [IntermediateGlobalMessageTagsEntry] { self.globalTagsTable.ensureInitialized(globalTagMask) let globalEntries = self.globalTagsTable.earlierEntries(globalTagMask, index: index, count: count) var entries: [IntermediateGlobalMessageTagsEntry] = [] for entry in globalEntries { switch entry { case let .hole(index): entries.append(.hole(index)) case let .message(index): let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { if let entry = self.globalTagsIntermediateEntry(readIntermediateEntry(key, value: value)) { entries.append(entry) } else { assertionFailure() } } else { assertionFailure() } } } return entries } func laterEntries(globalTagMask: GlobalMessageTags, index: MessageIndex?, count: Int) -> [IntermediateGlobalMessageTagsEntry] { self.globalTagsTable.ensureInitialized(globalTagMask) let globalEntries = self.globalTagsTable.laterEntries(globalTagMask, index: index, count: count) var entries: [IntermediateGlobalMessageTagsEntry] = [] for entry in globalEntries { switch entry { case let .hole(index): entries.append(.hole(index)) case let .message(index): let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { if let entry = self.globalTagsIntermediateEntry(readIntermediateEntry(key, value: value)) { entries.append(entry) } else { assertionFailure() } } else { assertionFailure() } } } return entries } func maxReadIndex(_ peerId: PeerId) -> MessageHistoryAnchorIndex? { if let combinedState = self.readStateTable.getCombinedState(peerId), let state = combinedState.states.first , state.1.count != 0 { switch state.1 { case let .idBased(maxIncomingReadId, _, _, _): return self.anchorIndex(MessageId(peerId: peerId, namespace: state.0, id: maxIncomingReadId)) case let .indexBased(maxIncomingReadIndex, _, _): return MessageHistoryAnchorIndex(index: maxIncomingReadIndex, exact: true) } } return nil } func anchorIndex(_ messageId: MessageId) -> MessageHistoryAnchorIndex? { let (lower, upper) = self.messageHistoryIndexTable.adjacentItems(messageId, bindUpper: false) if let lower = lower, case let .Hole(hole) = lower, messageId.id >= hole.min && messageId.id <= hole.maxIndex.id.id { return MessageHistoryAnchorIndex(index: MessageIndex(id: messageId, timestamp: lower.index.timestamp), exact: false) } if let upper = upper, case let .Hole(hole) = upper, messageId.id >= hole.min && messageId.id <= hole.maxIndex.id.id { return MessageHistoryAnchorIndex(index: MessageIndex(id: messageId, timestamp: upper.index.timestamp), exact: false) } if let lower = lower { return MessageHistoryAnchorIndex(index: MessageIndex(id: messageId, timestamp: lower.index.timestamp), exact: true) } else if let upper = upper { return MessageHistoryAnchorIndex(index: MessageIndex(id: messageId, timestamp: upper.index.timestamp), exact: true) } return MessageHistoryAnchorIndex(index: MessageIndex(id: messageId, timestamp: 1), exact: false) } func findMessageId(peerId: PeerId, timestamp: Int32) -> MessageId? { var result: MessageId? self.valueBox.range(self.table, start: self.key(MessageIndex(id: MessageId(peerId: peerId, namespace: 0, id: 0), timestamp: timestamp)), end: self.key(MessageIndex(id: MessageId(peerId: peerId, namespace: Int32.max, id: Int32.max), timestamp: timestamp)), values: { key, value in let entry = self.readIntermediateEntry(key, value: value) if case .Message = entry { result = entry.index.id return false } return true }, limit: 0) return result } func findClosestMessageId(peerId: PeerId, timestamp: Int32) -> MessageId? { var result: MessageId? self.valueBox.range(self.table, start: self.key(MessageIndex(id: MessageId(peerId: peerId, namespace: 0, id: 0), timestamp: timestamp)), end: self.lowerBound(peerId), values: { key, value in let entry = self.readIntermediateEntry(key, value: value) if case .Message = entry { result = entry.index.id return false } return true }, limit: 0) if result == nil { self.valueBox.range(self.table, start: self.key(MessageIndex(id: MessageId(peerId: peerId, namespace: 0, id: 0), timestamp: timestamp)), end: self.upperBound(peerId), values: { key, value in let entry = self.readIntermediateEntry(key, value: value) if case .Message = entry { result = entry.index.id return false } return true }, limit: 0) } return result } func incomingMessageStatsInIndices(_ peerId: PeerId, namespace: MessageId.Namespace, indices: [MessageIndex]) -> (Int, Bool) { var count: Int = 0 var holes = false for index in indices { let key = self.key(index) if let value = self.valueBox.get(self.table, key: key) { let entry = self.readIntermediateEntry(key, value: value) if case let .Message(message) = entry { if message.id.namespace == namespace && message.flags.contains(.Incoming) { count += 1 } } else { holes = true } } } return (count, holes) } func incomingMessageCountInRange(_ peerId: PeerId, namespace: MessageId.Namespace, fromIndex: MessageIndex, toIndex: MessageIndex) -> (Int, Bool, [MessageId]) { var count: Int = 0 var messageIds: [MessageId] = [] var holes = false self.valueBox.range(self.table, start: self.key(fromIndex).predecessor, end: self.key(toIndex).successor, values: { key, value in let entry = self.readIntermediateEntry(key, value: value) if case let .Message(message) = entry { if message.id.namespace == namespace && message.flags.contains(.Incoming) { count += 1 messageIds.append(message.id) } } else { holes = true } return true }, limit: 0) return (count, holes, messageIds) } func outgoingMessageCountInRange(_ peerId: PeerId, namespace: MessageId.Namespace, fromIndex: MessageIndex, toIndex: MessageIndex) -> [MessageId] { var messageIds: [MessageId] = [] self.valueBox.range(self.table, start: self.key(fromIndex).predecessor, end: self.key(toIndex).successor, values: { key, value in let entry = self.readIntermediateEntry(key, value: value) if case let .Message(message) = entry { if message.id.namespace == namespace && !message.flags.contains(.Incoming) { messageIds.append(message.id) } } return true }, limit: 0) return messageIds } func allIndices(_ peerId: PeerId) -> [MessageIndex] { var indices: [MessageIndex] = [] self.valueBox.range(self.table, start: self.key(MessageIndex.lowerBound(peerId: peerId)).predecessor, end: self.key(MessageIndex.upperBound(peerId: peerId)).successor, keys: { key in let index = MessageIndex(id: MessageId(peerId: PeerId(key.getInt64(0)), namespace: key.getInt32(8 + 4), id: key.getInt32(8 + 4 + 4)), timestamp: key.getInt32(8)) indices.append(index) return true }, limit: 0) return indices } func allIndicesWithAuthor(_ peerId: PeerId, authorId: PeerId) -> [MessageIndex] { var indices: [MessageIndex] = [] self.valueBox.range(self.table, start: self.key(MessageIndex.lowerBound(peerId: peerId)).predecessor, end: self.key(MessageIndex.upperBound(peerId: peerId)).successor, values: { key, value in if extractIntermediateEntryAuthor(value: value) == authorId { let index = MessageIndex(id: MessageId(peerId: PeerId(key.getInt64(0)), namespace: key.getInt32(8 + 4), id: key.getInt32(8 + 4 + 4)), timestamp: key.getInt32(8)) indices.append(index) } return true }, limit: 0) return indices } func getMessageCountInRange(peerId: PeerId, tagMask: MessageTags, lowerBound: MessageIndex, upperBound: MessageIndex) -> Int32 { return self.tagsTable.getMessageCountInRange(tagMask: tagMask, peerId: peerId, lowerBound: lowerBound, upperBound: upperBound) } func enumerateMedia(lowerBound: MessageIndex?, limit: Int) -> ([PeerId: Set], [MediaId: Media], MessageIndex?) { var mediaRefs: [MediaId: Media] = [:] var result: [PeerId: Set] = [:] var lastIndex: MessageIndex? var count = 0 self.valueBox.range(self.table, start: self.key(lowerBound == nil ? MessageIndex.absoluteLowerBound() : lowerBound!), end: self.key(MessageIndex.absoluteUpperBound()), values: { key, value in count += 1 let entry = self.readIntermediateEntry(key, value: value) lastIndex = entry.index if case let .Message(message) = entry { var parsedMedia: [Media] = [] let embeddedMediaData = message.embeddedMediaData.sharedBufferNoCopy() if embeddedMediaData.length > 4 { var embeddedMediaCount: Int32 = 0 embeddedMediaData.read(&embeddedMediaCount, offset: 0, length: 4) for _ in 0 ..< embeddedMediaCount { var mediaLength: Int32 = 0 embeddedMediaData.read(&mediaLength, offset: 0, length: 4) if let media = Decoder(buffer: MemoryBuffer(memory: embeddedMediaData.memory + embeddedMediaData.offset, capacity: Int(mediaLength), length: Int(mediaLength), freeWhenDone: false)).decodeRootObject() as? Media { parsedMedia.append(media) } embeddedMediaData.skip(Int(mediaLength)) } } for mediaId in message.referencedMedia { if let media = self.messageMediaTable.get(mediaId, embedded: { _ in return nil }) { parsedMedia.append(media) } } for media in parsedMedia { if let id = media.id { mediaRefs[id] = media if result[message.id.peerId] == nil { result[message.id.peerId] = Set() } result[message.id.peerId]!.insert(id) } } } return true }, limit: limit) return (result, mediaRefs, count == 0 ? nil : lastIndex) } func debugList(_ peerId: PeerId, peerTable: PeerTable) -> [RenderedMessageHistoryEntry] { var operationsByPeerId: [PeerId : [MessageHistoryOperation]] = [:] var unsentMessageOperations: [IntermediateMessageHistoryUnsentOperation] = [] var updatedPeerReadStateOperations: [PeerId : PeerReadStateSynchronizationOperation?] = [:] var globalTagsOperations: [GlobalMessageHistoryTagsOperation] = [] return self.laterEntries(peerId, index: nil, count: 1000, operationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations).map({ entry -> RenderedMessageHistoryEntry in switch entry { case let .Hole(hole): return .Hole(hole) case let .Message(message): return .RenderedMessage(self.renderMessage(message, peerTable: peerTable)) } }) } func debugList(_ tagMask: MessageTags, peerId: PeerId, peerTable: PeerTable) -> [RenderedMessageHistoryEntry] { var operationsByPeerId: [PeerId : [MessageHistoryOperation]] = [:] var unsentMessageOperations: [IntermediateMessageHistoryUnsentOperation] = [] var updatedPeerReadStateOperations: [PeerId : PeerReadStateSynchronizationOperation?] = [:] var globalTagsOperations: [GlobalMessageHistoryTagsOperation] = [] return self.laterEntries(tagMask, peerId: peerId, index: nil, count: 1000, operationsByPeerId: &operationsByPeerId, unsentMessageOperations: &unsentMessageOperations, updatedPeerReadStateOperations: &updatedPeerReadStateOperations, globalTagsOperations: &globalTagsOperations).map({ entry -> RenderedMessageHistoryEntry in switch entry { case let .Hole(hole): return .Hole(hole) case let .Message(message): return .RenderedMessage(self.renderMessage(message, peerTable: peerTable)) } }) } }