discourse/lib/rate_limiter/limit_exceeded.rb

43 lines
1.1 KiB
Ruby

# frozen_string_literal: true
class RateLimiter
def self.time_left(available_in)
if available_in <= 3
I18n.t("rate_limiter.short_time")
elsif available_in < 1.minute.to_i
I18n.t("rate_limiter.seconds", count: available_in)
elsif available_in < 1.hour.to_i
I18n.t("rate_limiter.minutes", count: (available_in / 1.minute.to_i))
else
I18n.t("rate_limiter.hours", count: (available_in / 1.hour.to_i))
end
end
# A rate limit has been exceeded.
class LimitExceeded < StandardError
attr_reader :type, :available_in, :error_code
def initialize(available_in, type = nil, error_code = nil)
@available_in = available_in
@type = type
@error_code = error_code
end
def time_left
@time_left ||= RateLimiter.time_left(@available_in)
end
def description
if @type.present?
type_key = @type.tr("-", "_")
msg = I18n.t("rate_limiter.by_type.#{type_key}", time_left: time_left, default: "")
return msg if msg.present?
end
I18n.t("rate_limiter.too_many_requests", time_left: time_left)
end
end
end