Added ability to react on different states of UILongPressGestureRecog…

ChattoAdditions/Source/Chat Items/BaseMessage/BaseMessagePresenter.swift

@@ -36,7 +36,8 @@ public protocol BaseMessageInteractionHandlerProtocol {
     associatedtype ViewModelT
     func userDidTapOnFailIcon(viewModel viewModel: ViewModelT)
     func userDidTapOnBubble(viewModel viewModel: ViewModelT)
-    func userDidLongPressOnBubble(viewModel viewModel: ViewModelT)
+    func userDidBeginLongPressOnBubble(viewModel viewModel: ViewModelT)
+    func userDidEndLongPressOnBubble(viewModel viewModel: ViewModelT)
 }
 
 public class BaseMessagePresenter<BubbleViewT, ViewModelBuilderT, InteractionHandlerT where
@@ -102,9 +103,13 @@ public class BaseMessagePresenter<BubbleViewT, ViewModelBuilderT, InteractionHan
                 guard let sSelf = self else { return }
                 sSelf.onCellBubbleTapped()
             }
-            cell.onBubbleLongPressed = { [weak self] (cell) in
+            cell.onBubbleLongPressBegan = { [weak self] (cell) in
                 guard let sSelf = self else { return }
-                sSelf.onCellBubbleLongPressed()
+                sSelf.onCellBubbleLongPressBegan()
+            }
+            cell.onBubbleLongPressEnded = { [weak self] (cell) in
+                guard let sSelf = self else { return }
+                sSelf.onCellBubbleLongPressEnded()
             }
             cell.onFailedButtonTapped = { [weak self] (cell) in
                 guard let sSelf = self else { return }
@@ -160,8 +165,12 @@ public class BaseMessagePresenter<BubbleViewT, ViewModelBuilderT, InteractionHan
         self.interactionHandler?.userDidTapOnBubble(viewModel: self.messageViewModel)
     }
 
-    public func onCellBubbleLongPressed() {
-        self.interactionHandler?.userDidLongPressOnBubble(viewModel: self.messageViewModel)
+    public func onCellBubbleLongPressBegan() {
+        self.interactionHandler?.userDidBeginLongPressOnBubble(viewModel: self.messageViewModel)
+    }
+
+    public func onCellBubbleLongPressEnded() {
+        self.interactionHandler?.userDidEndLongPressOnBubble(viewModel: self.messageViewModel)
     }
 
     public func onCellFailedButtonTapped() {

ChattoAdditions/Source/Chat Items/BaseMessage/Views/BaseMessageCollectionViewCell.swift

@@ -312,17 +312,19 @@ public class BaseMessageCollectionViewCell<BubbleViewType where BubbleViewType:U
         self.onBubbleTapped?(cell: self)
     }
 
-    public var onBubbleLongPressed: ((cell: BaseMessageCollectionViewCell) -> Void)?
+    public var onBubbleLongPressBegan: ((cell: BaseMessageCollectionViewCell) -> Void)?
+    public var onBubbleLongPressEnded: ((cell: BaseMessageCollectionViewCell) -> Void)?
     @objc
     private func bubbleLongPressed(longPressGestureRecognizer: UILongPressGestureRecognizer) {
-        if longPressGestureRecognizer.state == .Began {
-            self.bubbleLongPressed()
+        switch longPressGestureRecognizer.state {
+        case .Began:
+            self.onBubbleLongPressBegan?(cell: self)
+        case .Ended, .Cancelled:
+            self.onBubbleLongPressEnded?(cell: self)
+        default:
+            break
         }
     }
-
-    func bubbleLongPressed() {
-        self.onBubbleLongPressed?(cell: self)
-    }
 }
 
 struct BaseMessageLayoutModel {

ChattoAdditions/Tests/Chat Items/BaseMessage/BaseMessagePresenterTests.swift

@@ -57,10 +57,17 @@ class BaseMessagePresenterTests: XCTestCase {
         XCTAssertTrue(self.interactionHandler.didHandleTapOnBubble)
     }
 
-    func testThat_WhenCellIsLongPressedOnBubble_ThenInteractionHandlerHandlesEvent() {
+    func testThat_WhenCellIsBeginLongPressOnBubble_ThenInteractionHandlerHandlesEvent() {
         let cell = PhotoMessageCollectionViewCell(frame: CGRect.zero)
         self.presenter.configureCell(cell, decorationAttributes: self.decorationAttributes)
-        cell.bubbleLongPressed()
-        XCTAssertTrue(self.interactionHandler.didHandleLongPressOnBubble)
+        cell.onBubbleLongPressBegan?(cell: cell)
+        XCTAssertTrue(self.interactionHandler.didHandleBeginLongPressOnBubble)
+    }
+
+    func testThat_WhenCellIsEndLongPressOnBubble_ThenInteractionHandlerHandlesEvent() {
+        let cell = PhotoMessageCollectionViewCell(frame: CGRect.zero)
+        self.presenter.configureCell(cell, decorationAttributes: self.decorationAttributes)
+        cell.onBubbleLongPressEnded?(cell: cell)
+        XCTAssertTrue(self.interactionHandler.didHandleEndLongPressOnBubble)
     }
 }

ChattoAdditions/Tests/Chat Items/PhotoMessages/PhotoMessagePresenterTests.swift

@@ -88,8 +88,13 @@ class PhotoMessageTestHandler: BaseMessageInteractionHandlerProtocol {
         self.didHandleTapOnBubble = true
     }
 
-    var didHandleLongPressOnBubble = false
-    func userDidLongPressOnBubble(viewModel viewModel: ViewModelT) {
-        self.didHandleLongPressOnBubble = true
+    var didHandleBeginLongPressOnBubble = false
+    func userDidBeginLongPressOnBubble(viewModel viewModel: ViewModelT) {
+        self.didHandleBeginLongPressOnBubble = true
+    }
+
+    var didHandleEndLongPressOnBubble = false
+    func userDidEndLongPressOnBubble(viewModel viewModel: ViewModelT) {
+        self.didHandleEndLongPressOnBubble = true
     }
 }

ChattoAdditions/Tests/Chat Items/TextMessages/TextMessagePresenterTests.swift

@@ -97,7 +97,11 @@ class TextMessageTestHandler: BaseMessageInteractionHandlerProtocol {
 
     }
 
-    func userDidLongPressOnBubble(viewModel viewModel: ViewModelT) {
+    func userDidBeginLongPressOnBubble(viewModel viewModel: ViewModelT) {
+
+    }
+
+    func userDidEndLongPressOnBubble(viewModel viewModel: ViewModelT) {
 
     }
 }

ChattoApp/ChattoApp/Source/BaseMessageHandler.swift

@@ -46,7 +46,11 @@ class BaseMessageHandler {
 
     }
 
-    func userDidLongPressOnBubble(viewModel viewModel: DemoMessageViewModelProtocol) {
-        print("userDidLongPressOnBubble")
+    func userDidBeginLongPressOnBubble(viewModel viewModel: DemoMessageViewModelProtocol) {
+        print("userDidBeginLongPressOnBubble")
+    }
+
+    func userDidEndLongPressOnBubble(viewModel viewModel: DemoMessageViewModelProtocol) {
+        print("userDidEndLongPressOnBubble")
     }
 }

ChattoApp/ChattoApp/Source/Photo Messages/DemoPhotoMessageHandler.swift

@@ -39,7 +39,11 @@ class DemoPhotoMessageHandler: BaseMessageInteractionHandlerProtocol {
         self.baseHandler.userDidTapOnBubble(viewModel: viewModel)
     }
 
-    func userDidLongPressOnBubble(viewModel viewModel: DemoPhotoMessageViewModel) {
-        self.baseHandler.userDidLongPressOnBubble(viewModel: viewModel)
+    func userDidBeginLongPressOnBubble(viewModel viewModel: DemoPhotoMessageViewModel) {
+        self.baseHandler.userDidBeginLongPressOnBubble(viewModel: viewModel)
+    }
+
+    func userDidEndLongPressOnBubble(viewModel viewModel: DemoPhotoMessageViewModel) {
+        self.baseHandler.userDidEndLongPressOnBubble(viewModel: viewModel)
     }
 }

ChattoApp/ChattoApp/Source/Text Messages/DemoTextMessageHandler.swift

@@ -38,7 +38,11 @@ class DemoTextMessageHandler: BaseMessageInteractionHandlerProtocol {
         self.baseHandler.userDidTapOnBubble(viewModel: viewModel)
     }
 
-    func userDidLongPressOnBubble(viewModel viewModel: DemoTextMessageViewModel) {
-        self.baseHandler.userDidLongPressOnBubble(viewModel: viewModel)
+    func userDidBeginLongPressOnBubble(viewModel viewModel: DemoTextMessageViewModel) {
+        self.baseHandler.userDidBeginLongPressOnBubble(viewModel: viewModel)
+    }
+
+    func userDidEndLongPressOnBubble(viewModel viewModel: DemoTextMessageViewModel) {
+        self.baseHandler.userDidEndLongPressOnBubble(viewModel: viewModel)
     }
 }