add / remove contacts is now done with directly passing the uiview ob…

…jects
Luke47 · Aug 24, 2015 · 30e1a3f · 30e1a3f
1 parent 1894b47
commit 30e1a3f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/Classes/THContactPickerView.h b/Classes/THContactPickerView.h
@@ -30,8 +30,9 @@
 @property (nonatomic, assign) NSInteger maxNumberOfLines;	// maximum number of lines the view will display before scrolling
 @property (nonatomic, strong) UIFont *font;
 
-- (void)addContact:(id)contact withName:(NSString *)name;
+- (UIView *)addContact:(id)contact withName:(NSString *)name;
 - (void)removeContact:(id)contact;
+- (void)removeContactView:(THContactView *)contactView;
 - (void)removeAllContacts;
 - (BOOL)resignFirstResponder;
 

diff --git a/Classes/THContactPickerView.m b/Classes/THContactPickerView.m
@@ -145,11 +145,11 @@ - (void)setBackgroundColor:(UIColor *)backgroundColor{
     [super setBackgroundColor:backgroundColor];
 }
 
-- (void)addContact:(id)contact withName:(NSString *)name {
+- (UIView *)addContact:(id)contact withName:(NSString *)name {
     id contactKey = [NSValue valueWithNonretainedObject:contact];
     if ([self.contactKeys containsObject:contactKey]){
         NSLog(@"Cannot add the same object twice to ContactPickerView");
-        return;
+        return nil;
     }
 
     if (self.contactKeys.count == 1 && self.limitToOne){
@@ -189,6 +189,9 @@ - (void)addContact:(id)contact withName:(NSString *)name {
 		[self scrollToBottomWithAnimation:YES];
 		// after scroll animation [self selectTextView] will be called
 	}];
+
+    // Return the bubble
+    return contactView;
 }
 
 - (void)selectTextView {
@@ -285,6 +288,7 @@ - (void)scrollToBottomWithAnimation:(BOOL)animated {
     }
 }
 
+// This will be public!
 - (void)removeContactView:(THContactView *)contactView {
     id contact = [self contactForContactView:contactView];