From f27d0ef992e3fdab5fbe2fa64cbfea606fa2fc11 Mon Sep 17 00:00:00 2001
From: Michael Kirk <michael@signal.org>
Date: Thu, 20 Dec 2018 12:25:58 -0700
Subject: [PATCH] Fix SAE crash

---
 .../ViewControllers/SharingThreadPickerViewController.m        | 2 ++
 SignalMessaging/utils/ThreadUtil.h                             | 1 +
 SignalMessaging/utils/ThreadUtil.m                             | 3 ++-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/SignalMessaging/ViewControllers/SharingThreadPickerViewController.m b/SignalMessaging/ViewControllers/SharingThreadPickerViewController.m
index 07cf5b808..2b3ae178e 100644
--- a/SignalMessaging/ViewControllers/SharingThreadPickerViewController.m
+++ b/SignalMessaging/ViewControllers/SharingThreadPickerViewController.m
@@ -232,6 +232,7 @@ typedef void (^SendMessageBlock)(SendCompletionBlock completion);
 
 - (void)attachmentApproval:(AttachmentApprovalViewController *)approvalViewController
      didApproveAttachments:(NSArray<SignalAttachment *> *)attachments
+               messageText:(NSString *_Nullable)messageText
 {
     [ThreadUtil addThreadToProfileWhitelistIfEmptyContactThread:self.thread];
     [self tryToSendMessageWithBlock:^(SendCompletionBlock sendCompletion) {
@@ -243,6 +244,7 @@ typedef void (^SendMessageBlock)(SendCompletionBlock completion);
         // SAE runs as long as it needs.
         // TODO ALBUMS - send album via SAE
         outgoingMessage = [ThreadUtil sendMessageNonDurablyWithAttachment:attachments.firstObject
+                                                              messageBody:messageText
                                                                  inThread:self.thread
                                                          quotedReplyModel:nil
                                                             messageSender:self.messageSender
diff --git a/SignalMessaging/utils/ThreadUtil.h b/SignalMessaging/utils/ThreadUtil.h
index 2d7908058..1ffdfc6a7 100644
--- a/SignalMessaging/utils/ThreadUtil.h
+++ b/SignalMessaging/utils/ThreadUtil.h
@@ -73,6 +73,7 @@ NS_ASSUME_NONNULL_BEGIN
 
 // Used by SAE, otherwise we should use the durable `enqueue` counterpart
 + (TSOutgoingMessage *)sendMessageNonDurablyWithAttachment:(SignalAttachment *)attachment
+                                               messageBody:(nullable NSString *)messageBody
                                                   inThread:(TSThread *)thread
                                           quotedReplyModel:(nullable OWSQuotedReplyModel *)quotedReplyModel
                                              messageSender:(OWSMessageSender *)messageSender
diff --git a/SignalMessaging/utils/ThreadUtil.m b/SignalMessaging/utils/ThreadUtil.m
index 86e1b9287..56fd180ff 100644
--- a/SignalMessaging/utils/ThreadUtil.m
+++ b/SignalMessaging/utils/ThreadUtil.m
@@ -214,6 +214,7 @@ NS_ASSUME_NONNULL_BEGIN
 }
 
 + (TSOutgoingMessage *)sendMessageNonDurablyWithAttachment:(SignalAttachment *)attachment
+                                               messageBody:(nullable NSString *)messageBody
                                                   inThread:(TSThread *)thread
                                           quotedReplyModel:(nullable OWSQuotedReplyModel *)quotedReplyModel
                                              messageSender:(OWSMessageSender *)messageSender
@@ -232,7 +233,7 @@ NS_ASSUME_NONNULL_BEGIN
     TSOutgoingMessage *message =
         [[TSOutgoingMessage alloc] initOutgoingMessageWithTimestamp:[NSDate ows_millisecondTimeStamp]
                                                            inThread:thread
-                                                        messageBody:attachment.captionText
+                                                        messageBody:messageBody
                                                       attachmentIds:[NSMutableArray new]
                                                    expiresInSeconds:expiresInSeconds
                                                     expireStartedAt:0