76 lines
2.6 KiB
Kotlin
76 lines
2.6 KiB
Kotlin
// from https://proandroiddev.com/viewmodel-with-dagger2-architecture-components-2e06f06c9455
|
|
|
|
package at.connyduck.pixelcat.dagger
|
|
|
|
import androidx.lifecycle.ViewModel
|
|
import androidx.lifecycle.ViewModelProvider
|
|
import at.connyduck.pixelcat.components.compose.ComposeViewModel
|
|
import at.connyduck.pixelcat.components.login.LoginViewModel
|
|
import at.connyduck.pixelcat.components.main.MainViewModel
|
|
import at.connyduck.pixelcat.components.notifications.NotificationsViewModel
|
|
import at.connyduck.pixelcat.components.profile.ProfileViewModel
|
|
import at.connyduck.pixelcat.components.search.SearchViewModel
|
|
import at.connyduck.pixelcat.components.timeline.TimelineViewModel
|
|
import dagger.Binds
|
|
import dagger.MapKey
|
|
import dagger.Module
|
|
import dagger.multibindings.IntoMap
|
|
import javax.inject.Inject
|
|
import javax.inject.Provider
|
|
import javax.inject.Singleton
|
|
import kotlin.reflect.KClass
|
|
|
|
@Singleton
|
|
class ViewModelFactory @Inject constructor(private val viewModels: MutableMap<Class<out ViewModel>, Provider<ViewModel>>) : ViewModelProvider.Factory {
|
|
@Suppress("UNCHECKED_CAST")
|
|
override fun <T : ViewModel> create(modelClass: Class<T>): T = viewModels[modelClass]?.get() as T
|
|
}
|
|
|
|
@Target(AnnotationTarget.FUNCTION, AnnotationTarget.PROPERTY_GETTER, AnnotationTarget.PROPERTY_SETTER)
|
|
@Retention(AnnotationRetention.RUNTIME)
|
|
@MapKey
|
|
internal annotation class ViewModelKey(val value: KClass<out ViewModel>)
|
|
|
|
@Module
|
|
abstract class ViewModelModule {
|
|
|
|
@Binds
|
|
internal abstract fun bindViewModelFactory(factory: ViewModelFactory): ViewModelProvider.Factory
|
|
|
|
@Binds
|
|
@IntoMap
|
|
@ViewModelKey(LoginViewModel::class)
|
|
internal abstract fun loginViewModel(viewModel: LoginViewModel): ViewModel
|
|
|
|
@Binds
|
|
@IntoMap
|
|
@ViewModelKey(MainViewModel::class)
|
|
internal abstract fun mainViewModel(viewModel: MainViewModel): ViewModel
|
|
|
|
@Binds
|
|
@IntoMap
|
|
@ViewModelKey(TimelineViewModel::class)
|
|
internal abstract fun timelineViewModel(viewModel: TimelineViewModel): ViewModel
|
|
|
|
@Binds
|
|
@IntoMap
|
|
@ViewModelKey(SearchViewModel::class)
|
|
internal abstract fun searchViewModel(viewModel: SearchViewModel): ViewModel
|
|
|
|
@Binds
|
|
@IntoMap
|
|
@ViewModelKey(NotificationsViewModel::class)
|
|
internal abstract fun notificationsViewModel(viewModel: NotificationsViewModel): ViewModel
|
|
|
|
@Binds
|
|
@IntoMap
|
|
@ViewModelKey(ProfileViewModel::class)
|
|
internal abstract fun profileViewModel(viewModel: ProfileViewModel): ViewModel
|
|
|
|
@Binds
|
|
@IntoMap
|
|
@ViewModelKey(ComposeViewModel::class)
|
|
internal abstract fun composeViewModel(viewModel: ComposeViewModel): ViewModel
|
|
// Add more ViewModels here
|
|
}
|