jens@18: //
jens@18: //  MYAddressField.m
jens@18: //  YourMove
jens@18: //
jens@18: //  Created by Jens Alfke on 7/16/08.
jens@18: //  Copyright 2008 Jens Alfke. All rights reserved.
jens@18: //
jens@18: 
jens@18: #import "MYAddressField.h"
jens@18: #import "RegexKitLite.h"
jens@18: #import <AddressBook/AddressBook.h>
jens@18: 
jens@18: 
jens@18: @interface MYAddressField ()
jens@18: @property (retain) MYAddressItem *selectedAddress;
jens@18: @end
jens@18: 
jens@18: 
jens@18: @implementation MYAddressField
jens@18: 
jens@18: 
jens@18: @synthesize defaultAddresses=_defaultAddresses, addressProperty=_property, selectedAddress=_selectedAddress;
jens@18: 
jens@18: 
jens@18: - (void) _computeAddresses
jens@18: {
jens@18:     NSMutableArray *newAddresses = $marray();
jens@18:     if( _property ) {
jens@18:         if( _prefix.length ) {
jens@18:             // Find all the people in the address book matching _prefix:
jens@18:             ABAddressBook *ab = [ABAddressBook sharedAddressBook];
jens@18:             ABSearchElement *search = [ABPerson searchElementForProperty: _property
jens@18:                                                                    label: nil
jens@18:                                                                      key: nil
jens@18:                                                                    value: nil
jens@18:                                                               comparison: kABNotEqual];
jens@18:             for( ABPerson *person in [ab recordsMatchingSearchElement: search] ) {
jens@18:                 ABMultiValue *values = [person valueForProperty: _property];
jens@18:                 NSString *first = [person valueForProperty: kABFirstNameProperty];
jens@18:                 NSString *last = [person valueForProperty: kABLastNameProperty];
jens@18:                 BOOL nameMatches = _prefix==nil || ([first.lowercaseString hasPrefix: _prefix] 
jens@18:                                                     || [last.lowercaseString hasPrefix: _prefix]);
jens@18:                 for( int i=0; i<values.count; i++ ) {
jens@18:                     NSString *address = [values valueAtIndex: i];
jens@18:                     if( nameMatches || [address.lowercaseString hasPrefix: _prefix] ) {
jens@18:                         MYAddressItem *item = [[MYAddressItem alloc] initWithPerson: person
jens@18:                                                                         addressType: _property
jens@18:                                                                             address: address];
jens@18:                         [newAddresses addObject: item];
jens@18:                         [item release];
jens@18:                     }
jens@18:                 }
jens@18:             }
jens@18:         } else if( _defaultAddresses ) {
jens@18:             [newAddresses addObjectsFromArray: _defaultAddresses];
jens@18:         }
jens@18:     }
jens@18:     
jens@18:     [newAddresses sortUsingSelector: @selector(compare:)];
jens@18:     
jens@18:     if( ifSetObj(&_addresses,newAddresses) )
jens@18:         [self reloadData];
jens@18: }
jens@18: 
jens@18: - (NSArray*) addresses
jens@18: {
jens@18:     if( ! _addresses )
jens@18:         [self _computeAddresses];
jens@18:     return _addresses;
jens@18: }
jens@18: 
jens@18: 
jens@18: - (void) awakeFromNib
jens@18: {
jens@18:     if( ! _addresses )
jens@18:         _addresses = [[NSMutableArray alloc] init];
jens@18:     _property = [kABEmailProperty retain];
jens@18:     self.completes = NO;
jens@18:     self.usesDataSource = YES;
jens@18:     self.dataSource = self;
jens@18:     self.delegate = self;
jens@18: }
jens@18: 
jens@18: - (void) dealloc
jens@18: {
jens@18:     [_addresses release];
jens@18:     [_property release];
jens@18:     [_prefix release];
jens@18:     [_selectedAddress release];
jens@18:     [super dealloc];
jens@18: }
jens@18: 
jens@18: 
jens@18: - (BOOL) isExpanded
jens@18: {
jens@18:     id ax = NSAccessibilityUnignoredDescendant(self);
jens@18:     return [[ax accessibilityAttributeValue: NSAccessibilityExpandedAttribute] boolValue];
jens@18: }
jens@18: 
jens@18: 
jens@18: - (void) setExpanded: (BOOL)expanded
jens@18: {
jens@18:     id ax = NSAccessibilityUnignoredDescendant(self);
jens@18:     [ax accessibilitySetValue: $object(expanded) forAttribute: NSAccessibilityExpandedAttribute];
jens@18: }
jens@18: 
jens@18: 
jens@18: - (void) controlTextDidChange: (NSNotification*)n
jens@18: {
jens@18:     if( _prefix.length == 0 )
jens@18:         self.expanded = YES;
jens@18:     
jens@18:     if( ifSetObj(&_prefix, self.stringValue.lowercaseString) )
jens@18:         [self _computeAddresses];
jens@18:     MYAddressItem *item = [[MYAddressItem alloc] initWithString: self.stringValue 
jens@18:                                                     addressType: _property];
jens@18:     self.selectedAddress = item;
jens@18:     [item release];
jens@18: 
jens@18:     if( _prefix.length == 0 )
jens@18:         self.expanded = NO;
jens@18: 
jens@18:     //Log(@"Address selection = %@",self.selectedAddress);
jens@18: }
jens@18: 
jens@18: - (void)comboBoxSelectionDidChange:(NSNotification *)notification
jens@18: {
jens@18:     int sel = self.indexOfSelectedItem;
jens@18:     self.selectedAddress = sel>=0 ?[self.addresses objectAtIndex: sel] :nil;
jens@18:     //Log(@"Address selection = %@",self.selectedAddress);
jens@18: }
jens@18: 
jens@18: 
jens@18: #pragma mark -
jens@18: #pragma mark DATA SOURCE:
jens@18: 
jens@18: 
jens@18: - (NSInteger)numberOfItemsInComboBox:(NSComboBox *)aComboBox
jens@18: {
jens@18:     return self.addresses.count;
jens@18: }
jens@18: 
jens@18: - (id)comboBox:(NSComboBox *)aComboBox objectValueForItemAtIndex:(NSInteger)index
jens@18: {
jens@18:     return [[self.addresses objectAtIndex: index] description];
jens@18: }
jens@18: 
jens@18: 
jens@18: @end
jens@18: 
jens@18: 
jens@18: 
jens@18: 
jens@18: @implementation MYAddressItem
jens@18: 
jens@18: - (id) initWithName: (NSString*)name
jens@18:         addressType: (NSString*)addressType address: (NSString*)address
jens@18: {
jens@18:     self = [super init];
jens@18:     if( self ) {
jens@18:         _name = name.length ?[name copy] :nil;
jens@18:         _addressType = [addressType copy];
jens@18:         _address = [address copy];
jens@18:     }
jens@18:     return self;
jens@18: }
jens@18:     
jens@18: - (id) initWithPerson: (ABPerson*)person
jens@18:           addressType: (NSString*)addressType address: (NSString*)address
jens@18: {
jens@18:     NSString *first = [person valueForProperty: kABFirstNameProperty] ?: @"";
jens@18:     NSString *last = [person valueForProperty: kABLastNameProperty] ?: @"";
jens@18:     NSString *name = $sprintf(@"%@ %@", first,last);
jens@18:     
jens@18:     self = [self initWithName: name addressType: addressType address: address];
jens@18:     if( self )
jens@18:         _uuid = person.uniqueId.copy;
jens@18:     return self;
jens@18: }
jens@18: 
jens@18: - (id) initWithString: (NSString*)str addressType: (NSString*)addressType
jens@18: {
jens@18:     #define kJustAddrRegex "[-a-zA-Z0-9%_+.]+(?:@[-a-zA-Z0-9.]+)"
jens@18:     static NSString* const kNameAndAddrRegex = @"^\\s*(\\S+)?\\s*<("kJustAddrRegex")>\\s*$";
jens@18:     static NSString* const kAddrRegex = @"^\\s*("kJustAddrRegex")\\s*$";
jens@18: 
jens@18:     NSString *name = nil;
jens@18:     NSString *address = [str stringByMatching: kNameAndAddrRegex capture: 2];
jens@18:     if( address ) {
jens@18:         name = [str stringByMatching: kNameAndAddrRegex capture: 1];
jens@18:     } else {
jens@18:         address = [str stringByMatching: kAddrRegex];
jens@18:     }
jens@18:     if( ! address ) {
jens@18:         [self release];
jens@18:         return nil;
jens@18:     }
jens@18:     return [self initWithName: name addressType: addressType address: address];
jens@18: }
jens@18: 
jens@18: @synthesize name=_name, addressType=_addressType, address=_address;
jens@18: 
jens@18: - (ABPerson*) person
jens@18: {
jens@18:     if( _uuid )
jens@18:         return (ABPerson*) [[ABAddressBook sharedAddressBook] recordForUniqueId: _uuid];
jens@18:     else
jens@18:         return nil;
jens@18: }
jens@18: 
jens@18: - (NSString*) description
jens@18: {
jens@18:     return $sprintf(@"%@%@<%@>", _name,(_name ?@" ":@""),_address);
jens@18: }
jens@18: 
jens@18: - (NSComparisonResult) compare: (MYAddressItem*)other
jens@18: {
jens@18:     NSString *str1 = _name ?:_address;
jens@18:     NSString *str2 = other->_name ?: other->_address;
jens@18:     return [str1 localizedCaseInsensitiveCompare: str2];
jens@18: }
jens@18: 
jens@18: @end