Dactyloidae/mobile/ios/Client/Frontend/AuthenticationManager/ChangePasscodeViewController.swift

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

import Foundation
import SwiftKeychainWrapper
import Shared

let NotificationPasscodeDidChange   = "NotificationPasscodeDidChange"

/// Displayed to the user when changing an existing passcode.
class ChangePasscodeViewController: PagingPasscodeViewController, PasscodeInputViewDelegate {
    fileprivate var newPasscode: String?
    fileprivate var oldPasscode: String?

    override init() {
        super.init()
        self.title = AuthenticationStrings.changePasscode
        self.panes = [
            PasscodePane(title: AuthenticationStrings.enterPasscode, passcodeSize: authenticationInfo?.passcode?.count ?? 6),
            PasscodePane(title: AuthenticationStrings.enterNewPasscode, passcodeSize: 6),
            PasscodePane(title: AuthenticationStrings.reenterPasscode, passcodeSize: 6),
        ]
    }
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(animated)
        panes.forEach { $0.codeInputView.delegate = self }

        // Don't show the keyboard or allow typing if we're locked out. Also display the error.
        if authenticationInfo?.isLocked() ?? false {
            displayLockoutError()
            panes.first?.codeInputView.isUserInteractionEnabled = false
        } else {
            panes.first?.codeInputView.becomeFirstResponder()
        }
    }

    func passcodeInputView(_ inputView: PasscodeInputView, didFinishEnteringCode code: String) {
        switch currentPaneIndex {
        case 0:
            // Constraint: We need to make sure that the first passcode they've entered matches the one stored in the keychain
            if code != authenticationInfo?.passcode {
                panes[currentPaneIndex].shakePasscode()
                failIncorrectPasscode(inputView)
                return
            }
            oldPasscode = code
            authenticationInfo?.recordValidation()

            // Clear out any previous errors if we are allowed to proceed
            errorToast?.removeFromSuperview()
            scrollToNextAndSelect()
        case 1:
            // Constraint: The new passcode cannot match their old passcode.
            if oldPasscode == code {
                failMustBeDifferent()

                // Scroll back and reset the input fields
                resetAllInputFields()
                return
            }
            newPasscode = code
            errorToast?.removeFromSuperview()
            scrollToNextAndSelect()
        case 2:
            if newPasscode != code {
                failMismatchPasscode()

                // Scroll back and reset input fields
                resetAllInputFields()
                scrollToPreviousAndSelect()
                newPasscode = nil
                return
            }
            changePasscodeToCode(code)
            dismissAnimated()
        default:
            break
        }
    }

    fileprivate func changePasscodeToCode(_ code: String) {
        authenticationInfo?.updatePasscode(code)
        KeychainWrapper.sharedAppContainerKeychain.setAuthenticationInfo(authenticationInfo)
        let notificationCenter = NotificationCenter.default
        notificationCenter.post(name: Notification.Name(rawValue: NotificationPasscodeDidChange), object: nil)
    }
}