db123/uptodownbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: quota pre-check used max format size, blocked before format selection

Browse Source 
Changed pre-check to only reject when used >= limit (already exhausted),
not when the max format size alone would exceed the limit. The actual
quota is applied at 50% progress using the selected format's size.
This commit is contained in:
db123
2026-06-25 22:08:05 +03:30
parent f8f9d99e69
commit f39e4e6fb0
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/handler/download.go
View File

@@ -113,21 +113,17 @@ 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 {
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
}
}
}
ss := &stepState{
URL: url,