module RuboCop::Cop::AutocorrectLogic

def autocorrect?
  autocorrect_requested? && correctable? && autocorrect_enabled?
end

def autocorrect_enabled?
  # allow turning off autocorrect on a cop by cop basis
  return true unless cop_config
  return false if cop_config['AutoCorrect'] == false
  if @options.fetch(:safe_auto_correct, false)
    return cop_config.fetch('SafeAutoCorrect', true)
  end
  true
end

def autocorrect_requested?
  @options.fetch(:auto_correct, false)
end

def correctable?
  support_autocorrect? || disable_uncorrectable?
end

def disable_offense(node)
  range = node.location.expression
  eol_comment = " # rubocop:disable #{cop_name}"
  needed_line_length = range.column +
                       (range.source_line + eol_comment).length
  if needed_line_length <= max_line_length
    disable_offense_at_end_of_line(range_of_first_line(range),
                                   eol_comment)
  else
    disable_offense_before_and_after(range_by_lines(range))
  end
end

def disable_offense_at_end_of_line(range, eol_comment)
  ->(corrector) { corrector.insert_after(range, eol_comment) }
end

def disable_offense_before_and_after(range_by_lines)
  lambda do |corrector|
    range_with_newline = range_by_lines.resize(range_by_lines.size + 1)
    leading_whitespace = range_by_lines.source_line[/^\s*/]
    corrector.insert_before(
      range_with_newline,
      "#{leading_whitespace}# rubocop:disable #{cop_name}\n"
    )
    corrector.insert_after(
      range_with_newline,
      "#{leading_whitespace}# rubocop:enable #{cop_name}\n"
    )
  end
end

def disable_uncorrectable?
  @options[:disable_uncorrectable] == true
end

def max_line_length
  config.for_cop('Metrics/LineLength')['Max'] || 80
end

def range_by_lines(range)
  begin_of_first_line = range.begin_pos - range.column
  last_line = range.source_buffer.source_line(range.last_line)
  last_line_offset = last_line.length - range.last_column
  end_of_last_line = range.end_pos + last_line_offset
  Parser::Source::Range.new(range.source_buffer,
                            begin_of_first_line,
                            end_of_last_line)
end

def range_of_first_line(range)
  begin_of_first_line = range.begin_pos - range.column
  end_of_first_line = begin_of_first_line + range.source_line.length
  Parser::Source::Range.new(range.source_buffer,
                            begin_of_first_line,
                            end_of_first_line)
end

def support_autocorrect?
  respond_to?(:autocorrect)
end