Dactyloidae/mobile/ios/Shared/Extensions/KeychainWrapperExtensions.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 XCGLogger
import SwiftKeychainWrapper

private let log = Logger.keychainLogger

public extension KeychainWrapper {
    static var sharedAppContainerKeychain: KeychainWrapper {
        let baseBundleIdentifier = AppInfo.baseBundleIdentifier
        let accessGroupPrefix = Bundle.main.object(forInfoDictionaryKey: "MozDevelopmentTeam") as! String
        let accessGroupIdentifier = AppInfo.keychainAccessGroupWithPrefix(accessGroupPrefix)
        return KeychainWrapper(serviceName: baseBundleIdentifier, accessGroup: accessGroupIdentifier)
    }
}

public extension KeychainWrapper {
    func ensureStringItemAccessibility(_ accessibility: SwiftKeychainWrapper.KeychainItemAccessibility, forKey key: String) {
        if self.hasValue(forKey: key) {
            if self.accessibilityOfKey(key) != .afterFirstUnlock {
                log.debug("updating item \(key) with \(accessibility)")

                guard let value = self.string(forKey: key) else {
                    log.error("failed to get item \(key)")
                    return
                }

                if !self.removeObject(forKey: key) {
                    log.warning("failed to remove item \(key)")
                }

                if !self.set(value, forKey: key, withAccessibility: accessibility) {
                    log.warning("failed to update item \(key)")
                }
            }
        }
    }

    func ensureObjectItemAccessibility(_ accessibility: SwiftKeychainWrapper.KeychainItemAccessibility, forKey key: String) {
        if self.hasValue(forKey: key) {
            if self.accessibilityOfKey(key) != .afterFirstUnlock {
                log.debug("updating item \(key) with \(accessibility)")

                guard let value = self.object(forKey: key) else {
                    log.error("failed to get item \(key)")
                    return
                }

                if !self.removeObject(forKey: key) {
                    log.warning("failed to remove item \(key)")
                }

                if !self.set(value, forKey: key, withAccessibility: accessibility) {
                    log.warning("failed to update item \(key)")
                }
            }
        }
    }
}