discourse/lib/discourse_dev/topic.rb

132 lines
3.8 KiB
Ruby

# frozen_string_literal: true
require 'discourse_dev/record'
require 'faker'
module DiscourseDev
class Topic < Record
def initialize(private_messages: false, recipient: nil, ignore_current_count: false)
@settings = DiscourseDev.config.topic
@private_messages = private_messages
@recipient = recipient
@ignore_current_count = ignore_current_count
super(::Topic, @settings[:count])
end
def data
max_views = 0
case Faker::Number.between(from: 0, to: 5)
when 0
max_views = 10
when 1
max_views = 100
when 2
max_views = SiteSetting.topic_views_heat_low
when 3
max_views = SiteSetting.topic_views_heat_medium
when 4
max_views = SiteSetting.topic_views_heat_high
when 5
max_views = SiteSetting.topic_views_heat_high + SiteSetting.topic_views_heat_medium
end
if @category
merge_attributes = {
category: @category.id,
tags: tags
}
else
merge_attributes = {
archetype: "private_message",
target_usernames: [@recipient]
}
end
{
title: title[0, SiteSetting.max_topic_title_length],
raw: Faker::DiscourseMarkdown.sandwich(sentences: 5),
created_at: Faker::Time.between(from: DiscourseDev.config.start_date, to: DateTime.now),
topic_opts: {
import_mode: true,
views: Faker::Number.between(from: 1, to: max_views),
custom_fields: { dev_sample: true }
},
skip_validations: true
}.merge(merge_attributes)
end
def title
if current_count < I18n.t("faker.discourse.topics").count
Faker::Discourse.unique.topic
else
Faker::Lorem.unique.sentence(word_count: 5, supplemental: true, random_words_to_add: 4).chomp(".")
end
end
def tags
return unless SiteSetting.tagging_enabled
@tags = []
Faker::Number.between(from: @settings.dig(:tags, :min), to: @settings.dig(:tags, :max)).times do
@tags << Faker::Discourse.tag
end
@tags.uniq
end
def create!
if @private_messages && !::User.find_by_username(@recipient)
puts "Cannot create PMs for missing user with username: #{@recipient}"
exit 1
end
@category = @private_messages ? nil : Category.random
user = self.user
topic_data = Faker::DiscourseMarkdown.with_user(user.id) { data }
post = PostCreator.new(user, topic_data).create!
if override = @settings.dig(:replies, :overrides).find { |o| o[:title] == topic_data[:title] }
reply_count = override[:count]
else
reply_count = Faker::Number.between(from: @settings.dig(:replies, :min), to: @settings.dig(:replies, :max))
end
topic = post.topic
Post.new(topic, reply_count).populate!
topic
end
def populate!
topics = super(ignore_current_count: @ignore_current_count)
delete_unwanted_sidekiq_jobs
topics
end
def user
return ::User.find_by_username(@recipient) if @private_messages
return User.random if @category.groups.blank?
group_ids = @category.groups.pluck(:id)
user_ids = ::GroupUser.where(group_id: group_ids).pluck(:user_id)
user_count = user_ids.count
position = Faker::Number.between(from: 0, to: user_count - 1)
::User.find(user_ids[position] || Discourse::SYSTEM_USER_ID)
end
def current_count
category_definition_topic_ids = ::Category.pluck(:topic_id)
::Topic.where(archetype: :regular).where.not(id: category_definition_topic_ids).count
end
def delete_unwanted_sidekiq_jobs
Sidekiq::ScheduledSet.new.each do |job|
job.delete if job.item["class"] == "Jobs::UserEmail"
end
end
end
end