session-ios/Session/Utilities/BackgroundPoller.swift

import PromiseKit
import SessionSnodeKit

@objc(LKBackgroundPoller)
public final class BackgroundPoller : NSObject {
    private static var promises: [Promise<Void>] = []

    private override init() { }

    @objc(pollWithCompletionHandler:)
    public static func poll(completionHandler: @escaping (UIBackgroundFetchResult) -> Void) {
        promises = []
        promises.append(pollForMessages())
        promises.append(contentsOf: pollForClosedGroupMessages())
        let v2OpenGroupServers = Set(Storage.shared.getAllV2OpenGroups().values.map { $0.server })
        v2OpenGroupServers.forEach { server in
            let poller = OpenGroupPollerV2(for: server)
            poller.stop()
            promises.append(poller.poll(isBackgroundPoll: true))
        }
        when(resolved: promises).done { _ in
            completionHandler(.newData)
        }.catch { error in
            SNLog("Background poll failed due to error: \(error)")
            completionHandler(.failed)
        }
    }
    
    private static func pollForMessages() -> Promise<Void> {
        let userPublicKey = getUserHexEncodedPublicKey()
        return getMessages(for: userPublicKey)
    }
    
    private static func pollForClosedGroupMessages() -> [Promise<Void>] {
        let publicKeys = Storage.shared.getUserClosedGroupPublicKeys()
        return publicKeys.map { getClosedGroupMessages(for: $0) }
    }
    
    private static func getMessages(for publicKey: String) -> Promise<Void> {
        return SnodeAPI.getSwarm(for: publicKey).then(on: DispatchQueue.main) { swarm -> Promise<Void> in
            guard let snode = swarm.randomElement() else { throw SnodeAPI.Error.generic }
            return attempt(maxRetryCount: 4, recoveringOn: DispatchQueue.main) {
                SnodeAPI.getRawMessages(from: snode, associatedWith: publicKey).then(on: DispatchQueue.main) { rawResponse -> Promise<Void> in
                    let (messages, lastRawMessage) = SnodeAPI.parseRawMessagesResponse(rawResponse, from: snode, associatedWith: publicKey)
                    var processedMessages: [JSON] = []
                    let promises = messages.compactMap { json -> Promise<Void>? in
                        // Use a best attempt approach here; we don't want to fail the entire process if one of the
                        // messages failed to parse.
                        guard let envelope = SNProtoEnvelope.from(json),
                            let data = try? envelope.serializedData() else { return nil }
                        let job = MessageReceiveJob(data: data, serverHash: json["hash"] as? String, isBackgroundPoll: true)
                        processedMessages.append(json)
                        return job.execute()
                    }
                    // Now that the MessageReceiveJob's have been created we can update the `lastMessageHash` value & `receivedMessageHashes`
                    SnodeAPI.updateLastMessageHashValueIfPossible(for: snode, namespace: SnodeAPI.defaultNamespace, associatedWith: publicKey, from: lastRawMessage)
                    SnodeAPI.updateReceivedMessages(from: processedMessages, associatedWith: publicKey)
                    
                    return when(fulfilled: promises) // The promise returned by MessageReceiveJob never rejects
                }
            }
        }
    }
    
    private static func getClosedGroupMessages(for publicKey: String) -> Promise<Void> {
        return SnodeAPI.getSwarm(for: publicKey).then(on: DispatchQueue.main) { swarm -> Promise<Void> in
            guard let snode = swarm.randomElement() else { throw SnodeAPI.Error.generic }
            return attempt(maxRetryCount: 4, recoveringOn: DispatchQueue.main) {
                var namespaces: [Int] = []
                let promises: [SnodeAPI.RawResponsePromise] = {
                    if SnodeAPI.hardfork >= 19 && SnodeAPI.softfork >= 1 {
                        namespaces = [ SnodeAPI.closedGroupNamespace ]
                        return [ SnodeAPI.getRawMessages(from: snode, associatedWith: publicKey, authenticated: false) ]
                    }
                    if SnodeAPI.hardfork >= 19 {
                        namespaces = [ SnodeAPI.defaultNamespace, SnodeAPI.closedGroupNamespace ]
                        return [ SnodeAPI.getRawClosedGroupMessagesFromDefaultNamespace(from: snode, associatedWith: publicKey),
                                 SnodeAPI.getRawMessages(from: snode, associatedWith: publicKey, authenticated: false)]
                    }
                    namespaces = [ SnodeAPI.defaultNamespace ]
                    return [ SnodeAPI.getRawClosedGroupMessagesFromDefaultNamespace(from: snode, associatedWith: publicKey) ]
                }()

                return when(resolved: promises).then(on: DispatchQueue.main) { results -> Promise<Void> in
                    var promises: [Promise<Void>] = []
                    var index = 0
                    for result in results {
                        if case .fulfilled(let rawResponse) = result {
                            let (messages, lastRawMessage) = SnodeAPI.parseRawMessagesResponse(rawResponse, from: snode, associatedWith: publicKey)
                            var processedMessages: [JSON] = []
                            let jobPromises = messages.compactMap { json -> Promise<Void>? in
                                // Use a best attempt approach here; we don't want to fail the entire process if one of the
                                // messages failed to parse.
                                guard let envelope = SNProtoEnvelope.from(json),
                                    let data = try? envelope.serializedData() else { return nil }
                                let job = MessageReceiveJob(data: data, serverHash: json["hash"] as? String, isBackgroundPoll: true)
                                processedMessages.append(json)
                                return job.execute()
                            }
                            // Now that the MessageReceiveJob's have been created we can update the `lastMessageHash` value & `receivedMessageHashes`
                            SnodeAPI.updateLastMessageHashValueIfPossible(for: snode, namespace: namespaces[index], associatedWith: publicKey, from: lastRawMessage)
                            SnodeAPI.updateReceivedMessages(from: processedMessages, associatedWith: publicKey)
                            promises += jobPromises
                        }
                        index += 1
                    }
                    return when(fulfilled: promises) // The promise returned by MessageReceiveJob never rejects
                }
            }
        }
    }
}