//
// 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

        let bencher: MessageBackup.ArchiveBencher

        /// Nil if not a paid backups account.
        private let currentBackupAttachmentUploadEra: String?
        private let backupAttachmentUploadManager: BackupAttachmentUploadManager

        init(
            backupPurpose: MessageBackupPurpose,
            bencher: MessageBackup.ArchiveBencher,
            currentBackupAttachmentUploadEra: String?,
            backupAttachmentUploadManager: BackupAttachmentUploadManager,
            tx: DBWriteTransaction
        ) {
            self.backupPurpose = backupPurpose
            self.bencher = bencher
            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
        }
    }
}