gitlab-org--gitlab-foss/app/models/concerns/spammable.rb

85 lines
2 KiB
Ruby
Raw Normal View History

module Spammable
extend ActiveSupport::Concern
module ClassMethods
def attr_spammable(attr, options = {})
spammable_attrs << [attr.to_s, options]
end
end
included do
has_one :user_agent_detail, as: :subject, dependent: :destroy
2016-08-09 13:43:47 -04:00
attr_accessor :spam
attr_accessor :spam_log
2016-08-09 13:43:47 -04:00
2017-02-14 14:07:11 -05:00
after_validation :check_for_spam, on: [:create, :update]
cattr_accessor :spammable_attrs, instance_accessor: false do
[]
end
2016-08-09 13:43:47 -04:00
delegate :ip_address, :user_agent, to: :user_agent_detail, allow_nil: true
end
def submittable_as_spam_by?(current_user)
current_user && current_user.admin? && submittable_as_spam?
end
2016-08-09 13:43:47 -04:00
def submittable_as_spam?
if user_agent_detail
user_agent_detail.submittable? && current_application_settings.akismet_enabled
else
false
end
end
def spam?
@spam
end
2016-08-09 13:43:47 -04:00
def check_for_spam
error_msg = if Gitlab::Recaptcha.enabled?
"Your #{spammable_entity_type} has been recognized as spam. "\
"You can still submit it by solving Captcha."
else
"Your #{spammable_entity_type} has been recognized as spam and has been discarded."
end
self.errors.add(:base, error_msg) if spam?
end
def spammable_entity_type
self.class.name.underscore
end
def spam_title
2016-08-09 13:43:47 -04:00
attr = self.class.spammable_attrs.find do |_, options|
options.fetch(:spam_title, false)
end
2016-08-09 13:43:47 -04:00
public_send(attr.first) if attr && respond_to?(attr.first.to_sym)
end
def spam_description
2016-08-09 13:43:47 -04:00
attr = self.class.spammable_attrs.find do |_, options|
options.fetch(:spam_description, false)
end
2016-08-09 13:43:47 -04:00
public_send(attr.first) if attr && respond_to?(attr.first.to_sym)
end
def spammable_text
2016-08-09 13:43:47 -04:00
result = self.class.spammable_attrs.map do |attr|
public_send(attr.first)
end
result.reject(&:blank?).join("\n")
end
# Override in Spammable if further checks are necessary
def check_for_spam?
2016-08-09 13:43:47 -04:00
true
end
end