Workaround for fixing the issue where NavigationView does not correctly control `IsChecked` of the submenus

Helper.kt

class Helper {
    companion object {
        fun setupWithNavController(navigationView: NavigationView, navController: NavController) {
            navigationView.setNavigationItemSelectedListener { item ->
                val handled = NavigationUI.onNavDestinationSelected(item, navController)
                if (handled) {
                    val parent = navigationView.parent
                    if (parent is DrawerLayout) {
                        parent.closeDrawer(navigationView)
                    } else {
                        findBottomSheetBehavior(navigationView)?.setState(BottomSheetBehavior.STATE_HIDDEN)
                    }
                }
                handled
            }
            val weakReference = WeakReference(navigationView)
            navController.addOnDestinationChangedListener(
                object : NavController.OnDestinationChangedListener {
                    override fun onDestinationChanged(
                        controller: NavController,
                        destination: NavDestination, arguments: Bundle?
                    ) {
                        val view = weakReference.get()
                        if (view == null) {
                            navController.removeOnDestinationChangedListener(this)
                            return
                        }
                        checkDestination(destination, view.menu)
                    }
                })
        }

        private fun checkDestination(destination: NavDestination, menu: Menu) {
            menu.forEach { item ->
                if(item.hasSubMenu())
                    checkDestination(destination, item.subMenu)
                else
                    item.isChecked = matchDestination(destination, item.itemId)
            }
        }

        private fun matchDestination(destination: NavDestination, destId: Int): Boolean {
            var currentDestination: NavDestination? = destination
            while (currentDestination!!.id != destId && currentDestination.parent != null) {
                currentDestination = currentDestination.parent
            }
            return currentDestination.id == destId
        }

        private fun findBottomSheetBehavior(view: View): BottomSheetBehavior<*>? {
            val params = view.layoutParams
            if (params !is CoordinatorLayout.LayoutParams) {
                val parent = view.parent
                return if (parent is View) {
                    findBottomSheetBehavior(parent as View)
                } else null
            }
            val behavior = params.behavior
            return if (behavior !is BottomSheetBehavior<*>) {
                // We hit a CoordinatorLayout, but the View doesn't have the BottomSheetBehavior
                null
            } else behavior
        }
    }
}

MainActivity.kt

class MainActivity : AppCompatActivity() {

    private lateinit var appBarConfiguration: AppBarConfiguration

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_main)
        val toolbar: Toolbar = findViewById(R.id.toolbar)
        setSupportActionBar(toolbar)

        val drawerLayout: DrawerLayout = findViewById(R.id.drawer_layout)
        val navView: NavigationView = findViewById(R.id.nav_view)
        val navController = findNavController(R.id.nav_host_fragment)
        // Passing each menu ID as a set of Ids because each
        // menu should be considered as top level destinations.
        appBarConfiguration = AppBarConfiguration(
            setOf(
                R.id.nav_bookmark, R.id.nav_hold, R.id.nav_ranking, R.id.nav_search, R.id.nav_keyword
            ), drawerLayout
        )
        setupActionBarWithNavController(navController, appBarConfiguration)
        Helper.setupWithNavController(navView, navController)
    }

    override fun onSupportNavigateUp(): Boolean {
        val navController = findNavController(R.id.nav_host_fragment)
        return navController.navigateUp(appBarConfiguration) || super.onSupportNavigateUp()
    }
}

This code is almost the same as NavigationUI.setupWithNavController(NavigationView, NavController) except I added checkDestination.