Ensure attachment masking is updated to reflect cell relayout.

// FREEBIE

Signal/src/ViewControllers/ConversationView/Cells/OWSMessageCell.m

@@ -17,12 +17,57 @@
 
 NS_ASSUME_NONNULL_BEGIN
 
+@interface BubbleMaskingView : UIView
+
+@property (nonatomic) BOOL isOutgoing;
+@property (nonatomic, nullable, weak) UIView *maskedSubview;
+
+@end
+
+@implementation BubbleMaskingView
+
+- (void)setFrame:(CGRect)frame
+{
+    BOOL didSizeChange = !CGSizeEqualToSize(self.frame.size, frame.size);
+
+    [super setFrame:frame];
+
+    if (didSizeChange) {
+        [self updateMask];
+    }
+}
+
+- (void)setBounds:(CGRect)bounds
+{
+    BOOL didSizeChange = !CGSizeEqualToSize(self.bounds.size, bounds.size);
+
+    [super setBounds:bounds];
+
+    if (didSizeChange) {
+        [self updateMask];
+    }
+}
+
+- (void)updateMask
+{
+    UIView *_Nullable maskedSubview = self.maskedSubview;
+    if (!maskedSubview) {
+        return;
+    }
+    maskedSubview.frame = self.bounds;
+    [JSQMessagesMediaViewBubbleImageMasker applyBubbleImageMaskToMediaView:maskedSubview isOutgoing:self.isOutgoing];
+}
+
+@end
+
+#pragma mark -
+
 @interface OWSMessageCell ()
 
 // The nullable properties are created as needed.
 // The non-nullable properties are so frequently used that it's easier
 // to always keep one around.
-@property (nonatomic) UIView *payloadView;
+@property (nonatomic) BubbleMaskingView *payloadView;
 @property (nonatomic) UILabel *dateHeaderLabel;
 @property (nonatomic) UILabel *textLabel;
 @property (nonatomic, nullable) UIImageView *bubbleImageView;
@@ -60,7 +105,8 @@ NS_ASSUME_NONNULL_BEGIN
 
     self.layoutMargins = UIEdgeInsetsZero;
 
-    self.payloadView = [UIView containerView];
+    self.payloadView = [BubbleMaskingView new];
+    self.payloadView.layoutMargins = UIEdgeInsetsZero;
     [self.contentView addSubview:self.payloadView];
 
     self.footerView = [UIView containerView];
@@ -550,9 +596,12 @@ NS_ASSUME_NONNULL_BEGIN
 
 - (void)cropViewToBubbbleShape:(UIView *)view
 {
-    [self layoutIfNeeded];
+    OWSAssert(view);
-    view.frame = self.payloadView.bounds;
+    OWSAssert(view.superview == self.payloadView);
-    [JSQMessagesMediaViewBubbleImageMasker applyBubbleImageMaskToMediaView:view isOutgoing:!self.isIncoming];
+
+    self.payloadView.isOutgoing = !self.isIncoming;
+    self.payloadView.maskedSubview = view;
+    [self.payloadView updateMask];
 }
 
 //// TODO:
@@ -722,7 +771,6 @@ NS_ASSUME_NONNULL_BEGIN
     [NSLayoutConstraint deactivateConstraints:self.footerConstraints];
     self.footerConstraints = nil;
 
-    // The text label is used so frequently that we always keep one around.
     self.dateHeaderLabel.text = nil;
     self.dateHeaderLabel.hidden = YES;
     self.textLabel.text = nil;
@@ -731,6 +779,7 @@ NS_ASSUME_NONNULL_BEGIN
     self.footerLabel.hidden = YES;
     self.bubbleImageView.image = nil;
     self.bubbleImageView.hidden = YES;
+    self.payloadView.maskedSubview = nil;
 
     [self.stillImageView removeFromSuperview];
     self.stillImageView = nil;

Signal/src/ViewControllers/DebugUI/DebugUIMessages.m

@@ -66,6 +66,14 @@ NS_ASSUME_NONNULL_BEGIN
                         actionBlock:^{
                             [DebugUIMessages sendTextMessages:3000 thread:thread];
                         }],
+        [OWSTableItem itemWithTitle:@"Send 10 tiny text messages (1/sec.)"
+                        actionBlock:^{
+                            [DebugUIMessages sendTinyTextMessages:10 thread:thread];
+                        }],
+        [OWSTableItem itemWithTitle:@"Send 100 tiny text messages (1/sec.)"
+                        actionBlock:^{
+                            [DebugUIMessages sendTinyTextMessages:100 thread:thread];
+                        }],
         [OWSTableItem itemWithTitle:@"Send 10 tiny attachments"
                         actionBlock:^{
                             [DebugUIMessages sendTinyAttachments:10 thread:thread];
@@ -265,6 +273,25 @@ NS_ASSUME_NONNULL_BEGIN
     });
 }
 
++ (void)sendTinyTextMessageInThread:(TSThread *)thread counter:(int)counter
+{
+    NSString *randomText = [[self randomText] substringToIndex:arc4random_uniform(4)];
+    NSString *text = [[[@(counter) description] stringByAppendingString:@" "] stringByAppendingString:randomText];
+    OWSMessageSender *messageSender = [Environment getCurrent].messageSender;
+    [ThreadUtil sendMessageWithText:text inThread:thread messageSender:messageSender];
+}
+
++ (void)sendTinyTextMessages:(int)counter thread:(TSThread *)thread
+{
+    if (counter < 1) {
+        return;
+    }
+    [self sendTinyTextMessageInThread:thread counter:counter];
+    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)1.f * NSEC_PER_SEC), dispatch_get_main_queue(), ^{
+        [self sendTinyTextMessages:counter - 1 thread:thread];
+    });
+}
+
 + (void)ensureRandomFileWithURL:(NSString *)url
                        filename:(NSString *)filename
                         success:(nullable void (^)(NSString *filePath))success