mobile_nebula/ios/Runner/DNUpdate.swift

import Foundation
import os.log

class DNUpdater {
    private let apiClient = APIClient()
    private let timer = RepeatingTimer(timeInterval: 30) // 15 * 60 is 15 minutes
    private let log = Logger(subsystem: "net.defined.mobileNebula", category: "DNUpdater")
    
    func updateAll(onUpdate: @escaping (Site) -> ()) {
        _ = SiteList{ (sites, _) -> () in
            // NEVPN seems to force us onto the main thread and we are about to make network calls that
            // could block for a while. Push ourselves onto another thread to avoid blocking the UI.
            Task.detached(priority: .userInitiated) {
                sites?.values.forEach { site in
                    if (site.connected == true) {
                        // The vpn service is in charge of updating the currently connected site
                        return
                    }

                    self.updateSite(site: site, onUpdate: onUpdate)
                }
            }
        }
    }
    
    func updateAllLoop(onUpdate: @escaping (Site) -> ()) {
        timer.eventHandler = {
            self.updateAll(onUpdate: onUpdate)
        }
        timer.resume()
    }
    
    func updateSingleLoop(site: Site, onUpdate: @escaping (Site) -> ()) {
        timer.eventHandler = {
            self.updateSite(site: site, onUpdate: onUpdate)
        }
        timer.resume()
    }
    
    func updateSite(site: Site, onUpdate: @escaping (Site) -> ()) {
        do {
            if (!site.managed) {
                return
            }
            
            let credentials = try site.getDNCredentials()
          
            let newSite: IncomingSite?
            do {
                newSite = try apiClient.tryUpdate(
                    siteName: site.name,
                    hostID: credentials.hostID,
                    privateKey: credentials.privateKey,
                    counter: credentials.counter,
                    trustedKeys: credentials.trustedKeys
                )
            } catch (APIClientError.invalidCredentials) {
                if (!credentials.invalid) {
                    try site.invalidateDNCredentials()
                    log.notice("Invalidated credentials in site: \(site.name, privacy: .public)")
                }
                
                return
            }
            
            newSite?.save(manager: site.manager) { error in
                if (error != nil) {
                    self.log.error("failed to save update: \(error!.localizedDescription, privacy: .public)")
                }
                
                // reload nebula even if we couldn't save the vpn profile
                onUpdate(Site(incoming: newSite!))
            }
            
            if (credentials.invalid) {
                try site.validateDNCredentials()
                log.notice("Revalidated credentials in site \(site.name, privacy: .public)")
            }
            
        } catch {
            log.error("Error while updating \(site.name, privacy: .public): \(error.localizedDescription, privacy: .public)")
        }
    }
}

// From https://medium.com/over-engineering/a-background-repeating-timer-in-swift-412cecfd2ef9
class RepeatingTimer {

    let timeInterval: TimeInterval

    init(timeInterval: TimeInterval) {
        self.timeInterval = timeInterval
    }

    private lazy var timer: DispatchSourceTimer = {
        let t = DispatchSource.makeTimerSource()
        t.schedule(deadline: .now(), repeating: self.timeInterval)
        t.setEventHandler(handler: { [weak self] in
            self?.eventHandler?()
        })
        return t
    }()

    var eventHandler: (() -> Void)?

    private enum State {
        case suspended
        case resumed
    }

    private var state: State = .suspended

    deinit {
        timer.setEventHandler {}
        timer.cancel()
        /*
         If the timer is suspended, calling cancel without resuming
         triggers a crash. This is documented here https://forums.developer.apple.com/thread/15902
         */
        resume()
        eventHandler = nil
    }

    func resume() {
        if state == .resumed {
            return
        }
        state = .resumed
        timer.resume()
    }

    func suspend() {
        if state == .suspended {
            return
        }
        state = .suspended
        timer.suspend()
    }
}