From 83a470d441a1f2f52f642041e164a478650aae15 Mon Sep 17 00:00:00 2001
From: Matthew Chen <matthew@signal.org>
Date: Wed, 11 Apr 2018 11:38:33 -0400
Subject: [PATCH] Fix previews of oversize text messages.

---
 .../src/Messages/Interactions/TSMessage.m     | 32 +++++++++++++------
 1 file changed, 23 insertions(+), 9 deletions(-)

diff --git a/SignalServiceKit/src/Messages/Interactions/TSMessage.m b/SignalServiceKit/src/Messages/Interactions/TSMessage.m
index c86a95b83..2df353e27 100644
--- a/SignalServiceKit/src/Messages/Interactions/TSMessage.m
+++ b/SignalServiceKit/src/Messages/Interactions/TSMessage.m
@@ -4,6 +4,7 @@
 
 #import "TSMessage.h"
 #import "AppContext.h"
+#import "MIMETypeUtil.h"
 #import "NSDate+OWS.h"
 #import "NSString+SSK.h"
 #import "TSAttachment.h"
@@ -206,7 +207,7 @@ static const NSUInteger OWSMessageSchemaVersion = 4;
             stringWithFormat:@"Media Message with attachmentId: %@ and caption: '%@'", attachmentId, self.body];
     } else if ([self hasAttachments]) {
         NSString *attachmentId = self.attachmentIds[0];
-        return [NSString stringWithFormat:@"Media Message with attachmentId:%@", attachmentId];
+        return [NSString stringWithFormat:@"Media Message with attachmentId: %@", attachmentId];
     } else {
         return [NSString stringWithFormat:@"%@ with body: %@", [self class], self.body];
     }
@@ -214,12 +215,31 @@ static const NSUInteger OWSMessageSchemaVersion = 4;
 
 // TODO: This method contains view-specific logic and probably belongs in NotificationsManager, not in SSK.
 - (NSString *)previewTextWithTransaction:(YapDatabaseReadTransaction *)transaction
-{
+{    
+    NSString *_Nullable bodyDescription = nil;
+    if (self.body.length > 0) {
+        bodyDescription = self.body;
+    }
+    
     NSString *_Nullable attachmentDescription = nil;
     if ([self hasAttachments]) {
         NSString *attachmentId = self.attachmentIds[0];
         TSAttachment *attachment = [TSAttachment fetchObjectWithUniqueID:attachmentId transaction:transaction];
-        if (attachment) {
+        if ([OWSMimeTypeOversizeTextMessage isEqualToString:attachment.contentType]) {
+            // Handle oversize text attachments.
+            if ([attachment isKindOfClass:[TSAttachmentStream class]]) {
+                TSAttachmentStream *attachmentStream = (TSAttachmentStream *)attachment;
+                NSData *_Nullable data = [NSData dataWithContentsOfFile:attachmentStream.filePath];
+                if (data) {
+                    NSString *_Nullable text = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
+                    if (text) {
+                        return text.filterStringForDisplay;
+                    }
+                }
+            }
+            
+            return @"";
+        } else if (attachment) {
             attachmentDescription = attachment.description;
         } else {
             attachmentDescription = NSLocalizedString(@"UNKNOWN_ATTACHMENT_LABEL",
@@ -227,12 +247,6 @@ static const NSUInteger OWSMessageSchemaVersion = 4;
         }
     }
 
-    NSString *_Nullable bodyDescription = nil;
-    if (self.body.length > 0) {
-        // TODO: Filter this text using something like DisplayableText.
-        bodyDescription = self.body;
-    }
-
     if (attachmentDescription.length > 0 && bodyDescription.length > 0) {
         // Attachment with caption.
         if ([CurrentAppContext() isRTL]) {