discourse/plugins/chat/lib/discourse_dev/thread.rb

# frozen_string_literal: true

require "discourse_dev/record"
require "faker"

module DiscourseDev
  class Thread < Record
    def initialize(channel_id:, message_count: nil, ignore_current_count: false)
      @channel_id = channel_id
      @message_count = message_count&.to_i || 30
      @ignore_current_count = ignore_current_count
      super(::Chat::Thread, 1)
    end

    def data
      channel = ::Chat::Channel.find(@channel_id)
      return if !channel

      if !channel.threading_enabled
        puts "Enabling threads in channel #{channel.id}"
        channel.update!(threading_enabled: true)
      end

      membership =
        ::Chat::UserChatChannelMembership.where(chat_channel: channel).order("RANDOM()").first
      user = membership.user

      om =
        Chat::MessageCreator.create(
          user: user,
          content: Faker::Lorem.paragraph,
          chat_channel: channel,
        ).chat_message

      { original_message_user: user, original_message: om, channel: channel }
    end

    def create!
      super do |thread|
        thread.original_message.update!(thread: thread)
        user =
          ::Chat::UserChatChannelMembership
            .where(chat_channel: thread.channel)
            .order("RANDOM()")
            .first
            .user
        @message_count.times do
          Chat::MessageCreator.create(
            {
              user: user,
              chat_channel: thread.channel,
              content: Faker::Lorem.paragraph,
              thread_id: thread.id,
            },
          )
        end
      end
    end
  end
end