Updated right button to display image with proper constraint

zoyi · Nov 23, 2017 · 139da40 · 139da40
1 parent 41a017a
commit 139da40
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/CHSlackTextViewController.podspec b/CHSlackTextViewController.podspec
@@ -1,4 +1,4 @@
-@version = "1.10.12"
+@version = "1.10.13"
 
 Pod::Spec.new do |s|
   s.name         		= "CHSlackTextViewController"

diff --git a/Source/SLKTextInputbar.m b/Source/SLKTextInputbar.m
@@ -105,6 +105,7 @@ - (void)slk_commonInit
 
     [self slk_registerTo:self.layer forSelector:@selector(position)];
     [self slk_registerTo:self.leftButton.imageView forSelector:@selector(image)];
+    [self slk_registerTo:self.rightButton.imageView forSelector:@selector(image)];
     [self slk_registerTo:self.rightButton.titleLabel forSelector:@selector(font)];
 
     self.barState = SLKInputBarStateNormal;
@@ -780,6 +781,15 @@ - (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(N
             [self slk_updateConstraintConstants];
         }
     }
+    else if ([object isEqual:self.rightButton.imageView] && [keyPath isEqualToString:NSStringFromSelector(@selector(image))]) {
+
+      UIImage *newImage = change[NSKeyValueChangeNewKey];
+      UIImage *oldImage = change[NSKeyValueChangeOldKey];
+
+      if (![newImage isEqual:oldImage]) {
+        [self slk_updateConstraintConstants];
+      }
+    }
     else if ([object isEqual:self.rightButton.titleLabel] && [keyPath isEqualToString:NSStringFromSelector(@selector(font))]) {
 
         [self slk_updateConstraintConstants];
@@ -817,6 +827,7 @@ - (void)dealloc
 
     [self slk_unregisterFrom:self.layer forSelector:@selector(position)];
     [self slk_unregisterFrom:self.leftButton.imageView forSelector:@selector(image)];
+    [self slk_unregisterFrom:self.rightButton.imageView forSelector:@selector(image)];
     [self slk_unregisterFrom:self.rightButton.titleLabel forSelector:@selector(font)];
 }