From 1d65d6dc42b06cd89a67023eaea642906b4935bf Mon Sep 17 00:00:00 2001
From: Michael Kirk <michael.code@endoftheworl.de>
Date: Fri, 21 Apr 2017 18:21:15 -0400
Subject: [PATCH] clip upload mask to bounds of generic file bubble

// FREEBIE
---
 .../TSGenericAttachmentAdapter.m               | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

diff --git a/Signal/src/Models/TSMessageAdapaters/TSGenericAttachmentAdapter.m b/Signal/src/Models/TSMessageAdapaters/TSGenericAttachmentAdapter.m
index 3186a1d02..56e571c65 100644
--- a/Signal/src/Models/TSMessageAdapaters/TSGenericAttachmentAdapter.m
+++ b/Signal/src/Models/TSMessageAdapaters/TSGenericAttachmentAdapter.m
@@ -8,8 +8,7 @@
 #import "UIColor+JSQMessages.h"
 #import "UIColor+OWS.h"
 #import "UIFont+OWS.h"
-#import <JSQMessagesViewController/JSQMessagesBubbleImage.h>
-#import <JSQMessagesViewController/JSQMessagesBubbleImageFactory.h>
+#import <JSQMessagesViewController/JSQMessagesMediaViewBubbleImageMasker.h>
 #import <MobileCoreServices/MobileCoreServices.h>
 #import <SignalServiceKit/MimeTypeUtil.h>
 
@@ -101,15 +100,12 @@ NS_ASSUME_NONNULL_BEGIN
         CGSize viewSize = [self mediaViewDisplaySize];
         UIColor *textColor = (self.incoming ? [UIColor blackColor] : [UIColor whiteColor]);
 
-        JSQMessagesBubbleImageFactory *bubbleFactory = [[JSQMessagesBubbleImageFactory alloc] init];
-        JSQMessagesBubbleImage *bubbleImageData = (self.incoming
-                ? [bubbleFactory incomingMessagesBubbleImageWithColor:[UIColor jsq_messageBubbleLightGrayColor]]
-                : [bubbleFactory outgoingMessagesBubbleImageWithColor:[UIColor ows_materialBlueColor]]);
-        UIImage *bubbleImage = [bubbleImageData messageBubbleImage];
-        OWSAssert(bubbleImage);
-        UIImageView *bubbleImageView = [[UIImageView alloc] initWithImage:bubbleImage];
-        _cachedMediaView = bubbleImageView;
-        _cachedMediaView.frame = CGRectMake(0.f, 0.f, viewSize.width, viewSize.height);
+        _cachedMediaView = [[UIView alloc] initWithFrame:CGRectMake(0.f, 0.f, viewSize.width, viewSize.height)];
+
+        _cachedMediaView.backgroundColor
+            = self.incoming ? [UIColor jsq_messageBubbleLightGrayColor] : [UIColor ows_materialBlueColor];
+        [JSQMessagesMediaViewBubbleImageMasker applyBubbleImageMaskToMediaView:_cachedMediaView
+                                                                    isOutgoing:!self.incoming];
 
         const CGFloat kBubbleTailWidth = 6.f;
         CGRect contentFrame = CGRectMake(self.incoming ? kBubbleTailWidth : 0.f,