seed_reaper 0.0.0 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1cf72255d5cd6a5556a0d2292007d0cb7768e9d4811351c765423388e719600e
-  data.tar.gz: 0e55d2a5a1efac5163e8cf110d73f923fbd6fa059252f434387329df5b2b1ed5
+  metadata.gz: 0347bce9f9ed607a63163bd7e25b290b3a961e9d18b6f9e95a07c917a216a892
+  data.tar.gz: 539a54c298af39892b4944d2dd0b18db062029170cf914958f89d5db2a1102ad
 SHA512:
-  metadata.gz: 9b2d4b4e50e3dcca8db054497fc8ce889d6543dcd4edeef1c9211b0ac075363cf37a315b43ad35b9bad131feb2caa1204812da561f2a28b99032a19ae4dbbb0d
-  data.tar.gz: 21bf09230c93d2f0e8de754ffcb0d2ceeb8e4a35036a90ddade1d217306ca9b05eb129944e39e4a759e338ce3ef390fb00e3c36cc63e6bed8a572afb2355a7db
+  metadata.gz: 7c8b1f2f7a6b2c21de8ab5c87f329a20d4a21cf622c9816d68cbf69b2d218a63cac468f3176986ffe6ac8b6ecdcd1c9329b1ab61448bb6714407cb9cb516c056
+  data.tar.gz: 2dbfe636a01fc0d62bbd54edee05a4daf50db7779687ccb2b288b015d4d26bb55796ac0b94c7de88b850abf1068371c89d5af254198d8baca91efb1051946cbd

data/lib/Rakefile

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+require 'seed_reaper'
+
+path = File.expand_path(__dir__)
+Dir.glob("#{path}/tasks/**/*.rake").each { |f| load f }

data/lib/seed_reaper/config_evaluator.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module SeedReaper
+  class ConfigEvaluator
+    def initialize(config)
+      @config = config
+    end
+
+    def schema
+      return nil unless @config
+      return @config.reject { |k| k == :meta } if @config.is_a?(Hash)
+      return @config.reject { |c| c.is_a?(Hash) && c.has_key?(:meta) } if @config.is_a?(Array)
+
+      @config
+    end
+
+    %i[count joins].each do |meta_field|
+      define_method meta_field do
+        meta(meta_field)
+      end
+    end
+
+    private
+
+    def meta(field)
+      return nil unless @config
+      return @config.dig(:meta, field) if @config.is_a?(Hash)
+      return nil unless @config.is_a?(Array)
+
+      @config.select do |c|
+        c.is_a?(Hash) && c.has_key?(:meta)
+      end.first&.dig(:meta, field)
+    end
+  end
+end

data/lib/seed_reaper/railtie.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require 'seed_reaper'
+
+module SeedReaper
+  class Railtie < Rails::Railtie
+    railtie_name :seed_reaper
+
+    rake_tasks do
+      path = File.expand_path(__dir__)
+      Dir.glob("#{path}/tasks/**/*.rake").each { |f| load f }
+    end
+  end
+end

data/lib/seed_reaper/seed_writer.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module SeedReaper
+  class SeedWriter
+    def initialize(config)
+      @config = config
+    end
+
+    def write!
+      FileUtils.rm_rf('db/seeds/.', secure: true)
+
+      @config.each_with_index do |element, i|
+        File.open("db/seeds/#{i.to_s.rjust(@config.count.digits.count, '0')}_#{file_name(element)}.seeds.rb", 'w') do |file|
+          file.write(Seedifier.new(element).seedify)
+        end
+      end
+    end
+
+    private
+
+    def file_name(element)
+      return element if element.is_a?(Symbol)
+
+      element.first[0]
+    end
+  end
+end

data/lib/seed_reaper/seedifier.rb

@@ -0,0 +1,105 @@
+# frozen_string_literal: true
+
+require 'active_support/inflector'
+require_relative 'config_evaluator'
+
+module SeedReaper
+  class Seedifier
+    def initialize(config)
+      @config = config
+    end
+
+    def seedify
+      seedify_collection(base_model, base_config)
+    end
+
+    private
+
+    def base_model
+      return @config.to_s.camelize.constantize if @config.is_a?(Symbol)
+
+      @config.first[0].to_s.camelize.constantize
+    end
+
+    def base_config
+      return nil if @config.is_a?(Symbol)
+
+      @config.first[1]
+    end
+
+    def seedify_collection(collection, config)
+      ce = ConfigEvaluator.new(config)
+      evaluated_collection(collection, ce).reduce('') do |str, instance|
+        str += serialize(instance) + seedify_associations(instance, ce.schema)
+      end
+    end
+
+    def seedify_associations(instance, config)
+      return '' if config.nil?
+
+      if config.is_a?(Symbol)
+        instance.send(config).tap do |association|
+          return '' if association.blank?
+
+          association = [association] unless association.respond_to?(:each)
+          return seedify_collection(association, nil)
+        end
+      elsif config.is_a?(Array)
+        config.map do |sub_config|
+          seedify_associations(instance, sub_config)
+        end.join
+      elsif config.is_a?(Hash)
+        config.map do |(association_name, sub_config)|
+          association = instance.send(association_name)
+          next '' if association.blank?
+
+          association = [association] unless association.respond_to?(:each)
+          seedify_collection(association, sub_config)
+        end.join
+      else
+        fail "Your input config contains a #{config.class_name}. You may only use symbols, arrays and hashes."
+      end
+    end
+
+    def evaluated_collection(collection, config_evaluator)
+      limited_collection(
+        joined_collection(collection, config_evaluator.joins),
+        config_evaluator.count
+      )
+    end
+
+    def limited_collection(collection, count)
+      return collection unless collection.respond_to?(:limit)
+      return collection.limit(count) if count
+
+      collection.all
+    end
+
+    def joined_collection(collection, joins)
+      return collection unless collection.respond_to?(:joins)
+      return collection.joins(joins) if joins
+
+      collection
+    end
+
+    def serialize(instance)
+      "#{instance.class}.new(\n#{serialize_attrs(instance)}\n).save!(validate: false)\n\n"
+    end
+
+    def serialize_attrs(instance)
+      instance.attributes.to_h.reduce('') do |attr_str, (k, v)|
+        attr_str += "\s\s#{k}: #{serialize_value(v)},\n"
+      end[0...-2]
+    end
+
+    def serialize_value(value)
+      if value.nil?
+        "nil"
+      elsif value.is_a?(Integer)
+        value
+      else
+        "\"#{value}\""
+      end
+    end
+  end
+end

data/lib/seed_reaper/tasks/seed_reaper.rake

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'yaml'
+require 'seed_reaper'
+
+namespace :seed_reaper do
+  desc 'Write seeds to db/seeds/ based on config/seed_reaper.yml.'
+  task :write do
+    SeedReaper::SeedWriter.new(
+      YAML.load(
+        File.read('config/seed_reaper.yml')
+      )
+    ).write!
+  end
+end

data/lib/seed_reaper.rb

@@ -1,27 +1,8 @@
 # frozen_string_literal: true
 
-include 'seedifier'
+require 'seed_reaper/seedifier'
+require 'seed_reaper/seed_writer'
+require 'seed_reaper/railtie' if defined?(Rails)
 
-class SeedReaper
-  def initialize(config)
-    @config = config
-  end
-
-  def write!
-    FileUtils.rm_rf('db/seeds/.', secure: true)
-
-    @config.each_with_index do |element, i|
-      File.open("db/seeds/#{i.to_s.rjust(@config.count.digits.count, '0')}_#{file_name(element)}.seeds.rb", 'w') do |file|
-        file.write(Seedifier.new(element).seedify)
-      end
-    end
-  end
-
-  private
-
-  def file_name(element)
-    return element if element.is_a?(Symbol)
-
-    element.first[0]
-  end
+module SeedReaper
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: seed_reaper
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.0.2
 platform: ruby
 authors:
 - David Butts
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-02-14 00:00:00.000000000 Z
+date: 2022-02-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -143,9 +143,13 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/config_evaluator.rb
+- lib/Rakefile
 - lib/seed_reaper.rb
-- lib/seedifier.rb
+- lib/seed_reaper/config_evaluator.rb
+- lib/seed_reaper/railtie.rb
+- lib/seed_reaper/seed_writer.rb
+- lib/seed_reaper/seedifier.rb
+- lib/seed_reaper/tasks/seed_reaper.rake
 homepage: https://github.com/dbutts2/seed_reaper
 licenses:
 - MIT

data/lib/config_evaluator.rb

@@ -1,33 +0,0 @@
-# frozen_string_literal: true
-
-class ConfigEvaluator
-  def initialize(config)
-    @config = config
-  end
-
-  def schema
-    return nil unless @config
-    return @config.reject { |k| k == :meta } if @config.is_a?(Hash)
-    return @config.reject { |c| c.is_a?(Hash) && c.has_key?(:meta) } if @config.is_a?(Array)
-
-    @config
-  end
-
-  %i[count joins].each do |meta_field|
-    define_method meta_field do
-      meta(meta_field)
-    end
-  end
-
-  private
-
-  def meta(field)
-    return nil unless @config
-    return @config.dig(:meta, field) if @config.is_a?(Hash)
-    return nil unless @config.is_a?(Array)
-
-    @config.select do |c|
-      c.is_a?(Hash) && c.has_key?(:meta)
-    end.first&.dig(:meta, field)
-  end
-end

data/lib/seedifier.rb

@@ -1,103 +0,0 @@
-# frozen_string_literal: true
-
-require 'active_support/inflector'
-require 'config_evaluator'
-
-class Seedifier
-  def initialize(config)
-    @config = config
-  end
-
-  def seedify
-    seedify_collection(base_model, base_config)
-  end
-
-  private
-
-  def base_model
-    return @config.to_s.camelize.constantize if @config.is_a?(Symbol)
-
-    @config.first[0].to_s.camelize.constantize
-  end
-
-  def base_config
-    return nil if @config.is_a?(Symbol)
-
-    @config.first[1]
-  end
-
-  def seedify_collection(collection, config)
-    ce = ConfigEvaluator.new(config)
-    evaluated_collection(collection, ce).reduce('') do |str, instance|
-      str += serialize(instance) + seedify_associations(instance, ce.schema)
-    end
-  end
-
-  def seedify_associations(instance, config)
-    return '' if config.nil?
-
-    if config.is_a?(Symbol)
-      instance.send(config).tap do |association|
-        return '' if association.blank?
-
-        association = [association] unless association.respond_to?(:each)
-        return seedify_collection(association, nil)
-      end
-    elsif config.is_a?(Array)
-      config.map do |sub_config|
-        seedify_associations(instance, sub_config)
-      end.join
-    elsif config.is_a?(Hash)
-      config.map do |(association_name, sub_config)|
-        association = instance.send(association_name)
-        next '' if association.blank?
-
-        association = [association] unless association.respond_to?(:each)
-        seedify_collection(association, sub_config)
-      end.join
-    else
-      fail "Your input config contains a #{config.class_name}. You may only use symbols, arrays and hashes."
-    end
-  end
-
-  def evaluated_collection(collection, config_evaluator)
-    limited_collection(
-      joined_collection(collection, config_evaluator.joins),
-      config_evaluator.count
-    )
-  end
-
-  def limited_collection(collection, count)
-    return collection unless collection.respond_to?(:limit)
-    return collection.limit(count) if count
-
-    collection.all
-  end
-
-  def joined_collection(collection, joins)
-    return collection unless collection.respond_to?(:joins)
-    return collection.joins(joins) if joins
-
-    collection
-  end
-
-  def serialize(instance)
-    "#{instance.class}.new(\n#{serialize_attrs(instance)}\n).save!(validate: false)\n\n"
-  end
-
-  def serialize_attrs(instance)
-    instance.attributes.to_h.reduce('') do |attr_str, (k, v)|
-      attr_str += "\s\s#{k}: #{serialize_value(v)},\n"
-    end[0...-2]
-  end
-
-  def serialize_value(value)
-    if value.nil?
-      "nil"
-    elsif value.is_a?(Integer)
-      value
-    else
-      "\"#{value}\""
-    end
-  end
-end