Fix toolbar color button not updated when cursor change

Author: starshipcoder

aztec/src/main/kotlin/org/wordpress/aztec/toolbar/AztecToolbar.kt

@@ -393,6 +393,7 @@ class AztecToolbar : FrameLayout, IAztecToolbar, OnMenuItemClickListener {
 
                 editor.setSpanColor(color)
                 colorPickerView?.setColor(color)
+                updateToolbarColorButton(color)
             }
         })
 
@@ -407,16 +408,7 @@ class AztecToolbar : FrameLayout, IAztecToolbar, OnMenuItemClickListener {
         this.colorPickerView = colorPickerView
         colorPickerView.setColorPickerListener(object : ColorPickerListener {
             override fun onPickColor(color: Int) {
-                if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
-                    findViewById<View>(ToolbarAction.TEXTCOLOR.buttonId).apply {
-                        when (this) {
-                            is ImageButton -> {
-                                this.imageTintList = ColorStateList.valueOf(color)
-                            }
-                            else -> this.setBackgroundColor(color)
-                        }
-                    }
-                }
+                updateToolbarColorButton(color)
 
                 editor?.let {
 
@@ -458,6 +450,18 @@ class AztecToolbar : FrameLayout, IAztecToolbar, OnMenuItemClickListener {
         })
     }
 
+    private fun updateToolbarColorButton(color: Int?) {
+        val color2 = color ?: Color.BLACK
+        findViewById<View>(ToolbarAction.TEXTCOLOR.buttonId).apply {
+            when (this) {
+                is ImageButton -> {
+                    this.imageTintList = ColorStateList.valueOf(color2)
+                }
+                else -> this.setBackgroundColor(color2)
+            }
+        }
+    }
+
     private fun initView(attrs: AttributeSet?) {
         val styles = context.obtainStyledAttributes(attrs, R.styleable.AztecToolbar, 0, R.style.AztecToolbarStyle)
         isAdvanced = styles.getBoolean(R.styleable.AztecToolbar_advanced, false)