Implemented GetPeers intent

Signed-off-by: Alessio Nossa <alessio.nossa@gmail.com>

Sources/WireGuardIntentsExtension/Info.plist

@@ -28,8 +28,12 @@
 		<dict>
 			<key>IntentsRestrictedWhileLocked</key>
 			<array/>
-			<key>IntentsSupported</key>
+			<key>IntentsRestrictedWhileProtectedDataUnavailable</key>
 			<array/>
+			<key>IntentsSupported</key>
+			<array>
+				<string>GetPeersIntent</string>
+			</array>
 		</dict>
 		<key>NSExtensionPointIdentifier</key>
 		<string>com.apple.intents-service</string>

Sources/WireGuardIntentsExtension/IntentHandler.swift

@@ -5,11 +5,17 @@ import Intents
 
 class IntentHandler: INExtension {
 
-    override func handler(for intent: INIntent) -> Any {
-        // This is the default implementation.  If you want different objects to handle different intents,
-        // you can override this and return the handler you want for that particular intent.
+    override init() {
+        super.init()
+        Logger.configureGlobal(tagged: "INTENTS", withFilePath: FileManager.logFileURL?.path)
+    }
 
-        return self
+    override func handler(for intent: INIntent) -> Any {
+        guard intent is GetPeersIntent else {
+            fatalError("Unhandled intent type: \(intent)")
+        }
+
+        return IntentHandling()
     }
 
 }

Sources/WireGuardIntentsExtension/IntentHandling.swift

@@ -0,0 +1,120 @@
+// SPDX-License-Identifier: MIT
+// Copyright © 2018-2021 WireGuard LLC. All Rights Reserved.
+
+import Intents
+
+class IntentHandling: NSObject {
+
+    public enum IntentError: Error {
+        case failedDecode
+        case wrongTunnel
+        case unknown
+    }
+
+    var tunnelsManager: TunnelsManager?
+
+    var onTunnelsManagerReady: ((TunnelsManager) -> Void)?
+
+    override init() {
+        super.init()
+
+        TunnelsManager.create { [weak self] result in
+            guard let self = self else { return }
+
+            switch result {
+            case .failure(let error):
+                wg_log(.error, message: error.localizedDescription)
+            case .success(let tunnelsManager):
+                self.tunnelsManager = tunnelsManager
+
+                self.onTunnelsManagerReady?(tunnelsManager)
+                self.onTunnelsManagerReady = nil
+            }
+        }
+    }
+
+    init(tunnelsManager: TunnelsManager) {
+        super.init()
+
+        self.tunnelsManager = tunnelsManager
+    }
+}
+
+extension IntentHandling {
+
+    private func allTunnelNames(completion: @escaping ([String]?) -> Void) {
+        let getTunnelsNameBlock: (TunnelsManager) -> Void = { tunnelsManager in
+            let tunnelsNames = tunnelsManager.mapTunnels { $0.name }
+            return completion(tunnelsNames)
+        }
+
+        if let tunnelsManager = tunnelsManager {
+            getTunnelsNameBlock(tunnelsManager)
+        } else {
+            if onTunnelsManagerReady != nil {
+                wg_log(.error, message: "Overriding onTunnelsManagerReady action in allTunnelNames function. This should not happen.")
+            }
+            onTunnelsManagerReady = getTunnelsNameBlock
+        }
+    }
+
+    private func allTunnelPeers(for tunnelName: String, completion: @escaping (Result<[String], IntentError>) -> Void) {
+        let getPeersFromConfigBlock: (TunnelsManager) -> Void = { tunnelsManager in
+            guard let tunnel = tunnelsManager.tunnel(named: tunnelName) else {
+                return completion(.failure(.wrongTunnel))
+            }
+
+            guard let publicKeys = tunnel.tunnelConfiguration?.peers.map({ $0.publicKey.base64Key }) else {
+                return completion(.failure(.unknown))
+            }
+            return completion(.success(publicKeys))
+        }
+
+        if let tunnelsManager = tunnelsManager {
+            getPeersFromConfigBlock(tunnelsManager)
+        } else {
+            if onTunnelsManagerReady != nil {
+                wg_log(.error, message: "Overriding onTunnelsManagerReady action in allTunnelPeers function. This should not happen.")
+            }
+            onTunnelsManagerReady = getPeersFromConfigBlock
+        }
+    }
+}
+
+extension IntentHandling: GetPeersIntentHandling {
+
+    @available(iOSApplicationExtension 14.0, *)
+    func provideTunnelOptionsCollection(for intent: GetPeersIntent, with completion: @escaping (INObjectCollection<NSString>?, Error?) -> Void) {
+
+        self.allTunnelNames { tunnelsNames in
+            let tunnelsNamesObjects = (tunnelsNames ?? []).map { NSString(string: $0) }
+
+            let objectCollection = INObjectCollection(items: tunnelsNamesObjects)
+            completion(objectCollection, nil)
+        }
+    }
+
+    func handle(intent: GetPeersIntent, completion: @escaping (GetPeersIntentResponse) -> Void) {
+        guard let tunnel = intent.tunnel else {
+            return completion(GetPeersIntentResponse(code: .failure, userActivity: nil))
+        }
+
+        self.allTunnelPeers(for: tunnel) { peersResult in
+            switch peersResult {
+            case .success(let peers):
+                let response = GetPeersIntentResponse(code: .success, userActivity: nil)
+                response.peersPublicKeys = peers
+                completion(response)
+
+            case .failure(let error):
+                switch error {
+                case .wrongTunnel:
+                    completion(GetPeersIntentResponse(code: .wrongTunnel, userActivity: nil))
+                default:
+                    completion(GetPeersIntentResponse(code: .failure, userActivity: nil))
+                }
+            }
+        }
+    }
+
+}

Sources/WireGuardIntentsExtension/WireGuardIntentsExtension_iOS.entitlements

@@ -2,6 +2,10 @@
 <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
 <plist version="1.0">
 <dict>
+	<key>com.apple.developer.networking.networkextension</key>
+	<array>
+		<string>packet-tunnel-provider</string>
+	</array>
 	<key>com.apple.security.application-groups</key>
 	<array>
 		<string>group.$(APP_ID_IOS)</string>