ABContactsHelper.m

/*
 Erica Sadun, http://ericasadun.com
 iPhone Developer's Cookbook, 3.0 Edition
 BSD License, Use at your own risk
 */

#import "ABContactsHelper.h"
#import "ABStandin.h"

@implementation ABContactsHelper

#pragma mark Address Book

+ (ABAddressBookRef) addressBook
{
    return [ABStandin addressBook];
}

+ (void) refresh
{
    [ABStandin currentAddressBook];
}

+ (NSArray *) contacts
{
    return [ABContactsHelper contactsWithAddressBook:[ABStandin addressBook]];
}

+ (NSArray *) currentContacts
{
    return [ABContactsHelper contactsWithAddressBook:[ABStandin currentAddressBook]];
}

+ (NSArray *) contactsWithAddressBook:(ABAddressBookRef) addressBook
{
    NSArray *thePeople = (__bridge_transfer NSArray *)ABAddressBookCopyArrayOfAllPeople(addressBook);
    NSMutableArray *array = [NSMutableArray arrayWithCapacity:thePeople.count];
    for (id person in thePeople)
        [array addObject:[ABContact contactWithRecord:(__bridge ABRecordRef)person]];
    return array;
}

+ (int) contactsCount
{
    ABAddressBookRef addressBook = [ABStandin addressBook];
    return ABAddressBookGetPersonCount(addressBook);
}

+ (int) contactsWithImageCount
{
    ABAddressBookRef addressBook = [ABStandin addressBook];
    NSArray *thePeople = (__bridge_transfer NSArray *)ABAddressBookCopyArrayOfAllPeople(addressBook);
    int ncount = 0;
    for (id person in thePeople) 
    {
        ABRecordRef abPerson = (__bridge ABRecordRef) person;
        if (ABPersonHasImageData(abPerson)) ncount++;
    }
    return ncount;
}

+ (int) contactsWithoutImageCount
{
    ABAddressBookRef addressBook = [ABStandin addressBook];
    NSArray *thePeople = (__bridge_transfer NSArray *)ABAddressBookCopyArrayOfAllPeople(addressBook);
    int ncount = 0;
    for (id person in thePeople) 
    {
        ABRecordRef abPerson = (__bridge ABRecordRef) person;
        if (!ABPersonHasImageData(abPerson)) ncount++;
    }
    return ncount;
}

// Groups
+ (int) numberOfGroups
{
    ABAddressBookRef addressBook = [ABStandin addressBook];
    NSArray *groups = (__bridge_transfer NSArray *)ABAddressBookCopyArrayOfAllGroups(addressBook);
    int ncount = groups.count;
    return ncount;
}

+ (NSArray *) groups
{
    ABAddressBookRef addressBook = [ABStandin addressBook];
    NSArray *groups = (__bridge_transfer NSArray *)ABAddressBookCopyArrayOfAllGroups(addressBook);
    NSMutableArray *array = [NSMutableArray arrayWithCapacity:groups.count];
    for (id group in groups)
    {
        ABRecordRef abGroup = (__bridge ABRecordRef) group;
        [array addObject:[ABGroup groupWithRecord:abGroup]];
    }
    return array;
}

#pragma mark Sorting

// Sorting
+ (BOOL) firstNameSorting
{
    return (ABPersonGetCompositeNameFormat() == kABPersonCompositeNameFormatFirstNameFirst);
}

#pragma mark Contact Management

// Thanks to Eridius for suggestions re: error
+ (BOOL) addContact: (ABContact *) aContact withError: (NSError **) error
{
    ABAddressBookRef addressBook = [ABStandin addressBook];
    BOOL success;
    CFErrorRef cfError = NULL;
    
    if (!aContact) return NO;

    success = ABAddressBookAddRecord(addressBook, aContact.record, &cfError);
    if (!success)
    {
        if (error)
            *error = (__bridge_transfer NSError *)cfError;
        return NO;
    }

    return YES;
}

+ (BOOL) addGroup: (ABGroup *) aGroup withError: (NSError **) error
{
    ABAddressBookRef addressBook = [ABStandin addressBook];
    BOOL success;
    CFErrorRef cfError = NULL;
    
    success = ABAddressBookAddRecord(addressBook, aGroup.record, &cfError);
    if (!success)
    {
        if (error)
            *error = (__bridge_transfer NSError *)cfError;
        return NO;
    }
    
    return NO;
}

#pragma mark Searches

+ (NSArray *) contactsMatchingPredicate: (NSPredicate *) pred
{
    NSArray *contacts = [ABContactsHelper currentContacts];
    return [contacts filteredArrayUsingPredicate:pred];
}

+ (NSArray *) contactsMatchingName: (NSString *) fname
{
    NSPredicate *pred;
    NSArray *contacts = [ABContactsHelper currentContacts];
    pred = [NSPredicate predicateWithFormat:@"firstname beginswith[cd] %@ OR lastname beginswith[cd] %@ OR nickname beginswith[cd] %@ OR middlename beginswith[cd] %@", fname, fname, fname, fname];
    return [contacts filteredArrayUsingPredicate:pred];
}

+ (NSArray *) contactsMatchingName: (NSString *) fname andName: (NSString *) lname
{
    NSPredicate *pred;
    NSArray *contacts = [ABContactsHelper currentContacts];
    pred = [NSPredicate predicateWithFormat:@"firstname beginswith[cd] %@ OR lastname beginswith[cd] %@ OR nickname beginswith[cd] %@ OR middlename beginswith[cd] %@", fname, fname, fname, fname];
    contacts = [contacts filteredArrayUsingPredicate:pred];
    pred = [NSPredicate predicateWithFormat:@"firstname beginswith[cd] %@ OR lastname beginswith[cd] %@ OR nickname beginswith[cd] %@ OR middlename beginswith[cd] %@", lname, lname, lname, lname];
    contacts = [contacts filteredArrayUsingPredicate:pred];
    return contacts;
}

+ (NSArray *) contactsMatchingPhone: (NSString *) number
{
    NSPredicate *pred;
    NSArray *contacts = [ABContactsHelper currentContacts];
    pred = [NSPredicate predicateWithFormat:@"phonenumbers contains[cd] %@", number];
    return [contacts filteredArrayUsingPredicate:pred];
}

+ (NSArray *) contactsMatchingEmail: (NSString *) email
{
    NSPredicate *pred;
    NSArray *contacts = [ABContactsHelper currentContacts];
    pred = [NSPredicate predicateWithFormat:@"emailaddresses contains[cd] %@", email];
    return [contacts filteredArrayUsingPredicate:pred];
}

// Thanks Frederic Bronner
+ (NSArray *) contactsMatchingOrganization: (NSString *) organization
{
	NSPredicate *pred;
	NSArray *contacts = [ABContactsHelper currentContacts];
	pred = [NSPredicate predicateWithFormat:@"organization beginswith[cd] %@", organization];
	return [contacts filteredArrayUsingPredicate:pred];
}


+ (NSArray *) groupsMatchingName: (NSString *) fname
{
    NSPredicate *pred;
    NSArray *groups = [ABContactsHelper groups];
    pred = [NSPredicate predicateWithFormat:@"name contains[cd] %@ ", fname];
    return [groups filteredArrayUsingPredicate:pred];
}
@end