Lockbox.m

//
//  Lockbox.m
//
//  Created by Mark H. Granoff on 4/19/12.
//  Copyright (c) 2012 Hawk iMedia. All rights reserved.
//

#import "Lockbox.h"
#import <Security/Security.h>

#define kDelimiter @"-|-"

static NSString *_bundleId = nil;

@implementation Lockbox

+(void)initialize
{
    _bundleId = [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString*)kCFBundleIdentifierKey];
}

+(NSMutableDictionary *)_service
{
    NSMutableDictionary* dict = [NSMutableDictionary dictionary];
    
#if __has_feature(objc_arc)
    [dict setObject: (__bridge id) kSecClassGenericPassword  forKey: (__bridge id) kSecClass];
#else
    [dict setObject: (id) kSecClassGenericPassword  forKey: (id) kSecClass];
#endif
    return dict;
}

+(NSMutableDictionary *)_query
{
    NSMutableDictionary* query = [NSMutableDictionary dictionary];
    
#if __has_feature(objc_arc)
    [query setObject: (__bridge id) kSecClassGenericPassword forKey: (__bridge id) kSecClass];
    [query setObject: (id) kCFBooleanTrue           forKey: (__bridge id) kSecReturnData];
#else
    [query setObject: (id) kSecClassGenericPassword forKey: (id) kSecClass];
    [query setObject: (id) kCFBooleanTrue           forKey: (id) kSecReturnData];
#endif
    return query;
}

// Prefix a bare key like "MySecureKey" with the bundle id, so the actual key stored
// is unique to this app, e.g. "com.mycompany.myapp.MySecretKey"
+(NSString *)_hierarchicalKey:(NSString *)key
{
    return [_bundleId stringByAppendingFormat:@".%@", key];
}

+(BOOL)setObject:(NSString *)obj forKey:(NSString *)key
{
    OSStatus status;
    
    NSString *hierKey = [self _hierarchicalKey:key];

    // If the object is nil, delete the item
    if (!obj) {
        NSMutableDictionary *query = [self _query];
#if __has_feature(objc_arc)
        [query setObject:hierKey forKey:(__bridge id)kSecAttrService];
        status = SecItemDelete((__bridge CFDictionaryRef)query);
#else
        [query setObject:hierKey forKey:(id)kSecAttrService];
        status = SecItemDelete((CFDictionaryRef)query);
#endif
        return (status == errSecSuccess);
    }
    
    NSMutableDictionary *dict = [self _service];
    
#if __has_feature(objc_arc)
    [dict setObject: hierKey forKey: (__bridge id) kSecAttrService];
    [dict setObject: [obj dataUsingEncoding:NSUTF8StringEncoding] forKey: (__bridge id) kSecValueData];
    status = SecItemAdd ((__bridge CFDictionaryRef) dict, NULL);
#else
    [dict setObject: hierKey forKey: (id) kSecAttrService];
    [dict setObject: [obj dataUsingEncoding:NSUTF8StringEncoding] forKey: (id) kSecValueData];
    status = SecItemAdd ((CFDictionaryRef) dict, NULL);
#endif
    if (status == errSecDuplicateItem) {
        NSMutableDictionary *query = [self _query];
#if __has_feature(objc_arc)
        [query setObject:hierKey forKey:(__bridge id)kSecAttrService];
        status = SecItemDelete((__bridge CFDictionaryRef)query);
#else
        [query setObject:hierKey forKey:(id)kSecAttrService];
        status = SecItemDelete((CFDictionaryRef) query);
#endif
        if (status == errSecSuccess)
#if __has_feature(objc_arc)
            status = SecItemAdd((__bridge CFDictionaryRef) dict, NULL);        
#else
            status = SecItemAdd((CFDictionaryRef) dict, NULL);
#endif
    }
    if (status != errSecSuccess)
        NSLog(@"SecItemAdd failed for key %@: %ld", hierKey, status);
    
    return (status == errSecSuccess);
}

+(NSString *)objectForKey:(NSString *)key
{
    NSString *hierKey = [self _hierarchicalKey:key];

    NSMutableDictionary *query = [self _query];
#if __has_feature(objc_arc)
    [query setObject:hierKey forKey: (__bridge id)kSecAttrService];
#else
    [query setObject:hierKey forKey: (id)kSecAttrService];
#endif

    CFDataRef data = nil;
    OSStatus status =
#if __has_feature(objc_arc)
        SecItemCopyMatching ( (__bridge CFDictionaryRef) query, (CFTypeRef *) &data );
#else
        SecItemCopyMatching((CFDictionaryRef) query, (CFTypeRef *)&data);
#endif
    if (status != errSecSuccess)
        NSLog(@"SecItemCopyMatching failed for key %@: %ld", hierKey, status);
    
    if (!data)
        return nil;

    NSString *s = [[NSString alloc] 
                    initWithData: 
#if __has_feature(objc_arc)
                   (__bridge_transfer NSData *)data 
#else
                   (NSData *)data
#endif
                    encoding: NSUTF8StringEncoding];

#if !__has_feature(objc_arc)
    [s autorelease];
    CFRelease(data);
#endif
    
    return s;    
}

+(BOOL)setString:(NSString *)value forKey:(NSString *)key
{
    return [self setObject:value forKey:key];
}

+(NSString *)stringForKey:(NSString *)key
{
    return [self objectForKey:key];
}

+(BOOL)setArray:(NSArray *)value forKey:(NSString *)key
{
    NSString *components = [value componentsJoinedByString:kDelimiter];
    return [self setObject:components forKey:key];
}

+(NSArray *)arrayForKey:(NSString *)key
{
    NSArray *array = nil;
    NSString *components = [self objectForKey:key];
    if (components)
        array = [NSArray arrayWithArray:[components componentsSeparatedByString:kDelimiter]];
    
    return array;
}

+(BOOL)setSet:(NSSet *)value forKey:(NSString *)key
{
    return [self setArray:[value allObjects] forKey:key];
}

+(NSSet *)setForKey:(NSString *)key
{
    NSSet *set = nil;
    NSArray *array = [self arrayForKey:key];
    if (array)
        set = [NSSet setWithArray:array];
    
    return set;
}

+(BOOL)setDate:(NSDate *)value forKey:(NSString *)key
{
    if (!value)
        return [self setObject:nil forKey:key];
    NSNumber *rti = [NSNumber numberWithDouble:[value timeIntervalSinceReferenceDate]];
    return [self setObject:[rti stringValue] forKey:key];
}

+(NSDate *)dateForKey:(NSString *)key
{
    NSString *dateString = [self objectForKey:key];
    if (dateString)
        return [NSDate dateWithTimeIntervalSinceReferenceDate:[dateString doubleValue]];
    return nil;
}

@end