チャット中の相手を取得し表示する

Author: geekshu625

chat-iOS/Views/SelectChat/SelectChatModel.swift

@@ -5,16 +5,63 @@
 //  Created by 松木周 on 2020/07/15.
 //
 
-import Foundation
+import Firebase
 
 protocol SelectChatModelProtocol {
     var presenter: SelectChatModelOutput! { get set }
+    func setUpFirestore()
+    func fetchCurrentChatRoom()
 }
 
 protocol SelectChatModelOutput: class {
+    func successFetchCurrentChatRooms(currentChatRooms rooms: [Room])
+    func onError(error: Error?)
 }
 
 final class SelectChatModel: SelectChatModelProtocol {
 
     weak var presenter: SelectChatModelOutput!
+    private var firestore: Firestore!
+    private var listener: ListenerRegistration?
+
+    init() {
+        self.setUpFirestore()
+    }
+
+    deinit {
+        listener?.remove()
+    }
+
+    func setUpFirestore() {
+        self.firestore = Firestore.firestore()
+        let settings = FirestoreSettings()
+        self.firestore.settings = settings
+    }
+
+    func fetchCurrentChatRoom() {
+
+        guard let curretnUserId = Auth.auth().currentUser?.uid else { return }
+
+        self.listener = self.firestore.collection("rooms").whereField("members", arrayContains: curretnUserId).addSnapshotListener { [weak self] (documentSnapshot, error) in
+
+            if let error = error {
+                self?.presenter.onError(error: error)
+                return
+            }
+
+            guard let documents = documentSnapshot?.documents else {
+                print("The document doesn't exist.")
+                return
+            }
+
+            let rooms = documents.compactMap { queryDocumentSnapshot -> Room? in
+                return try? queryDocumentSnapshot.data(as: Room.self)
+            }
+
+            self?.presenter.successFetchCurrentChatRooms(currentChatRooms: rooms)
+        }
+        
+    }
+
 }
+

chat-iOS/Views/SelectChat/SelectChatTableViewCell.swift

@@ -24,5 +24,11 @@ class SelectChatTableViewCell: UITableViewCell {
 
         // Configure the view for the selected state
     }
+
+    func generateCell(withRoom room: Room) {
+        userNameLabel.text = room.name
+        lastMessageLabel.text = room.name
+        //TODO:- Storageから画像を取得して表示する
+    }
 
 }

chat-iOS/Views/SelectChat/SelectChatViewController.swift

@@ -12,12 +12,14 @@ class SelectChatViewController: UIViewController {
     @IBOutlet weak var selectChatTableView: UITableView!
 
     private let reuseCellId = "SelectChatTableViewCell"
+    private var currentChatRooms: [Room] = []
     private var presenter: SelectChatViewPresenterProtocol!
 
     override func viewDidLoad() {
         super.viewDidLoad()
 
         setupSelectChatTableView()
+        self.presenter.didLoadViewController()
     }
 
     func inject(with presenter: SelectChatViewPresenterProtocol) {
@@ -44,17 +46,30 @@ extension SelectChatViewController: UITableViewDelegate {
 extension SelectChatViewController: UITableViewDataSource {
 
     func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
-        return 10
+        return currentChatRooms.count
     }
 
     func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
         let cell = tableView.dequeueReusableCell(withIdentifier: reuseCellId) as! SelectChatTableViewCell
-
+        cell.generateCell(withRoom: currentChatRooms[indexPath.row])
         return cell
     }
 
 }
 
 extension SelectChatViewController: SelectChatViewPresenterOutput {
+
+    func setCurrentChatUsers() {
+        self.currentChatRooms = self.presenter.currentChatRooms
+        self.selectChatTableView.reloadData()
+    }
+
+    func showAlert(withMessage message: String) {
+        let alert = UIAlertController(title: "エラーが発生しました", message: message, preferredStyle: .alert)
+        let defaultAction: UIAlertAction = UIAlertAction(title: "OK", style: .default, handler: nil)
+
+        alert.addAction(defaultAction)
+        present(alert, animated: false, completion: nil)
+    }
 
 }

chat-iOS/Views/SelectChat/SelectChatViewPresenter.swift

@@ -9,18 +9,45 @@ import Foundation
 
 protocol SelectChatViewPresenterProtocol {
     var view: SelectChatViewPresenterOutput! { get set }
+    var currentChatRooms: [Room] { get }
+    func didLoadViewController()
 }
 
 protocol SelectChatViewPresenterOutput: class {
+    func setCurrentChatUsers()
+    func showAlert(withMessage message: String)
 }
 
 final class SelectChatViewPresenter: SelectChatViewPresenterProtocol, SelectChatModelOutput {
 
     weak var view: SelectChatViewPresenterOutput!
     private var model: SelectChatModelProtocol
 
+    private(set) var _currentChatRooms: [Room] = []
+
     init(model: SelectChatModelProtocol) {
         self.model = model
+        self.model.presenter = self
+    }
+
+    var currentChatRooms: [Room] {
+        return _currentChatRooms
+    }
+
+    func didLoadViewController() {
+        self.model.fetchCurrentChatRoom()
+    }
+
+    func successFetchCurrentChatRooms(currentChatRooms rooms: [Room]) {
+        self._currentChatRooms = rooms
+        self.view.setCurrentChatUsers()
+    }
+
+    func onError(error: Error?) {
+        DispatchQueue.main.async { [weak self] in
+            let message = error?.localizedDescription ?? "チャット相手を取得できませんでした"
+            self?.view.showAlert(withMessage: message)
+        }
     }
 
 }