class RuboCop::Cop::Style::CommentAnnotation

def annotation_range(comment, margin, length)
  start = comment.loc.expression.begin_pos + margin.length
  source_buffer = comment.loc.expression.source_buffer
  Parser::Source::Range.new(source_buffer, start, start + length)
end

def autocorrect(comment)
  margin, first_word, colon, space, note = split_comment(comment)
  start = comment.loc.expression.begin_pos + margin.length
  return if note.nil?
  lambda do |corrector|
    length = first_word.length + colon.to_s.length + space.to_s.length
    range = Parser::Source::Range.new(comment.loc.expression.source,
                                      start,
                                      start + length)
    corrector.replace(range, "#{first_word.upcase}: ")
  end
end

def correct_annotation?(first_word, colon, space, note)
  keyword?(first_word) && (colon && space && note || !colon && !note)
end

def first_comment_line?(comments, ix)
  ix == 0 || comments[ix - 1].loc.line < comments[ix].loc.line - 1
end

def investigate(processed_source)
  processed_source.comments.each_with_index do |comment, ix|
    next unless first_comment_line?(processed_source.comments, ix)
    margin, first_word, colon, space, note = split_comment(comment)
    next unless annotation?(comment) &&
                !correct_annotation?(first_word, colon, space, note)
    length = first_word.length + colon.to_s.length + space.to_s.length
    add_offense(comment, annotation_range(comment, margin, length),
                format(note ? MSG : MISSING_NOTE, first_word))
  end
end