class RuboCop::Cop::Layout::TrailingBlankLines


# EOF
class Foo; end
# good
class Foo; end # EOF
# bad
# EOF
class Foo; end
# bad
# ‘final_newline` looks for one newline at the end of files.
@example EnforcedStyle: final_newline (default)
# EOF
class Foo; end
# good
class Foo; end # EOF
# bad
# EOF
class Foo; end
# bad
# at the end of files.
# `final_blank_line` looks for one blank line followed by a new line
@example EnforcedStyle: final_blank_line
source code.
This cop looks for trailing blank lines and a final newline in the

def autocorrect(range)

def autocorrect(range)
  lambda do |corrector|
    corrector.replace(range, style == :final_newline ? "\n" : "\n\n")
  end
end

def ends_in_end?(processed_source)

def ends_in_end?(processed_source)
  buffer = processed_source.buffer
  return true if buffer.source.strip.start_with?('__END__')
  return false if processed_source.tokens.empty?
  extra = buffer.source[processed_source.tokens.last.end_pos..-1]
  extra && extra.strip.start_with?('__END__')
end

def investigate(processed_source)

def investigate(processed_source)
  buffer = processed_source.buffer
  return if buffer.source.empty?
  # The extra text that comes after the last token could be __END__
  # followed by some data to read. If so, we don't check it because
  # there could be good reasons why it needs to end with a certain
  # number of newlines.
  return if ends_in_end?(processed_source)
  whitespace_at_end = buffer.source[/\s*\Z/]
  blank_lines = whitespace_at_end.count("\n") - 1
  wanted_blank_lines = style == :final_newline ? 0 : 1
  return unless blank_lines != wanted_blank_lines
  offense_detected(buffer, wanted_blank_lines, blank_lines,
                   whitespace_at_end)
end

def message(wanted_blank_lines, blank_lines)

def message(wanted_blank_lines, blank_lines)
  case blank_lines
  when -1
    'Final newline missing.'
  when 0
    'Trailing blank line missing.'
  else
    instead_of = if wanted_blank_lines.zero?
                   ''
                 else
                   "instead of #{wanted_blank_lines} "
                 end
    format('%<current>d trailing blank lines %<prefer>sdetected.',
           current: blank_lines, prefer: instead_of)
  end
end

def offense_detected(buffer, wanted_blank_lines, blank_lines,

def offense_detected(buffer, wanted_blank_lines, blank_lines,
                     whitespace_at_end)
  begin_pos = buffer.source.length - whitespace_at_end.length
  autocorrect_range = range_between(begin_pos, buffer.source.length)
  begin_pos += 1 unless whitespace_at_end.empty?
  report_range = range_between(begin_pos, buffer.source.length)
  add_offense(autocorrect_range,
              location: report_range,
              message: message(wanted_blank_lines, blank_lines))
end