mirror of
https://github.com/discourse/discourse.git
synced 2024-12-19 14:24:05 +08:00
41 lines
1.1 KiB
Ruby
41 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
|