class RuboCop::Cop::Style::EachWithObject

1, 2].each_with_object({}) { |e, a| a[e

e }
# good
[1, 2].inject({}) { |a, e| a = e; a }
# bad
@example
parameter is assigned to within the block.
However, we can’t replace with each_with_object if the accumulator
the need to return the object at the end.
returned at the end and so could be replaced by each_with_object without
This cop looks for inject / reduce calls where the passed in object is

def accumulator_param_assigned_to?(body, args)

then we can't convert to each_with_object
if the accumulator parameter is assigned to in the block,
def accumulator_param_assigned_to?(body, args)
  first_arg, = *args
  accumulator_var, = *first_arg
  body.each_descendant.any? do |n|
    next unless n.assignment?
    lhs, _rhs = *n
    lhs.equal?(accumulator_var)
  end
end

def autocorrect(node)

rubocop:disable Metrics/AbcSize
def autocorrect(node)
  lambda do |corrector|
    corrector.replace(node.send_node.loc.selector, 'each_with_object')
    first_arg, second_arg = *node.arguments
    corrector.replace(first_arg.loc.expression, second_arg.source)
    corrector.replace(second_arg.loc.expression, first_arg.source)
    return_value = return_value(node.body)
    if return_value_occupies_whole_line?(return_value)
      corrector.remove(whole_line_expression(return_value))
    else
      corrector.remove(return_value.loc.expression)
    end
  end
end

def first_argument_returned?(args, return_value)

def first_argument_returned?(args, return_value)
  first_arg, = *args
  accumulator_var, = *first_arg
  return_var, = *return_value
  accumulator_var == return_var
end

def on_block(node)

def on_block(node)
  each_with_object_candidate?(node) do |method, args, body|
    _, method_name, method_arg = *method
    return if simple_method_arg?(method_arg)
    return_value = return_value(body)
    return unless return_value
    return unless first_argument_returned?(args, return_value)
    return if accumulator_param_assigned_to?(body, args)
    add_offense(node, location: method.loc.selector,
                      message: format(MSG, method: method_name))
  end
end

def return_value(body)

def return_value(body)
  return unless body
  return_value = body.begin_type? ? body.children.last : body
  return_value if return_value && return_value.lvar_type?
end

def return_value_occupies_whole_line?(node)

def return_value_occupies_whole_line?(node)
  whole_line_expression(node).source.strip == node.source
end

def simple_method_arg?(method_arg)

def simple_method_arg?(method_arg)
  method_arg && method_arg.basic_literal?
end

def whole_line_expression(node)

def whole_line_expression(node)
  range_by_whole_lines(node.loc.expression, include_final_newline: true)
end