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) } }