gistfile1.txt

import android.app.Application
import androidx.lifecycle.ViewModel
import androidx.lifecycle.ViewModelProvider
import caux.martin.go4lunch.MainApplication
import caux.martin.go4lunch.repositories.UserRepository
import caux.martin.go4lunch.repositories.places.PlacesRepository
import caux.martin.go4lunch.viewModel.LocationViewModel
import caux.martin.go4lunch.viewModel.WorkmatesViewModel

class ViewModelFactory private constructor(
    private val permissionChecker: PermissionChecker,
    private val application: Application
) : ViewModelProvider.NewInstanceFactory() {

    @Suppress("UNCHECKED_CAST")
    override fun <T : ViewModel> create(modelClass: Class<T>): T {
        return when (modelClass) {
            LocationViewModel::class.java -> LocationViewModel(permissionChecker, application)
            WorkmatesViewModel::class.java -> WorkmatesViewModel(UserRepository,PlacesRepository)
            else -> throw IllegalArgumentException("Unknown ViewModel class")
        } as T
    }

    companion object {
        val INSTANCE = ViewModelFactory(
            PermissionChecker(MainApplication.instance),
            MainApplication.instance
        )
    }
}