seed_reaper 0.0.3 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3b3718d655f542f022d098a6a3d6d117957a3c27c755cb4ea3291d2ef07467de
-  data.tar.gz: c50dd4f1bb5d335593988810b42d117b23502a289a65aafa73d107060fe293f1
+  metadata.gz: 33aaf207d0c92c6c1c42f679d275cafa0c4cb52537b75bc7b05ab59b8c53beb4
+  data.tar.gz: 4d88d55befff9eb3fd5ddcb290325541bb866ee023a5173310700c77100eab74
 SHA512:
-  metadata.gz: 12e1027c7f5664cb3813abfd3531580e64012b7c0606a47e69f36937ed4e1c1b9fffc066af63572e6c115d591ce144ca66d1e31a15c8efd1c84ab1ee1c18b222
-  data.tar.gz: 897fb366266c4822fabf13032bd273fc00c1385bbf51060cada640a7ccc8435c5335e24c4150902a3eb81cfbccf903157d325577b208fa2a3199e0c1542c5743
+  metadata.gz: 1a13d8c2f213c592b57079b573789a66512290595bbbe8a2f5efb992e69eae30e7219349b8ebe8bb5dbc352e03c42f8eb14c6c57a51c29ab8cf5a6defa3cbaf8
+  data.tar.gz: d49d7e67e0c2955b814db386ba23f65abf01b6f67e3c119041050d212677df6217a6c54c1ceadaefb8db05fcc9193b65f3c685561f4183ccea8c67e3001c5e79

data/lib/seed_reaper/config_evaluator.rb

@@ -14,7 +14,33 @@ module SeedReaper
       @config
     end
 
-    %i[count joins].each do |meta_field|
+    def belongs_to_schema(instance)
+      return nil unless schema
+      return schema.select { |k| instance._reflections[k.to_s].belongs_to? } if schema.is_a?(Hash)
+      return schema.select do |c|
+        if c.is_a?(Hash)
+          instance._reflections[c.first[0].to_s].belongs_to?
+        else
+          instance._reflections[c.to_s].belongs_to?
+        end
+      end if schema.is_a?(Array)
+      return schema if instance._reflections[schema.to_s].belongs_to?
+    end
+
+    def non_belongs_to_schema(instance)
+      return nil unless schema
+      return schema.reject { |k| instance._reflections[k.to_s].belongs_to? } if schema.is_a?(Hash)
+      return schema.reject do |c|
+        if c.is_a?(Hash)
+          instance._reflections[c.first[0].to_s].belongs_to?
+        else
+          instance._reflections[c.to_s].belongs_to?
+        end
+      end if schema.is_a?(Array)
+      return schema unless instance._reflections[schema.to_s].belongs_to?
+    end
+
+    %i[count joins nullify].each do |meta_field|
       define_method meta_field do
         meta(meta_field)
       end

data/lib/seed_reaper/seedifier.rb

@@ -2,11 +2,13 @@
 
 require 'active_support/inflector'
 require_relative 'config_evaluator'
+require_relative 'value_serializer'
 
 module SeedReaper
   class Seedifier
     def initialize(config)
       @config = config
+      @serialized = []
     end
 
     def seedify
@@ -30,7 +32,10 @@ module SeedReaper
     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)
+        str +=
+          seedify_associations(instance, ce.belongs_to_schema(instance)) +
+          serialize(instance, ce) +
+          seedify_associations(instance, ce.non_belongs_to_schema(instance))
       end
     end
 
@@ -82,24 +87,20 @@ module SeedReaper
       collection
     end
 
-    def serialize(instance)
-      "#{instance.class}.new(\n#{serialize_attrs(instance)}\n).save!(validate: false)\n\n"
+    def serialize(instance, config_evaluator)
+      seed = "#{instance.class}.new(\n#{serialize_attrs(instance, config_evaluator)}\n).save!(validate: false)\n\n"
+      hashed_seed = Digest::SHA2.hexdigest(seed)
+      return '' if @serialized.include?(hashed_seed)
+
+      @serialized << hashed_seed
+      seed
     end
 
-    def serialize_attrs(instance)
+    def serialize_attrs(instance, config_evaluator)
       instance.attributes.to_h.reduce('') do |attr_str, (k, v)|
-        attr_str += "\s\s#{k}: #{serialize_value(v)},\n"
+        nullify = [config_evaluator.nullify].flatten.compact.map(&:to_s).include?(k.to_s)
+        attr_str += "\s\s#{k}: #{ValueSerializer.new(v).serialized(nullify: nullify)},\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/value_serializer.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module SeedReaper
+  class ValueSerializer
+    def initialize(value)
+      @value = value
+    end
+
+    def serialized(nullify: false)
+      @serialized ||=
+        if nullify || @value.nil?
+          "nil"
+        elsif @value.is_a?(Integer)
+          @value
+        else
+          "%q{#{@value}}"
+        end
+    end
+  end
+end

data/lib/seed_reaper/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module SeedReaper
+  VERSION = '0.0.6'
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: seed_reaper
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.6
 platform: ruby
 authors:
 - David Butts
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-02-17 00:00:00.000000000 Z
+date: 2022-02-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -150,6 +150,8 @@ files:
 - lib/seed_reaper/seed_writer.rb
 - lib/seed_reaper/seedifier.rb
 - lib/seed_reaper/tasks/seed_reaper.rake
+- lib/seed_reaper/value_serializer.rb
+- lib/seed_reaper/version.rb
 homepage: https://github.com/dbutts2/seed_reaper
 licenses:
 - MIT