From bd4b4f0041292ca2bb7246c666ba77065c1853e3 Mon Sep 17 00:00:00 2001
From: Michael Kirk <michael.code@endoftheworl.de>
Date: Sat, 21 Oct 2017 18:31:29 -0700
Subject: [PATCH] Use proper color when messages fail to send.

// FREEBIE
---
 .../ConversationView/Cells/OWSMessageCell.m       | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/Signal/src/ViewControllers/ConversationView/Cells/OWSMessageCell.m b/Signal/src/ViewControllers/ConversationView/Cells/OWSMessageCell.m
index e4754fd71..2c0ddf7c5 100644
--- a/Signal/src/ViewControllers/ConversationView/Cells/OWSMessageCell.m
+++ b/Signal/src/ViewControllers/ConversationView/Cells/OWSMessageCell.m
@@ -226,9 +226,18 @@ NS_ASSUME_NONNULL_BEGIN
 
     DDLogError(@"%p loadForDisplay: %@", self, NSStringForOWSMessageCellType(self.cellType));
 
-    BOOL isIncoming = self.isIncoming;
-    JSQMessagesBubbleImage *bubbleImageData
-        = isIncoming ? [self.bubbleFactory incoming] : [self.bubbleFactory outgoing];
+    JSQMessagesBubbleImage *bubbleImageData;
+    if ([self.viewItem.interaction isKindOfClass:[TSOutgoingMessage class]]) {
+        TSOutgoingMessage *outgoingMessage = (TSOutgoingMessage *)self.viewItem.interaction;
+        if (outgoingMessage.messageState == TSOutgoingMessageStateUnsent) {
+            bubbleImageData = self.bubbleFactory.outgoingFailed;
+        } else {
+            bubbleImageData = self.bubbleFactory.outgoing;
+        }
+    } else {
+        bubbleImageData = self.bubbleFactory.incoming;
+    }
+
     self.bubbleImageView.image = bubbleImageData.messageBubbleImage;
 
     [self updateDateHeader];