bullet_train-super_scaffolding 1.0.5 → 1.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 718da72ec75872187974897a5213689485797049c7ebdf1ddb564a30163cc3b1
-  data.tar.gz: 0d19c7282a403fee1baa0fb1c3c8e45cc72defe674ce58243099ea00aa245fff
+  metadata.gz: ce5d6f3d2cb0ff2a0ed85aa5928e2a9e390b17709b41889c0fb6c25b9d7c4293
+  data.tar.gz: af398b0b3cbc631f42e38d8643899678399bf1d7e819a1261f0142980029ec97
 SHA512:
-  metadata.gz: b5034959fedec36b6c8312e64ed757f25834399744d69f95a5dfb84f5f88b01b1e7bff080a0161048ae4ab8140147c4c818edd7e49356b424cbc2326940ad970
-  data.tar.gz: d1ddfc16f3035831d95531bb2ca8fa62b88d413c4aba4f181703f2e6d1c0d02fe37edf48c6313da8d5e35525a20b1d236d48cd69f774e12bd9507e7f9507b971
+  metadata.gz: 6a1450d9449295670cd943ad637da3405b6b509c09905520c93bfdbab32eb471d93eedd3df5915db3cf325d77a5bd2d0ea8b4add6cf29b9811aba12b4242d74b
+  data.tar.gz: f5c3879c59b9ed29c0982856e206e569756f6861f5f80784ef6ae04cdc5cd315771adfebdb18bcf3b7ae0a7aa32482ddf7c9f025f90befc7f7632b7c38f18cda

data/lib/bullet_train/super_scaffolding/version.rb

@@ -1,5 +1,5 @@
 module BulletTrain
   module SuperScaffolding
-    VERSION = "1.0.5"
+    VERSION = "1.0.6"
   end
 end

data/lib/scaffolding/block_manipulator.rb

@@ -0,0 +1,140 @@
+class Scaffolding::BlockManipulator
+  attr_accessor :lines
+
+  def initialize(filepath)
+    @filepath = filepath
+    @lines = File.readlines(filepath)
+  end
+
+  #
+  # Wrap a block of ruby code inside another block
+  #
+  # @param [String] starting A string to search for at the start of the block. Eg "<%= updates_for context, collection do"
+  # @param [Array] with An array with two String elements. The text that should wrap the block. Eg ["<%= action_model_select_controller do %>", "<% end %>"]
+  #
+  def wrap_block(starting:, with:)
+    with[0] += "\n" unless with[0].match?(/\n$/)
+    with[1] += "\n" unless with[1].match?(/\n$/)
+    starting_line = find_block_start(starting)
+    end_line = find_block_end(starting_from: starting_line)
+
+    final = []
+    block_indent = ""
+    spacer = "  "
+    @lines.each_with_index do |line, index|
+      line += "\n" unless line.match?(/\n$/)
+      if index < starting_line
+        final << line
+      elsif index == starting_line
+        block_indent = line.match(/^\s*/).to_s
+        final << block_indent + with[0]
+        final << (line.blank? ? "\n" : "#{spacer}#{line}")
+      elsif index > starting_line && index < end_line
+        final << (line.blank? ? "\n" : "#{spacer}#{line}")
+      elsif index == end_line
+        final << (line.blank? ? "\n" : "#{spacer}#{line}")
+        final << block_indent + with[1]
+      else
+        final << line
+      end
+    end
+
+    @lines = final
+    unless @lines.last.match?(/\n$/)
+      @lines.last += "\n"
+    end
+  end
+
+  def insert(content, within: nil, after: nil, before: nil, after_block: nil, append: false)
+    # Search for before like we do after, we'll just inject before it.
+    after ||= before
+
+    # If within is given, find the start and end lines of the block
+    content += "\n" unless content.match?(/\n$/)
+    start_line = 0
+    end_line = @lines.count - 1
+    if within.present?
+      start_line = find_block_start(within)
+      end_line = find_block_end(starting_from: start_line)
+      # start_line += 1 # ensure we actually insert the content _within_ the given block
+      # end_line += 1 if end_line == start_line
+    end
+    if after_block.present?
+      block_start = find_block_start(after_block)
+      block_end = find_block_end(starting_from: block_start)
+      start_line = block_end
+      end_line = @lines.count - 1
+    end
+    index = start_line
+    match = false
+    while index < end_line && !match
+      line = @lines[index]
+      if after.nil? || line.match?(after)
+        unless append
+          match = true
+          # We adjust the injection point if we really wanted to insert before.
+          insert_line(content, index - (before ? 1 : 0))
+        end
+      end
+      index += 1
+    end
+
+    return if match
+
+    # Match should always be false here.
+    if append && !match
+      insert_line(content, index - 1)
+    end
+  end
+
+  def insert_line(content, insert_at_index)
+    content += "\n" unless content.match?(/\n$/)
+    final = []
+    @lines.each_with_index do |line, index|
+      indent = line.match(/^\s*/).to_s
+      final << line
+      if index == insert_at_index
+        final << indent + content
+      end
+    end
+    @lines = final
+  end
+
+  def insert_block(block_content, after_block:)
+    block_start = find_block_start(after_block)
+    block_end = find_block_end(starting_from: block_start)
+    insert_line(block_content[0], block_end)
+    insert_line(block_content[1], block_end + 1)
+  end
+
+  def write
+    File.write(@filepath, @lines.join)
+  end
+
+  def find_block_start(starting_string)
+    matcher = Regexp.escape(starting_string)
+    starting_line = 0
+
+    @lines.each_with_index do |line, index|
+      if line.match?(matcher)
+        starting_line = index
+        break
+      end
+    end
+    starting_line
+  end
+
+  def find_block_end(starting_from:)
+    depth = 0
+    current_line = starting_from
+    @lines[starting_from..@lines.count].each_with_index do |line, index|
+      current_line = starting_from + index
+      depth += 1 if line.match?(/\s*<%.+ do .*%>/)
+      depth += 1 if line.match?(/\s*<% if .*%>/)
+      depth += 1 if line.match?(/\s*<% unless .*%>/)
+      depth -= 1 if line.match?(/\s*<%.* end .*%>/)
+      break current_line if depth == 0
+    end
+    current_line
+  end
+end

data/lib/scaffolding/script.rb

@@ -3,6 +3,7 @@
 
 require "scaffolding"
 require "scaffolding/transformer"
+require "scaffolding/block_manipulator"
 require "scaffolding/class_names_transformer"
 require "scaffolding/oauth_providers"
 require "scaffolding/routes_file_manipulator"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bullet_train-super_scaffolding
 version: !ruby/object:Gem::Version
-  version: 1.0.5
+  version: 1.0.6
 platform: ruby
 authors:
 - Andrew Culver
@@ -45,6 +45,7 @@ files:
 - lib/bullet_train/super_scaffolding/scaffolders/oauth_provider_scaffolder.rb
 - lib/bullet_train/super_scaffolding/version.rb
 - lib/scaffolding.rb
+- lib/scaffolding/block_manipulator.rb
 - lib/scaffolding/class_names_transformer.rb
 - lib/scaffolding/oauth_providers.rb
 - lib/scaffolding/routes_file_manipulator.rb