viewModelDagger.kt

@Suppress("UNCHECKED_CAST")
class ViewModelFactory @Inject constructor(
    private val creators: Map<Class<out ViewModel>, Provider<ViewModel>>
) : ViewModelProvider.Factory {

    override fun <T : ViewModel> create(modelClass: Class<T>): T {
        val creator = creators[modelClass]
            ?: creators.asIterable().find { (key, _) -> modelClass.isAssignableFrom(key) }?.value
            ?: throw IllegalArgumentException("Unknown ViewModel class $modelClass")

        return try {
            creator.get() as T
        } catch (cce: ClassCastException) {
            throw cce
        }
    }
}

@Module
object ViewModelFactoryModule {

    @Provides
    fun provideViewModelFactory(
        viewModelMap: Map<Class<out ViewModel>, @JvmSuppressWildcards Provider<ViewModel>>
    ): ViewModelProvider.Factory = ViewModelFactory(creators = viewModelMap)
}

@Module
abstract class ViewModelModule {

    @Binds
    @IntoMap
    @ViewModelKey(OurViewModel::class)
    abstract fun bindsOurViewModel(ourViewModel: OurViewModel): ViewModel
}