//

//  MYAddressField.m

//  YourMove

//

//  Created by Jens Alfke on 7/16/08.

//  Copyright 2008 Jens Alfke. All rights reserved.

//

#import "MYAddressField.h"

#import "RegexKitLite.h"

#import <AddressBook/AddressBook.h>

@interface MYAddressField ()

@property (retain) MYAddressItem *selectedAddress;

@end

@implementation MYAddressField

@synthesize defaultAddresses=_defaultAddresses, addressProperty=_property, selectedAddress=_selectedAddress;

- (void) _computeAddresses

{

    NSMutableArray *newAddresses = $marray();

    if( _property ) {

        if( _prefix.length ) {

            // Find all the people in the address book matching _prefix:

            ABAddressBook *ab = [ABAddressBook sharedAddressBook];

            ABSearchElement *search = [ABPerson searchElementForProperty: _property

                                                                   label: nil

                                                                     key: nil

                                                                   value: nil

                                                              comparison: kABNotEqual];

            for( ABPerson *person in [ab recordsMatchingSearchElement: search] ) {

                ABMultiValue *values = [person valueForProperty: _property];

                NSString *first = [person valueForProperty: kABFirstNameProperty];

                NSString *last = [person valueForProperty: kABLastNameProperty];

                BOOL nameMatches = _prefix==nil || ([first.lowercaseString hasPrefix: _prefix] 

                                                    || [last.lowercaseString hasPrefix: _prefix]);

                for( int i=0; i<values.count; i++ ) {

                    NSString *address = [values valueAtIndex: i];

                    if( nameMatches || [address.lowercaseString hasPrefix: _prefix] ) {

                        MYAddressItem *item = [[MYAddressItem alloc] initWithPerson: person

                                                                        addressType: _property

                                                                            address: address];

                        [newAddresses addObject: item];

                        [item release];

                    }

                }

            }

        } else if( _defaultAddresses ) {

            [newAddresses addObjectsFromArray: _defaultAddresses];

        }

    }

    [newAddresses sortUsingSelector: @selector(compare:)];

    if( ifSetObj(&_addresses,newAddresses) )

        [self reloadData];

}

- (NSArray*) addresses

{

    if( ! _addresses )

        [self _computeAddresses];

    return _addresses;

}

- (void) awakeFromNib

{

    if( ! _addresses )

        _addresses = [[NSMutableArray alloc] init];

    _property = [kABEmailProperty retain];

    self.completes = NO;

    self.usesDataSource = YES;

    self.dataSource = self;

    self.delegate = self;

}

- (void) dealloc

{

    [_addresses release];

    [_property release];

    [_prefix release];

    [_selectedAddress release];

    [super dealloc];

}

- (BOOL) isExpanded

{

    id ax = NSAccessibilityUnignoredDescendant(self);

    return [[ax accessibilityAttributeValue: NSAccessibilityExpandedAttribute] boolValue];

}

- (void) setExpanded: (BOOL)expanded

{

    id ax = NSAccessibilityUnignoredDescendant(self);

    [ax accessibilitySetValue: $object(expanded) forAttribute: NSAccessibilityExpandedAttribute];

}

- (void) controlTextDidChange: (NSNotification*)n

{

    if( _prefix.length == 0 )

        self.expanded = YES;

    if( ifSetObj(&_prefix, self.stringValue.lowercaseString) )

        [self _computeAddresses];

    MYAddressItem *item = [[MYAddressItem alloc] initWithString: self.stringValue 

                                                    addressType: _property];

    self.selectedAddress = item;

    [item release];

    if( _prefix.length == 0 )

        self.expanded = NO;

    //Log(@"Address selection = %@",self.selectedAddress);

}

- (void)comboBoxSelectionDidChange:(NSNotification *)notification

{

    int sel = self.indexOfSelectedItem;

    self.selectedAddress = sel>=0 ?[self.addresses objectAtIndex: sel] :nil;

    //Log(@"Address selection = %@",self.selectedAddress);

}

#pragma mark -

#pragma mark DATA SOURCE:

- (NSInteger)numberOfItemsInComboBox:(NSComboBox *)aComboBox

{

    return self.addresses.count;

}

- (id)comboBox:(NSComboBox *)aComboBox objectValueForItemAtIndex:(NSInteger)index

{

    return [[self.addresses objectAtIndex: index] description];

}

@end

@implementation MYAddressItem

- (id) initWithName: (NSString*)name

        addressType: (NSString*)addressType address: (NSString*)address

{

    self = [super init];

    if( self ) {

        _name = name.length ?[name copy] :nil;

        _addressType = [addressType copy];

        _address = [address copy];

    }

    return self;

}

- (id) initWithPerson: (ABPerson*)person

          addressType: (NSString*)addressType address: (NSString*)address

{

    NSString *first = [person valueForProperty: kABFirstNameProperty] ?: @"";

    NSString *last = [person valueForProperty: kABLastNameProperty] ?: @"";

    NSString *name = $sprintf(@"%@ %@", first,last);

    self = [self initWithName: name addressType: addressType address: address];

    if( self )

        _uuid = person.uniqueId.copy;

    return self;

}

- (id) initWithString: (NSString*)str addressType: (NSString*)addressType

{

    #define kJustAddrRegex "[-a-zA-Z0-9%_+.]+(?:@[-a-zA-Z0-9.]+)"

    static NSString* const kNameAndAddrRegex = @"^\\s*(\\S+)?\\s*<("kJustAddrRegex")>\\s*$";

    static NSString* const kAddrRegex = @"^\\s*("kJustAddrRegex")\\s*$";

    NSString *name = nil;

    NSString *address = [str stringByMatching: kNameAndAddrRegex capture: 2];

    if( address ) {

        name = [str stringByMatching: kNameAndAddrRegex capture: 1];

    } else {

        address = [str stringByMatching: kAddrRegex];

    }

    if( ! address ) {

        [self release];

        return nil;

    }

    return [self initWithName: name addressType: addressType address: address];

}

@synthesize name=_name, addressType=_addressType, address=_address;

- (ABPerson*) person

{

    if( _uuid )

        return (ABPerson*) [[ABAddressBook sharedAddressBook] recordForUniqueId: _uuid];

    else

        return nil;

}

- (NSString*) description

{

    return $sprintf(@"%@%@<%@>", _name,(_name ?@" ":@""),_address);

}

- (NSComparisonResult) compare: (MYAddressItem*)other

{

    NSString *str1 = _name ?:_address;

    NSString *str2 = other->_name ?: other->_address;

    return [str1 localizedCaseInsensitiveCompare: str2];

}

@end