64ab2b3d9f
- Removed unnecessary column from `SpamLog` - Moved creation of SpamLogs out of its own service and into SpamCheckService - Simplified code in SpamCheckService. - Moved move spam related code into Spammable concern
82 lines
1.6 KiB
Ruby
82 lines
1.6 KiB
Ruby
module Spammable
|
|
extend ActiveSupport::Concern
|
|
include Gitlab::AkismetHelper
|
|
|
|
module ClassMethods
|
|
def attr_spammable(*attrs)
|
|
attrs.each do |attr|
|
|
spammable_attrs << attr.to_s
|
|
end
|
|
end
|
|
end
|
|
|
|
included do
|
|
has_one :user_agent_detail, as: :subject, dependent: :destroy
|
|
attr_accessor :spam
|
|
after_validation :check_for_spam, on: :create
|
|
|
|
cattr_accessor :spammable_attrs, instance_accessor: false do
|
|
[]
|
|
end
|
|
end
|
|
|
|
def can_be_submitted?
|
|
if user_agent_detail
|
|
user_agent_detail.submittable?
|
|
else
|
|
false
|
|
end
|
|
end
|
|
|
|
def submit_spam
|
|
return unless akismet_enabled? && can_be_submitted?
|
|
spam!(user_agent_detail, spammable_text, owner)
|
|
end
|
|
|
|
def spam_detected?(env)
|
|
@spam = is_spam?(env, owner, spammable_text)
|
|
end
|
|
|
|
def spam?
|
|
@spam
|
|
end
|
|
|
|
def check_for_spam
|
|
self.errors.add(:base, "Your #{self.class.name.underscore} has been recognized as spam and has been discarded.") if spam?
|
|
end
|
|
|
|
def owner_id
|
|
if self.respond_to?(:author_id)
|
|
self.author_id
|
|
elsif self.respond_to?(:creator_id)
|
|
self.creator_id
|
|
end
|
|
end
|
|
|
|
# Override this method if an additional check is needed before calling Akismet
|
|
def check_for_spam?
|
|
akismet_enabled?
|
|
end
|
|
|
|
def spam_title
|
|
raise 'Implement in included model!'
|
|
end
|
|
|
|
def spam_description
|
|
raise 'Implement in included model!'
|
|
end
|
|
|
|
private
|
|
|
|
def spammable_text
|
|
result = []
|
|
self.class.spammable_attrs.each do |entry|
|
|
result << self.send(entry)
|
|
end
|
|
result.reject(&:blank?).join("\n")
|
|
end
|
|
|
|
def owner
|
|
User.find(owner_id)
|
|
end
|
|
end
|