Merge pull request #498 from DannyBen/refactor/indenter

Move IndentationHelper to concerns directory

Author: DannyBen

.rubocop.yml

@@ -22,7 +22,7 @@ Style/GuardClause:            { Exclude: [lib/bashly/config_validator.rb] }
 # False positive report of invalid use of a method that starts with `set_`
 Naming/AccessorMethodName:
   Exclude:
-    - 'lib/bashly/indenter.rb'
+    - 'lib/bashly/concerns/indentation_helper.rb'
 
 # FIXME: The `Command` class is too long
 Metrics/ClassLength: { Exclude: [lib/bashly/script/command.rb] }

lib/bashly/concerns/indentation_helper.rb

@@ -0,0 +1,45 @@
+module Bashly
+  # A helper class, used by the `Array#indent` extension.
+  # It will return the array of strings with all strings prefixed by `indentation`
+  # unless the line is within a heredoc block.
+  class IndentationHelper
+    attr_reader :marker, :indentation
+
+    def initialize(indentation)
+      @indentation = indentation
+      @marker = nil
+    end
+
+    def indent(line)
+      if inside_heredoc?
+        reset_marker if heredoc_closed?(line)
+        line
+      else
+        set_heredoc_state(line)
+        "#{indentation}#{line}"
+      end
+    end
+
+  private
+
+    def reset_marker
+      @marker = nil
+    end
+
+    def inside_heredoc?
+      !!marker
+    end
+
+    def set_heredoc_state(line)
+      @marker = extract_heredoc_marker(line)
+    end
+
+    def extract_heredoc_marker(line)
+      line =~ /<<-?\s*(\w+)/ ? $1 : nil
+    end
+
+    def heredoc_closed?(line)
+      inside_heredoc? && /^#{marker}\n?$/.match?(line)
+    end
+  end
+end

lib/bashly/extensions/array.rb

@@ -1,11 +1,11 @@
-require 'bashly/indenter'
+require 'bashly/concerns/indentation_helper'
 
 class Array
   def indent(offset)
     return self unless offset.positive?
 
     indentation = ' ' * offset
-    indenter = Indenter.new indentation
+    indenter = Bashly::IndentationHelper.new indentation
 
     map { |line| indenter.indent line }
   end