RichPushRequest.swift

import CioInternalCommon
import Foundation

/// Request to download and attach a rich push image. Only created by the handler when the push
/// has a valid image URL, so this request always uses the shared HTTP client and cancel() always cancels it.
class RichPushRequest {
    private let completionHandler: (PushNotification) -> Void
    private var push: PushNotification
    private let imageURL: URL
    private let httpClient: HttpClient

    private var isCompleted = false

    init(
        push: PushNotification,
        imageURL: URL,
        httpClient: HttpClient,
        completionHandler: @escaping (PushNotification) -> Void
    ) {
        self.completionHandler = completionHandler
        self.push = push
        self.imageURL = imageURL
        self.httpClient = httpClient
    }

    func start() {
        httpClient.downloadFile(url: imageURL, fileType: .richPushImage) { [weak self] localFilePath in
            guard let self = self else { return }

            if let localFilePath = localFilePath {
                self.push.cioRichPushImageFile = localFilePath
            }

            self.completeOnce()
        }
    }

    /// Call when the request must be aborted (e.g. NSE expiry). Cancels the download and completes.
    func cancel() {
        guard !isCompleted else { return }
        isCompleted = true
        httpClient.cancel(finishTasks: false)
        completionHandler(push)
    }

    private func completeOnce() {
        guard !isCompleted else { return }
        isCompleted = true
        completionHandler(push)
    }
}