class RuboCop::Cop::Style::NestedTernaryOperator

def autocorrect(corrector, if_node)
  replace_loc_and_whitespace(corrector, if_node.loc.question, "\n")
  replace_loc_and_whitespace(corrector, if_node.loc.colon, "\nelse\n")
  corrector.replace(if_node.if_branch, remove_parentheses(if_node.if_branch.source))
  corrector.wrap(if_node, 'if ', "\nend")
end

def if_node(node)
  node = node.parent
  return node if node.if_type?
  if_node(node)
end

def on_if(node)
  return unless node.ternary?
  node.each_descendant(:if).select(&:ternary?).each do |nested_ternary|
    add_offense(nested_ternary) do |corrector|
      if_node = if_node(nested_ternary)
      next if part_of_ignored_node?(if_node)
      autocorrect(corrector, if_node)
      ignore_node(if_node)
    end
  end
end

def remove_parentheses(source)
  return source unless source.start_with?('(')
  source.delete_prefix('(').delete_suffix(')')
end

def replace_loc_and_whitespace(corrector, range, replacement)
  corrector.replace(
    range_with_surrounding_space(range: range, whitespace: true),
    replacement
  )
end