2019-01-30 10:08:01 +00:00
|
|
|
/*
|
|
|
|
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 KeyBackupRecoverCoordinator: KeyBackupRecoverCoordinatorType {
|
|
|
|
|
|
|
|
// MARK: - Properties
|
|
|
|
|
|
|
|
// MARK: Private
|
|
|
|
|
|
|
|
private let session: MXSession
|
|
|
|
private let navigationRouter: NavigationRouterType
|
|
|
|
private let keyBackupVersion: MXKeyBackupVersion
|
|
|
|
|
|
|
|
// MARK: Public
|
|
|
|
|
|
|
|
var childCoordinators: [Coordinator] = []
|
|
|
|
|
|
|
|
weak var delegate: KeyBackupRecoverCoordinatorDelegate?
|
|
|
|
|
|
|
|
// MARK: - Setup
|
|
|
|
|
2020-06-15 14:14:31 +00:00
|
|
|
init(session: MXSession, keyBackupVersion: MXKeyBackupVersion, navigationRouter: NavigationRouterType? = nil) {
|
2019-01-30 10:08:01 +00:00
|
|
|
self.session = session
|
|
|
|
self.keyBackupVersion = keyBackupVersion
|
2020-06-15 14:14:31 +00:00
|
|
|
|
|
|
|
if let navigationRouter = navigationRouter {
|
|
|
|
self.navigationRouter = navigationRouter
|
|
|
|
} else {
|
|
|
|
self.navigationRouter = NavigationRouter(navigationController: RiotNavigationController())
|
|
|
|
}
|
2019-01-30 10:08:01 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// MARK: - Public
|
2020-06-15 14:14:31 +00:00
|
|
|
|
2019-01-30 10:08:01 +00:00
|
|
|
func start() {
|
2020-06-15 14:14:31 +00:00
|
|
|
|
2019-01-30 10:08:01 +00:00
|
|
|
let rootCoordinator: Coordinator & Presentable
|
|
|
|
|
2020-04-08 12:09:55 +00:00
|
|
|
// Check if we have the private key locally
|
|
|
|
if self.session.crypto.backup.hasPrivateKeyInCryptoStore {
|
2020-04-08 14:48:12 +00:00
|
|
|
rootCoordinator = self.createRecoverFromPrivateKeyCoordinator()
|
2019-01-30 10:08:01 +00:00
|
|
|
} else {
|
2020-04-08 14:48:12 +00:00
|
|
|
rootCoordinator = self.createRecoverWithUserInteractionCoordinator()
|
2019-01-30 10:08:01 +00:00
|
|
|
}
|
2020-06-15 14:14:31 +00:00
|
|
|
|
2019-01-30 10:08:01 +00:00
|
|
|
rootCoordinator.start()
|
|
|
|
|
|
|
|
self.add(childCoordinator: rootCoordinator)
|
|
|
|
|
2020-06-15 14:14:31 +00:00
|
|
|
if self.navigationRouter.modules.isEmpty == false {
|
|
|
|
self.navigationRouter.push(rootCoordinator, animated: true, popCompletion: { [weak self] in
|
|
|
|
self?.remove(childCoordinator: rootCoordinator)
|
|
|
|
})
|
|
|
|
} else {
|
|
|
|
self.navigationRouter.setRootModule(rootCoordinator) { [weak self] in
|
|
|
|
self?.remove(childCoordinator: rootCoordinator)
|
|
|
|
}
|
|
|
|
}
|
2019-01-30 10:08:01 +00:00
|
|
|
}
|
|
|
|
|
2020-06-15 14:14:31 +00:00
|
|
|
|
2019-01-30 10:08:01 +00:00
|
|
|
func toPresentable() -> UIViewController {
|
|
|
|
return self.navigationRouter.toPresentable()
|
|
|
|
}
|
|
|
|
|
|
|
|
// MARK: - Private
|
|
|
|
|
2020-04-08 14:48:12 +00:00
|
|
|
private func createRecoverWithUserInteractionCoordinator() -> Coordinator & Presentable {
|
|
|
|
let coordinator: Coordinator & Presentable
|
|
|
|
|
|
|
|
// Check if a passphrase has been set for given backup
|
|
|
|
if let megolmBackupAuthData = MXMegolmBackupAuthData(fromJSON: self.keyBackupVersion.authData), megolmBackupAuthData.privateKeySalt != nil {
|
|
|
|
coordinator = self.createRecoverFromPassphraseCoordinator()
|
|
|
|
} else {
|
|
|
|
coordinator = self.createRecoverFromRecoveryKeyCoordinator()
|
|
|
|
}
|
|
|
|
return coordinator
|
|
|
|
}
|
|
|
|
|
2020-04-08 12:09:55 +00:00
|
|
|
private func createRecoverFromPrivateKeyCoordinator() -> KeyBackupRecoverFromPrivateKeyCoordinator {
|
|
|
|
let coordinator = KeyBackupRecoverFromPrivateKeyCoordinator(keyBackup: self.session.crypto.backup, keyBackupVersion: self.keyBackupVersion)
|
|
|
|
coordinator.delegate = self
|
|
|
|
return coordinator
|
|
|
|
}
|
|
|
|
|
2019-01-30 10:08:01 +00:00
|
|
|
private func createRecoverFromPassphraseCoordinator() -> KeyBackupRecoverFromPassphraseCoordinator {
|
2019-01-30 14:55:25 +00:00
|
|
|
let coordinator = KeyBackupRecoverFromPassphraseCoordinator(keyBackup: self.session.crypto.backup, keyBackupVersion: self.keyBackupVersion)
|
2019-01-30 10:08:01 +00:00
|
|
|
coordinator.delegate = self
|
|
|
|
return coordinator
|
|
|
|
}
|
|
|
|
|
|
|
|
private func createRecoverFromRecoveryKeyCoordinator() -> KeyBackupRecoverFromRecoveryKeyCoordinator {
|
2019-01-30 14:55:25 +00:00
|
|
|
let coordinator = KeyBackupRecoverFromRecoveryKeyCoordinator(keyBackup: self.session.crypto.backup, keyBackupVersion: self.keyBackupVersion)
|
2019-01-30 10:08:01 +00:00
|
|
|
coordinator.delegate = self
|
|
|
|
return coordinator
|
|
|
|
}
|
|
|
|
|
|
|
|
private func showRecoverFromRecoveryKey() {
|
|
|
|
let coordinator = self.createRecoverFromRecoveryKeyCoordinator()
|
|
|
|
|
|
|
|
self.add(childCoordinator: coordinator)
|
|
|
|
|
|
|
|
self.navigationRouter.push(coordinator, animated: true) {
|
|
|
|
self.remove(childCoordinator: coordinator)
|
|
|
|
}
|
|
|
|
|
|
|
|
coordinator.start()
|
|
|
|
}
|
|
|
|
|
|
|
|
private func showRecoverSuccess() {
|
|
|
|
let keyBackupRecoverSuccessViewController = KeyBackupRecoverSuccessViewController.instantiate()
|
|
|
|
keyBackupRecoverSuccessViewController.delegate = self
|
|
|
|
self.navigationRouter.push(keyBackupRecoverSuccessViewController, animated: true, popCompletion: nil)
|
|
|
|
}
|
2020-04-08 14:48:12 +00:00
|
|
|
|
|
|
|
private func showRecoverFallback() {
|
|
|
|
let coordinator = self.createRecoverWithUserInteractionCoordinator()
|
|
|
|
self.add(childCoordinator: coordinator)
|
|
|
|
|
|
|
|
// Skip the previously displayed KeyBackupRecoverFromPrivateKeyCoordinator in the navigation stack
|
|
|
|
self.navigationRouter.setRootModule(coordinator)
|
|
|
|
|
|
|
|
coordinator.start()
|
|
|
|
}
|
2019-01-30 10:08:01 +00:00
|
|
|
}
|
|
|
|
|
2020-04-08 12:09:55 +00:00
|
|
|
// MARK: - KeyBackupRecoverFromPassphraseCoordinatorDelegate
|
|
|
|
extension KeyBackupRecoverCoordinator: KeyBackupRecoverFromPrivateKeyCoordinatorDelegate {
|
2020-04-08 14:48:12 +00:00
|
|
|
|
2020-04-08 12:09:55 +00:00
|
|
|
func keyBackupRecoverFromPrivateKeyCoordinatorDidRecover(_ coordinator: KeyBackupRecoverFromPrivateKeyCoordinatorType) {
|
|
|
|
self.showRecoverSuccess()
|
|
|
|
}
|
|
|
|
|
2020-04-08 14:48:12 +00:00
|
|
|
func keyBackupRecoverFromPrivateKeyCoordinatorDidPrivateKeyFail(_ coordinator: KeyBackupRecoverFromPrivateKeyCoordinatorType) {
|
|
|
|
// The private key did not work. Ask the user to enter their passphrase or recovery key
|
|
|
|
self.showRecoverFallback()
|
|
|
|
}
|
|
|
|
|
2020-04-08 12:09:55 +00:00
|
|
|
func keyBackupRecoverFromPrivateKeyCoordinatorDidCancel(_ coordinator: KeyBackupRecoverFromPrivateKeyCoordinatorType) {
|
|
|
|
self.delegate?.keyBackupRecoverCoordinatorDidCancel(self)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-01-30 10:08:01 +00:00
|
|
|
// MARK: - KeyBackupRecoverFromPassphraseCoordinatorDelegate
|
|
|
|
extension KeyBackupRecoverCoordinator: KeyBackupRecoverFromPassphraseCoordinatorDelegate {
|
|
|
|
func keyBackupRecoverFromPassphraseCoordinatorDidRecover(_ keyBackupRecoverFromPassphraseCoordinator: KeyBackupRecoverFromPassphraseCoordinatorType) {
|
|
|
|
self.showRecoverSuccess()
|
|
|
|
}
|
|
|
|
|
|
|
|
func keyBackupRecoverFromPassphraseCoordinatorDoNotKnowPassphrase(_ keyBackupRecoverFromPassphraseCoordinator: KeyBackupRecoverFromPassphraseCoordinatorType) {
|
|
|
|
self.showRecoverFromRecoveryKey()
|
|
|
|
}
|
|
|
|
|
|
|
|
func keyBackupRecoverFromPassphraseCoordinatorDidCancel(_ keyBackupRecoverFromPassphraseCoordinator: KeyBackupRecoverFromPassphraseCoordinatorType) {
|
|
|
|
self.delegate?.keyBackupRecoverCoordinatorDidCancel(self)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// MARK: - KeyBackupRecoverFromRecoveryKeyCoordinatorDelegate
|
|
|
|
extension KeyBackupRecoverCoordinator: KeyBackupRecoverFromRecoveryKeyCoordinatorDelegate {
|
|
|
|
func keyBackupRecoverFromPassphraseCoordinatorDidRecover(_ keyBackupRecoverFromRecoveryKeyCoordinator: KeyBackupRecoverFromRecoveryKeyCoordinatorType) {
|
|
|
|
self.showRecoverSuccess()
|
|
|
|
}
|
|
|
|
|
|
|
|
func keyBackupRecoverFromPassphraseCoordinatorDidCancel(_ keyBackupRecoverFromRecoveryKeyCoordinator: KeyBackupRecoverFromRecoveryKeyCoordinatorType) {
|
|
|
|
self.delegate?.keyBackupRecoverCoordinatorDidCancel(self)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// MARK: - KeyBackupRecoverSuccessViewControllerDelegate
|
|
|
|
extension KeyBackupRecoverCoordinator: KeyBackupRecoverSuccessViewControllerDelegate {
|
2020-04-08 12:09:55 +00:00
|
|
|
func keyBackupRecoverSuccessViewControllerDidTapDone(_ keyBackupRecoverSuccessViewController: KeyBackupRecoverSuccessViewController) {
|
2019-01-30 10:08:01 +00:00
|
|
|
self.delegate?.keyBackupRecoverCoordinatorDidRecover(self)
|
|
|
|
}
|
|
|
|
}
|