findViewTreeViewModelStoreOwner findViewTreeLifecycleOwner findViewTreeSavedStateRegistryOwner

findViewTreeViewModelStoreOwner.kt

private inline fun <reified VM : ViewModel> View.underViewMode(
    savedStateDefaultArgs: Bundle? = null,
    crossinline viewModelCreator: (
        savedStateHandle: SavedStateHandle,
        outerLifecycle: Lifecycle,
        outerLifecycleScope: CoroutineScope
    ) -> VM,
    block: VM.(
        outerLifecycle: Lifecycle,
        outerLifecycleScope: CoroutineScope
    ) -> Unit
) {
    findViewTreeViewModelStoreOwner()?.let { viewModelStoreOwner ->
        findViewTreeLifecycleOwner()?.let { lifecycleOwner ->
            findViewTreeSavedStateRegistryOwner()?.let { savedStateRegistryOwner ->
                val viewModel: VM = ViewModelProvider(
                    viewModelStoreOwner,
                    object :
                        AbstractSavedStateViewModelFactory(
                            savedStateRegistryOwner,
                            savedStateDefaultArgs ?: bundleOf()
                        ) {
                        @Suppress("UNCHECKED_CAST")
                        override fun <T : ViewModel> create(
                            key: String,
                            modelClass: Class<T>,
                            handle: SavedStateHandle
                        ): T {
                            return viewModelCreator(
                                handle,
                                lifecycleOwner.lifecycle,
                                lifecycleOwner.lifecycleScope
                            ) as T
                        }
                    }
                ).get()

                viewModel.block(
                    lifecycleOwner.lifecycle,
                    lifecycleOwner.lifecycleScope
                )
            }
        }
    }
}