Self info message on end poll

Author: kate-signal

Signal/translations/en.lproj/Localizable.strings

@@ -6541,6 +6541,9 @@
 /* Shown when another user ends a poll. Embeds {{ another user }} and {{ poll question }}. */
 "POLL_ENDED_BY_OTHER_CHAT_LIST_UPDATE" = "%@ ended the poll: %@";
 
+/* Shown when the local user ends a poll. Embeds {{ poll question }}. */
+"POLL_ENDED_BY_YOU_CHAT_LIST_UPDATE" = "You ended the poll: %@";
+
 /* Notification that {{contact}} ended a poll with question {{poll question}} */
 "POLL_ENDED_NOTIFICATION" = "%@ ended the poll “%@”";
 

SignalServiceKit/Messages/Interactions/Polls/OutgoingPollTerminate.swift

@@ -96,6 +96,8 @@ class OutgoingPollTerminateMessage: TSOutgoingMessage {
 
             try PollStore().revertPollTerminate(interactionId: interactionId, transaction: tx)
 
+            // TODO: delete info message.
+
             SSKEnvironment.shared.databaseStorageRef.touch(interaction: targetMessage, shouldReindex: false, tx: tx)
         } catch {
             Logger.error("Failed to revert poll terminate: \(error)")

SignalServiceKit/Messages/Interactions/Polls/PollMessageManager.swift

@@ -254,6 +254,19 @@ public class PollMessageManager {
                 message: preparedMessage,
                 transaction: tx
             )
+
+            guard let localAci = accountManager.localIdentifiers(tx: tx)?.aci else {
+                throw OWSAssertionError("User not registered")
+            }
+
+            insertInfoMessageForEndPoll(
+                timestamp: Date().ows_millisecondsSince1970,
+                groupThread: thread,
+                targetPollTimestamp: targetPoll.timestamp,
+                pollQuestion: poll.question,
+                terminateAuthor: localAci,
+                tx: tx
+            )
         }
     }
 

SignalServiceKit/Messages/Interactions/Polls/TSInfoMessage+Polls.swift

@@ -82,6 +82,11 @@ extension TSInfoMessage {
             return nil
         }
 
+        guard let localAci = DependenciesBridge.shared.tsAccountManager.localIdentifiers(tx: transaction)?.aci else {
+            Logger.error("Can't find local ACI")
+            return nil
+        }
+
         var pollTerminateAci: Aci?
         do {
             pollTerminateAci = try Aci.parseFrom(serviceIdBinary: endPollItem.authorServiceIdBinary)
@@ -95,6 +100,14 @@ extension TSInfoMessage {
             return nil
         }
 
+        if localAci == pollTerminateAci {
+            let formatString = OWSLocalizedString(
+                "POLL_ENDED_BY_YOU_CHAT_LIST_UPDATE",
+                comment: "Shown when the local user ends a poll. Embeds {{ poll question }}."
+            )
+            return String(format: formatString, question)
+        }
+
         let displayName = SSKEnvironment.shared.contactManagerRef.displayName(
             for: SignalServiceAddress(pollTerminateAci),
             tx: transaction