class RuboCop::Cop::MultilineLiteralBraceCorrector

def self.correct(corrector, node, processed_source)
  new(corrector, node, processed_source).call
end

def call
  if closing_brace_on_same_line?(node)
    correct_same_line_brace(corrector)
  else
    # When a comment immediately before the closing brace gets in the
    # way of an easy correction, the offense is reported but not auto-
    # corrected. The user must handle the delicate decision of where to
    # put the comment.
    return if new_line_needed_before_closing_brace?(node)
    correct_next_line_brace(corrector)
  end
end

def content_if_comment_present(corrector, node)
  range = range_with_surrounding_space(
    range: children(node).last.source_range,
    side: :right
  ).end.resize(1)
  if range.source == '#'
    select_content_to_be_inserted_after_last_element(corrector, node)
  else
    node.loc.end.source
  end
end

def correct_next_line_brace(corrector)
  corrector.remove(range_with_surrounding_space(range: node.loc.end, side: :left))
  corrector.insert_before(
    last_element_range_with_trailing_comma(node).end,
    content_if_comment_present(corrector, node)
  )
end

def correct_same_line_brace(corrector)
  corrector.insert_before(node.loc.end, "\n")
end

def initialize(corrector, node, processed_source)
  @corrector = corrector
  @node = node
  @processed_source = processed_source
end

def last_element_range_with_trailing_comma(node)
  trailing_comma_range = last_element_trailing_comma_range(node)
  if trailing_comma_range
    children(node).last.source_range.join(trailing_comma_range)
  else
    children(node).last.source_range
  end
end

def last_element_trailing_comma_range(node)
  range = range_with_surrounding_space(
    range: children(node).last.source_range,
    side: :right
  ).end.resize(1)
  range.source == ',' ? range : nil
end

def remove_trailing_content_of_comment(corrector, range)
  corrector.remove(range)
end

def select_content_to_be_inserted_after_last_element(corrector, node)
  range = range_between(
    node.loc.end.begin_pos,
    range_by_whole_lines(node.loc.expression).end.end_pos
  )
  remove_trailing_content_of_comment(corrector, range)
  range.source
end