custom_seeds 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2c18e6ab268e271ea6b6e850d485ae3ddafe97c2b35aaf28eccfee628215498c
-  data.tar.gz: 4f23f88fe7db39949b6cb99a49fdb53c7d7f5367f808a2f8207a1b95f6fc2e27
+  metadata.gz: e38b6613018be59c8754faa324c14bd806094b2cf32da6e9d8035e9187bb2bff
+  data.tar.gz: e7ba26d4d6f84f44f8d224043a960df3e2472da93fcdd46699ab7ed4f4d49016
 SHA512:
-  metadata.gz: c25f3fcfe10bea51714723bc3ecf21f344de803253b393f42e72f0d01ff59e563e57e43183d62f6dc978e2bc971265ad8683825c1f9cd18923c621e7f67436c9
-  data.tar.gz: ed8a157977947610ed55fb446ba911bc5d2eb451cda744dd191660a2f2ab0694631e9ac5eef99f7dae0a00f79a7bab9d3b1f3a7f135244367d72a31036250cd5
+  metadata.gz: 399e0ee4e87f51637aa49f5944ecf0a268f20a00094f6c4a78897596abce04ce4d7f57d3fe9642082f011d59f03f6782e6156ad237a14fa8467ad63acb1a777e
+  data.tar.gz: f5b4542a641a5a3510ac03c0b662e875f9d83d02a37e77b8e4e25d0ea044b72d01cd5192141d78609845decaf43f0346b2d98cef15a078e8b2df8bc3c7a67093

data/.gitignore

@@ -0,0 +1 @@
+*.gem

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    custom_seeds (0.1.0)
+    custom_seeds (0.2.0)
       progress_bar (= 1.3.1)
 
 GEM

data/README.md

@@ -45,6 +45,18 @@ CustomSeeds.define do
 end
 ```
 
+### CLI
+
+Run seeds using the `rseed` command:
+
+    $ rseed
+    $ rseed db/seeds/users/
+    $ rseed db/seeds/users/followed_products.rb
+
+Run all seeds:
+
+### Rake
+
 Run the file using the rake task:
 
 ```bash

data/bin/rseed

@@ -3,9 +3,27 @@
 
 require 'bundler/setup'
 require 'custom_seeds'
+require 'optparse'
+
+options = {}
+OptionParser.new do |opts|
+  opts.banner = 'Usage: rseed [options] FILE'
+
+  opts.on('-v', '--verbose', 'Log all sql commands run') do
+    options[:verbose] = true
+  end
+
+  opts.on('-d', '--dry-run', 'Rollback all changes after test run') do
+    options[:dry_run] = true
+  end
+end.parse!
+
+CustomSeeds::SeedFile.options(options)
+
+file_path = ARGV[0]
 
 # if argv[0] is a directory...
-if !ARGV[0] || File.directory?(ARGV[0])
+if !file_path || File.directory?(file_path)
   list = CustomSeeds::SeedList.new(directory: ARGV[0])
   list.each { |seed| load(seed.filename) }
 else

data/lib/custom_seeds/seed_file.rb

@@ -17,11 +17,27 @@ module CustomSeeds
     end
 
     def initialize
+      @dry_run = self.class.options[:dry_run]
+      @verbose = self.class.options[:verbose]
       @records = []
       @build_block = ->(_record) { raise 'No build block defined' }
       @log_block = nil
     end
 
+    def self.options(value = nil)
+      return @options if value.nil?
+
+      @options = value
+    end
+
+    def dry_run?
+      @dry_run
+    end
+
+    def verbose?
+      @verbose
+    end
+
     def colorize(string, color_code)
       "\e[#{MAP[color_code]}m#{string}\e[0m"
     end
@@ -51,6 +67,12 @@ module CustomSeeds
       @before_block = block
     end
 
+    def after(&block)
+      return @after_block if block.nil?
+
+      @after_block = block
+    end
+
     # memoize block
     def let(name, &block)
       @memoized ||= {}
@@ -76,21 +98,52 @@ module CustomSeeds
       @log_block = block
     end
 
-    def run
-      before&.call
-      progress_bar = ProgressBar.new(records.size)
+    def log_sql_statements(&block)
+      sql_statements = []
+      return block.call unless verbose?
 
-      records.each do |record|
-        build_block.call(record)
+      subscriber = ActiveSupport::Notifications.subscribe('sql.active_record') do |_, _, _, _, details|
+        sql_statements << details[:sql] unless details[:name] == 'SCHEMA'
+      end
 
-        if log_block
-          puts colorize(log_block.call(record), :blue)
-        else
-          progress_bar.increment!
-        end
+      block.call
+    ensure
+      ActiveSupport::Notifications.unsubscribe(subscriber)
+      sql_statements.each { |sql| puts sql }
+    end
+
+    def transaction(&block)
+      puts colorize('🚧 Dry Run. No changes will be applied.', :green) if dry_run?
+
+      ActiveRecord::Base.transaction do
+        block.call
+
+        raise ActiveRecord::Rollback if dry_run?
       end
 
-      puts colorize('✅ Seeding completed', :green)
+      puts colorize('🚧 Dry run complete', :green) if dry_run?
+    end
+
+    def run
+      log_sql_statements do
+        transaction do
+          before&.call
+          progress_bar = ProgressBar.new(records.size)
+
+          records.each do |record|
+            build_block.call(record)
+
+            if log_block
+              puts colorize(log_block.call(record), :blue)
+            else
+              progress_bar.increment!
+            end
+          end
+
+          after&.call
+          puts colorize('✅ Seeding completed', :green)
+        end
+      end
     end
   end
 end

data/lib/custom_seeds/version.rb

@@ -1,3 +1,3 @@
 module CustomSeeds
-  VERSION = '0.2.0'.freeze
+  VERSION = '0.3.0'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: custom_seeds
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Michael Wheeler
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-15 00:00:00.000000000 Z
+date: 2024-11-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: progress_bar
@@ -88,6 +88,7 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- ".gitignore"
 - ".tool-versions"
 - Gemfile
 - Gemfile.lock