From 190c3467ec78aaf30f535bf4554c2a4f525d8f18 Mon Sep 17 00:00:00 2001
From: lbarraga <lukas.barragantorres@ugent.be>
Date: Sun, 7 May 2023 14:03:46 +0200
Subject: [PATCH] #74 viewmodel of a feed

---
 .../sel/studeez/screens/home/FeedViewModel.kt | 35 ++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/be/ugent/sel/studeez/screens/home/FeedViewModel.kt b/app/src/main/java/be/ugent/sel/studeez/screens/home/FeedViewModel.kt
index 52c0be2..3b9fb0c 100644
--- a/app/src/main/java/be/ugent/sel/studeez/screens/home/FeedViewModel.kt
+++ b/app/src/main/java/be/ugent/sel/studeez/screens/home/FeedViewModel.kt
@@ -1,4 +1,37 @@
 package be.ugent.sel.studeez.screens.home
 
-class FeedViewModel {
+import androidx.lifecycle.viewModelScope
+import be.ugent.sel.studeez.data.SelectedTask
+import be.ugent.sel.studeez.data.local.models.FeedEntry
+import be.ugent.sel.studeez.domain.FeedDAO
+import be.ugent.sel.studeez.domain.LogService
+import be.ugent.sel.studeez.domain.TaskDAO
+import be.ugent.sel.studeez.navigation.StudeezDestinations
+import be.ugent.sel.studeez.screens.StudeezViewModel
+import dagger.hilt.android.lifecycle.HiltViewModel
+import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.launch
+import javax.inject.Inject
+
+@HiltViewModel
+class FeedViewModel @Inject constructor(
+    private val feedDAO: FeedDAO,
+    private val taskDAO: TaskDAO,
+    private val selectedTask: SelectedTask,
+    logService: LogService
+) : StudeezViewModel(logService) {
+
+    private val entries: Flow<List<FeedEntry>> = feedDAO.getFeedEntries()
+
+    fun getFeedEntries(): Flow<List<FeedEntry>> {
+        return entries
+    }
+
+    fun continueWithTask(open: (String) -> Unit, subjectId: String, taskId: String) {
+        viewModelScope.launch {
+            val task = taskDAO.getTask(subjectId, taskId)
+            selectedTask.set(task)
+            open(StudeezDestinations.TIMER_SELECTION_SCREEN)
+        }
+    }
 }
\ No newline at end of file