mobile_nebula/ios/Runner/Sites.swift

import NetworkExtension
import MobileNebula

class SiteContainer {
    var site: Site
    var updater: SiteUpdater
    
    init(site: Site, updater: SiteUpdater) {
        self.site = site
        self.updater = updater
    }
}

class Sites {
    private var containers = [String: SiteContainer]()
    private var messenger: FlutterBinaryMessenger?
    
    init(messenger: FlutterBinaryMessenger?) {
        self.messenger = messenger
    }

    func loadSites(completion: @escaping ([String: Site]?, Error?) -> ()) {
        _ = SiteList { (sites, err) in
            if (err != nil) {
                return completion(nil, err)
            }
            
            sites?.values.forEach{ site in
                var updater = self.containers[site.id]?.updater
                if (updater != nil) {
                    updater!.setSite(site: site)
                } else {
                    updater = SiteUpdater(messenger: self.messenger!, site: site)
                }
                self.containers[site.id] = SiteContainer(site: site, updater: updater!)
            }
            
            let justSites = self.containers.mapValues {
                return $0.site
            }
            completion(justSites, nil)
        }
    }
    
    func deleteSite(id: String, callback: @escaping (Error?) -> ()) {
        if let site = self.containers.removeValue(forKey: id) {
            _ = KeyChain.delete(key: "\(site.site.id).dnCredentials")
            _ = KeyChain.delete(key: "\(site.site.id).key")
            
            do {
                let fileManager = FileManager.default
                let siteDir = try SiteList.getSiteDir(id: site.site.id)
                try fileManager.removeItem(at: siteDir)
            } catch {
                print("Failed to delete site from fs: \(error.localizedDescription)")
            }
            
#if !targetEnvironment(simulator)
            site.site.manager!.removeFromPreferences(completionHandler: callback)
            return
#endif
        }
        
        // Nothing to remove
        callback(nil)
    }
    
    func getSite(id: String) -> Site? {
        return self.containers[id]?.site
    }
    
    func getUpdater(id: String) -> SiteUpdater? {
        return self.containers[id]?.updater
    }
    
    func getContainer(id: String) -> SiteContainer? {
        return self.containers[id]
    }
}

class SiteUpdater: NSObject, FlutterStreamHandler {
    private var eventSink: FlutterEventSink?;
    private var eventChannel: FlutterEventChannel;
    private var site: Site
    private var notification: Any?
    public var startFunc: (() -> Void)?
    private var configFd: Int32? = nil
    private var configObserver: DispatchSourceFileSystemObject? = nil
    
    init(messenger: FlutterBinaryMessenger, site: Site) {
        do {
            let configPath = try SiteList.getSiteConfigFile(id: site.id, createDir: false)
            self.configFd = open(configPath.path, O_EVTONLY)
            self.configObserver = DispatchSource.makeFileSystemObjectSource(
                fileDescriptor: self.configFd!,
                eventMask: .write
            )
            
        } catch {
            // SiteList.getSiteConfigFile should never throw because we are not creating it here
            self.configObserver = nil
        }
        
        eventChannel = FlutterEventChannel(name: "net.defined.nebula/\(site.id)", binaryMessenger: messenger)
        self.site = site
        super.init()
        
        eventChannel.setStreamHandler(self)
        
        self.configObserver?.setEventHandler(handler: self.configUpdated)
        self.configObserver?.setCancelHandler {
            if self.configFd != nil {
                close(self.configFd!)
            }
            self.configObserver = nil
        }
        
        self.configObserver?.resume()
    }
    
    func setSite(site: Site) {
        self.site = site
    }

    /// onListen is called when flutter code attaches an event listener
    func onListen(withArguments arguments: Any?, eventSink events: @escaping FlutterEventSink) -> FlutterError? {
        eventSink = events;
#if !targetEnvironment(simulator)
        self.notification = NotificationCenter.default.addObserver(forName: NSNotification.Name.NEVPNStatusDidChange, object: site.manager!.connection , queue: nil) { n in
            let connected = self.site.connected
            self.site.status = statusString[self.site.manager!.connection.status]
            self.site.connected = statusMap[self.site.manager!.connection.status]

            // Check to see if we just moved to connected and if we have a start function to call when that happens
            if self.site.connected! && connected != self.site.connected && self.startFunc != nil {
                self.startFunc!()
                self.startFunc = nil
            }

            self.update(connected: self.site.connected!)
        }
#endif
        return nil
    }

    /// onCancel is called when the flutter listener stops listening
    func onCancel(withArguments arguments: Any?) -> FlutterError? {
        if (self.notification != nil) {
            NotificationCenter.default.removeObserver(self.notification!)
        }
        return nil
    }
    
    /// update is a way to send information to the flutter listener and generally should not be used directly
    func update(connected: Bool, replaceSite: Site? = nil) {
        if (replaceSite != nil) {
            site = replaceSite!
        }
        site.connected = connected
        site.status = connected ? "Connected" : "Disconnected"
        
        let encoder = JSONEncoder()
        let data = try! encoder.encode(site)
        self.eventSink?(String(data: data, encoding: .utf8))
    }
    
    private func configUpdated() {
        if self.site.connected != true {
            return
        }
        
        guard let newSite = try? Site(manager: self.site.manager!) else {
            return
        }
            
        self.update(connected: newSite.connected ?? false, replaceSite: newSite)
    }
}