class Thor::Actions::InjectIntoFile

def initialize(base, destination, data, config)
  super(base, destination, {:verbose => true}.merge(config))
  @behavior, @flag = if @config.key?(:after)
                       [:after, @config.delete(:after)]
                     else
                       [:before, @config.delete(:before)]
                     end
  @replacement = data.is_a?(Proc) ? data.call : data
  @flag = Regexp.escape(@flag) unless @flag.is_a?(Regexp)
end

def invoke!
  say_status :invoke
  content = if @behavior == :after
              '\0' + replacement
            else
              replacement + '\0'
            end
  replace!(/#{flag}/, content, config[:force])
end

def replace!(regexp, string, force)
  unless base.options[:pretend]
    content = File.binread(destination)
    if force || !content.include?(replacement)
      content.gsub!(regexp, string)
      File.open(destination, 'wb') { |file| file.write(content) }
    end
  end
end

def revoke!
  say_status :revoke
  regexp = if @behavior == :after
             content = '\1\2'
             /(#{flag})(.*)(#{Regexp.escape(replacement)})/m
           else
             content = '\2\3'
             /(#{Regexp.escape(replacement)})(.*)(#{flag})/m
           end
  replace!(regexp, content, true)
end

def say_status(behavior)
  status = if behavior == :invoke
             if flag == /\A/
               :prepend
             elsif flag == /\z/
               :append
             else
               :insert
             end
           else
             :subtract
           end
  super(status, config[:verbose])
end