SavedStateViewModelFactory.kt

interface SavedStateViewModelFactory {
    fun <T : ViewModel?> create(key: String, modelClass: Class<T>, handle: SavedStateHandle): T
}

private fun SavedStateViewModelFactory.asViewModelProviderFactory(
    owner: SavedStateRegistryOwner
): ViewModelProvider.Factory {
    return object : AbstractSavedStateViewModelFactory(owner, owner.arguments) {
        override fun <T : ViewModel?> create(key: String, modelClass: Class<T>, handle: SavedStateHandle): T {
            return [email protected](key, modelClass, handle)
        }
    }
}

private val SavedStateRegistryOwner.arguments: Bundle
    get() = when (this) {
        is Activity -> intent?.extras
        is Fragment -> arguments
        else -> Bundle()
    }