Added the ability to set a custom view instead of the text (for example an UIImageView)

Author: ipodishima

src/NSBubbleData.h

@@ -24,8 +24,11 @@ typedef enum _NSBubbleType
 @property (readonly, nonatomic, strong) NSDate *date;
 @property (readonly, nonatomic) NSBubbleType type;
 @property (readonly, nonatomic, strong) NSString *text;
+@property (readonly, nonatomic, strong) UIView *view;
 
 - (id)initWithText:(NSString *)text andDate:(NSDate *)date andType:(NSBubbleType)type;
 + (id)dataWithText:(NSString *)text andDate:(NSDate *)date andType:(NSBubbleType)type;
+- (id)initWithView:(UIView *)view andDate:(NSDate *)date andType:(NSBubbleType)type;
++ (id)dataWithView:(UIView *)view andDate:(NSDate *)date andType:(NSBubbleType)type;
 
 @end

src/NSBubbleData.m

@@ -18,6 +18,7 @@ @implementation NSBubbleData
 @synthesize date = _date;
 @synthesize type = _type;
 @synthesize text = _text;
+@synthesize view = _view;
 
 + (id)dataWithText:(NSString *)text andDate:(NSDate *)date andType:(NSBubbleType)type
 {
@@ -38,12 +39,30 @@ - (id)initWithText:(NSString *)initText andDate:(NSDate *)initDate andType:(NSBu
     return self;
 }
 
++ (id)dataWithView:(UIView *)view andDate:(NSDate *)date andType:(NSBubbleType)type
+{
+    return [[[NSBubbleData alloc] initWithView:view andDate:date andType:type] autorelease];
+}
+
+- (id)initWithView:(UIView *)view andDate:(NSDate *)date andType:(NSBubbleType)type
+{
+    self = [super init];
+    if (self) {
+        _view = [view retain];
+        _date = [date retain];
+        _type = type;
+    }
+    return self;
+}
+
 - (void)dealloc
 {
     [_date release];
 	_date = nil;
 	[_text release];
 	_text = nil;
+    [_view release];
+    _view = nil;
     [super dealloc];
 }
 

src/UIBubbleTableView.m

@@ -129,9 +129,16 @@ - (void)reloadData
             dataInternal.type = NSBubbleDataTypeNormalBubble;
 
             // Calculating cell height
-            dataInternal.labelSize = [(dataInternal.data.text ? dataInternal.data.text : @"") sizeWithFont:[UIFont systemFontOfSize:[UIFont systemFontSize]] constrainedToSize:CGSizeMake(220, 9999) lineBreakMode:UILineBreakModeWordWrap];
+            CGFloat dataHeight = 0.0;
+            if (dataInternal.data.text)
+            {
+                dataInternal.labelSize = [(dataInternal.data.text ? dataInternal.data.text : @"") sizeWithFont:[UIFont systemFontOfSize:[UIFont systemFontSize]] constrainedToSize:CGSizeMake(220, 9999) lineBreakMode:UILineBreakModeWordWrap];
+                dataHeight = dataInternal.labelSize.height;
+            }
+            else
+                dataHeight = CGRectGetHeight(dataInternal.data.view.frame)+12.0 /*cf offsetY in cell. this not really cool implementation all these values*/;
 
-            dataInternal.height = dataInternal.labelSize.height + 5 + 11;
+            dataInternal.height = dataHeight + 5 + 11;
 
             dataInternal.header = nil;
 

src/UIBubbleTableViewCell.m

@@ -17,11 +17,13 @@
 
 @interface UIBubbleTableViewCell ()
 - (void) setupInternalData;
+@property (nonatomic, retain) UIView *oldCustomView;
 @end
 
 @implementation UIBubbleTableViewCell
 
 @synthesize dataInternal = _dataInternal;
+@synthesize oldCustomView = _oldCustomView;
 
 - (void)setFrame:(CGRect)frame
 {
@@ -33,6 +35,8 @@ - (void) dealloc
 {
     [_dataInternal release];
 	_dataInternal = nil;
+    [_oldCustomView release];
+    _oldCustomView = nil;
     [super dealloc];
 }
 
@@ -58,22 +62,39 @@ - (void) setupInternalData
     }
 
     NSBubbleType type = self.dataInternal.data.type;
+    CGFloat imageOffsetX = 5.0;
+    CGFloat imageOffsetY = 6.0;
 
-    float x = (type == BubbleTypeSomeoneElse) ? 20 : self.frame.size.width - 20 - self.dataInternal.labelSize.width;
+    CGFloat height = self.dataInternal.data.view ? (CGRectGetHeight(self.dataInternal.data.view.frame) + imageOffsetY*2.0) : self.dataInternal.labelSize.height;
+    CGFloat width = self.dataInternal.data.view ? (CGRectGetWidth(self.dataInternal.data.view.frame) + imageOffsetX*2.0) : self.dataInternal.labelSize.width;
+
+    float x = (type == BubbleTypeSomeoneElse) ? 20 : self.frame.size.width - 20 - width;
     float y = 5 + (self.dataInternal.header ? 30 : 0);
 
-    contentLabel.font = [UIFont systemFontOfSize:[UIFont systemFontSize]];
-    contentLabel.frame = CGRectMake(x, y, self.dataInternal.labelSize.width, self.dataInternal.labelSize.height);
-    contentLabel.text = self.dataInternal.data.text;
-    
+    [self.oldCustomView removeFromSuperview];
+    self.oldCustomView = nil;
+        
+    if (!self.dataInternal.data.view) {
+        contentLabel.hidden = NO;
+        contentLabel.font = [UIFont systemFontOfSize:[UIFont systemFontSize]];
+        contentLabel.frame = CGRectMake(x, y, self.dataInternal.labelSize.width, self.dataInternal.labelSize.height);
+        contentLabel.text = self.dataInternal.data.text;
+    }
+    else
+    {
+        contentLabel.hidden = YES;
+        self.oldCustomView = self.dataInternal.data.view;
+        [self.contentView addSubview:self.oldCustomView];
+        self.oldCustomView.frame = CGRectMake(x+imageOffsetX-1.0, y+imageOffsetY+3.0, CGRectGetWidth(self.oldCustomView.frame), CGRectGetHeight(self.oldCustomView.frame));
+    }
     if (type == BubbleTypeSomeoneElse)
     {
         bubbleImage.image = [[UIImage imageNamed:@"bubbleSomeone.png"] stretchableImageWithLeftCapWidth:21 topCapHeight:14];
-        bubbleImage.frame = CGRectMake(x - 18, y - 4, self.dataInternal.labelSize.width + 30, self.dataInternal.labelSize.height + 15);
+        bubbleImage.frame = CGRectMake(x - 18, y - 4, width + 30, height + 15);
     }
     else {
         bubbleImage.image = [[UIImage imageNamed:@"bubbleMine.png"] stretchableImageWithLeftCapWidth:15 topCapHeight:14];
-        bubbleImage.frame = CGRectMake(x - 9, y - 4, self.dataInternal.labelSize.width + 26, self.dataInternal.labelSize.height + 15);
+        bubbleImage.frame = CGRectMake(x - 9, y - 4, width + 26, height + 15);
     }
 }