discourse_dev 0.0.5 → 0.2.0

data/lib/discourse_dev.rb

@@ -2,12 +2,57 @@
 
 require 'i18n'
 
-Dir[File.dirname(__FILE__) + '/**/*.rb'].each {|file| require file }
-
-I18n.load_path += Dir[File.join(__dir__, 'faker', 'locales', '**/*.yml')]
-I18n.reload! if I18n.backend.initialized?
+Dir[File.dirname(__FILE__) + '/**/*.rb'].each { |file| require file }
 
 module DiscourseDev
   require 'discourse_dev/railtie'
   require 'discourse_dev/engine'
+
+  def self.auth_plugin_enabled?
+    config.auth_plugin_enabled
+  end
+
+  def self.config
+    @config ||= Config.new
+  end
+
+  def self.auth_plugin
+    return unless auth_plugin_enabled?
+
+    @auth_plugin ||= begin
+      path = File.join(root, 'auth', 'plugin.rb')
+      source = File.read(path)
+      metadata = Plugin::Metadata.parse(source)
+      Plugin::Instance.new(metadata, path)
+    end
+  end
+
+  def self.settings_file
+    File.join(root, "config", "settings.yml")
+  end
+
+  def self.client_locale_files(locale_str)
+    Dir[File.join(root, "config", "locales", "client*.#{locale_str}.yml")]
+  end
+
+  def self.root
+    File.expand_path("..", __dir__)
+  end
+end
+
+require "active_record/database_configurations"
+
+ActiveRecord::Tasks::DatabaseTasks.module_eval do
+  alias_method :rails_each_current_configuration, :each_current_configuration
+
+  private
+  def each_current_configuration(environment, name = nil)
+    rails_each_current_configuration(environment, name) { |db_config|
+      next if environment == "development" &&
+        ENV["SKIP_TEST_DATABASE"] == "1" &&
+        db_config.configuration_hash[:database] != "discourse_development"
+
+      yield db_config
+    }
+  end
 end

data/lib/discourse_dev/category.rb

@@ -7,8 +7,8 @@ require 'faker'
 module DiscourseDev
   class Category < Record
 
-    def initialize(count = DEFAULT_COUNT)
-      super(::Category, count)
+    def initialize
+      super(::Category, DiscourseDev.config.category[:count])
       @parent_category_ids = ::Category.where(parent_category_id: nil).pluck(:id)
     end
 
@@ -36,7 +36,7 @@ module DiscourseDev
     def permissions
       return @permissions if @permissions.present?
       return { everyone: :full } if Faker::Boolean.boolean(true_ratio: 0.75)
-        
+
       permission = {}
       group = Group.random
       permission[group.id] = Faker::Number.between(from: 1, to: 3)

data/lib/discourse_dev/config.rb

@@ -1,77 +1,149 @@
 # frozen_string_literal: true
 
 require 'rails'
+require 'highline/import'
 
 module DiscourseDev
   class Config
-    attr_reader :config, :default_config
+    attr_reader :config, :file_path
 
     def initialize
-      @default_config = YAML.load_file(File.join(File.expand_path(__dir__), "config.yml"))
-      file_path = File.join(Rails.root, "config", "dev.yml")
+      default_file_path = File.join(DiscourseDev.root, "config", "dev.yml")
+      @file_path = File.join(Rails.root, "config", "dev.yml")
+      default_config = YAML.load_file(default_file_path)
 
       if File.exists?(file_path)
-        @config = YAML.load_file(file_path)
+        user_config = YAML.load_file(file_path)
       else
-        @config = {}
+        puts "I did no detect a custom `config/dev.yml` file, creating one for you where you can amend defaults."
+        FileUtils.cp(default_file_path, file_path)
+        user_config = {}
       end
+
+      @config = default_config.deep_merge(user_config).deep_symbolize_keys
     end
 
     def update!
       update_site_settings
       create_admin_user
+      create_new_user
       set_seed
     end
 
+    private
+
     def update_site_settings
       puts "Updating site settings..."
 
-      site_settings = config["site_settings"] || {}
+      site_settings = config[:site_settings] || {}
 
       site_settings.each do |key, value|
         puts "#{key} = #{value}"
         SiteSetting.set(key, value)
       end
 
-      keys = site_settings.keys
-
-      default_config["site_settings"].each do |key, value|
-        next if keys.include?(key)
-
-        puts "#{key} = #{value}"
-        SiteSetting.set(key, value)
-      end
-
       SiteSetting.refresh!
     end
 
     def create_admin_user
       puts "Creating default admin user account..."
 
-      settings = config["admin"]
+      settings = config[:admin]
+
+      if settings.present?
+        create_admin_user_from_settings(settings)
+      else
+        create_admin_user_from_input
+      end
+    end
+
+    def create_new_user
+      settings = config[:new_user]
 
       if settings.present?
-        email = settings["email"]
+        email = settings[:email] || "new_user@example.com"
 
-        admin = ::User.create!(
+        new_user = ::User.create!(
           email: email,
-          username: settings["username"] || UserNameSuggester.suggest(email),
-          password: settings["password"]
+          username: settings[:username] || UserNameSuggester.suggest(email)
         )
-        admin.grant_admin!
-        if admin.trust_level < 1
-          admin.change_trust_level!(1)
-        end
-        admin.email_tokens.update_all confirmed: true
-        admin.activate
-      else
-        Rake::Task['admin:create'].invoke
+        new_user.email_tokens.update_all confirmed: true
+        new_user.activate
       end
     end
 
     def set_seed
-      seed = config["seed"] || default_config["seed"] || 1
+      seed = self.seed || 1
       Faker::Config.random = Random.new(seed)
     end
+
+    def start_date
+      DateTime.parse(config[:start_date])
+    end
+
+    def method_missing(name)
+      config[name.to_sym]
+    end
+
+    def create_admin_user_from_settings(settings)
+      email = settings[:email]
+
+      admin = ::User.create!(
+        email: email,
+        username: settings[:username] || UserNameSuggester.suggest(email),
+        password: settings[:password]
+      )
+      admin.grant_admin!
+      if admin.trust_level < 1
+        admin.change_trust_level!(1)
+      end
+      admin.email_tokens.update_all confirmed: true
+      admin.activate
+    end
+
+    def create_admin_user_from_input
+      begin
+        email = ask("Email:  ")
+        password = ask("Password (optional, press ENTER to skip):  ")
+        username = UserNameSuggester.suggest(email)
+
+        admin = ::User.new(
+          email: email,
+          username: username
+        )
+
+        if password.present?
+          admin.password = password
+        else
+          puts "Once site is running use https://localhost:9292/user/#{username}/become to access the account in development"
+        end
+
+        admin.name = ask("Full name:  ") if SiteSetting.full_name_required
+        saved = admin.save
+
+        if saved
+          File.open(file_path, 'a') do | file|
+            file.puts("admin:")
+            file.puts("  username: #{admin.username}")
+            file.puts("  email: #{admin.email}")
+            file.puts("  password: #{password}") if password.present?
+          end
+        else
+          say(admin.errors.full_messages.join("\n"))
+        end
+      end while !saved
+
+      admin.active = true
+      admin.save
+
+      admin.grant_admin!
+      if admin.trust_level < 1
+        admin.change_trust_level!(1)
+      end
+      admin.email_tokens.update_all confirmed: true
+      admin.activate
+
+      say("\nAdmin account created successfully with username #{admin.username}")
+    end
   end
 end

data/lib/discourse_dev/group.rb

@@ -7,10 +7,8 @@ require 'faker'
 module DiscourseDev
   class Group < Record
 
-    DEFAULT_COUNT = 15.freeze
-
-    def initialize(count = DEFAULT_COUNT)
-      super(::Group, count)
+    def initialize
+      super(::Group, DiscourseDev.config.group[:count])
     end
 
     def data
@@ -18,7 +16,8 @@ module DiscourseDev
         name: Faker::Discourse.unique.group,
         public_exit: Faker::Boolean.boolean,
         public_admission: Faker::Boolean.boolean,
-        primary_group: Faker::Boolean.boolean
+        primary_group: Faker::Boolean.boolean,
+        created_at: Faker::Time.between(from: DiscourseDev.config.start_date, to: DateTime.now),
       }
     end
 

data/lib/discourse_dev/post.rb

@@ -8,43 +8,102 @@ module DiscourseDev
 
     attr_reader :topic
 
-    def initialize(topic, count = DEFAULT_COUNT)
+    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::Markdown.sandwich(sentences: 5),
-        skip_validations: true
+        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!
-      PostCreator.new(user, data).create!
+      user = self.user
+      data = Faker::DiscourseMarkdown.with_user(user.id) { self.data }
+      post = PostCreator.new(user, data).create!
+      topic.reload
+      generate_likes(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)
+
       @count.times do |i|
         @index = i
         create!
       end
     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|
+        @index = 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
+
   end
 end

data/lib/discourse_dev/record.rb

@@ -11,7 +11,12 @@ module DiscourseDev
     attr_reader :model, :type
 
     def initialize(model, count = DEFAULT_COUNT)
-      Faker::Discourse.unique.clear
+      @@initialized ||= begin
+        Faker::Discourse.unique.clear
+        RateLimiter.disable
+        true
+      end
+
       @model = model
       @type = model.to_s
       @count = count

data/lib/discourse_dev/tag.rb

@@ -7,8 +7,8 @@ require 'faker'
 module DiscourseDev
   class Tag < Record
 
-    def initialize(count = DEFAULT_COUNT)
-      super(::Tag, count)
+    def initialize
+      super(::Tag, DiscourseDev.config.tag[:count])
     end
 
     def data

data/lib/discourse_dev/tasks/dev.rake

@@ -4,6 +4,9 @@ def check_environment!
   if !Rails.env.development?
     raise "Database commands are only supported in development environment"
   end
+
+  ENV['SKIP_TEST_DATABASE'] = "1"
+  ENV['SKIP_MULTISITE'] = "1"
 end
 
 desc 'Run db:migrate:reset task and populate sample content for development environment'
@@ -17,11 +20,12 @@ end
 
 desc 'Initialize development environment'
 task 'dev:config' => ['db:load_config'] do |_, args|
-  DiscourseDev::Config.new.update!
+  DiscourseDev.config.update!
 end
 
 desc 'Populate sample content for development environment'
 task 'dev:populate' => ['db:load_config'] do |_, args|
+  system("redis-cli flushall")
   Rake::Task['groups:populate'].invoke
   Rake::Task['users:populate'].invoke
   Rake::Task['categories:populate'].invoke

data/lib/discourse_dev/tasks/populate.rake

@@ -24,3 +24,8 @@ desc 'Creates sample topics'
 task 'topics:populate' => ['db:load_config'] do |_, args|
   DiscourseDev::Topic.populate!
 end
+
+desc 'Add replies to a topic'
+task 'replies:populate', [:topic_id, :count] => ['db:load_config'] do |_, args|
+  DiscourseDev::Post.add_replies!(args)
+end