deep-linking-sample/Apps/Wikipedia/WMF Framework/WMFArticle+Errors.swift

import Foundation


public enum ArticleError: Int32, Error {
    case none = 0
    case saveToDiskFailed = 1
    case apiFailed = 2
}

extension ArticleError: LocalizedError {
    public var errorDescription: String? {
        switch self {
        case .none:
            return nil
        case .saveToDiskFailed:
            return WMFLocalizedString("reading-lists-article-save-to-disk-failed", value: "Device limited exceeded, unable to sync article", comment: "Text of the alert label informing the user that article couldn't be saved due to insufficient storage available")
        case .apiFailed:
            return WMFLocalizedString("reading-lists-article-api-failure", value: "Unable to sync article", comment: "Text of the alert label informing the user that article couldn't be saved due to a server-side error")
        }
    }
    public var failureReason: String? {
        switch self {
        case .none:
            return nil
        case .saveToDiskFailed:
            return nil
        case .apiFailed:
            return nil
        }
    }
    public var recoverySuggestion: String? {
        switch self {
        case .none:
            return nil
        case .saveToDiskFailed:
            return WMFLocalizedString("reading-lists-article-save-to-disk-recovery-suggestion", value: "Clear some space on your device and try again", comment: "Recovery suggestion to clear space on the user's device to allow articles to download")
        case .apiFailed:
            return nil
        }
    }
}


extension WMFArticle {
    public var error: ArticleError {
        get {
            return ArticleError(rawValue: errorCodeNumber?.int32Value ?? 0) ?? .none
        }
        set {
            guard newValue != .none else {
                errorCodeNumber = nil
                return
            }
            errorCodeNumber = NSNumber(value: newValue.rawValue)
        }
    }
    
    public func retryDownload() {
        guard savedDate != nil else {
            return
        }
        isDownloaded = false
        errorCodeNumber = nil
        downloadAttemptCount = 0
        downloadRetryDate = nil
    }
}

extension NSManagedObjectContext {
    public func retryFailedArticleDownloads(with keys: [String]) throws {
        let batches = keys.chunked(into: 500)
        for batch in batches {
            let articleFetch = WMFArticle.fetchRequest()
            articleFetch.predicate = NSPredicate(format: "errorCodeNumber != NULL && key IN %@", batch)
            let articles = try fetch(articleFetch)
            for article in articles {
                article.retryDownload()
            }
            try save()
        }
    }
}