TabLayoutWithViewPager2.kt

// Extension function

fun TabLayout.setupWithViewPager2(
    viewPager: ViewPager2,
    callback: (TabLayout.Tab, Int) -> Unit
) {
    TabLayoutMediator(this, viewPager) { tab, position ->
        callback.invoke(tab, position)
    }.attach()
}


// Use, make sure to set an adapter to the view pager or you'll get an IllegalStateException. 
// view_pager is a ViewPager2 instance.
tab_layout.setupWithViewPager2(view_pager) { tab, position ->
    when (position) {
        0 -> tab.text = "TAB ONE"
        1 -> tab.text = "TAB TWO"
    }
}