Add mehtod to interact with dao class

yahyalmh · Sep 8, 2021 · 6dcb538 · 6dcb538
1 parent 1380b83
commit 6dcb538
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/app/src/main/java/com/yaya/apod/data/repo/ApodRepository.kt b/app/src/main/java/com/yaya/apod/data/repo/ApodRepository.kt
@@ -1,6 +1,7 @@
 package com.yaya.apod.data.repo
 
 import androidx.lifecycle.LiveData
+import androidx.lifecycle.asLiveData
 import com.yaya.apod.AppExecutors
 import com.yaya.apod.api.ApiResponse
 import com.yaya.apod.api.ApodApi
@@ -9,6 +10,7 @@ import com.yaya.apod.data.db.dao.ApodDao
 import com.yaya.apod.data.model.Apod
 import com.yaya.apod.di.NetworkModule
 import com.yaya.apod.util.DateUtil
+import kotlinx.coroutines.flow.Flow
 import javax.inject.Inject
 import javax.inject.Singleton
 
@@ -75,4 +77,12 @@ class ApodRepository @Inject constructor(
     fun getFavorite(): LiveData<MutableList<Apod>> {
         return apodDao.getFavorite()
     }
+
+    fun getApod(apodId: String): LiveData<Apod> {
+        return apodDao.getApod(apodId.toInt()).asLiveData()
+    }
+
+    fun isFavorite(apodId: String): Flow<Boolean> {
+        return apodDao.isFavorite(apodId.toInt())
+    }
 }