/* 
 The struct User must implement getter for objejectId and name and 
 function == because SenderProtocol must conform Equatable.
 */
struct User: SenderProtocol {
    var objectId: String?
    var name: String?
    public static func == (lhs: Person, rhs: Person) -> Bool {
        return lhs.objectId == rhs.objectId
    }
}

/* 
 The class MyMessage must implement getter for message, sender, timeStr, and dayStr.
 ChatUIKit provides a Date extension that export Date to time and day, see
 below the implementation.
 */
struct MyMessage: Message {
    var date: Date?
    var message: String?
    var sender: User?
    var timeStr: String? { get { return self.date?.chatUIKitTimeStr  } }
    var dayStr: String? { get { return self.date?.chatUIKitDayStr } }    
    public static func == (lhs: MyMessage, rhs: MyMessage) -> Bool {
        return lhs.objectId == rhs.objectId
    }

}

class MyChatViewController: ChatViewController {
	MyChatViewController(user: User) {
		super.init()
		let user = User.current()
        let chatDataSource = ChatDataSource<MyMessage, User>(owner: user, otherName:"Admin")
		self.dataSource = chatDataSource
        self.delegate = chatDataSource
        self.appointment = appointment
        self.chatPartner = chatPartner
        self.sendMessageBlock = { (_ message:String, _ succed:@escaping ()->Void) in
        	//  Your code probably should send the message to your server and only when gets
        	// the confirmation it will update the chat view controller or notifying about what
        	// was the problem.
        	guard let chatDataSource = self.dataSource as?
        		ChatDataSource<PersonMessage, Person> else { return }
            var message = MyMessage()
            message.message = message
            let user = User.current()
            message.sender = user
            if chatDataSource.chatList.contains( where: { $0 == message } ) { return }
            chatDataSource.chatList.append(message)
            chatViewController.tableView.reloadData()
            chatViewController.resetTablePosition()
            succed()
		}
    }
}

class MyContainerChatViewController: UIViewController {
    var chatViewController : ChatViewController?
    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
        if let identifier = segue.identifier,
            identifier  == "MyContainerToChatViewControllerSegue",
            let chatViewController = segue.destination as? ChatViewController {
            self.chatViewController = chatViewController
            let user = User.current()
            let chatDataSource = ChatDataSource<MyMessage, User>(owner: user, otherName:"Admin")
            chatViewController.dataSource = chatDataSource
            chatViewController.delegate = chatDataSource
            chatViewController.sendMessageBlock = {
            	(_ message:String, _ succed:@escaping ()->Void) in
        			// Same implementation as before.
            }
        }
    }

class MyContainerChatViewController: UIViewController {
    var chatViewController : ChatViewController?
	func messageArrive( message: MyMessage ){
        guard let chatDataSource = self.dataSource as? ChatDataSource<MyMessage, User> else {
        	return
        }
        if chatDataSource.chatList.contains( where: { $0 == message } ) { return }
        chatDataSource.chatList.append(message)
        chatViewController.tableView.reloadData()
        chatViewController.resetTablePosition()
	}
}