Also add overriding left/right button classes.

Author: kirb

Source/SLKTextInputbar.h

@@ -66,6 +66,16 @@ NS_ASSUME_NONNULL_BEGIN
 /// @name Initialization
 ///------------------------------------------------
 
+/**
+ Initializes a text input bar with a class to be used for the text view, and left and right buttons
+ 
+ @param textViewClass The class to be used when creating the text view. May be nil. If provided, the class must be a subclass of SLKTextView
+ @param leftButtonClass The class to be used when creating the left button. May be nil. If provided, the class must be a subclass of UIButton
+ @param rightButtonClass The class to be used when creating the right button. May be nil. If provided, the class must be a subclass of UIButton
+ @return An initialized SLKTextInputbar object or nil if the object could not be created.
+ */
+- (instancetype)initWithTextViewClass:(Class)textViewClass leftButtonClass:(Class)leftButtonClass rightButtonClass:(Class)rightButtonClass;
+
 /**
  Initializes a text input bar with a class to be used for the text view
  

Source/SLKTextInputbar.m

@@ -35,6 +35,8 @@ @interface SLKTextInputbar ()
 @property (nonatomic) CGPoint previousOrigin;
 
 @property (nonatomic, strong) Class textViewClass;
+@property (nonatomic, strong) Class leftButtonClass;
+@property (nonatomic, strong) Class rightButtonClass;
 
 @property (nonatomic, getter=isHidden) BOOL hidden; // Required override
 
@@ -46,9 +48,16 @@ @implementation SLKTextInputbar
 #pragma mark - Initialization
 
 - (instancetype)initWithTextViewClass:(Class)textViewClass
+{
+    return [self initWithTextViewClass:textViewClass leftButtonClass:nil rightButtonClass:nil];
+}
+
+- (instancetype)initWithTextViewClass:(Class)textViewClass leftButtonClass:(Class)leftButtonClass rightButtonClass:(Class)rightButtonClass
 {
     if (self = [super init]) {
         self.textViewClass = textViewClass;
+        self.leftButtonClass = leftButtonClass;
+        self.rightButtonClass = rightButtonClass;
         [self slk_commonInit];
     }
     return self;
@@ -162,7 +171,9 @@ - (SLKInputAccessoryView *)inputAccessoryView
 - (UIButton *)leftButton
 {
     if (!_leftButton) {
-        _leftButton = [UIButton buttonWithType:UIButtonTypeSystem];
+        Class buttonClass = _leftButtonClass ?: UIButton.class;
+        
+        _leftButton = [buttonClass buttonWithType:UIButtonTypeSystem];
         _leftButton.translatesAutoresizingMaskIntoConstraints = NO;
         _leftButton.titleLabel.font = [UIFont systemFontOfSize:15.0];
     }
@@ -172,7 +183,9 @@ - (UIButton *)leftButton
 - (UIButton *)rightButton
 {
     if (!_rightButton) {
-        _rightButton = [UIButton buttonWithType:UIButtonTypeSystem];
+        Class buttonClass = _rightButtonClass ?: UIButton.class;
+        
+        _leftButton = [buttonClass buttonWithType:UIButtonTypeSystem];
         _rightButton.translatesAutoresizingMaskIntoConstraints = NO;
         _rightButton.titleLabel.font = [UIFont boldSystemFontOfSize:15.0];
         _rightButton.enabled = NO;