discourse_dev 0.0.7 → 0.2.1

data/config/locales/client.en.yml

@@ -0,0 +1,6 @@
+en:
+  js:
+    login:
+      developmentauth:
+        title: DevelopmentAuth
+        name: DevelopmentAuth

data/discourse_dev.gemspec

@@ -22,6 +22,7 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler", "~> 2.0"
   spec.add_development_dependency "rake", ">= 12.3.3"
+  spec.add_development_dependency "rubocop-discourse"
 
   spec.required_ruby_version = '>= 2.6.0'
 end

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,31 +8,47 @@ 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!
-      begin
-        PostCreator.new(user, data).create!
-      rescue ActiveRecord::RecordNotSaved => e
-        puts e
+      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
 
@@ -45,8 +61,9 @@ module DiscourseDev
     end
 
     def populate!
-      @count.times do |i|
-        @index = i
+      generate_likes(topic.first_post)
+
+      @count.times do
         create!
       end
     end
@@ -68,14 +85,16 @@ module DiscourseDev
       puts "Creating #{count} replies in '#{topic.title}'"
 
       count.times do |i|
-        @index = i
         begin
-          reply = {
-            topic_id: topic.id,
-            raw: Faker::Markdown.sandwich(sentences: 5),
-            skip_validations: true
-          }
-          PostCreator.new(User.random, reply).create!
+          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

data/lib/discourse_dev/record.rb

@@ -11,11 +11,15 @@ 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
-      @index = nil
     end
 
     def create!
@@ -25,7 +29,7 @@ module DiscourseDev
 
     def populate!
       if current_count >= @count
-        puts "Already have #{@count}+ #{type.downcase} records."
+        puts "Already have #{current_count} #{type.downcase} records"
 
         Rake.application.top_level_tasks.each do |task_name|
           Rake::Task[task_name].reenable
@@ -33,12 +37,14 @@ module DiscourseDev
 
         Rake::Task['dev:repopulate'].invoke
         return
+      elsif current_count > 0
+        @count -= current_count
+        puts "There are #{current_count} #{type.downcase} records. Creating #{@count} more."
+      else
+        puts "Creating #{@count} sample #{type.downcase} records"
       end
 
-      puts "Creating #{@count} sample #{type.downcase} records"
-
-      @count.times do |i|
-        @index = i
+      @count.times do
         create!
         putc "."
       end
@@ -46,10 +52,6 @@ module DiscourseDev
       puts
     end
 
-    def index
-      @index
-    end
-
     def current_count
       model.count
     end