Add factory method for custom attachments in quoted messages

Author: martinmitrevski

CHANGELOG.md

@@ -3,6 +3,9 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
 # Upcoming
 
+### ✅ Added
+- Add factory method for custom attachments in quoted messages
+
 ### 🐞 Fixed
 - Fix marked read while the app is in the background
 - Fix recently saved images to camera roll don't show up

Sources/StreamChatSwiftUI/ChatChannel/MessageList/QuotedMessageView.swift

@@ -60,13 +60,18 @@ public struct QuotedMessageView<Factory: ViewFactory>: View {
     @Injected(\.images) private var images
     @Injected(\.fonts) private var fonts
     @Injected(\.colors) private var colors
-
+    @Injected(\.utils) private var utils
+    
     private let attachmentWidth: CGFloat = 36
 
     public var factory: Factory
     public var quotedMessage: ChatMessage
     public var fillAvailableSpace: Bool
     public var forceLeftToRight: Bool
+    
+    private var messageTypeResolver: MessageTypeResolving {
+        utils.messageTypeResolver
+    }
 
     public init(
         factory: Factory,
@@ -84,7 +89,9 @@ public struct QuotedMessageView<Factory: ViewFactory>: View {
         HStack(alignment: .top) {
             if !quotedMessage.attachmentCounts.isEmpty {
                 ZStack {
-                    if !quotedMessage.imageAttachments.isEmpty {
+                    if messageTypeResolver.hasCustomAttachment(message: quotedMessage) {
+                        factory.makeCustomAttachmentQuotedView(for: quotedMessage)
+                    } else if !quotedMessage.imageAttachments.isEmpty {
                         LazyLoadingImage(
                             source: quotedMessage.imageAttachments[0].imageURL,
                             width: attachmentWidth,

Sources/StreamChatSwiftUI/DefaultViewFactory.swift

@@ -830,6 +830,10 @@ extension ViewFactory {
         )
     }
 
+    public func makeCustomAttachmentQuotedView(for message: ChatMessage) -> some View {
+        EmptyView()
+    }
+    
     public func makeEditedMessageHeaderView(
         editedMessage: Binding<ChatMessage?>
     ) -> some View {

Sources/StreamChatSwiftUI/ViewFactory.swift

@@ -822,6 +822,12 @@ public protocol ViewFactory: AnyObject {
         isInComposer: Bool,
         scrolledId: Binding<String?>
     ) -> QuotedMessageViewType
+    
+    associatedtype CustomAttachmentQuotedViewType: View
+    /// Creates a quoted view for custom attachments. Returns `EmptyView` by default.
+    /// - Parameter message: the quoted message.
+    /// - Returns: view shown in quoted messages with custom attachments.
+    func makeCustomAttachmentQuotedView(for message: ChatMessage) -> CustomAttachmentQuotedViewType
 
     associatedtype EditedMessageHeaderViewType: View
     /// Creates the edited message header view in the composer.

StreamChatSwiftUITests/Tests/Utils/ViewFactory_Tests.swift

@@ -819,6 +819,17 @@ class ViewFactory_Tests: StreamChatTestCase {
         // Then
         XCTAssert(view is BottomReactionsView)
     }
+    
+    func test_viewFactory_makeCustomAttachmentQuotedView() {
+        // Given
+        let viewFactory = DefaultViewFactory.shared
+        
+        // When
+        let view = viewFactory.makeCustomAttachmentQuotedView(for: .mock())
+        
+        // Then
+        XCTAssert(view is EmptyView)
+    }
 }
 
 extension ChannelAction: Equatable {