From 1ef01a7876c99f91c103566a61a5839f938377d1 Mon Sep 17 00:00:00 2001
From: Anton Schukin <anton.schukin@corp.badoo.com>
Date: Mon, 14 Mar 2016 17:02:20 +0000
Subject: [PATCH] Presenters can responds on cell events directly and implement
 additional behaviour in addition to interaction handlers

---
 .../BaseMessage/BaseMessagePresenter.swift    | 20 +++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/ChattoAdditions/Source/Chat Items/BaseMessage/BaseMessagePresenter.swift b/ChattoAdditions/Source/Chat Items/BaseMessage/BaseMessagePresenter.swift
index 09b81b75e..5dc1662db 100644
--- a/ChattoAdditions/Source/Chat Items/BaseMessage/BaseMessagePresenter.swift	
+++ b/ChattoAdditions/Source/Chat Items/BaseMessage/BaseMessagePresenter.swift	
@@ -100,21 +100,21 @@ public class BaseMessagePresenter<BubbleViewT, ViewModelBuilderT, InteractionHan
             cell.messageViewModel = self.messageViewModel
             cell.onBubbleTapped = { [weak self] (cell) in
                 guard let sSelf = self else { return }
-                sSelf.interactionHandler?.userDidTapOnBubble(viewModel: sSelf.messageViewModel)
+                sSelf.onCellBubbleTapped()
             }
             cell.onBubbleLongPressed = { [weak self] (cell) in
                 guard let sSelf = self else { return }
-                sSelf.interactionHandler?.userDidLongPressOnBubble(viewModel: sSelf.messageViewModel)
+                sSelf.onCellBubbleLongPressed()
             }
             cell.onFailedButtonTapped = { [weak self] (cell) in
                 guard let sSelf = self else { return }
-                sSelf.interactionHandler?.userDidTapOnFailIcon(viewModel: sSelf.messageViewModel)
+                sSelf.onCellFailedButtonTapped()
             }
             additionalConfiguration?()
         }, animated: animated, completion: nil)
     }
 
-  public override func heightForCell(maximumWidth width: CGFloat, decorationAttributes: ChatItemDecorationAttributesProtocol?) -> CGFloat {
+    public override func heightForCell(maximumWidth width: CGFloat, decorationAttributes: ChatItemDecorationAttributesProtocol?) -> CGFloat {
         guard let decorationAttributes = decorationAttributes as? ChatItemDecorationAttributes else {
             assert(false, "Expecting decoration attributes")
             return 0
@@ -155,4 +155,16 @@ public class BaseMessagePresenter<BubbleViewT, ViewModelBuilderT, InteractionHan
         // Override in subclass
         return false
     }
+
+    public func onCellBubbleTapped() {
+        self.interactionHandler?.userDidTapOnBubble(viewModel: self.messageViewModel)
+    }
+
+    public func onCellBubbleLongPressed() {
+        self.interactionHandler?.userDidLongPressOnBubble(viewModel: self.messageViewModel)
+    }
+
+    public func onCellFailedButtonTapped() {
+        self.interactionHandler?.userDidTapOnFailIcon(viewModel: self.messageViewModel)
+    }
 }