feat: MessagesLayoutDelegate method for avatar size

martinpucik · martinpucik · commit ddfc814d325e · 2022-03-04T13:07:06.000+01:00
diff --git a/Sources/Layout/MessageSizeCalculator.swift b/Sources/Layout/MessageSizeCalculator.swift
@@ -164,6 +164,11 @@ open class MessageSizeCalculator: CellSizeCalculator {
     }
 
     open func avatarSize(for message: MessageType, at indexPath: IndexPath) -> CGSize {
+        let layoutDelegate = messagesLayout.messagesLayoutDelegate
+        let collectionView = messagesLayout.messagesCollectionView
+        if let size = layoutDelegate.avatarSize(for: message, at: indexPath, in: collectionView) {
+            return size
+        }
         let dataSource = messagesLayout.messagesDataSource
         let isFromCurrentSender = dataSource.isFromCurrentSender(message: message)
         return isFromCurrentSender ? outgoingAvatarSize : incomingAvatarSize
diff --git a/Sources/Protocols/MessagesLayoutDelegate.swift b/Sources/Protocols/MessagesLayoutDelegate.swift
@@ -129,6 +129,15 @@ public protocol MessagesLayoutDelegate: AnyObject {
     ///  alignment from MessageSizeCalculator will be used depending if the message is outgoing or incoming
     func messageBottomLabelAlignment(for message: MessageType, at indexPath: IndexPath, in messagesCollectionView: MessagesCollectionView) -> LabelAlignment?
 
+    /// Specifies the size for the `MessageContentCell`'s avatar image view.
+    /// - Parameters:
+    ///   - message: The `MessageType` that will be displayed for this cell.
+    ///   - indexPath: The `IndexPath` of the cell.
+    ///   - messagesCollectionView: The `MessagesCollectionView` in which this cell will be displayed.
+    /// - Returns: Optional CGSize for the avatar image view. If nil is returned or delegate method is not implemented,
+    /// size from `MessageSizeCalculator`'s `incomingAvatarSize` or `outgoingAvatarSize` will be used depending if the message is outgoing or incoming
+    func avatarSize(for message: MessageType, at indexPath: IndexPath, in messagesCollectionView: MessagesCollectionView) -> CGSize?
+
     /// Text cell size calculator for messages with MessageType.text.
     ///
     /// - Parameters:
@@ -270,6 +279,10 @@ public extension MessagesLayoutDelegate {
         return 0
     }
 
+    func avatarSize(for message: MessageType, at indexPath: IndexPath, in messagesCollectionView: MessagesCollectionView) -> CGSize? {
+        return nil
+    }
+
     func messageBottomLabelAlignment(for message: MessageType, at indexPath: IndexPath, in messagesCollectionView: MessagesCollectionView) -> LabelAlignment? {
         return nil
     }
