Allowing storyboard layout & custom main view class for BaseChatViewController

Chatto/Source/ChatController/BaseChatViewController.swift

@@ -58,6 +58,10 @@ open class BaseChatViewController: UIViewController, UICollectionViewDataSource,
         }
     }
 
+    // If set to false user is responsible to make sure that view provided in loadView() implements BaseChatViewContollerViewProtocol.
+    // Must be set before loadView is called.
+    public var substitutesMainViewAutomatically = true
+
     // Custom update on setting the data source. if triggeringUpdateType is nil it won't enqueue any update (you should do it later manually)
     public final func setChatDataSource(_ dataSource: ChatDataSourceProtocol?, triggeringUpdateType updateType: UpdateType?) {
         self._chatDataSource = dataSource
@@ -73,8 +77,13 @@ open class BaseChatViewController: UIViewController, UICollectionViewDataSource,
     }
 
     open override func loadView() {
-        self.view = BaseChatViewControllerView() // http://stackoverflow.com/questions/24596031/uiviewcontroller-with-inputaccessoryview-is-not-deallocated
-        self.view.backgroundColor = UIColor.white
+        if substitutesMainViewAutomatically {
+            self.view = BaseChatViewControllerView() // http://stackoverflow.com/questions/24596031/uiviewcontroller-with-inputaccessoryview-is-not-deallocated
+            self.view.backgroundColor = UIColor.white
+        } else {
+            super.loadView()
+        }
+
     }
 
     override open func viewDidLoad() {
@@ -184,7 +193,9 @@ open class BaseChatViewController: UIViewController, UICollectionViewDataSource,
             sSelf.isAdjustingInputContainer = false
         }
         self.keyboardTracker = KeyboardTracker(viewController: self, inputContainer: self.inputContainer, layoutBlock: layoutBlock, notificationCenter: self.notificationCenter)
-        (self.view as? BaseChatViewControllerView)?.bmaInputAccessoryView = self.keyboardTracker?.trackingView
+
+        (self.view as? BaseChatViewControllerViewProtocol)?.bmaInputAccessoryView = self.keyboardTracker?.trackingView
+
     }
 
     var notificationCenter = NotificationCenter.default

Chatto/Source/ChatController/Collaborators/BaseChatViewControllerView.swift

@@ -24,8 +24,13 @@
 
 import Foundation
 
+// If you wish to use your custom view instead of BaseChatViewControllerView, you must implement this protocol.
+public protocol BaseChatViewControllerViewProtocol: class {
+    var bmaInputAccessoryView: UIView? { get set }
+}
+
 // http://stackoverflow.com/questions/24596031/uiviewcontroller-with-inputaccessoryview-is-not-deallocated
-final class BaseChatViewControllerView: UIView {
+final class BaseChatViewControllerView: UIView, BaseChatViewControllerViewProtocol {
 
     var bmaInputAccessoryView: UIView?