discourse-akismet/plugin.rb

# frozen_string_literal: true

# name: discourse-akismet
# about: supports submitting posts to akismet for review
# version: 0.1.0
# authors: Michael Verdi, Robin Ward
# url: https://github.com/discourse/discourse-akismet

enabled_site_setting :akismet_enabled

load File.expand_path('../lib/discourse_akismet.rb', __FILE__)
load File.expand_path('../lib/discourse_akismet/users_bouncer.rb', __FILE__)
load File.expand_path('../lib/akismet.rb', __FILE__)

# Classes are not loaded at this point so we check for the file
reviewable_api_enabled = File.exist? File.expand_path('../../../app/models/reviewable.rb', __FILE__)

# We still want to run tests associated to the engine ;)
if reviewable_api_enabled && !Rails.env.test?
  register_asset "stylesheets/reviewable-akismet-post-styles.scss"
else
  load File.expand_path('../lib/discourse_akismet/engine.rb', __FILE__)
  register_asset "stylesheets/mod-queue-styles.scss"

  add_admin_route 'akismet.title', 'akismet'
  Discourse::Application.routes.append do
    mount ::DiscourseAkismet::Engine, at: '/admin/plugins/akismet'
  end
end

after_initialize do
  %W[
    check_for_spam_posts
    regular/check_users_for_spam
    regular/confirm_akismet_flagged_posts
    check_akismet_post
    update_akismet_status
  ].each do |filename|
    require_dependency File.expand_path("../jobs/#{filename}.rb", __FILE__)
  end

  # We want to include this even if the plugin is not enabled, that's why we use false here.
  add_to_serializer(:site, :reviewable_api_enabled, false) { reviewable_api_enabled }

  if reviewable_api_enabled
    require_dependency File.expand_path('../models/reviewable_akismet_post.rb', __FILE__)
    require_dependency File.expand_path('../models/reviewable_akismet_user.rb', __FILE__)
    require_dependency File.expand_path('../serializers/reviewable_akismet_post_serializer.rb', __FILE__)
    require_dependency File.expand_path('../serializers/reviewable_akismet_user_serializer.rb', __FILE__)
    register_reviewable_type ReviewableAkismetPost
    register_reviewable_type ReviewableAkismetUser

    add_model_callback(UserProfile, :before_save) do
      if bio_raw_changed? && bio_raw.present?
        DiscourseAkismet::UsersBouncer.new.enqueue_for_check(user)
      end
    end

  else
    add_to_class(:guardian, :can_review_akismet?) do
      user.try(:staff?)
    end

    add_to_serializer(:current_user, :akismet_review_count) do
      scope.can_review_akismet? ? DiscourseAkismet.needs_review.count : nil
    end
  end

  # Store extra data for akismet
  on(:post_created) do |post, params|
    if DiscourseAkismet.should_check_post?(post)
      DiscourseAkismet.move_to_state(post, 'new', params)

      # Enqueue checks for TL0 posts faster
      Jobs.enqueue(:check_akismet_post, post_id: post.id) if post.user.trust_level == 0
    end
  end

  # When staff agrees a flagged post is spam, send it to akismet
  on(:confirmed_spam_post) do |post|
    if SiteSetting.akismet_enabled?
      Jobs.enqueue(:update_akismet_status, post_id: post.id, status: 'spam')
    end
  end

  # If a user is anonymized, support anonymizing their IPs
  on(:user_anonymized) do |args|
    user = args[:user]
    opts = args[:opts]

    if user && opts && opts.has_key?(:anonymize_ip)
      sql = <<~SQL
        UPDATE post_custom_fields AS pcf
         SET value = :new_ip
         FROM posts AS p
         WHERE name = 'AKISMET_IP_ADDRESS'
           AND p.id = pcf.post_id
           AND p.user_id = :user_id
      SQL

      args = { user_id: user.id, new_ip: opts[:anonymize_ip] }

      # TODO remove post Discourse 2.1
      if defined? DB
        DB.exec sql, args
      else
        PostCustomField.exec_sql(sql, args)
      end
    end
  end

  if reviewable_api_enabled
    staff_actions = %i[confirmed_spam confirmed_ham ignored confirmed_spam_deleted]
    extend_list_method(UserHistory, :staff_actions, staff_actions)
  end
end