ChatInputBar and ChatInputItem extensions

ChattoAdditions/Source/Input/ChatInputBar.swift

@@ -51,6 +51,10 @@ open class ChatInputBar: ReusableXibView {
         return !inputBar.textView.text.isEmpty
     }
 
+    public var inputTextView: UITextView? {
+        return self.textView
+    }
+
     @IBOutlet weak var scrollView: HorizontalStackScrollView!
     @IBOutlet weak var textView: ExpandableTextView!
     @IBOutlet weak var sendButton: UIButton!

ChattoAdditions/Source/Input/ChatInputItem.swift

@@ -39,6 +39,7 @@ public protocol ChatInputItemProtocol: AnyObject {
 
     var supportsExpandableState: Bool { get }
     var expandedStateTopMargin: CGFloat { get }
+    var shouldSaveDraftMessage: Bool { get }
 
     func handleInput(_ input: AnyObject)
 }

ChattoAdditions/Source/Input/Photos/PhotosChatInputItem.swift

@@ -93,6 +93,10 @@ open class PhotosChatInputItem: ChatInputItemProtocol {
     }
 
     open func handleInput(_ input: AnyObject) {}
+
+    open var shouldSaveDraftMessage: Bool {
+        return false
+    }
 }
 
 // MARK: - PhotosInputViewDelegate

ChattoAdditions/Source/Input/Text/TextChatInputItem.swift

@@ -84,4 +84,8 @@ extension TextChatInputItem: ChatInputItemProtocol {
             self.textInputHandler?(text)
         }
     }
+
+    public var shouldSaveDraftMessage: Bool {
+        return false
+    }
 }

ChattoAdditions/Tests/Input/ChatInputItemTests.swift

@@ -30,6 +30,7 @@ class ChatInputItemTests: XCTestCase {
 
 @objc
 class MockInputItem: NSObject, ChatInputItemProtocol {
+    var shouldSaveDraftMessage: Bool = false
     var supportsExpandableState: Bool = false
     var expandedStateTopMargin: CGFloat = 0.0
     var selected = false

ChattoApp/ChattoApp/Source/Chat Items/Content Aware Input /ContentAwareInputItem.swift

@@ -60,6 +60,10 @@ open class ContentAwareInputItem {
 
 // MARK: - ChatInputItemProtocol
 extension ContentAwareInputItem: ChatInputItemProtocol {
+    public var shouldSaveDraftMessage: Bool {
+        return false
+    }
+
     public var supportsExpandableState: Bool {
         return true
     }