seed_reaper 0.0.4 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 84b0a2cbec760f86053dbc8df84816f0fec6f55ec3598e52d8f8ddb3ed5b61e9
-  data.tar.gz: 8a096dac0835f12535249231686b8ecd10a3b9bac6973adaee7c2c157541db0b
+  metadata.gz: 8c773e91b80565a52fcb3d9db04bbf20c388eb94d4c2b50d82cc5039a93f0c2b
+  data.tar.gz: af0b15bec5a9fdab53d79db47f00396a543e61cf656eae6f379f2cf3a871bac6
 SHA512:
-  metadata.gz: 5f7cf51a355a87a65567d0ce8659e897ebc127dff5ebe46ec1fd86ff68e1686b9e45e0008bdbfb8b6f4c52502a9a00891fbd189063fa074bf89c1681b6bd8432
-  data.tar.gz: f0b8bc62ed2b49937a6be284a6efb6bcfeaf1d049cd4b55f3c2274706ee61ef6e5c28d9dfe19dea5e21276244b4dbf38ac15ae5e2cbcfcf27891c02f8f183283
+  metadata.gz: '012080a2a7cc080d9e555bf60857a2e64ae5349d24bdd680ec2f004741801afd2363db99e52446df6d6cb03e2c4d0f79ce4c2131a1247ecd4a083bf2b0e4ecbf'
+  data.tar.gz: 665f2d89354342054b06443fca9165c73e142d61b05c93114980a23362d79e0c137214572e46e07105cc0c36a7e8d5bb716825d0c806cdf34d1d93e2ec9d8249

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

@@ -8,6 +8,7 @@ module SeedReaper
   class Seedifier
     def initialize(config)
       @config = config
+      @serialized = []
     end
 
     def seedify
@@ -31,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
 
@@ -83,13 +87,19 @@ 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}: #{ValueSerializer.new(v).serialized},\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
   end

data/lib/seed_reaper/value_serializer.rb

@@ -6,14 +6,14 @@ module SeedReaper
       @value = value
     end
 
-    def serialized
+    def serialized(nullify: false)
       @serialized ||=
-        if @value.nil?
+        if nullify || @value.nil?
           "nil"
         elsif @value.is_a?(Integer)
           @value
         else
-          "%q{#{@value}}"
+          "%q{#{@value.to_s.gsub(/({|})/) { |c| "\\#{c}" }}}"
         end
     end
   end

data/lib/seed_reaper/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: seed_reaper
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.7
 platform: ruby
 authors:
 - David Butts
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-02-17 00:00:00.000000000 Z
+date: 2022-02-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport