66 lines
2.2 KiB
Swift
66 lines
2.2 KiB
Swift
//
|
|
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
public import LibSignalClient
|
|
|
|
extension MessageBackup {
|
|
|
|
/// Base context class used for archiving (creating a backup).
|
|
///
|
|
/// Requires a write tx on init; we want to hold the write lock while
|
|
/// creating a backup to avoid races with e.g. message processing.
|
|
///
|
|
/// But only exposes a read tx, because we explicitly do not want
|
|
/// archiving to be updating the database, just reading from it.
|
|
/// (The exception to this is enqueuing attachment uploads.)
|
|
open class ArchivingContext {
|
|
|
|
private let _tx: DBWriteTransaction
|
|
|
|
public var tx: DBReadTransaction { _tx }
|
|
|
|
/// The purpose for this backup. Determines minor behavior variations, such as
|
|
/// whether we include expiring messages or not.
|
|
public let backupPurpose: MessageBackupPurpose
|
|
|
|
/// Nil if not a paid backups account.
|
|
private let currentBackupAttachmentUploadEra: String?
|
|
private let backupAttachmentUploadManager: BackupAttachmentUploadManager
|
|
|
|
init(
|
|
backupPurpose: MessageBackupPurpose,
|
|
currentBackupAttachmentUploadEra: String?,
|
|
backupAttachmentUploadManager: BackupAttachmentUploadManager,
|
|
tx: DBWriteTransaction
|
|
) {
|
|
self.backupPurpose = backupPurpose
|
|
self.currentBackupAttachmentUploadEra = currentBackupAttachmentUploadEra
|
|
self.backupAttachmentUploadManager = backupAttachmentUploadManager
|
|
self._tx = tx
|
|
}
|
|
|
|
func enqueueAttachmentForUploadIfNeeded(_ referencedAttachment: ReferencedAttachment) throws {
|
|
guard let currentBackupAttachmentUploadEra else {
|
|
return
|
|
}
|
|
try backupAttachmentUploadManager.enqueueIfNeeded(
|
|
referencedAttachment,
|
|
currentUploadEra: currentBackupAttachmentUploadEra,
|
|
tx: _tx
|
|
)
|
|
}
|
|
}
|
|
|
|
/// Base context class used for restoring from a backup.
|
|
open class RestoringContext {
|
|
|
|
public let tx: DBWriteTransaction
|
|
|
|
init(tx: DBWriteTransaction) {
|
|
self.tx = tx
|
|
}
|
|
}
|
|
}
|