planter 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b66f9b16f720b193d4dee6e0a23382b22a073982634bccbad2d52092516090eb
-  data.tar.gz: 5d66d7905379b7b148586b52b85afb2d715501543e88f0d84a88973b734d9f6b
+  metadata.gz: d5ec6b0fc5c0ec804e4a1ea2f1c050bcad9eadcc4cea08a8816a6b3ff17a1c37
+  data.tar.gz: 3639096bccfaf6aac225c78c7acfc919610802f4569fa239ffe4ffe7b648edcc
 SHA512:
-  metadata.gz: dde579bb6c055ce0c819148d1d818b39bf44ef9e75c692799dd760c7f9a96a0828d959f3204e6741fa55c41a6b9c2d8045f746142c139b5f20f1ea450ba7b259
-  data.tar.gz: 9f208c04796ae9f80ce6d46b4ec94e56da4fc2c62c36abf3198d806621bf2c7d79d9f52fd2d30ad0ee73c94206e7f0997b953ff93dbc93b83ff8064ee7ab85ed
+  metadata.gz: 2da58e993a59e4019e83f9885826672158bf7483b7310ef07f86b54d93ceb3675bf7509c656a976372debac167aa2c12d6e12214d71be0acff253e0dd3d682b5
+  data.tar.gz: 80ce1e443cc6eafaf9eb1746f3d617fcf4b33bae7621d28e1f36a2a00a1f57a2e2638a2a629e02b0bdda049208b458a72e3bf43b4098d054e17f41ff8351d9de

data/README.md

@@ -136,8 +136,8 @@ things to note.
 
 - The seeder will always be appended at the end of the array. If this is not the
 correct order, you'll need to adjust the array manually.
-- When adjusting the array, always keep the closing bracket on its own line, or
-the generator won't know where to put the new seeders.
+- The generator can append to either multiline or inline `%i[...]` seeders
+arrays.
 
 You can also tell the generator which seeding style to use.
 

data/lib/generators/planter/seeder_generator.rb

@@ -50,9 +50,7 @@ module Planter
 
         create_csv(seeder) if selected_seeding_method == :csv
 
-        inject_into_file "config/initializers/planter.rb",
-          "    #{seeder}\n",
-          before: /^\s*\]\s*$/
+        register_seeder(seeder)
       end
 
       def seeder_contents
@@ -94,6 +92,110 @@ module Planter
         end.join
       end
 
+      def register_seeder(seeder)
+        contents = ::File.read(initializer_full_path)
+        ::File.write(initializer_full_path, add_seeder_to_initializer(contents, seeder))
+      end
+
+      def add_seeder_to_initializer(contents, seeder)
+        assignment = contents.match(seeders_assignment_pattern)
+        unless assignment
+          raise Thor::Error, "Could not find config.seeders = %i[...] in #{initializer_path}"
+        end
+
+        opening_bracket_index = assignment.end(0) - 1
+        closing_bracket_index = closing_bracket_index_for(contents, opening_bracket_index)
+        unless closing_bracket_index
+          raise Thor::Error,
+            "Could not find the closing bracket for config.seeders in #{initializer_path}"
+        end
+
+        insert_seeder(contents, seeder, assignment[1], opening_bracket_index, closing_bracket_index)
+      end
+
+      def insert_seeder(
+        contents,
+        seeder,
+        assignment_indent,
+        opening_bracket_index,
+        closing_bracket_index
+      )
+        body = contents[(opening_bracket_index + 1)...closing_bracket_index]
+        updated = contents.dup
+
+        if body.include?("\n")
+          insert_multiline_seeder(
+            updated,
+            contents,
+            seeder,
+            assignment_indent,
+            body,
+            closing_bracket_index
+          )
+        elsif body.strip.empty?
+          updated.insert(closing_bracket_index, seeder)
+        else
+          updated.insert(closing_bracket_index, " #{seeder}")
+        end
+
+        updated
+      end
+
+      def insert_multiline_seeder(
+        updated,
+        contents,
+        seeder,
+        assignment_indent,
+        body,
+        closing_bracket_index
+      )
+        closing_line_start = contents.rindex("\n", closing_bracket_index) + 1
+        closing_line_prefix = contents[closing_line_start...closing_bracket_index]
+
+        if closing_line_prefix.match?(/\A[ \t]*\z/)
+          updated.insert(closing_line_start, "#{seeder_indent(body, assignment_indent)}#{seeder}\n")
+        else
+          updated.insert(closing_bracket_index, " #{seeder}")
+        end
+      end
+
+      def seeder_indent(body, assignment_indent)
+        item_line = body.lines.find { |line| line.match?(/\S/) }
+        indent = item_line&.match(/\A[ \t]*/).to_s
+        indent.empty? ? "#{assignment_indent}  " : indent
+      end
+
+      def closing_bracket_index_for(contents, opening_bracket_index)
+        depth = 1
+        index = opening_bracket_index + 1
+
+        while index < contents.length
+          if contents[index] == "\\"
+            index += 2
+            next
+          elsif contents[index] == "["
+            depth += 1
+          elsif contents[index] == "]"
+            depth -= 1
+            return index if depth.zero?
+          end
+
+          index += 1
+        end
+      end
+
+      def seeders_assignment_pattern
+        /^([ \t]*)config\.seeders\s*=\s*%i\[/
+      end
+
+      def initializer_path
+        "config/initializers/planter.rb"
+      end
+
+      def initializer_full_path
+        ::File.join(destination_root, initializer_path)
+      end
+
       def selected_seeding_method
         return unless options["seeding_method"]
 

data/lib/planter/version.rb

@@ -21,7 +21,7 @@ module Planter
     # Patch version.
     #
     # @return [Integer]
-    PATCH = 0
+    PATCH = 1
 
     module_function
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: planter
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Evan Gray