custom_seeds 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9602b763569053bcf931f449a65f4a2137eebec44c7503db21e040fe0cef193f
-  data.tar.gz: 9823e21e7d44e4c3bf2189b672600c4828525f221d091612da00b1f1f66e8a7c
+  metadata.gz: e38b6613018be59c8754faa324c14bd806094b2cf32da6e9d8035e9187bb2bff
+  data.tar.gz: e7ba26d4d6f84f44f8d224043a960df3e2472da93fcdd46699ab7ed4f4d49016
 SHA512:
-  metadata.gz: 94217e6c650f77bd1b247260fc5105fed80aeb75b4f3ab49bdb8c4ce81bd07631634b7e2c806ed514f4ab5963b52dd28875ff3a29f494779b4a4756c957488a9
-  data.tar.gz: eeac7048aba9934c5efc21b47e6f567503f3d1c47eea2b2dbf62e2a776e3f69bbcabe94e1d8eb86f64468bad17e091ee3b34a9f7e76c2d78e17ebdc4da25ad88
+  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
@@ -34,6 +34,7 @@ GEM
     rspec-support (3.10.3)
 
 PLATFORMS
+  arm64-darwin-23
   x86_64-darwin-21
   x86_64-darwin-23
 

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

@@ -0,0 +1,32 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+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 !file_path || File.directory?(file_path)
+  list = CustomSeeds::SeedList.new(directory: ARGV[0])
+  list.each { |seed| load(seed.filename) }
+else
+  # ...otherwise, load the file
+  load ARGV[0]
+end

data/custom_seeds.gemspec

@@ -13,8 +13,8 @@ Gem::Specification.new do |spec|
   spec.files         = `[ -d ".git" ] > /dev/null && type git > /dev/null && git ls-files -z`.split("\x0").reject do |f|
     f.match(%r{^(test|spec|features)/})
   end
-  spec.bindir        = 'exe'
-  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.bindir        = 'bin'
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
 
   spec.add_dependency 'progress_bar', '1.3.1'

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/seed_list.rb

@@ -13,8 +13,8 @@ module CustomSeeds
       end
     end
 
-    def initialize(directory: Rails.root.join('db/seeds/'))
-      @directory = directory
+    def initialize(directory: nil)
+      @directory = directory || ::Rails.root.join('db/seeds/')
     end
 
     def seeds

data/lib/custom_seeds/version.rb

@@ -1,3 +1,3 @@
 module CustomSeeds
-  VERSION = '0.1.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.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Michael Wheeler
 autorequire:
-bindir: exe
+bindir: bin
 cert_chain: []
-date: 2024-10-07 00:00:00.000000000 Z
+date: 2024-11-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: progress_bar
@@ -83,15 +83,18 @@ dependencies:
 description:
 email:
 - mwheeler@g2.com
-executables: []
+executables:
+- rseed
 extensions: []
 extra_rdoc_files: []
 files:
+- ".gitignore"
 - ".tool-versions"
 - Gemfile
 - Gemfile.lock
 - README.md
 - Rakefile
+- bin/rseed
 - custom_seeds.gemspec
 - lib/custom_seeds.rb
 - lib/custom_seeds/railtie.rb