From 880236e1ce3981c8d1013eedb231a2b455220e55 Mon Sep 17 00:00:00 2001
From: lbarraga <lukas.barragantorres@ugent.be>
Date: Sun, 16 Apr 2023 23:33:18 +0200
Subject: [PATCH] #40 viewmodel for overview screen

---
 .../timer_overview/TimerOverviewViewModel.kt  | 32 ++++++++++++++++++-
 1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/be/ugent/sel/studeez/screens/timer_overview/TimerOverviewViewModel.kt b/app/src/main/java/be/ugent/sel/studeez/screens/timer_overview/TimerOverviewViewModel.kt
index 62b30ab..c2be1e9 100644
--- a/app/src/main/java/be/ugent/sel/studeez/screens/timer_overview/TimerOverviewViewModel.kt
+++ b/app/src/main/java/be/ugent/sel/studeez/screens/timer_overview/TimerOverviewViewModel.kt
@@ -1,4 +1,34 @@
 package be.ugent.sel.studeez.screens.timer_overview
 
-class TimerOverviewViewModel {
+import be.ugent.sel.studeez.data.local.models.timer_info.TimerInfo
+import be.ugent.sel.studeez.domain.ConfigurationService
+import be.ugent.sel.studeez.domain.LogService
+import be.ugent.sel.studeez.domain.TimerDAO
+import be.ugent.sel.studeez.screens.StudeezViewModel
+import dagger.hilt.android.lifecycle.HiltViewModel
+import kotlinx.coroutines.flow.Flow
+import javax.inject.Inject
+
+@HiltViewModel
+class TimerOverviewViewModel @Inject constructor(
+    private val configurationService: ConfigurationService,
+    private val timerDAO: TimerDAO,
+    logService: LogService
+) : StudeezViewModel(logService) {
+
+    fun getUserTimers() : Flow<List<TimerInfo>> {
+        return timerDAO.getUserTimers()
+    }
+
+    fun getDefaultTimers(): List<TimerInfo> {
+        return configurationService.getDefaultTimers()
+    }
+
+    fun update(timerInfo: TimerInfo) = timerDAO.updateTimer(timerInfo)
+
+    fun delete(timerInfo: TimerInfo) =timerDAO.deleteTimer(timerInfo)
+
+    fun save(timerInfo: TimerInfo) = timerDAO.saveTimer(timerInfo)
+
+
 }
\ No newline at end of file