alpha-wallet-ios/Trust/Lock/Lock.swift

// Copyright SIX DAY LLC. All rights reserved.

import UIKit
import SAMKeychain
import KeychainSwift

protocol LockInterface {
    func isPasscodeSet() -> Bool
}

class Lock: LockInterface {
    private struct Keys {
        static let service = "trust.lock"
        static let account = "trust.account"
    }
    private let passcodeAttempts = "passcodeAttempts"
    private let maxAttemptTime = "maxAttemptTime"
    private let keychain = KeychainSwift(keyPrefix: Constants.keychainKeyPrefix)
    func isPasscodeSet() -> Bool {
        return currentPasscode() != nil
    }
    func currentPasscode() -> String? {
        return SAMKeychain.password(forService: Keys.service, account: Keys.account)
    }
    func isPasscodeValid(passcode: String) -> Bool {
        return passcode == currentPasscode()
    }
    func setPasscode(passcode: String) {
        SAMKeychain.setPassword(passcode, forService: Keys.service, account: Keys.account)
    }
    func deletePasscode() {
        SAMKeychain.deletePassword(forService: Keys.service, account: Keys.account)
        resetPasscodeAttemptHistory()
    }
    func numberOfAttempts() -> Int {
        guard let attempts = keychain.get(passcodeAttempts) else {
            return 0
        }
        return Int(attempts)!
    }
    func resetPasscodeAttemptHistory() {
         keychain.delete(passcodeAttempts)
    }
    func recordIncorrectPasscodeAttempt() {
        var numberOfAttemptsSoFar = numberOfAttempts()
        numberOfAttemptsSoFar += 1
        keychain.set(String(numberOfAttemptsSoFar), forKey: passcodeAttempts)
    }
    func recordedMaxAttemptTime() -> Date {
        //This method is called only when we knew that maxAttemptTime is set. So no worries with !.
        let timeString = keychain.get(maxAttemptTime)!
        return dateFormatter().date(from: timeString)!
    }
    func incorrectMaxAttemptTimeIsSet() -> Bool {
        guard let timeString = keychain.get(maxAttemptTime), !timeString.isEmpty  else {
            return false
        }
        return true
    }
    func recordIncorrectMaxAttemptTime() {
        let timeString = dateFormatter().string(from: Date())
        keychain.set(timeString, forKey: maxAttemptTime)
    }
    func removeIncorrectMaxAttemptTime() {
        keychain.delete(maxAttemptTime)
    }
    private func dateFormatter() -> DateFormatter {
        let dateFormatter = DateFormatter()
        dateFormatter.dateStyle = DateFormatter.Style.short
        dateFormatter.timeStyle = DateFormatter.Style.short
        return dateFormatter
    }
    func clear() {
        deletePasscode()
        resetPasscodeAttemptHistory()
        removeIncorrectMaxAttemptTime()
    }
}