discourse_dev 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2b40ff76d084fee2f0eeef50807149a399e6c4bbdf256259dd23a6925d6bdeb7
-  data.tar.gz: 748af95af548a6d79ee21d96dd68688d5c554cb78391a416b63980805d4b25a3
+  metadata.gz: b494dc262cd9afaa6f8f3906f3f9f23cb452cf852504dc095f411ece98858119
+  data.tar.gz: c98f3d163fa2e68b6bedd53984924432ecd1234534425d2cba565614d1fa3a59
 SHA512:
-  metadata.gz: '0914512344fa239f36c6ae42c4890ef4cad9fd991a234c5f8b301751a84a89633d20f035e64e526935031767842b27e6a36758e8ea982a3672e83e22c2826b4b'
-  data.tar.gz: d9cfa147a8616124232baef18d01203cd09b6c88a5d5c4c7b9a7527e4ef56d4f79c07c0d2ea13809907ba81b3cc38c4b669307d2a6ebef57b4d21364b96ba1b4
+  metadata.gz: 37b9f8b3f1fd4a555e260de5f4470236c2f68bc9141b5b68171d210b933137e1f6364de8213a6e189af25d5f36deb26b6641bcef6e9f8518e147615f366b7b67
+  data.tar.gz: d7b3b6502018d94908f0364f191834be6000f0d4341e66504214cfa1d1b1738897f34b8534c037aa3ac8382811b97ad981b484844179c6b9bd6f83fa19fd9995

data/config/routes.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+DiscourseDev::Engine.routes.draw do
+  get ':username_or_email/become' => 'admin/impersonate#create', constraints: AdminConstraint.new
+end

data/lib/discourse_dev.rb

@@ -8,5 +8,6 @@ I18n.load_path += Dir[File.join(__dir__, 'faker', 'locales', '**/*.yml')]
 I18n.reload! if I18n.backend.initialized?
 
 module DiscourseDev
-  require 'discourse_dev/railtie' if defined?(Rails)
+  require 'discourse_dev/railtie'
+  require 'discourse_dev/engine'
 end

data/lib/discourse_dev/engine.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module DiscourseDev
+  class Engine < Rails::Engine; end
+end

data/lib/discourse_dev/post.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require 'discourse_dev/record'
+require 'faker'
+
+module DiscourseDev
+  class Post < Record
+
+    attr_reader :topic
+
+    def initialize(topic, count = DEFAULT_COUNT)
+      super(::Post, count)
+      @topic = topic
+
+      category = topic.category
+      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
+      end
+    end
+
+    def data
+      {
+        topic_id: topic.id,
+        raw: Faker::Markdown.sandwich(sentences: 5),
+        skip_validations: true
+      }
+    end
+
+    def create!
+      PostCreator.new(user, data).create!
+    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!
+      @count.times do |i|
+        @index = i
+        create!
+      end
+    end
+  end
+end

data/lib/discourse_dev/record.rb

@@ -24,6 +24,17 @@ module DiscourseDev
     end
 
     def populate!
+      if current_count >= @count
+        puts "Already have #{@count}+ #{type.downcase} records."
+
+        Rake.application.top_level_tasks.each do |task_name|
+          Rake::Task[task_name].reenable
+        end
+
+        Rake::Task['dev:repopulate'].invoke
+        return
+      end
+
       puts "Creating #{@count} sample #{type.downcase} records"
 
       @count.times do |i|
@@ -39,6 +50,10 @@ module DiscourseDev
       @index
     end
 
+    def current_count
+      model.count
+    end
+
     def self.populate!
       self.new.populate!
     end

data/lib/discourse_dev/tasks/dev.rake

@@ -28,3 +28,16 @@ task 'dev:populate' => ['db:load_config'] do |_, args|
   Rake::Task['tags:populate'].invoke
   Rake::Task['topics:populate'].invoke
 end
+
+desc 'Repopulate sample datas in development environment'
+task 'dev:repopulate' => ['db:load_config'] do |_, args|
+  require 'highline/import'
+
+  answer = ask("Do you want to repopulate the database with fresh data? It will recreate DBs and run migration from scratch before generating all the samples. (Y/n)  ")
+
+  if (answer == "" || answer.downcase == 'y')
+    Rake::Task['dev:reset'].invoke
+  else
+    puts "You can run `dev:reset` rake task to do this repopulate action anytime."
+  end
+end

data/lib/discourse_dev/topic.rb

@@ -41,7 +41,9 @@ module DiscourseDev
 
     def create!
       @category = Category.random
-      PostCreator.new(user, data).create!
+      post = PostCreator.new(user, data).create!
+
+      Post.new(post.topic, Faker::Number.between(from: 0, to: 5)).populate!
     end
 
     def user
@@ -53,5 +55,10 @@ module DiscourseDev
       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
   end
 end

data/lib/discourse_dev/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DiscourseDev
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: discourse_dev
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Vinoth Kannan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-03-11 00:00:00.000000000 Z
+date: 2021-03-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faker
@@ -65,12 +65,15 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- config/routes.rb
 - discourse_dev.gemspec
 - lib/discourse_dev.rb
 - lib/discourse_dev/category.rb
 - lib/discourse_dev/config.rb
 - lib/discourse_dev/config.yml
+- lib/discourse_dev/engine.rb
 - lib/discourse_dev/group.rb
+- lib/discourse_dev/post.rb
 - lib/discourse_dev/railtie.rb
 - lib/discourse_dev/record.rb
 - lib/discourse_dev/tag.rb