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