yggdrasil-ios/Yggdrasil Network Cross-Platform/ConfigurationProxy.swift

//
//  ConfigurationProxy.swift
//  YggdrasilNetwork
//
//  Created by Neil Alexander on 07/01/2019.
//

#if canImport(UIKit)
import UIKit
#elseif canImport(AppKit)
import AppKit
#endif
import SwiftUI
import Yggdrasil
import NetworkExtension
import Foundation
import CoreData
import Combine

#if os(iOS)
class PlatformItemSource: NSObject, UIActivityItemSource {
    func activityViewControllerPlaceholderItem(_ activityViewController: UIActivityViewController) -> Any {
        return "yggdrasil.conf"
    }
    
    func activityViewController(_ activityViewController: UIActivityViewController, itemForActivityType activityType: UIActivity.ActivityType?) -> Any? {
        return nil
    }
}
#elseif os(OSX)
class PlatformItemSource: NSObject {}
#endif

class ConfigurationProxy: PlatformItemSource {
    private var manager: NETunnelProviderManager?
    private var json: Data? = nil {
        didSet {
            //summary = MobileSummaryFromConfig(json)
        }
    }
    private var dict: [String: Any]? = [:]
    private var timer: Timer?
    //public var summary: MobileParsedConfig?
    
    init(manager: NETunnelProviderManager) {
        self.manager = manager
        super.init()
        
        self.json = MobileGenerateConfigJSON()
        do {
            try self.convertToDict()
        } catch {
            NSLog("ConfigurationProxy: Error deserialising JSON (\(error))")
        }
#if os(iOS)
        self.set("name", inSection: "NodeInfo", to: UIDevice.current.name)
#elseif os(OSX)
        self.set("name", inSection: "NodeInfo", to: Host.current().localizedName ?? "")
#endif
        self.fix()
    }
    
    init(json: Data, manager: NETunnelProviderManager? = nil) throws {
        self.manager = manager
        super.init()
        
        self.json = json
        try self.convertToDict()
        self.fix()
    }
    
    private func fix() {
        self.set("Listen", to: [] as [String])
        self.set("AdminListen", to: "none")
        self.set("IfName", to: "dummy")
        
        if self.get("AutoStart") == nil {
            self.set("AutoStart", to: ["Any": false, "WiFi": false, "Mobile": false, "Ethernet": false] as [String: Bool])
        }
        
        let multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
        if multicastInterfaces.count != 1 {
            self.set("MulticastInterfaces", to: [
                [
                    "Regex": "en.*",
                    "Beacon": true,
                    "Listen": true,
                ] as [String : Any]
            ])
        }
    }
    
    public var multicastBeacons: Bool {
        get {
            let multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
            if multicastInterfaces.count == 0 {
                return false
            }
            return multicastInterfaces[0]["Beacon"] as? Bool ?? true
        }
        set {
            var multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
            multicastInterfaces[0]["Beacon"] = newValue
            self.set("MulticastInterfaces", to: multicastInterfaces)
            self.saveSoon()
        }
    }
    
    public var multicastListen: Bool {
        get {
            let multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
            if multicastInterfaces.count == 0 {
                return false
            }
            return multicastInterfaces[0]["Listen"] as? Bool ?? true
        }
        set {
            var multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
            multicastInterfaces[0]["Listen"] = newValue
            self.set("MulticastInterfaces", to: multicastInterfaces)
            self.saveSoon()
        }
    }
    
    public var multicastPassword: String {
        get {
            let multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
            if multicastInterfaces.count == 0 {
                return ""
            }
            return multicastInterfaces[0]["Password"] as? String ?? ""
        }
        set {
            var multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
            multicastInterfaces[0]["Password"] = newValue
            self.set("MulticastInterfaces", to: multicastInterfaces)
            self.saveSoon()
        }
    }
    
    public var autoStartAny: Bool {
        get {
            return self.get("Any", inSection: "AutoStart") as? Bool ?? false
        }
        set {
            self.set("Any", inSection: "AutoStart", to: newValue)
            self.saveSoon()
        }
    }
    
    public var autoStartWiFi: Bool {
        get {
            return self.get("WiFi", inSection: "AutoStart") as? Bool ?? false
        }
        set {
            self.set("WiFi", inSection: "AutoStart", to: newValue)
            self.saveSoon()
        }
    }
    
    public var autoStartEthernet: Bool {
        get {
            return self.get("Ethernet", inSection: "AutoStart") as? Bool ?? false
        }
        set {
            self.set("Ethernet", inSection: "AutoStart", to: newValue)
            self.saveSoon()
        }
    }
    
    public var autoStartMobile: Bool {
        get {
            return self.get("Mobile", inSection: "AutoStart") as? Bool ?? false
        }
        set {
            self.set("Mobile", inSection: "AutoStart", to: newValue)
            self.saveSoon()
        }
    }
    
    public var deviceName: String {
        get {
            return self.get("name", inSection: "NodeInfo") as? String ?? ""
        }
        set {
            self.set("name", inSection: "NodeInfo", to: newValue)
            self.saveSoon()
        }
    }
    
    public var peers: [String] {
        get {
            return self.get("Peers") as? [String] ?? []
        }
        set {
            self.set("Peers", to: newValue)
            self.saveSoon()
        }
    }
    
    private func get(_ key: String) -> Any? {
        if let dict = self.dict {
            if dict.keys.contains(key) {
                return dict[key]
            }
        }
        return nil
    }
    
    private func get(_ key: String, inSection section: String) -> Any? {
        if let dict = self.get(section) as? [String: Any] {
            if dict.keys.contains(key) {
                return dict[key]
            }
        }
        return nil
    }
    
    private func add(_ value: Any, in key: String) {
        if self.dict != nil {
            if self.dict![key] as? [Any] != nil {
                var temp = self.dict![key] as? [Any] ?? []
                temp.append(value)
                self.dict!.updateValue(temp, forKey: key)
            }
        }
    }
    
    private func remove(_ value: String, from key: String) {
        if self.dict != nil {
            if self.dict![key] as? [String] != nil {
                var temp = self.dict![key] as? [String] ?? []
                if let index = temp.firstIndex(of: value) {
                    temp.remove(at: index)
                }
                self.dict!.updateValue(temp, forKey: key)
            }
        }
    }
    
    private func remove(index: Int, from key: String) {
        if self.dict != nil {
            if self.dict![key] as? [Any] != nil {
                var temp = self.dict![key] as? [Any] ?? []
                temp.remove(at: index)
                self.dict!.updateValue(temp, forKey: key)
            }
        }
    }
    
    private func set(_ key: String, to value: Any) {
        if self.dict != nil {
            self.dict![key] = value
        }
    }
    
    private func set(_ key: String, inSection section: String, to value: Any?) {
        if self.dict != nil {
            if self.dict!.keys.contains(section), let value = value {
                var temp = self.dict![section] as? [String: Any] ?? [:]
                temp.updateValue(value, forKey: key)
                self.dict!.updateValue(temp, forKey: section)
            }
        }
    }
    
    func data() -> Data? {
        do {
            try self.convertToJson()
            return self.json
        } catch {
            return nil
        }
    }
    
    private func saveSoon() {
        self.timer?.invalidate()
        self.timer = Timer.scheduledTimer(
            timeInterval: 1.0,
            target: self,
            selector: #selector(saveFromTimer),
            userInfo: nil,
            repeats: false
        )
    }
    
    @objc private func saveFromTimer() {
        if var manager = self.manager {
            try? self.save(to: &manager)
        }
        self.timer = nil
    }
    
    func save(to manager: inout NETunnelProviderManager) throws {
        self.fix()
        if let data = self.data() {
            let providerProtocol = NETunnelProviderProtocol()
            providerProtocol.providerBundleIdentifier = "eu.neilalexander.yggdrasil.extension"
            providerProtocol.providerConfiguration = [ "json": data ]
            providerProtocol.serverAddress = "yggdrasil"
            providerProtocol.username = self.get("PublicKey") as? String ?? self.get("SigningPublicKey") as? String ?? "(unknown public key)"
            
            NSLog(String(data: data, encoding: .utf8) ?? "(unknown)")
            
            let disconnectrule = NEOnDemandRuleDisconnect()
            var rules: [NEOnDemandRule] = [disconnectrule]
            if self.get("Any", inSection: "AutoStart") as? Bool ?? false {
                let wifirule = NEOnDemandRuleConnect()
                wifirule.interfaceTypeMatch = .any
                rules.insert(wifirule, at: 0)
            }
#if os(macOS)
            if self.get("Ethernet", inSection: "AutoStart") as? Bool ?? false {
                let wifirule = NEOnDemandRuleConnect()
                wifirule.interfaceTypeMatch = .ethernet
                rules.insert(wifirule, at: 0)
            }
#endif
            if self.get("WiFi", inSection: "AutoStart") as? Bool ?? false {
                let wifirule = NEOnDemandRuleConnect()
                wifirule.interfaceTypeMatch = .wiFi
                rules.insert(wifirule, at: 0)
            }
#if canImport(UIKit)
            if self.get("Mobile", inSection: "AutoStart") as? Bool ?? false {
                let mobilerule = NEOnDemandRuleConnect()
                mobilerule.interfaceTypeMatch = .cellular
                rules.insert(mobilerule, at: 0)
            }
#endif
            manager.onDemandRules = rules
            manager.isOnDemandEnabled = rules.count > 1
            providerProtocol.disconnectOnSleep = rules.count > 1
            
            manager.protocolConfiguration = providerProtocol
            
            manager.saveToPreferences(completionHandler: { error in
                if let error = error {
                    print(error)
                } else {
                    print("Saved successfully")
                }
            })
        }
    }
    
    private func convertToDict() throws {
        self.dict = try JSONSerialization.jsonObject(with: self.json!, options: []) as? [String: Any]
    }
    
    private func convertToJson() throws {
        self.json = try JSONSerialization.data(withJSONObject: self.dict as Any, options: .prettyPrinted)
    }
    
#if canImport(UIKit)
    override func activityViewControllerPlaceholderItem(_ activityViewController: UIActivityViewController) -> Any {
        return "yggdrasil.conf"
    }
    
    override func activityViewController(_ activityViewController: UIActivityViewController, itemForActivityType activityType: UIActivity.ActivityType?) -> Any? {
        return self.data()
    }
    
    func activityViewController(_ activityViewController: UIActivityViewController, subjectForActivityType activityType: UIActivity.ActivityType?) -> String {
        if let pubkey = self.get("PublicKey") as? String {
            return "yggdrasil-\(pubkey).conf.json"
        }
        return "yggdrasil.conf.json"
    }
#endif
}