2016-03-20 00:57:30 +11:00
|
|
|
class Rack::Attack
|
2016-10-23 04:38:47 +11:00
|
|
|
# Rate limits for the API
|
|
|
|
throttle('api', limit: 150, period: 5.minutes) do |req|
|
|
|
|
req.ip if req.path.match(/\A\/api\//)
|
2016-09-24 21:53:54 +10:00
|
|
|
end
|
|
|
|
|
2016-10-23 04:38:47 +11:00
|
|
|
self.throttled_response = lambda do |env|
|
|
|
|
now = Time.now.utc
|
|
|
|
match_data = env['rack.attack.match_data']
|
|
|
|
|
|
|
|
headers = {
|
|
|
|
'X-RateLimit-Limit' => match_data[:limit].to_s,
|
|
|
|
'X-RateLimit-Remaining' => '0',
|
2016-11-26 04:20:47 +11:00
|
|
|
'X-RateLimit-Reset' => (now + (match_data[:period] - now.to_i % match_data[:period])).iso8601(6)
|
2016-10-23 04:38:47 +11:00
|
|
|
}
|
|
|
|
|
|
|
|
[429, headers, [{ error: 'Throttled' }.to_json]]
|
2016-03-26 00:12:24 +11:00
|
|
|
end
|
2016-03-20 00:57:30 +11:00
|
|
|
end
|