Add config to skip edited label for some messages

Author: martinmitrevski

DemoAppSwiftUI/AppDelegate.swift

@@ -63,7 +63,13 @@ class AppDelegate: NSObject, UIApplicationDelegate {
         #endif
 
         let utils = Utils(
-            messageListConfig: MessageListConfig(dateIndicatorPlacement: .messageList, userBlockingEnabled: true),
+            messageListConfig: MessageListConfig(
+                dateIndicatorPlacement: .messageList,
+                userBlockingEnabled: true,
+                skipEditedMessageLabel: { message in
+                    message.extraData["ai_generated"]?.boolValue == true
+                }
+            ),
             composerConfig: ComposerConfig(isVoiceRecordingEnabled: true)
         )
         streamChat = StreamChat(chatClient: chatClient, utils: utils)

Sources/StreamChatSwiftUI/ChatChannel/MessageList/MessageListConfig.swift

@@ -31,7 +31,8 @@ public struct MessageListConfig {
         localLinkDetectionEnabled: Bool = true,
         isMessageEditedLabelEnabled: Bool = true,
         markdownSupportEnabled: Bool = true,
-        userBlockingEnabled: Bool = false
+        userBlockingEnabled: Bool = false,
+        skipEditedMessageLabel: @escaping (ChatMessage) -> Bool = { _ in false }
     ) {
         self.messageListType = messageListType
         self.typingIndicatorPlacement = typingIndicatorPlacement
@@ -56,6 +57,7 @@ public struct MessageListConfig {
         self.isMessageEditedLabelEnabled = isMessageEditedLabelEnabled
         self.markdownSupportEnabled = markdownSupportEnabled
         self.userBlockingEnabled = userBlockingEnabled
+        self.skipEditedMessageLabel = skipEditedMessageLabel
     }
 
     public let messageListType: MessageListType
@@ -81,6 +83,7 @@ public struct MessageListConfig {
     public let isMessageEditedLabelEnabled: Bool
     public let markdownSupportEnabled: Bool
     public let userBlockingEnabled: Bool
+    public let skipEditedMessageLabel: (ChatMessage) -> Bool
 }
 
 /// Contains information about the message paddings.

Sources/StreamChatSwiftUI/ChatChannel/MessageList/MessageListHelperViews.swift

@@ -66,7 +66,9 @@ struct MessageDateView: View {
 
     var text: String {
         var text = dateFormatter.string(from: message.createdAt)
-        let showMessageEditedLabel = utils.messageListConfig.isMessageEditedLabelEnabled
+        let messageListConfig = utils.messageListConfig
+        let showMessageEditedLabel = messageListConfig.isMessageEditedLabelEnabled
+            && !messageListConfig.skipEditedMessageLabel(message)
             && message.textUpdatedAt != nil
             && !message.isDeleted
         if showMessageEditedLabel {

StreamChatSwiftUITests/Tests/ChatChannel/MessageContainerView_Tests.swift

@@ -54,6 +54,32 @@ class MessageContainerView_Tests: StreamChatTestCase {
         assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
     }
 
+    func test_messageContainerEditedAIGenerated_snapshot() {
+        // Given
+        let key = "ai_generated"
+        let utils = Utils(
+            dateFormatter: EmptyDateFormatter(),
+            messageListConfig: .init(skipEditedMessageLabel: { message in
+                message.extraData[key]?.boolValue == true
+        }))
+        streamChat = StreamChat(chatClient: chatClient, utils: utils)
+        let message = ChatMessage.mock(
+            id: .unique,
+            cid: .unique,
+            text: "Message sent by current user",
+            author: .mock(id: Self.currentUserId, name: "Martin"),
+            extraData: [key: true],
+            isSentByCurrentUser: true,
+            textUpdatedAt: Date()
+        )
+
+        // When
+        let view = testMessageViewContainer(message: message)
+
+        // Then
+        assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
+    }
+    
     func test_messageContainerCurrentUserColor_snapshot() {
         // Given
         let utils = Utils(dateFormatter: EmptyDateFormatter())