Add TabLayout#updateCustomTab(...)

Author: DylanCaiCoding

viewbinding-ktx/src/main/java/com/dylanc/viewbinding/TabLayout.kt

@@ -23,12 +23,16 @@ import androidx.viewbinding.ViewBinding
 import com.google.android.material.tabs.TabLayout
 
 inline fun <reified VB : ViewBinding> TabLayout.Tab.setCustomView(block: VB.() -> Unit) {
+  requireNotNull(parent) { "Tab not attached to a TabLayout" }
   inflateBinding<VB>(LayoutInflater.from(parent!!.context)).apply(block).let { binding ->
     customView = binding.root
     customView?.tag = binding
   }
 }
 
+inline fun <reified VB : ViewBinding> TabLayout.updateCustomTab(index: Int, block: VB.() -> Unit) =
+  getTabAt(index)?.customView?.getBinding<VB>()?.also(block)
+
 inline fun <reified VB : ViewBinding> TabLayout.doOnCustomTabSelected(
   crossinline onTabUnselected: VB.(TabLayout.Tab) -> Unit = {},
   crossinline onTabReselected: VB.(TabLayout.Tab) -> Unit = {},

viewbinding-nonreflection-ktx/src/main/java/com/dylanc/viewbinding/nonreflection/TabLayout.kt

@@ -24,9 +24,13 @@ import androidx.viewbinding.ViewBinding
 import com.google.android.material.tabs.TabLayout
 
 fun <VB : ViewBinding> TabLayout.Tab.setCustomView(inflate: (LayoutInflater) -> VB, block: VB.() -> Unit) {
+  requireNotNull(parent) { "Tab not attached to a TabLayout" }
   customView = inflate(LayoutInflater.from(parent!!.context)).apply(block).root
 }
 
+fun <VB : ViewBinding> TabLayout.updateCustomTab(bind: (View) -> VB, index: Int, block: VB.() -> Unit) =
+  getTabAt(index)?.customView?.getBinding(bind)?.also(block)
+
 fun <VB : ViewBinding> TabLayout.doOnCustomTabSelected(
   bind: (View) -> VB,
   onTabUnselected: (VB.(TabLayout.Tab) -> Unit)? = null,