cloudroam
2025-04-11 cb99bd7dad1b305a434c5c6c99ca65e782eb0f34
app/src/main/java/com/example/firstapp/ui/dashboard/DashboardViewModel.kt
@@ -4,6 +4,7 @@
import androidx.lifecycle.AndroidViewModel
import androidx.lifecycle.asLiveData
import androidx.lifecycle.viewModelScope
import androidx.room.RewriteQueriesToDropUnusedColumns
import com.example.firstapp.AppDatabase
import com.example.firstapp.database.entity.Code
import com.example.firstapp.database.repository.CodeRepository
@@ -26,6 +27,7 @@
    fun getYearlyHeatmap(date: Long) = repository.getYearlyHeatmap(date).asLiveData()
    fun getWeeklyStats(date: Long, weekCount: Int = 6) = 
        repository.getWeeklyStats(date, weekCount).asLiveData()
    
@@ -39,4 +41,10 @@
    fun insert(code: Code) = viewModelScope.launch {
        repository.insert(code)
    }
    fun getCurrentWeekStats(date: Long) =
        repository.getCurrentWeekStats(date).asLiveData()
    fun getCurrentYearStats(date: Long) =
        repository.getCurrentYearStats(date).asLiveData()
}