03a122f0ce
* Implemented ActiveJob for RateLimitChecker Job * Changes as per the review * Further changes as per the review
43 lines
1.2 KiB
Ruby
43 lines
1.2 KiB
Ruby
class RateLimitChecker
|
|
attr_accessor :user
|
|
def initialize(user = nil)
|
|
@user = user
|
|
end
|
|
|
|
class UploadRateLimitReached < StandardError; end
|
|
|
|
def limit_by_situation(situation)
|
|
result = case situation
|
|
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
|
|
ping_admins if result == true
|
|
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 situation.
|
|
EmailMessage.where(to: address).
|
|
where("sent_at > ?", 2.minutes.ago).size > 5
|
|
end
|
|
|
|
def ping_admins
|
|
RateLimitCheckerJob.perform_later(user.id)
|
|
end
|
|
|
|
def ping_admins_without_delay
|
|
RateLimitCheckerJob.perform_now(user.id)
|
|
end
|
|
end
|