savedstate_viewmodelfactory_assisted_dagger.kt

class ViewModelFactory @Inject constructor(
    private val viewModelMap: MutableMap<Class<out ViewModel>, ViewModelAssistedFactory<out ViewModel>>,
    owner: SavedStateRegistryOwner,
    defaultArgs: Bundle?
) : AbstractSavedStateVMFactory(owner, defaultArgs) {

    override fun <T : ViewModel?> create(
        key: String, 
        modelClass: Class<T>, 
        handle: SavedStateHandle): T {
        return viewModelMap[modelClass]?.create(handle) as? T 
            ?: throw IllegalStateException("Unknown ViewModel class")
    }
}