trying koin viewModel

Author: Pedrobam

app/src/main/java/com/arctouch/codechallenge/App.kt

@@ -3,6 +3,7 @@ package com.arctouch.codechallenge
 import android.app.Application
 import com.arctouch.codechallenge.api.TmdbApi
 import com.arctouch.codechallenge.data.Cache
+import com.arctouch.codechallenge.di.apiModule
 import com.arctouch.codechallenge.di.viewModelModule
 import io.reactivex.android.schedulers.AndroidSchedulers
 import io.reactivex.schedulers.Schedulers
@@ -25,7 +26,7 @@ class App : Application() {
         startKoin {
             androidLogger()
             androidContext(this@App)
-            modules(listOf(viewModelModule))
+            modules(listOf(apiModule, viewModelModule))
         }
         api = createWebService()
         getGenres()

app/src/main/java/com/arctouch/codechallenge/di/ApiModule.kt

@@ -9,6 +9,7 @@ import retrofit2.converter.moshi.MoshiConverterFactory
 
 val apiModule = module {
     single { createWebService() }
+    single<TmbRepository> { TmbRepositoryImpl(get()) }
 }
 
 fun createWebService(): TmdbApi {

app/src/main/java/com/arctouch/codechallenge/di/ViewModelModule.kt

@@ -5,6 +5,6 @@ import com.arctouch.codechallenge.ui.home.HomeViewModel
 import org.koin.dsl.module
 
 val viewModelModule = module {
-    single { HomeViewModel() }
+    single { HomeViewModel(get()) }
     single { DetailsViewModel() }
 }

app/src/main/java/com/arctouch/codechallenge/ui/home/HomeViewModel.kt

@@ -6,13 +6,15 @@ import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
 import com.arctouch.codechallenge.App
 import com.arctouch.codechallenge.data.Cache
+import com.arctouch.codechallenge.di.TmbRepository
 import com.arctouch.codechallenge.model.Movie
 import com.arctouch.codechallenge.model.UpcomingMoviesResponse
 import kotlinx.coroutines.launch
+import org.koin.core.KoinComponent
+import org.koin.core.inject
 
-class HomeViewModel : ViewModel() {
+class HomeViewModel(private val repository: TmbRepository) : ViewModel(), KoinComponent {
 
-    private val api = App.api
     private val _upcomingMovies = MutableLiveData<MutableList<Movie>>()
     val upcomingMovies: LiveData<MutableList<Movie>> = _upcomingMovies
     private val _searchLiveData = MutableLiveData<MutableList<Movie>>()
@@ -28,7 +30,7 @@ class HomeViewModel : ViewModel() {
     fun getUpComingMovies() {
         page = page.plus(1)
         viewModelScope.launch {
-            val upcomingMoviesResponse = api.upcomingMovies(page = page)
+            val upcomingMoviesResponse = repository.upcomingMovies(page = page)
             val moviesWithGenres = addGenres(upcomingMoviesResponse)
             val list = _upcomingMovies.value
             list?.addAll(moviesWithGenres)
@@ -38,7 +40,7 @@ class HomeViewModel : ViewModel() {
 
     fun searchMovies(query: String) {
         viewModelScope.launch {
-            val upcomingMoviesResponse = api.getMoviesByName(query)
+            val upcomingMoviesResponse = repository.getMoviesByName(query)
             val moviesWithGenres = addGenres(upcomingMoviesResponse)
             _searchLiveData.postValue(moviesWithGenres.toMutableList())
         }