discourse/lib/discourse_dev/post.rb

# frozen_string_literal: true

require 'discourse_dev/record'
require 'faker'

module DiscourseDev
  class Post < Record

    attr_reader :topic

    def initialize(topic, count)
      super(::Post, count)
      @topic = topic

      category = topic.category
      @max_likes_count = DiscourseDev.config.post[:max_likes_count]
      unless 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
        @max_likes_count = @user_count - 1
      end
    end

    def data
      {
        topic_id: topic.id,
        raw: Faker::DiscourseMarkdown.sandwich(sentences: 5),
        created_at: Faker::Time.between(from: topic.last_posted_at, to: DateTime.now),
        skip_validations: true,
        skip_guardian: true
      }
    end

    def create!
      user = self.user
      data = Faker::DiscourseMarkdown.with_user(user.id) { self.data }
      post = PostCreator.new(user, data).create!
      topic.reload
      generate_likes(post)
      post
    end

    def generate_likes(post)
      user_ids = [post.user_id]

      Faker::Number.between(from: 0, to: @max_likes_count).times do
        user = self.user
        next if user_ids.include?(user.id)

        PostActionCreator.new(user, post, PostActionType.types[:like], created_at: Faker::Time.between(from: post.created_at, to: DateTime.now)).perform
        user_ids << user.id
      end
    end

    def user
      return User.random if topic.category.groups.blank?
      return Discourse.system_user if @user_ids.blank?

      position = Faker::Number.between(from: 0, to: @user_count - 1)
      ::User.find(@user_ids[position])
    end

    def populate!
      generate_likes(topic.first_post)

      super(ignore_current_count: true)
    end

    def current_count
      topic.posts_count - 1
    end

    def self.add_replies!(args)
      if !args[:topic_id]
        puts "Topic ID is required. Aborting."
        return
      end

      if !::Topic.find_by_id(args[:topic_id])
        puts "Topic ID does not match topic in DB, aborting."
        return
      end

      topic = ::Topic.find_by_id(args[:topic_id])
      count = args[:count] ? args[:count].to_i : 50

      puts "Creating #{count} replies in '#{topic.title}'"

      count.times do |i|
        begin
          user = User.random
          reply = Faker::DiscourseMarkdown.with_user(user.id) do
            {
              topic_id: topic.id,
              raw: Faker::DiscourseMarkdown.sandwich(sentences: 5),
              skip_validations: true
            }
          end
          PostCreator.new(user, reply).create!
        rescue ActiveRecord::RecordNotSaved => e
          puts e
        end
      end

      puts "Done!"
    end

    def self.random
      super(::Post)
    end

  end
end