From 2f6a2a14a0d2b86ee79470341bcdae39694ecd7d Mon Sep 17 00:00:00 2001
From: Kino <kroy@sfu.ca>
Date: Sat, 18 Jul 2020 14:32:51 -0700
Subject: [PATCH] Configure avatarView, currentSender

---
 Example/Sources/Views/MessagesView.swift | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/Example/Sources/Views/MessagesView.swift b/Example/Sources/Views/MessagesView.swift
index 936f6e96..d0799b55 100644
--- a/Example/Sources/Views/MessagesView.swift
+++ b/Example/Sources/Views/MessagesView.swift
@@ -7,8 +7,8 @@
 //
 
 #if canImport(SwiftUI)
-
 import SwiftUI
+#endif
 import MessageKit
 
 class MessageSwiftUIVC: MessagesViewController {
@@ -56,7 +56,7 @@ struct MessagesView: UIViewControllerRepresentable {
 @available(iOS 13.0.0, *)
 extension MessagesView.Coordinator: MessagesDataSource {
     func currentSender() -> SenderType {
-        return MockUser(senderId: "", displayName: "")
+        return SampleData.shared.currentSender
     }
     
     func messageForItem(at indexPath: IndexPath, in messagesCollectionView: MessagesCollectionView) -> MessageType {
@@ -70,7 +70,10 @@ extension MessagesView.Coordinator: MessagesDataSource {
 
 @available(iOS 13.0.0, *)
 extension MessagesView.Coordinator: MessagesLayoutDelegate, MessagesDisplayDelegate {
-    
+    func configureAvatarView(_ avatarView: AvatarView, for message: MessageType, at indexPath: IndexPath, in messagesCollectionView: MessagesCollectionView) {
+        let avatar = SampleData.shared.getAvatarFor(sender: message.sender)
+        avatarView.set(avatar: avatar)
+    }
 }
 
 @available(iOS 13.0.0, *)
@@ -91,5 +94,3 @@ struct MessagesView_Previews: PreviewProvider {
         MessageViewPreviewWrapper()
     }
 }
-
-#endif