Create encryption keys export presenter

2026-04-19 08:03:50 +02:00 · 2019-02-15 19:04:03 +01:00
parent 6b1644d415
commit f747e38b8d
2 changed files with 159 additions and 0 deletions
@@ -0,0 +1,147 @@
+/*
+ Copyright 2019 New Vector Ltd
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ */
+
+import Foundation
+
+final class EncryptionKeysExportPresenter: NSObject {
+    
+    // MARK: - Constants
+    
+    private enum Constants {
+        static let keyExportFileName = "riot-keys.txt"
+    }
+    
+    // MARK: - Properties
+    
+    // MARK: Private
+
+    private let session: MXSession
+    private let activityViewPresenter: ActivityIndicatorPresenterType
+    private let keyExportFileURL: URL    
+    
+    private weak var presentingViewController: UIViewController?
+    private weak var sourceView: UIView?
+    private var encryptionKeysExportView: MXKEncryptionKeysExportView?
+    private var documentInteractionController: UIDocumentInteractionController?
+    
+    // MARK: Public
+    
+    // MARK: - Setup
+
+    init(session: MXSession) {
+        self.session = session
+        self.activityViewPresenter = ActivityIndicatorPresenter()
+        self.keyExportFileURL = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent(Constants.keyExportFileName)
+        super.init()
+    }
+    
+    deinit {
+        self.deleteKeyExportFile()
+    }
+    
+    // MARK: - Public
+
+    func present(from viewController: UIViewController, sourceView: UIView?) {
+        self.presentingViewController = viewController
+        self.sourceView = sourceView
+        
+        let keysExportView: MXKEncryptionKeysExportView = MXKEncryptionKeysExportView(matrixSession: self.session)
+
+        // Make sure the file is empty
+        self.deleteKeyExportFile()
+
+        keysExportView.show(in: viewController,
+                         toExportKeysToFile: self.keyExportFileURL,
+                         onLoading: { [weak self] (loading) in
+
+                            guard let sself = self else {
+                                return
+                            }
+                            
+                            if loading {
+                                sself.activityViewPresenter.removeCurrentActivityIndicator(animated: false)
+                                sself.activityViewPresenter.presentActivityIndicator(on: viewController.view, animated: true)
+                            } else {
+                                sself.activityViewPresenter.removeCurrentActivityIndicator(animated: true)
+                            }
+        }, onComplete: { [weak self] (success) in
+            guard let sself = self else {
+                return
+            }
+
+            guard success else {
+                sself.encryptionKeysExportView = nil
+                return
+            }
+
+            sself.presentInteractionDocumentController()
+        })
+        
+        self.encryptionKeysExportView = keysExportView
+    }
+    
+    // MARK: - Private
+
+    private func presentInteractionDocumentController() {
+        
+        let sourceRect: CGRect
+        
+        guard let presentingView = self.presentingViewController?.view else {
+            self.encryptionKeysExportView = nil
+            return
+        }
+        
+        if let sourceView = self.sourceView {
+            sourceRect = sourceView.convert(sourceView.bounds, to: presentingView)
+        } else {
+            sourceRect = presentingView.bounds
+        }
+        
+        let documentInteractionController = UIDocumentInteractionController(url: self.keyExportFileURL)
+        documentInteractionController.delegate = self
+
+        if documentInteractionController.presentOptionsMenu(from: sourceRect, in: presentingView, animated: true) {
+            self.documentInteractionController = documentInteractionController
+        } else {
+            self.encryptionKeysExportView = nil
+            self.deleteKeyExportFile()
+        }
+    }
+
+    @objc private func deleteKeyExportFile() {
+        
+        let fileManager = FileManager.default
+        
+        if fileManager.fileExists(atPath: self.keyExportFileURL.path) {
+            try? fileManager.removeItem(atPath: self.keyExportFileURL.path)
+        }
+    }
+}
+
+// MARK: - UIDocumentInteractionControllerDelegate
+extension EncryptionKeysExportPresenter: UIDocumentInteractionControllerDelegate {
+
+    // Note: This method is not called in all cases (see http://stackoverflow.com/a/21867096).
+    func documentInteractionController(_ controller: UIDocumentInteractionController, didEndSendingToApplication application: String?) {
+        self.deleteKeyExportFile()
+        self.documentInteractionController = nil
+    }
+    
+    func documentInteractionControllerDidDismissOptionsMenu(_ controller: UIDocumentInteractionController) {
+        self.encryptionKeysExportView = nil
+        self.documentInteractionController = nil
+    }
+}