43 lines
1.2 KiB
Ruby
43 lines
1.2 KiB
Ruby
class RateLimitChecker
|
|
attr_reader :user, :action
|
|
|
|
def initialize(user = nil)
|
|
@user = user
|
|
end
|
|
|
|
class UploadRateLimitReached < StandardError; end
|
|
|
|
def limit_by_action(action)
|
|
result = case action
|
|
when "comment_creation"
|
|
user.comments.where("created_at > ?", 30.seconds.ago).size > 9
|
|
when "published_article_creation"
|
|
user.articles.published.where("created_at > ?", 30.seconds.ago).size > 9
|
|
when "image_upload"
|
|
Rails.cache.read("#{user.id}_image_upload").to_i > 9
|
|
else
|
|
false
|
|
end
|
|
if result
|
|
@action = action
|
|
ping_admins
|
|
end
|
|
result
|
|
end
|
|
|
|
def track_image_uploads
|
|
count = Rails.cache.read("#{@user.id}_image_upload").to_i
|
|
count += 1
|
|
Rails.cache.write("#{@user.id}_image_upload", count, expires_in: 30.seconds)
|
|
end
|
|
|
|
def limit_by_email_recipient_address(address)
|
|
# This is related to the recipient, not the "user" initiator, like in action.
|
|
EmailMessage.where(to: address).
|
|
where("sent_at > ?", 2.minutes.ago).size > 5
|
|
end
|
|
|
|
def ping_admins
|
|
RateLimitCheckerJob.perform_later(user.id, action)
|
|
end
|
|
end
|