metatext/DB/Sources/DB/Content/ContextItemsInfo.swift
Justin Mazzocchi b32a85aebc
wip
2021-02-07 17:46:51 -08:00

58 lines
2.6 KiB
Swift

// Copyright © 2020 Metabolist. All rights reserved.
import Foundation
import GRDB
import Mastodon
struct ContextItemsInfo: Codable, Hashable, FetchableRecord {
let parent: StatusInfo
let ancestors: [StatusInfo]
let descendants: [StatusInfo]
}
extension ContextItemsInfo {
static func addingIncludes<T: DerivableRequest>(_ request: T) -> T where T.RowDecoder == StatusRecord {
StatusInfo.addingIncludes(request)
.including(all: StatusInfo.addingIncludes(StatusRecord.ancestors).forKey(CodingKeys.ancestors))
.including(all: StatusInfo.addingIncludes(StatusRecord.descendants).forKey(CodingKeys.descendants))
}
static func request(_ request: QueryInterfaceRequest<StatusRecord>) -> QueryInterfaceRequest<Self> {
addingIncludes(request).asRequest(of: self)
}
func items(filters: [Filter]) -> [CollectionSection] {
let regularExpression = filters.regularExpression(context: .thread)
return [ancestors, [parent], descendants].map { section in
section.filtered(regularExpression: regularExpression)
.enumerated()
.map { index, statusInfo in
let isContextParent = statusInfo.record.id == parent.record.id
let isReplyInContext: Bool
if isContextParent {
isReplyInContext = !ancestors.isEmpty
&& statusInfo.record.inReplyToId == ancestors.last?.record.id
} else {
isReplyInContext = index > 0
&& section[index - 1].record.id == statusInfo.record.inReplyToId
}
let hasReplyFollowing = (section.count > index + 1
&& section[index + 1].record.inReplyToId == statusInfo.record.id)
|| (statusInfo == ancestors.last && parent.record.inReplyToId == statusInfo.record.id)
return .status(
.init(info: statusInfo),
.init(showContentToggled: statusInfo.showContentToggled,
showAttachmentsToggled: statusInfo.showAttachmentsToggled,
isContextParent: isContextParent,
isReplyInContext: isReplyInContext,
hasReplyFollowing: hasReplyFollowing),
statusInfo.reblogRelationship ?? statusInfo.relationship)
}
}
.map { CollectionSection(items: $0) }
}
}