class RuboCop::Cop::LineBreakCorrector

def break_line_before(range:, node:, corrector:, configured_width:,
                      indent_steps: 1)
  corrector.insert_before(
    range,
    "\n#{' ' * (node.loc.keyword.column + (indent_steps * configured_width))}"
  )
end

def correct_trailing_body(configured_width:, corrector:, node:,
                          processed_source:)
  @processed_source = processed_source
  range = first_part_of(node.to_a.last)
  eol_comment = processed_source.comment_at_line(node.source_range.line)
  break_line_before(range: range, node: node, corrector: corrector,
                    configured_width: configured_width)
  move_comment(eol_comment: eol_comment, node: node, corrector: corrector)
  remove_semicolon(node, corrector)
end

def move_comment(eol_comment:, node:, corrector:)
  return unless eol_comment
  text = eol_comment.loc.expression.source
  corrector.insert_before(node, "#{text}\n#{' ' * node.loc.keyword.column}")
  corrector.remove(eol_comment)
end

def remove_semicolon(node, corrector)
  return unless semicolon(node)
  corrector.remove(semicolon(node).pos)
end

def semicolon(node)
  @semicolon ||= {}.compare_by_identity
  @semicolon[node] ||= processed_source.tokens_within(node).find(&:semicolon?)
end