diff --git a/app/src/main/java/com/example/shiftalarm/ShiftRepository.kt b/app/src/main/java/com/example/shiftalarm/ShiftRepository.kt index e401d88..b33abc1 100644 --- a/app/src/main/java/com/example/shiftalarm/ShiftRepository.kt +++ b/app/src/main/java/com/example/shiftalarm/ShiftRepository.kt @@ -93,6 +93,21 @@ class ShiftRepository(private val context: Context) { } suspend fun updateRemainingAnnualLeave() { + val annualLeave = dao.getAnnualLeave() + val usedDays = calculateUsedAnnualLeave() + + if (annualLeave != null) { + val remainingDays = annualLeave.totalDays - usedDays + dao.insertAnnualLeave(annualLeave.copy(remainingDays = remainingDays)) + } else { + // AnnualLeave가 없으면 기본값 15일로 생성 + dao.insertAnnualLeave(AnnualLeave( + id = 1, + totalDays = 15f, + remainingDays = 15f - usedDays + )) + } + } val annualLeave = dao.getAnnualLeave() annualLeave?.let { val usedDays = calculateUsedAnnualLeave()