diff --git a/internal/handler/download.go b/internal/handler/download.go index f07110c..7da6e1e 100644 --- a/internal/handler/download.go +++ b/internal/handler/download.go @@ -113,19 +113,15 @@ func (h *Handler) handleURLInput(ctx context.Context, chatID int64, text string, return } + // Reject if user has already exhausted quota (used >= limit) with a minimum threshold. q, err := h.Repo.GetOrCreateQuotas(userID) if err == nil { q = h.resetQuotasIfNeeded(q) - estimatedMB := info.EstimatedSize / (1024 * 1024) - if estimatedMB > 0 { - if q.DailyUsedMB+estimatedMB > q.DailyLimitMB || - q.WeeklyUsedMB+estimatedMB > q.WeeklyLimitMB || - q.MonthlyUsedMB+estimatedMB > q.MonthlyLimitMB { - h.sendText(ctx, chatID, fmt.Sprintf( - "Quota exceeded. Daily: %d/%d MB, Weekly: %d/%d MB, Monthly: %d/%d MB.", - q.DailyUsedMB, q.DailyLimitMB, q.WeeklyUsedMB, q.WeeklyLimitMB, q.MonthlyUsedMB, q.MonthlyLimitMB)) - return - } + if q.DailyUsedMB >= q.DailyLimitMB || q.WeeklyUsedMB >= q.WeeklyLimitMB || q.MonthlyUsedMB >= q.MonthlyLimitMB { + h.sendText(ctx, chatID, fmt.Sprintf( + "Quota exceeded. Daily: %d/%d MB, Weekly: %d/%d MB, Monthly: %d/%d MB.", + q.DailyUsedMB, q.DailyLimitMB, q.WeeklyUsedMB, q.WeeklyLimitMB, q.MonthlyUsedMB, q.MonthlyLimitMB)) + return } }