65 lines
1.7 KiB
Ruby
65 lines
1.7 KiB
Ruby
|
# 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
|
||
|
if !SiteSetting.enable_experimental_chat_threaded_discussions
|
||
|
raise "You need to enable_experimental_chat_threaded_discussions to run this task"
|
||
|
end
|
||
|
|
||
|
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
|