discourse/plugins/chat/lib/chat_message_rate_limiter.rb

# frozen_string_literal: true

class Chat::ChatMessageRateLimiter
  def self.run!(user)
    instance = self.new(user)
    instance.run!
  end

  def initialize(user)
    @user = user
  end

  def run!
    return if @user.staff?

    allowed_message_count =
      (
        if @user.trust_level == TrustLevel[0]
          SiteSetting.chat_allowed_messages_for_trust_level_0
        else
          SiteSetting.chat_allowed_messages_for_other_trust_levels
        end
      )
    return if allowed_message_count.zero?

    @rate_limiter = RateLimiter.new(@user, "create_chat_message", allowed_message_count, 30.seconds)
    silence_user if @rate_limiter.remaining.zero?
    @rate_limiter.performed!
  end

  def clear!
    # Used only for testing. Need to clear the rate limiter between tests.
    @rate_limiter.clear! if defined?(@rate_limiter)
  end

  private

  def silence_user
    silenced_for_minutes = SiteSetting.chat_auto_silence_duration
    return if silenced_for_minutes.zero?

    UserSilencer.silence(
      @user,
      Discourse.system_user,
      silenced_till: silenced_for_minutes.minutes.from_now,
      reason: I18n.t("chat.errors.rate_limit_exceeded"),
    )
  end
end