Clear state for fragment when using bottom navigation

What I did was just use the same androidx.navigation.ui.NavigationUI.setupWithNavController logic but change the saveState and other logic specific to my use case. You could apply this when navigating to one specific fragment.

        this.findViewById<BottomNavigationView>(R.id.bottom_navigation).apply {

        setOnItemSelectedListener { item ->
            val builder = NavOptions.Builder().setLaunchSingleTop(true)
            val destinationId = item.itemId
            item.isChecked = true

            if (
                navController.currentDestination!!.parent!!.findNode(item.itemId)
                        is ActivityNavigator.Destination
            ) {
                builder.setEnterAnim(R.anim.nav_default_enter_anim)
                    .setExitAnim(R.anim.nav_default_exit_anim)
                    .setPopEnterAnim(R.anim.nav_default_pop_enter_anim)
                    .setPopExitAnim(R.anim.nav_default_pop_exit_anim)
            } else {
                builder.setEnterAnim(R.animator.nav_default_enter_anim)
                    .setExitAnim(R.animator.nav_default_exit_anim)
                    .setPopEnterAnim(R.animator.nav_default_pop_enter_anim)
                    .setPopExitAnim(R.animator.nav_default_pop_exit_anim)
            }

            if (item.order and Menu.CATEGORY_SECONDARY == 0) {
                builder.setPopUpTo(
                    navController.graph.findStartDestination().id,
                    inclusive = false,
                    saveState = false
                )
            }

            val options = builder.build()
            return@setOnItemSelectedListener try {
                navController.navigate(destinationId, null, options)
                // Return true only if the destination we've navigated to matches the MenuItem
                (navController.currentDestination?.id ?: false) == destinationId
            } catch (e: IllegalArgumentException) {
                false
            }
        }

        // Do nothing on reselect
        setOnItemReselectedListener {}

        val weakReference = WeakReference(this)
        navController.addOnDestinationChangedListener(
            object : NavController.OnDestinationChangedListener {
                override fun onDestinationChanged(
                    controller: NavController,
                    destination: NavDestination,
                    arguments: Bundle?
                ) {

                    // Hide BottomNavigationView from top level fragments
                    if (topLevelDestinations.any { it == destination.id }) {
                        this@apply.visibility = View.VISIBLE
                    } else this@apply.visibility = View.GONE

                    // Highlight item in BottomNavigationView
                    val view = weakReference.get()
                    if (view == null) {
                        navController.removeOnDestinationChangedListener(this)
                        return
                    }
                    view.menu.forEach { item ->
                        if (destination.id == item.itemId) {
                            item.isChecked = true
                        }
                    }
                }
            })
    }