AttachmentValidationBackfillRunner.swift

//
// Copyright 2024 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//

import Foundation
import SignalServiceKit

/// Manages the BGProcessingTask for doing the backfill of attachments that were
/// validated using an old version of the validator and need revalidation.
class AttachmentValidationBackfillRunner: BGProcessingTaskRunner {

    private let db: SDSDatabaseStorage
    private let store: AttachmentValidationBackfillStore
    private let migrator: () -> any AttachmentValidationBackfillMigrator

    init(
        db: SDSDatabaseStorage,
        store: AttachmentValidationBackfillStore,
        migrator: @escaping () -> any AttachmentValidationBackfillMigrator,
    ) {
        self.db = db
        self.store = store
        self.migrator = migrator
    }

    // MARK: - BGProcessingTaskRunner

    static let taskIdentifier = "AttachmentValidationBackfillMigrator"
    static let logPrefix: String? = nil
    static let requiresNetworkConnectivity = false
    static let requiresExternalPower = false

    func run() async throws(CancellationError) {
        try await self.runInBatches(
            willBegin: {},
            runNextBatch: { await migrator().runNextBatch() },
        )
    }

    func startCondition() -> BGProcessingTaskStartCondition {
        return db.read { tx in
            if store.needsToRun(tx: tx) {
                return .asSoonAsPossible
            } else {
                return .never
            }
        }
    }
}