fix 123

cloudroam

2025-03-24 f25e015f2741649a0e429bacbb19e1e7b177067b

 app/src/main/java/com/example/firstapp/ui/dashboard/DashboardViewModel.kt

@@ -1,13 +1,48 @@
package com.example.firstapp.ui.dashboard

import androidx.lifecycle.LiveData
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.ViewModel
import android.app.Application
import androidx.lifecycle.AndroidViewModel
import androidx.lifecycle.asLiveData
import androidx.lifecycle.viewModelScope
import com.example.firstapp.AppDatabase
import com.example.firstapp.database.entity.Code
import com.example.firstapp.database.repository.CodeRepository
import kotlinx.coroutines.launch

class DashboardViewModel : ViewModel() {
class DashboardViewModel(application: Application) : AndroidViewModel(application) {

    private val _text = MutableLiveData<String>().apply {
        value = "This is dashboard Fragment"
    private val repository: CodeRepository

    init {
        val codeDao = AppDatabase.getInstance(application).codeDao()
        repository = CodeRepository(codeDao)
    }
    val text: LiveData<String> = _text

    fun getPackages(date: Long, dateType: String) =
        repository.getPackages(date, dateType).asLiveData()

    fun getCourierStats(date: Long, dateType: String) = 
        repository.getCourierStats(date, dateType).asLiveData()

    fun getYearlyHeatmap(date: Long) = repository.getYearlyHeatmap(date).asLiveData()

    fun getWeeklyStats(date: Long, weekCount: Int = 6) = 
        repository.getWeeklyStats(date, weekCount).asLiveData()
    
    fun getMonthlyStats(date: Long) = repository.getMonthlyStats(date).asLiveData()
    
    fun getYearlyStats(date: Long) = repository.getYearlyStats(date).asLiveData()

    fun getYearMonthlyStats(date: Long) = 
        repository.getYearMonthlyStats(date).asLiveData()

    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()
}