keychain-swift/keychain/TegKeychain.swift

//
//  TegKeychain.swift
//  keychain
//
//  Created by Evgenii Neumerzhitckii on 5/02/2015.
//  Copyright (c) 2015 The Exchange Group Pty Ltd. All rights reserved.
//

import UIKit
import Security

public class TegKeychain {
  
  public class func set(key: String, value: String) -> Bool {
    if let currentData = value.dataUsingEncoding(NSUTF8StringEncoding) {
      return set(key, value: currentData)
    }
    
    return false
  }

  public class func set(key: String, value: NSData) -> Bool {
    let query = [
      TegKeychainConstants.klass       : TegKeychainConstants.classGenericPassword,
      TegKeychainConstants.attrAccount : key,
      TegKeychainConstants.valueData   : value ]
    
    SecItemDelete(query as CFDictionaryRef)
    
    let status: OSStatus = SecItemAdd(query as CFDictionaryRef, nil)
    
    return status == noErr
  }

  public class func get(key: String) -> String? {
    if let currentData = getData(key) {
      if let currentString = NSString(data: currentData,
        encoding: NSUTF8StringEncoding) as? String {

        return currentString
      }
    }

    return nil
  }

  public class func getData(key: String) -> NSData? {
    let query = [
      TegKeychainConstants.klass       : kSecClassGenericPassword,
      TegKeychainConstants.attrAccount : key,
      TegKeychainConstants.returnData  : kCFBooleanTrue,
      TegKeychainConstants.matchLimit  : kSecMatchLimitOne ]
    
    var dataTypeRef :Unmanaged<AnyObject>?
    
    let status: OSStatus = SecItemCopyMatching(query, &dataTypeRef)
    
    if status == noErr {
      if let currentDataTypeRef = dataTypeRef {
        return currentDataTypeRef.takeRetainedValue() as? NSData
      }
    }
    
    return nil
  }

  public class func delete(key: String) -> Bool {
    let query = [
      TegKeychainConstants.klass       : kSecClassGenericPassword,
      TegKeychainConstants.attrAccount : key ]
    
    let status: OSStatus = SecItemDelete(query as CFDictionaryRef)
    
    return status == noErr
  }

  public class func clear() -> Bool {
    let query = [ kSecClass as String : kSecClassGenericPassword ]
    
    let status: OSStatus = SecItemDelete(query as CFDictionaryRef)
    
    return status == noErr
  }
}