Otus-Android#3 viewModel factory

Author: AkashevPavel

app/src/main/java/otus/homework/coroutines/CatsViewModel.kt

@@ -0,0 +1,23 @@
+package otus.homework.coroutines
+
+import androidx.lifecycle.ViewModel
+import androidx.lifecycle.ViewModelProvider
+import androidx.lifecycle.viewmodel.initializer
+import androidx.lifecycle.viewmodel.viewModelFactory
+
+class CatsViewModel(
+    private val catsService: CatsService,
+    private val imageService: ImageService
+) : ViewModel() {
+
+    companion object {
+        fun provideFactory(
+            catsService: CatsService,
+            imageService: ImageService
+        ): ViewModelProvider.Factory = viewModelFactory {
+            initializer {
+                CatsViewModel(catsService, imageService)
+            }
+        }
+    }
+}

app/src/main/java/otus/homework/coroutines/MainActivity.kt

@@ -2,13 +2,21 @@ package otus.homework.coroutines
 
 import android.os.Bundle
 import androidx.appcompat.app.AppCompatActivity
+import androidx.lifecycle.ViewModelProvider
 
 class MainActivity : AppCompatActivity() {
 
     lateinit var catsPresenter: CatsPresenter
 
     private val diContainer = DiContainer()
 
+    private val viewModel by lazy {
+        ViewModelProvider(
+            this,
+            CatsViewModel.provideFactory(diContainer.catsService, diContainer.imageService)
+        )[CatsViewModel::class.java]
+    }
+
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)