133 lines
4.1 KiB
Objective-C
133 lines
4.1 KiB
Objective-C
//
|
|
// GroupContactsResult.m
|
|
// Signal
|
|
//
|
|
// Created by Frederic Jacobs on 17/02/15.
|
|
// Copyright (c) 2015 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "GroupContactsResult.h"
|
|
|
|
#import "Contact.h"
|
|
#import "Environment.h"
|
|
#import "ContactsManager.h"
|
|
#import "SignalKeyingStorage.h"
|
|
|
|
@interface GroupContactsResult()
|
|
|
|
@property NSMutableArray *unknownNumbers;
|
|
@property NSMutableArray *knownNumbers;
|
|
|
|
@property NSMutableDictionary *associatedContactDict;
|
|
|
|
@end
|
|
|
|
@implementation GroupContactsResult
|
|
|
|
- (instancetype)initWithMembersId:(NSArray*)memberIdentifiers without:(NSArray*)removeIds{
|
|
self = [super init];
|
|
|
|
ContactsManager *manager = [Environment.getCurrent contactsManager];
|
|
|
|
NSMutableSet *remainingNumbers = [NSMutableSet setWithArray:memberIdentifiers];
|
|
|
|
NSMutableArray *knownNumbers = [NSMutableArray array];
|
|
NSMutableArray *associatedContacts = [NSMutableArray array];
|
|
|
|
for (NSString *identifier in memberIdentifiers) {
|
|
if([identifier isEqualToString:[SignalKeyingStorage.localNumber toE164]]) {
|
|
// remove local number
|
|
|
|
[remainingNumbers removeObject:identifier];
|
|
continue;
|
|
}
|
|
|
|
if (removeIds && [removeIds containsObject:identifier]){
|
|
//Remove ids
|
|
[remainingNumbers removeObject:identifier];
|
|
continue;
|
|
}
|
|
|
|
PhoneNumber *number = [PhoneNumber phoneNumberFromE164:identifier];
|
|
|
|
if (!number) {
|
|
continue;
|
|
}
|
|
|
|
Contact *contact = [manager latestContactForPhoneNumber:number];
|
|
|
|
if (!contact) {
|
|
continue;
|
|
}
|
|
|
|
[knownNumbers addObject:identifier];
|
|
[associatedContacts addObject:contact];
|
|
|
|
[remainingNumbers removeObject:identifier];
|
|
}
|
|
|
|
_unknownNumbers = [NSMutableArray arrayWithArray:[remainingNumbers allObjects]];
|
|
[_unknownNumbers sortUsingComparator:^NSComparisonResult(id obj1, id obj2) {
|
|
return [obj1 compare:obj2 options:0];
|
|
}];
|
|
|
|
|
|
// Populate mapping dictionary.
|
|
_associatedContactDict = [NSMutableDictionary dictionary];
|
|
for (NSUInteger i = 0 ; i < [knownNumbers count]; i++) {
|
|
NSString *identifier = [knownNumbers objectAtIndex:i];
|
|
Contact *contact = [associatedContacts objectAtIndex:i];
|
|
|
|
[_associatedContactDict setObject:contact forKey:identifier];
|
|
}
|
|
|
|
// Known Numbers
|
|
_knownNumbers = [NSMutableArray arrayWithArray:knownNumbers];
|
|
[_knownNumbers sortedArrayUsingComparator:^NSComparisonResult(id obj1, id obj2) {
|
|
Contact *contact1 = [_associatedContactDict objectForKey:obj1];
|
|
Contact *contact2 = [_associatedContactDict objectForKey:obj2];
|
|
|
|
return [[manager class] contactComparator](contact1, contact2);
|
|
}];
|
|
|
|
return self;
|
|
}
|
|
|
|
- (NSUInteger)numberOfMembers {
|
|
return [self.knownNumbers count] + [self.unknownNumbers count];
|
|
}
|
|
|
|
- (BOOL)isContactAtIndexPath:(NSIndexPath*)indexPath {
|
|
if ((NSUInteger)indexPath.row < [self.unknownNumbers count]) {
|
|
return NO;
|
|
} else {
|
|
return YES;
|
|
}
|
|
}
|
|
|
|
- (Contact*)contactForIndexPath:(NSIndexPath*)indexPath {
|
|
if ([self isContactAtIndexPath:indexPath]) {
|
|
NSString *identifier = [_knownNumbers objectAtIndex:[self knownNumbersIndexForIndexPath:indexPath]];
|
|
return [_associatedContactDict objectForKey:identifier];
|
|
} else {
|
|
NSAssert(NO, @"Trying to retrieve contact from array at an index that is not a contact.");
|
|
return nil;
|
|
}
|
|
}
|
|
|
|
- (NSString*)identifierForIndexPath:(NSIndexPath*)indexPath {
|
|
if ([self isContactAtIndexPath:indexPath]) {
|
|
return [_knownNumbers objectAtIndex:[self knownNumbersIndexForIndexPath:indexPath]];
|
|
} else {
|
|
return [_unknownNumbers objectAtIndex:(NSUInteger)indexPath.row];
|
|
}
|
|
}
|
|
|
|
- (NSUInteger)knownNumbersIndexForIndexPath:(NSIndexPath*)indexPath {
|
|
NSAssert(((NSUInteger)indexPath.row >= [_unknownNumbers count]), @"Wrong index for known number");
|
|
|
|
return (NSUInteger)indexPath.row - [_unknownNumbers count];
|
|
}
|
|
|
|
@end
|