94 lines
2.8 KiB
Swift
94 lines
2.8 KiB
Swift
//
|
|
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
protocol DeleteForMeAddressableMessageFinder {
|
|
func findLocalMessage(
|
|
threadUniqueId: String,
|
|
addressableMessage: DeleteForMeSyncMessage.Incoming.AddressableMessage,
|
|
tx: any DBReadTransaction
|
|
) -> TSMessage?
|
|
|
|
func threadContainsAnyAddressableMessages(
|
|
threadUniqueId: String,
|
|
tx: any DBReadTransaction
|
|
) -> Bool
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
final class DeleteForMeAddressableMessageFinderImpl: DeleteForMeAddressableMessageFinder {
|
|
private let tsAccountManager: TSAccountManager
|
|
|
|
init(tsAccountManager: TSAccountManager) {
|
|
self.tsAccountManager = tsAccountManager
|
|
}
|
|
|
|
func findLocalMessage(
|
|
threadUniqueId: String,
|
|
addressableMessage: DeleteForMeSyncMessage.Incoming.AddressableMessage,
|
|
tx: any DBReadTransaction
|
|
) -> TSMessage? {
|
|
let authorAddress: SignalServiceAddress
|
|
switch addressableMessage.author {
|
|
case .localUser:
|
|
guard let localAddress = tsAccountManager.localIdentifiers(tx: tx)?.aciAddress else {
|
|
return nil
|
|
}
|
|
authorAddress = localAddress
|
|
case .otherUser(let signalRecipient):
|
|
authorAddress = signalRecipient.address
|
|
}
|
|
|
|
return InteractionFinder.findMessage(
|
|
withTimestamp: addressableMessage.sentTimestamp,
|
|
threadId: threadUniqueId,
|
|
author: authorAddress,
|
|
transaction: SDSDB.shimOnlyBridge(tx)
|
|
)
|
|
}
|
|
|
|
func threadContainsAnyAddressableMessages(
|
|
threadUniqueId: String,
|
|
tx: any DBReadTransaction
|
|
) -> Bool {
|
|
var foundAddressableMessage = false
|
|
|
|
do {
|
|
try DeleteForMeMostRecentAddressableMessageCursor(
|
|
threadUniqueId: threadUniqueId,
|
|
sdsTx: SDSDB.shimOnlyBridge(tx)
|
|
).iterate { interaction in
|
|
owsPrecondition(
|
|
interaction is TSIncomingMessage || interaction is TSOutgoingMessage,
|
|
"Unexpected interaction type! \(type(of: interaction))"
|
|
)
|
|
|
|
foundAddressableMessage = true
|
|
return false
|
|
}
|
|
} catch {
|
|
owsFailDebug("Failed to enumerate interactions!")
|
|
return false
|
|
}
|
|
|
|
return foundAddressableMessage
|
|
}
|
|
}
|
|
|
|
// MARK: - Mock
|
|
|
|
#if TESTABLE_BUILD
|
|
|
|
open class MockDeleteForMeAddressableMessageFinder: DeleteForMeAddressableMessageFinder {
|
|
func findLocalMessage(threadUniqueId: String, addressableMessage: DeleteForMeSyncMessage.Incoming.AddressableMessage, tx: any DBReadTransaction) -> TSMessage? {
|
|
return nil
|
|
}
|
|
|
|
func threadContainsAnyAddressableMessages(threadUniqueId: String, tx: any DBReadTransaction) -> Bool {
|
|
return false
|
|
}
|
|
}
|
|
|
|
#endif
|