553 lines
20 KiB
Swift
553 lines
20 KiB
Swift
//
|
|
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import UIKit
|
|
|
|
@objc
|
|
public class OnboardingCountryState: NSObject {
|
|
public let countryName: String
|
|
public let callingCode: String
|
|
public let countryCode: String
|
|
|
|
@objc
|
|
public init(countryName: String,
|
|
callingCode: String,
|
|
countryCode: String) {
|
|
self.countryName = countryName
|
|
self.callingCode = callingCode
|
|
self.countryCode = countryCode
|
|
}
|
|
|
|
public static var defaultValue: OnboardingCountryState {
|
|
AssertIsOnMainThread()
|
|
|
|
var countryCode: String = PhoneNumber.defaultCountryCode()
|
|
if let lastRegisteredCountryCode = OnboardingController.lastRegisteredCountryCode(),
|
|
lastRegisteredCountryCode.count > 0 {
|
|
countryCode = lastRegisteredCountryCode
|
|
}
|
|
|
|
let callingCodeNumber: NSNumber = PhoneNumberUtil.sharedThreadLocal().nbPhoneNumberUtil.getCountryCode(forRegion: countryCode)
|
|
let callingCode = "\(COUNTRY_CODE_PREFIX)\(callingCodeNumber)"
|
|
|
|
var countryName = NSLocalizedString("UNKNOWN_COUNTRY_NAME", comment: "Label for unknown countries.")
|
|
if let countryNameDerived = PhoneNumberUtil.countryName(fromCountryCode: countryCode) {
|
|
countryName = countryNameDerived
|
|
}
|
|
|
|
return OnboardingCountryState(countryName: countryName, callingCode: callingCode, countryCode: countryCode)
|
|
}
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
@objc
|
|
public class OnboardingPhoneNumber: NSObject {
|
|
public let e164: String
|
|
public let userInput: String
|
|
|
|
@objc
|
|
public init(e164: String,
|
|
userInput: String) {
|
|
self.e164 = e164
|
|
self.userInput = userInput
|
|
}
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
@objc
|
|
public class OnboardingController: NSObject {
|
|
|
|
// MARK: - Dependencies
|
|
|
|
private var tsAccountManager: TSAccountManager {
|
|
return TSAccountManager.sharedInstance()
|
|
}
|
|
|
|
private var accountManager: AccountManager {
|
|
return AppEnvironment.shared.accountManager
|
|
}
|
|
|
|
private var contactsManager: OWSContactsManager {
|
|
return Environment.shared.contactsManager
|
|
}
|
|
|
|
private var backup: OWSBackup {
|
|
return AppEnvironment.shared.backup
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
@objc
|
|
public override init() {
|
|
super.init()
|
|
}
|
|
|
|
// MARK: - Factory Methods
|
|
|
|
@objc
|
|
public func initialViewController() -> UIViewController {
|
|
AssertIsOnMainThread()
|
|
|
|
let view = OnboardingSplashViewController(onboardingController: self)
|
|
return view
|
|
}
|
|
|
|
// MARK: - Transitions
|
|
|
|
public func onboardingSplashDidComplete(viewController: UIViewController) {
|
|
pushSeedVC(from: viewController)
|
|
}
|
|
|
|
public func onboardingPermissionsWasSkipped(viewController: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
pushSeedVC(from: viewController)
|
|
}
|
|
|
|
public func onboardingPermissionsDidComplete(viewController: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
pushSeedVC(from: viewController)
|
|
}
|
|
|
|
public func pushSeedVC(from viewController: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
let seedVC = SeedVC(onboardingController: self)
|
|
viewController.navigationController?.pushViewController(seedVC, animated: true)
|
|
}
|
|
|
|
public func pushDisplayNameVC(from viewController: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
let displayNameVC = DisplayNameVC(onboardingController: self)
|
|
viewController.navigationController?.pushViewController(displayNameVC, animated: true)
|
|
}
|
|
|
|
public func onboardingRegistrationSucceeded(viewController: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
let view = OnboardingVerificationViewController(onboardingController: self)
|
|
viewController.navigationController?.pushViewController(view, animated: true)
|
|
}
|
|
|
|
public func onboardingDidRequireCaptcha(viewController: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
guard let navigationController = viewController.navigationController else {
|
|
owsFailDebug("Missing navigationController.")
|
|
return
|
|
}
|
|
|
|
// The service could demand CAPTCHA from the "phone number" view or later
|
|
// from the "code verification" view. The "Captcha" view should always appear
|
|
// immediately after the "phone number" view.
|
|
while navigationController.viewControllers.count > 1 &&
|
|
!(navigationController.topViewController is DisplayNameVC) {
|
|
navigationController.popViewController(animated: false)
|
|
}
|
|
|
|
let view = OnboardingCaptchaViewController(onboardingController: self)
|
|
navigationController.pushViewController(view, animated: true)
|
|
}
|
|
|
|
@objc
|
|
public func verificationDidComplete(fromView view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
// At this point, the user has been prompted for contact access
|
|
// and has valid service credentials.
|
|
// We start the contact fetch/intersection now so that by the time
|
|
// they get to HomeView we can show meaningful contact in the suggested
|
|
// contact bubble.
|
|
contactsManager.fetchSystemContactsOnceIfAlreadyAuthorized()
|
|
|
|
if tsAccountManager.isReregistering() {
|
|
showHomeView(view: view)
|
|
} else {
|
|
checkCanImportBackup(fromView: view)
|
|
}
|
|
}
|
|
|
|
private func showProfileView(fromView view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
guard let navigationController = view.navigationController else {
|
|
owsFailDebug("Missing navigationController")
|
|
return
|
|
}
|
|
|
|
ProfileViewController.present(forRegistration: navigationController)
|
|
}
|
|
|
|
private func showBackupRestoreView(fromView view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
guard let navigationController = view.navigationController else {
|
|
owsFailDebug("Missing navigationController")
|
|
return
|
|
}
|
|
|
|
let restoreView = BackupRestoreViewController()
|
|
navigationController.setViewControllers([restoreView], animated: true)
|
|
}
|
|
|
|
private func checkCanImportBackup(fromView view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
backup.checkCanImport({ (canImport) in
|
|
Logger.info("canImport: \(canImport)")
|
|
|
|
if (canImport) {
|
|
self.backup.setHasPendingRestoreDecision(true)
|
|
|
|
self.showBackupRestoreView(fromView: view)
|
|
} else {
|
|
self.showHomeView(view: view)
|
|
}
|
|
}, failure: { (_) in
|
|
self.showBackupCheckFailedAlert(fromView: view)
|
|
})
|
|
}
|
|
|
|
private func showBackupCheckFailedAlert(fromView view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
let alert = UIAlertController(title: NSLocalizedString("CHECK_FOR_BACKUP_FAILED_TITLE",
|
|
comment: "Title for alert shown when the app failed to check for an existing backup."),
|
|
message: NSLocalizedString("CHECK_FOR_BACKUP_FAILED_MESSAGE",
|
|
comment: "Message for alert shown when the app failed to check for an existing backup."),
|
|
preferredStyle: .alert)
|
|
alert.addAction(UIAlertAction(title: NSLocalizedString("REGISTER_FAILED_TRY_AGAIN", comment: ""),
|
|
style: .default) { (_) in
|
|
self.checkCanImportBackup(fromView: view)
|
|
})
|
|
alert.addAction(UIAlertAction(title: NSLocalizedString("CHECK_FOR_BACKUP_DO_NOT_RESTORE", comment: "The label for the 'do not restore backup' button."),
|
|
style: .destructive) { (_) in
|
|
self.showHomeView(view: view)
|
|
})
|
|
view.presentAlert(alert)
|
|
}
|
|
|
|
public func onboardingDidRequire2FAPin(viewController: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
guard let navigationController = viewController.navigationController else {
|
|
owsFailDebug("Missing navigationController")
|
|
return
|
|
}
|
|
|
|
let view = Onboarding2FAViewController(onboardingController: self)
|
|
navigationController.pushViewController(view, animated: true)
|
|
}
|
|
|
|
@objc
|
|
public func profileWasSkipped(fromView view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
showHomeView(view: view)
|
|
}
|
|
|
|
@objc
|
|
public func profileDidComplete(fromView view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
Logger.info("")
|
|
|
|
showHomeView(view: view)
|
|
}
|
|
|
|
private func showHomeView(view: UIViewController) {
|
|
AssertIsOnMainThread()
|
|
|
|
guard let navigationController = view.navigationController else {
|
|
owsFailDebug("Missing navigationController")
|
|
return
|
|
}
|
|
|
|
// In production, this view will never be presented in a modal.
|
|
// During testing (debug UI, etc.), it may be a modal.
|
|
let isModal = navigationController.presentingViewController != nil
|
|
if isModal {
|
|
view.dismiss(animated: true, completion: {
|
|
SignalApp.shared().showHomeView()
|
|
})
|
|
} else {
|
|
SignalApp.shared().showHomeView()
|
|
}
|
|
}
|
|
|
|
// MARK: - State
|
|
|
|
public private(set) var countryState: OnboardingCountryState = .defaultValue
|
|
|
|
public private(set) var phoneNumber: OnboardingPhoneNumber?
|
|
|
|
public private(set) var captchaToken: String?
|
|
|
|
public private(set) var verificationCode: String?
|
|
|
|
public private(set) var twoFAPin: String?
|
|
|
|
@objc
|
|
public func update(countryState: OnboardingCountryState) {
|
|
AssertIsOnMainThread()
|
|
|
|
self.countryState = countryState
|
|
}
|
|
|
|
@objc
|
|
public func update(phoneNumber: OnboardingPhoneNumber) {
|
|
AssertIsOnMainThread()
|
|
|
|
self.phoneNumber = phoneNumber
|
|
}
|
|
|
|
@objc
|
|
public func update(captchaToken: String) {
|
|
AssertIsOnMainThread()
|
|
|
|
self.captchaToken = captchaToken
|
|
}
|
|
|
|
@objc
|
|
public func update(verificationCode: String) {
|
|
AssertIsOnMainThread()
|
|
|
|
self.verificationCode = verificationCode
|
|
}
|
|
|
|
@objc
|
|
public func update(twoFAPin: String) {
|
|
AssertIsOnMainThread()
|
|
|
|
self.twoFAPin = twoFAPin
|
|
}
|
|
|
|
// MARK: - Debug
|
|
|
|
private static let kKeychainService_LastRegistered = "kKeychainService_LastRegistered"
|
|
private static let kKeychainKey_LastRegisteredCountryCode = "kKeychainKey_LastRegisteredCountryCode"
|
|
private static let kKeychainKey_LastRegisteredPhoneNumber = "kKeychainKey_LastRegisteredPhoneNumber"
|
|
|
|
private class func debugValue(forKey key: String) -> String? {
|
|
AssertIsOnMainThread()
|
|
|
|
guard OWSIsDebugBuild() else {
|
|
return nil
|
|
}
|
|
|
|
do {
|
|
let value = try CurrentAppContext().keychainStorage().string(forService: kKeychainService_LastRegistered, key: key)
|
|
return value
|
|
} catch {
|
|
// The value may not be present in the keychain.
|
|
return nil
|
|
}
|
|
}
|
|
|
|
private class func setDebugValue(_ value: String, forKey key: String) {
|
|
AssertIsOnMainThread()
|
|
|
|
guard OWSIsDebugBuild() else {
|
|
return
|
|
}
|
|
|
|
do {
|
|
try CurrentAppContext().keychainStorage().set(string: value, service: kKeychainService_LastRegistered, key: key)
|
|
} catch {
|
|
owsFailDebug("Error: \(error)")
|
|
}
|
|
}
|
|
|
|
public class func lastRegisteredCountryCode() -> String? {
|
|
return debugValue(forKey: kKeychainKey_LastRegisteredCountryCode)
|
|
}
|
|
|
|
private class func setLastRegisteredCountryCode(value: String) {
|
|
setDebugValue(value, forKey: kKeychainKey_LastRegisteredCountryCode)
|
|
}
|
|
|
|
public class func lastRegisteredPhoneNumber() -> String? {
|
|
return debugValue(forKey: kKeychainKey_LastRegisteredPhoneNumber)
|
|
}
|
|
|
|
private class func setLastRegisteredPhoneNumber(value: String) {
|
|
setDebugValue(value, forKey: kKeychainKey_LastRegisteredPhoneNumber)
|
|
}
|
|
|
|
// MARK: - Registration
|
|
|
|
public func tryToRegister(fromViewController: UIViewController,
|
|
smsVerification: Bool) {
|
|
guard let phoneNumber = phoneNumber else {
|
|
owsFailDebug("Missing phoneNumber.")
|
|
return
|
|
}
|
|
|
|
// We eagerly update this state, regardless of whether or not the
|
|
// registration request succeeds.
|
|
OnboardingController.setLastRegisteredCountryCode(value: countryState.countryCode)
|
|
OnboardingController.setLastRegisteredPhoneNumber(value: phoneNumber.userInput)
|
|
|
|
let captchaToken = self.captchaToken
|
|
ModalActivityIndicatorViewController.present(fromViewController: fromViewController,
|
|
canCancel: true) { (modal) in
|
|
|
|
self.tsAccountManager.register(withPhoneNumber: phoneNumber.e164,
|
|
captchaToken: captchaToken,
|
|
success: {
|
|
DispatchQueue.main.async {
|
|
modal.dismiss(completion: {
|
|
self.registrationSucceeded(viewController: fromViewController)
|
|
})
|
|
}
|
|
}, failure: { (error) in
|
|
Logger.error("Error: \(error)")
|
|
|
|
DispatchQueue.main.async {
|
|
modal.dismiss(completion: {
|
|
self.registrationFailed(viewController: fromViewController, error: error as NSError)
|
|
})
|
|
}
|
|
}, smsVerification: smsVerification)
|
|
}
|
|
}
|
|
|
|
private func registrationSucceeded(viewController: UIViewController) {
|
|
onboardingRegistrationSucceeded(viewController: viewController)
|
|
}
|
|
|
|
private func registrationFailed(viewController: UIViewController, error: NSError) {
|
|
if error.code == 402 {
|
|
Logger.info("Captcha requested.")
|
|
|
|
onboardingDidRequireCaptcha(viewController: viewController)
|
|
} else if error.code == 400 {
|
|
OWSAlerts.showAlert(title: NSLocalizedString("REGISTRATION_ERROR", comment: ""),
|
|
message: NSLocalizedString("REGISTRATION_NON_VALID_NUMBER", comment: ""))
|
|
|
|
} else {
|
|
OWSAlerts.showAlert(title: error.localizedDescription,
|
|
message: error.localizedRecoverySuggestion)
|
|
}
|
|
}
|
|
|
|
// MARK: - Verification
|
|
|
|
public enum VerificationOutcome {
|
|
case success
|
|
case invalidVerificationCode
|
|
case invalid2FAPin
|
|
}
|
|
|
|
public func tryToVerify(fromViewController: UIViewController,
|
|
completion : @escaping (VerificationOutcome) -> Void) {
|
|
AssertIsOnMainThread()
|
|
|
|
guard let phoneNumber = phoneNumber else {
|
|
owsFailDebug("Missing phoneNumber.")
|
|
return
|
|
}
|
|
guard let verificationCode = verificationCode else {
|
|
completion(.invalidVerificationCode)
|
|
return
|
|
}
|
|
|
|
// Ensure the account manager state is up-to-date.
|
|
//
|
|
// TODO: We could skip this in production.
|
|
tsAccountManager.phoneNumberAwaitingVerification = phoneNumber.e164
|
|
|
|
let twoFAPin = self.twoFAPin
|
|
ModalActivityIndicatorViewController.present(fromViewController: fromViewController,
|
|
canCancel: true) { (modal) in
|
|
|
|
self.accountManager.register(verificationCode: verificationCode, pin: twoFAPin)
|
|
.done { (_) in
|
|
DispatchQueue.main.async {
|
|
modal.dismiss(completion: {
|
|
self.verificationDidComplete(fromView: fromViewController)
|
|
})
|
|
}
|
|
}.catch({ (error) in
|
|
Logger.error("Error: \(error)")
|
|
|
|
DispatchQueue.main.async {
|
|
modal.dismiss(completion: {
|
|
self.verificationFailed(fromViewController: fromViewController,
|
|
error: error as NSError,
|
|
completion: completion)
|
|
})
|
|
}
|
|
}).retainUntilComplete()
|
|
}
|
|
}
|
|
|
|
private func verificationFailed(fromViewController: UIViewController, error: NSError,
|
|
completion : @escaping (VerificationOutcome) -> Void) {
|
|
AssertIsOnMainThread()
|
|
|
|
if error.domain == OWSSignalServiceKitErrorDomain &&
|
|
error.code == OWSErrorCode.registrationMissing2FAPIN.rawValue {
|
|
|
|
Logger.info("Missing 2FA PIN.")
|
|
|
|
completion(.invalid2FAPin)
|
|
|
|
onboardingDidRequire2FAPin(viewController: fromViewController)
|
|
} else {
|
|
if error.domain == OWSSignalServiceKitErrorDomain &&
|
|
error.code == OWSErrorCode.userError.rawValue {
|
|
completion(.invalidVerificationCode)
|
|
}
|
|
|
|
Logger.verbose("error: \(error.domain) \(error.code)")
|
|
OWSAlerts.showAlert(title: NSLocalizedString("REGISTRATION_VERIFICATION_FAILED_TITLE", comment: "Alert view title"),
|
|
message: error.localizedDescription,
|
|
fromViewController: fromViewController)
|
|
}
|
|
}
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
public extension UIView {
|
|
public func addBottomStroke() -> UIView {
|
|
return addBottomStroke(color: Theme.middleGrayColor, strokeWidth: CGHairlineWidth())
|
|
}
|
|
|
|
public func addBottomStroke(color: UIColor, strokeWidth: CGFloat) -> UIView {
|
|
let strokeView = UIView()
|
|
strokeView.backgroundColor = color
|
|
addSubview(strokeView)
|
|
strokeView.autoSetDimension(.height, toSize: strokeWidth)
|
|
strokeView.autoPinWidthToSuperview()
|
|
strokeView.autoPinEdge(toSuperviewEdge: .bottom)
|
|
return strokeView
|
|
}
|
|
}
|