96 lines
3.2 KiB
Swift
96 lines
3.2 KiB
Swift
// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved.
|
|
|
|
import Foundation
|
|
import GRDB
|
|
import SessionUtilitiesKit
|
|
|
|
public struct Quote: Codable, Equatable, Hashable, FetchableRecord, PersistableRecord, TableRecord, ColumnExpressible {
|
|
public static var databaseTableName: String { "quote" }
|
|
public static let interactionForeignKey = ForeignKey([Columns.interactionId], to: [Interaction.Columns.id])
|
|
internal static let originalInteractionForeignKey = ForeignKey(
|
|
[Columns.timestampMs, Columns.authorId],
|
|
to: [Interaction.Columns.timestampMs, Interaction.Columns.authorId]
|
|
)
|
|
internal static let profileForeignKey = ForeignKey([Columns.authorId], to: [Profile.Columns.id])
|
|
internal static let interaction = belongsTo(Interaction.self, using: interactionForeignKey)
|
|
private static let profile = hasOne(Profile.self, using: profileForeignKey)
|
|
private static let quotedInteraction = hasOne(Interaction.self, using: originalInteractionForeignKey)
|
|
public static let attachment = hasOne(Attachment.self, using: Attachment.quoteForeignKey)
|
|
|
|
public typealias Columns = CodingKeys
|
|
public enum CodingKeys: String, CodingKey, ColumnExpression {
|
|
case interactionId
|
|
case authorId
|
|
case timestampMs
|
|
case body
|
|
case attachmentId
|
|
}
|
|
|
|
/// The id for the interaction this Quote belongs to
|
|
public let interactionId: Int64
|
|
|
|
/// The id for the author this Quote belongs to
|
|
public let authorId: String
|
|
|
|
/// The timestamp in milliseconds since epoch when the quoted interaction was sent
|
|
public let timestampMs: Int64
|
|
|
|
/// The body of the quoted message if the user is quoting a text message or an attachment with a caption
|
|
public let body: String?
|
|
|
|
/// The id for the attachment this Quote is associated with
|
|
public let attachmentId: String?
|
|
|
|
// MARK: - Relationships
|
|
|
|
public var interaction: QueryInterfaceRequest<Interaction> {
|
|
request(for: Quote.interaction)
|
|
}
|
|
|
|
public var profile: QueryInterfaceRequest<Profile> {
|
|
request(for: Quote.profile)
|
|
}
|
|
|
|
public var attachment: QueryInterfaceRequest<Attachment> {
|
|
request(for: Quote.attachment)
|
|
}
|
|
|
|
public var originalInteraction: QueryInterfaceRequest<Interaction> {
|
|
request(for: Quote.quotedInteraction)
|
|
}
|
|
|
|
// MARK: - Interaction
|
|
|
|
public init(
|
|
interactionId: Int64,
|
|
authorId: String,
|
|
timestampMs: Int64,
|
|
body: String?,
|
|
attachmentId: String?
|
|
) {
|
|
self.interactionId = interactionId
|
|
self.authorId = authorId
|
|
self.timestampMs = timestampMs
|
|
self.body = body
|
|
self.attachmentId = attachmentId
|
|
}
|
|
}
|
|
|
|
// MARK: - Protobuf
|
|
|
|
public extension Quote {
|
|
init?(_ db: Database, proto: SNProtoDataMessage, interactionId: Int64, thread: SessionThread) throws {
|
|
guard
|
|
let quoteProto = proto.quote,
|
|
quoteProto.id != 0,
|
|
!quoteProto.author.isEmpty
|
|
else { return nil }
|
|
|
|
self.interactionId = interactionId
|
|
self.timestampMs = Int64(quoteProto.id)
|
|
self.authorId = quoteProto.author
|
|
self.body = nil
|
|
self.attachmentId = nil
|
|
}
|
|
}
|