From 76fe3a16a9f347ad1535bdfcbc5a39d27a88f178 Mon Sep 17 00:00:00 2001
From: Vinoth Kannan <svkn.87@gmail.com>
Date: Thu, 15 Jul 2021 11:11:41 +0530
Subject: [PATCH] DEV: trigger new discourse events `after_create_dev_record` &
 `after_populate_dev_records`. (#13733)

After every new random record created using the `dev:populate` rake task a new Discourse event will be triggered. So the plugins can modify the records if needed.
---
 lib/discourse_dev/record.rb | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/lib/discourse_dev/record.rb b/lib/discourse_dev/record.rb
index b6196aebcfb..5713178f641 100644
--- a/lib/discourse_dev/record.rb
+++ b/lib/discourse_dev/record.rb
@@ -18,18 +18,20 @@ module DiscourseDev
       end
 
       @model = model
-      @type = model.to_s
+      @type = model.to_s.downcase.to_sym
       @count = count
     end
 
     def create!
       record = model.create!(data)
       yield(record) if block_given?
+      DiscourseEvent.trigger(:after_create_dev_record, record, type)
+      record
     end
 
     def populate!
       if current_count >= @count
-        puts "Already have #{current_count} #{type.downcase} records"
+        puts "Already have #{current_count} #{type} records"
 
         Rake.application.top_level_tasks.each do |task_name|
           Rake::Task[task_name].reenable
@@ -39,17 +41,20 @@ module DiscourseDev
         return
       elsif current_count > 0
         @count -= current_count
-        puts "There are #{current_count} #{type.downcase} records. Creating #{@count} more."
+        puts "There are #{current_count} #{type} records. Creating #{@count} more."
       else
-        puts "Creating #{@count} sample #{type.downcase} records"
+        puts "Creating #{@count} sample #{type} records"
       end
 
+      records = []
       @count.times do
-        create!
+        records << create!
         putc "."
       end
 
+      DiscourseEvent.trigger(:after_populate_dev_records, records, type)
       puts
+      records
     end
 
     def current_count